Check return value of mcs_recv_connect_initial().

[rdpsrv] / rdpsrv.c

diff --git a/rdpsrv.c b/rdpsrv.c
index 8355426ebc0755b51cef60570dd9ac7bd77cc155..c5b4169403f7ef271fa7f237e52257bae1925f28 100644 (file)
--- a/rdpsrv.c
+++ b/rdpsrv.c
@@ -10,24 +10,16 @@
 
 const int tcp_port_rdp = 3389;
 int create_server_socket();
-int serve_client(int sock);
+int serve_client();
 
 int main()
 {
        int server_sock = create_server_socket();
        for ( ;; ) {
-               struct sockaddr_in sin;
-               socklen_t len;
-               int sock = accept(server_sock, (struct sockaddr *)&sin, &len);
-       
-               if (sock == -1) {
-                       perror("accept()");
-                       exit(1);
-               }
-
-               serve_client(sock);
-
-               close(sock);
+               iso_recv_connect(server_sock);
+               printf("Got connection.\n");
+               serve_client();
+               printf("Client closed.\n");
        }
 }
 
@@ -67,13 +59,18 @@ int create_server_socket()
 }
 
 
-int serve_client(int sock)
+int serve_client()
 {
+       if (!mcs_recv_connect_initial())
+               error("MCS_CONNECT_INITIAL recv failed");
+       mcs_send_connect_response();
+       
        for ( ;; ) {
-               unsigned char buf[4096];
                short channel;
                
                /* receive ISO packets */
                mcs_recv(&channel);
+
+               printf("Packet on ch %u\n", channel);
        }
 }