diff --git a/samples/socket/EchoServer.kt b/samples/socket/EchoServer.kt new file mode 100644 index 00000000000..bec850a660e --- /dev/null +++ b/samples/socket/EchoServer.kt @@ -0,0 +1,71 @@ +import kotlinx.cinterop.* +import sockets.* + +fun main(args: Array) { + if (args.size < 1) { + println("Usage: ./echo_server ") + return + } + + val port = atoi(args[0]).toShort() + + memScoped { + + val bufferLength = 100L + val buffer = allocArray(bufferLength) + val serverAddr = alloc() + + val listenFd = socket(AF_INET, SOCK_STREAM, 0) + .ensureUnixCallResult { it >= 0 } + + with(serverAddr) { + memset(this.ptr, 0, sockaddr_in.size) + sin_family.value = AF_INET.toByte() + sin_addr.s_addr.value = htons(0).toInt() + sin_port.value = htons(port) + } + + bind(listenFd, serverAddr.ptr.reinterpret(), sockaddr_in.size.toInt()) + .ensureUnixCallResult { it == 0 } + + listen(listenFd, 10) + .ensureUnixCallResult { it == 0 } + + val commFd = accept(listenFd, null, null) + .ensureUnixCallResult { it >= 0 } + + while (true) { + val length = read(commFd, buffer.ptr, bufferLength) + .ensureUnixCallResult { it >= 0 } + + if (length == 0L) { + break + } + + write(commFd, buffer.ptr, length) + .ensureUnixCallResult { it >= 0 } + } + } +} + +// Not available through interop because declared as macro: +fun htons(value: Short) = ((value.toInt() ushr 8) or (value.toInt() shl 8)).toShort() + +fun throwUnixError(): Nothing { + perror(null) // TODO: store error message to exception instead. + throw Error("UNIX call failed") +} + +inline fun Int.ensureUnixCallResult(predicate: (Int) -> Boolean): Int { + if (!predicate(this)) { + throwUnixError() + } + return this +} + +inline fun Long.ensureUnixCallResult(predicate: (Long) -> Boolean): Long { + if (!predicate(this)) { + throwUnixError() + } + return this +} \ No newline at end of file diff --git a/samples/socket/sockets.def b/samples/socket/sockets.def new file mode 100644 index 00000000000..58f6641c85d --- /dev/null +++ b/samples/socket/sockets.def @@ -0,0 +1,5 @@ +headers = sys/socket.h netdb.h stdio.h string.h unistd.h stdlib.h arpa/inet.h +excludedFunctions = _IO_flockfile _IO_funlockfile _IO_ftrylockfile _IO_cleanup_region_start _IO_cleanup_region_end \ + mkstemp_dprotected_np + +compilerOpts = -D_POSIX_SOURCE