some fixes for scp v0

sesman/libscp_v0.c

@@ -30,8 +30,10 @@
 #include "os_calls.h"
 /* client API */
 /******************************************************************************/
-enum SCP_CLIENT_STATES_E scp_v0c_connect(struct SCP_CONNECTION* c, struct SCP_SESSION* s, SCP_DISPLAY* d)
+enum SCP_CLIENT_STATES_E scp_v0c_connect(struct SCP_CONNECTION* c, struct SCP_SESSION* s)
 {
+  uint32_t version;
+  uint32_t size;
   uint16_t sz;
 
   init_stream(c->in_s, c->in_s->size);
@@ -41,7 +43,18 @@ enum SCP_CLIENT_STATES_E scp_v0c_connect(struct SCP_CONNECTION* c, struct SCP_SE
   g_tcp_set_no_delay(c->in_sck);
   s_push_layer(c->out_s, channel_hdr, 8);
 
+  if (s->type==SCP_SESSION_TYPE_XVNC)
+  {
+    out_uint16_be(c->out_s, 0); // code
+  }
+  else if (s->type==SCP_SESSION_TYPE_XRDP)
+  {
     out_uint16_be(c->out_s, 10); // code
+  }
+  else
+  {
+    return SCP_CLIENT_STATE_INTERNAL_ERR;
+  }
   sz = g_strlen(s->username);
   out_uint16_be(c->out_s, sz);
   out_uint8a(c->out_s, s->username, sz);
@@ -64,7 +77,45 @@ enum SCP_CLIENT_STATES_E scp_v0c_connect(struct SCP_CONNECTION* c, struct SCP_SE
     return SCP_CLIENT_STATE_NETWORK_ERR;
   }
 
-  return SCP_CLIENT_STATE_OK;
+  if (0!=tcp_force_recv(c->in_sck, c->in_s->data, 8))
+  {
+    return SCP_CLIENT_STATE_NETWORK_ERR;
+  }
+
+  in_uint32_be(c->in_s, version);
+  if (0 != version)
+  {
+    return SCP_CLIENT_STATE_VERSION_ERR;
+  }
+
+  in_uint32_be(c->in_s, size);
+  if (size < 14)
+  {
+    return SCP_CLIENT_STATE_SIZE_ERR;
+  }
+
+  init_stream(c->in_s, c->in_s->size);
+  if (0!=tcp_force_recv(c->in_sck, c->in_s->data, size - 8))
+  {
+    return SCP_CLIENT_STATE_NETWORK_ERR;
+  }
+
+  in_uint16_be(c->in_s, sz);
+  if (3!=sz)
+  {
+    return SCP_CLIENT_STATE_SEQUENCE_ERR;
+  }
+
+  in_uint16_be(c->in_s, sz);
+  if (1!=sz)
+  {
+    return SCP_CLIENT_STATE_CONNECTION_DENIED;
+  }
+
+  in_uint16_be(c->in_s, sz);
+  s->display=sz;
+
+  return SCP_CLIENT_STATE_END;
 }
 
 /* server API */

sesman/libscp_v0.h

@@ -39,7 +39,7 @@
  * @param d display
  *
  */
-enum SCP_CLIENT_STATES_E scp_v0c_connect(struct SCP_CONNECTION* c, struct SCP_SESSION* s, SCP_DISPLAY* d);
+enum SCP_CLIENT_STATES_E scp_v0c_connect(struct SCP_CONNECTION* c, struct SCP_SESSION* s);
 
 /* server API */
 /**