wrenner/socket_example.wren at main

 // socket_example.wren
 import "socket" for Socket, SocketInstance
 foreign class Host {
   foreign static signalDone()
 }
 class Time {
   static sleep(ms) {
     var start = System.clock
     while ((System.clock - start) * 1000 < ms) {
       Fiber.yield()
     }
   }
 }
 var mainFiber = Fiber.new {
   System.print("=== Simple Socket Test ===")
   // Test 1: Create a server socket
   System.print("\n1. Creating server socket...")
   var serverResult = SocketInstance.new()
   if (!serverResult) {
     System.print("Failed to create server socket")
     Host.signalDone()
     return
   }
   var serverErr = serverResult[0]
   var serverSock = serverResult[1]
   if (serverErr) {
     System.print("Server socket error: %(serverErr)")
     Host.signalDone()
     return
   }
   System.print("Server socket created successfully")
   // Test 2: Bind the server socket
   System.print("\n2. Binding server socket to port 11000...")
   var bindResult = serverSock.bind("127.0.0.1", 11000)
   var bindErr = bindResult[0]
   var bindSuccess = bindResult[1]
   if (bindErr) {
     System.print("Bind error: %(bindErr)")
     serverSock.close()
     Host.signalDone()
     return
   }
   System.print("Server socket bound successfully")
   // Test 3: Listen on the server socket
   System.print("\n3. Starting to listen...")
   var listenResult = serverSock.listen(5)
   var listenErr = listenResult[0]
   var listenSuccess = listenResult[1]
   if (listenErr) {
     System.print("Listen error: %(listenErr)")
     serverSock.close()
     Host.signalDone()
     return
   }
   System.print("Server listening successfully")
   // Test 4: Create and connect a client socket
   System.print("\n4. Creating client socket and connecting...")
   var clientResult = SocketInstance.connect("127.0.0.1", 11000)
   if (!clientResult) {
     System.print("Failed to create client socket")
     serverSock.close()
     Host.signalDone()
     return
   }
   var clientErr = clientResult[0]
   var clientSock = clientResult[1]
   if (clientErr) {
     System.print("Client connection error: %(clientErr)")
     serverSock.close()
     Host.signalDone()
     return
   }
   System.print("Client connected successfully")
   // Test 5: Accept connection on server
   System.print("\n5. Accepting connection on server...")
   var acceptAttempts = 0
   var accepted = false
   var acceptedSock = null
   while (acceptAttempts < 100 && !accepted) {
     var acceptResult = serverSock.accept()
     var acceptErr = acceptResult[0]
     var acceptedFd = acceptResult[1]
     if (!acceptErr && acceptedFd) {
       acceptedSock = SocketInstance.fromFd(acceptedFd)
       accepted = true
       System.print("Connection accepted successfully")
     } else {
       acceptAttempts = acceptAttempts + 1
       Fiber.yield()
     }
   }
   if (!accepted) {
     System.print("Failed to accept connection after %(acceptAttempts) attempts")
     clientSock.close()
     serverSock.close()
     Host.signalDone()
     return
   }
   // Test 6: Send data from client to server
   System.print("\n6. Sending data from client to server...")
   var testMessage = "Hello from client!"
   var writeErr = clientSock.write(testMessage)
   if (writeErr) {
     System.print("Client write error: %(writeErr)")
   } else {
     System.print("Client sent: %(testMessage)")
   }
   // Test 7: Read data on server side
   System.print("\n7. Reading data on server side...")
   var readAttempts = 0
   var dataReceived = false
   while (readAttempts < 100 && !dataReceived) {
     var readResult = acceptedSock.read(1024)
     var readErr = readResult[0]
     var readData = readResult[1]
     if (!readErr && readData && readData.count > 0) {
       System.print("Server received: %(readData)")
       dataReceived = true
       // Echo back
       var echoErr = acceptedSock.write(readData)
       if (echoErr) {
         System.print("Server echo error: %(echoErr)")
       } else {
         System.print("Server echoed data back")
       }
     } else {
       readAttempts = readAttempts + 1
       Fiber.yield()
     }
   }
   if (!dataReceived) {
     System.print("No data received after %(readAttempts) attempts")
   }
   // Test 8: Read echo on client side
   System.print("\n8. Reading echo on client side...")
   readAttempts = 0
   var echoReceived = false
   while (readAttempts < 100 && !echoReceived) {
     var readResult = clientSock.read(1024)
     var readErr = readResult[0]
     var readData = readResult[1]
     if (!readErr && readData && readData.count > 0) {
       System.print("Client received echo: %(readData)")
       echoReceived = true
     } else {
       readAttempts = readAttempts + 1
       Fiber.yield()
     }
   }
   // Clean up
   System.print("\n9. Cleaning up...")
   acceptedSock.close()
   clientSock.close()
   serverSock.close()
   System.print("\n=== All tests completed successfully! ===")
   Host.signalDone()
 }