adapt peer.c to Unix

test/peer.c

@@ -1,9 +1,150 @@
 #include <stdio.h>
-#include <conio.h>
 
+#ifdef _WIN32
+#include <conio.h>
 #define WIN32_LEAN_AND_MEAN
 #include <winsock2.h>
 #pragma comment(lib, "ws2_32.lib")
+#else
+#include <unistd.h>
+#include <stdlib.h>
+#include <errno.h>
+#include <time.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/ioctl.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <signal.h>
+#include <termios.h>
+#define SOCKET int
+#define INVALID_SOCKET (-1)
+#define SOCKET_ERROR (-1)
+#define closesocket close
+#define ioctlsocket ioctl
+#define SOCKADDR struct sockaddr
+#define SOCKADDR_IN struct sockaddr_in
+#define getch cgetc
+
+static void Sleep(u_int msec)
+{
+  struct timespec ts;
+  ts.tv_sec = msec / 1000;
+  ts.tv_nsec = (long)(msec % 1000) * 1000000;
+  nanosleep(&ts, NULL);
+}
+
+/* kbhit() and cgetc() for Unix: */
+
+static int __conio_initialized;
+
+static struct termios tty,otty;
+
+static __inline__ void makecooked(void)
+{
+	tcsetattr(0, TCSANOW, &otty);
+}
+
+static void sighand( int num )
+{
+	signal(SIGINT, SIG_IGN);
+	signal(SIGTERM, SIG_IGN);
+	signal(SIGHUP, SIG_IGN);
+	exit(0);
+}
+
+static void exitfn(void)
+{
+	if (__conio_initialized) makecooked();
+	printf("\n");
+}
+
+static void makeraw(void)
+{
+	static int first_call = 1;
+
+	if (first_call) {
+		first_call = 0;
+		if (tcgetattr(0, &tty)) {  /* input terminal */
+			fprintf(stderr, "cannot get terminal attributes: %s\n", strerror(errno));
+			exit(1);
+		}
+		otty=tty;  /* save it */
+#ifdef NO_MAKERAW /* makeraw code from NetBSD 1.6.2 (/usr/src/lib/libc/termios/cfmakeraw.c) */
+		tty.c_iflag &= ~(IMAXBEL|IGNBRK|BRKINT|PARMRK|ISTRIP|INLCR|IGNCR|ICRNL|IXON);
+		tty.c_oflag &= ~OPOST;
+		tty.c_lflag &= ~(ECHO|ECHONL|ICANON|ISIG|IEXTEN);
+		tty.c_cflag &= ~(CSIZE|PARENB);
+		tty.c_cflag |= CS8;
+#else
+		cfmakeraw(&tty);
+#endif
+		tty.c_lflag |= ISIG;  /* enable signals from ^C etc. */
+		tty.c_oflag |= ONLCR | OPOST; /* translate '\n' to '\r\n' on output */
+	}
+	tcsetattr(0, TCSANOW, &tty);
+}
+
+static void __init_conio(void)
+{
+	if (! __conio_initialized) {
+		signal(SIGINT, sighand);
+		signal(SIGTERM, sighand);
+		signal(SIGHUP, sighand);
+		atexit(exitfn);
+		makeraw();
+		__conio_initialized = 1;
+	}
+}
+
+static void __makeraw(void)
+{
+	if (! __conio_initialized) __init_conio();
+	else makeraw();
+}
+
+static void __makecooked(void)
+{
+	if (__conio_initialized) {
+		makecooked();
+	}
+}
+
+static char cgetc (void)
+{
+	char c;
+
+	__makeraw();
+	read(0, &c, 1);
+	return c;
+}
+
+static unsigned char kbhit (void)
+{
+	int retval;
+	fd_set fds;
+	struct timeval tv;
+
+	__makeraw();
+
+ again:
+	FD_ZERO(&fds);
+	FD_SET(0, &fds);
+	tv.tv_sec = tv.tv_usec = 0;
+	retval = select(1, &fds, NULL, NULL, &tv);
+	if (retval == -1) {
+		if (errno == EINTR) goto again;
+		__makecooked();
+		exit(1);
+	}
+	if (FD_ISSET(0, &fds)) {
+		return 1;
+	}
+	return 0;
+}
+
+#endif
 
 #define LEN 200
 
@@ -25,11 +166,13 @@ static void dump(unsigned char *buf, unsigned len)
 void main(void)
 {
   printf("Init\n");
+#ifdef _WIN32
   WSADATA wsa;
   if (WSAStartup(MAKEWORD(2, 2), &wsa) != 0)
   {
     return;
   }
+#endif
 
   SOCKET udp = socket(AF_INET, SOCK_DGRAM , IPPROTO_UDP);
   if (udp == INVALID_SOCKET)
@@ -91,15 +234,15 @@ void main(void)
       key = '\0';
     }
 
-    if (key == 'u')
+    if (key == 'u' || key == 'U')
     {
       if (remote.sin_addr.s_addr == INADDR_NONE)
       {
-        printf("Peer Unknown As Yet\n", len);
+        printf("Peer Unknown As Yet\n");
       }
       else
       {
-        unsigned i;
+        int i;
 
         len = LEN;
         for (i = 0; i < len; ++i)
@@ -115,15 +258,15 @@ void main(void)
       }
     }
 
-    if (key == 't')
+    if (key == 't' || key == 'T')
     {
       if (tcp == INVALID_SOCKET)
       {
-        printf("No Connection\n", len);
+        printf("No Connection\n");
       }
       else
       {
-        unsigned i;
+        int i;
 
         len = LEN;
         for (i = 0; i < len; ++i)
@@ -143,10 +286,17 @@ void main(void)
     len = recvfrom(udp, buf, sizeof(buf), 0, (SOCKADDR *)&remote, &remote_size);
     if (len == SOCKET_ERROR)
     {
+#ifdef _WIN32
       if (WSAGetLastError() != WSAEWOULDBLOCK)
       {
         return;
       }
+#else
+      if (errno != EAGAIN && errno != EWOULDBLOCK)
+      {
+        return;
+      }
+#endif
     }
     else if (len)
     {
@@ -161,10 +311,17 @@ void main(void)
       tcp = accept(srv, (SOCKADDR *)&conn, &conn_size);
       if (tcp == INVALID_SOCKET)
       {
+#ifdef _WIN32
         if (WSAGetLastError() != WSAEWOULDBLOCK)
         {
           return;
         }
+#else
+      if (errno != EAGAIN && errno != EWOULDBLOCK)
+      {
+        return;
+      }
+#endif
       }
       else
       {
@@ -182,10 +339,17 @@ void main(void)
       len = recv(tcp, buf, sizeof(buf), 0);
       if (len == SOCKET_ERROR)
       {
+#ifdef _WIN32
         if (WSAGetLastError() != WSAEWOULDBLOCK)
         {
           return;
         }
+#else
+      if (errno != EAGAIN && errno != EWOULDBLOCK)
+      {
+        return;
+      }
+#endif
       }
       else if (len)
       {
@@ -202,11 +366,13 @@ void main(void)
 
     Sleep(10);
   }
-  while (key != 'x');
+  while (key != 'x' && key != 'X');
 
   closesocket(udp);
   closesocket(tcp);
   closesocket(srv);
+#ifdef _WIN32
   WSACleanup();
+#endif
   printf("Done\n");
 }