X-Git-Url: https://git.sesse.net/?p=rdpsrv;a=blobdiff_plain;f=mcs.c;h=e3690516e9fb75ec8a4cb244c9f25d8a5d7a97ac;hp=cee23643e00473371fc8d4be9a307e7dc7e8b785;hb=f7c77f1d801d69794ea81a97060f260a0af51e51;hpb=908d0bab155aee5757f902486dd1bbba859637f4

diff --git a/mcs.c b/mcs.c
index cee2364..e369051 100644
--- a/mcs.c
+++ b/mcs.c
@@ -215,35 +215,20 @@ mcs_send_aurq(void)
 	iso_send(s);
 }
 
-/* Expect a AUcf message (ASN.1 PER) */
-static BOOL
-mcs_recv_aucf(uint16 * mcs_userid)
+/* Send a AUcf message (ASN.1 PER) */
+static void
+mcs_send_aucf(uint16 mcs_userid)
 {
-	uint8 opcode, result;
 	STREAM s;
 
-	s = iso_recv();
-	if (s == NULL)
-		return False;
-
-	in_uint8(s, opcode);
-	if ((opcode >> 2) != MCS_AUCF)
-	{
-		error("expected AUcf, got %d\n", opcode);
-		return False;
-	}
-
-	in_uint8(s, result);
-	if (result != 0)
-	{
-		error("AUrq: %d\n", result);
-		return False;
-	}
-
-	if (opcode & 2)
-		in_uint16_be(s, *mcs_userid);
+	s = iso_init(5);
 
-	return s_check_end(s);
+	out_uint8(s, (MCS_AUCF << 2) | 2);  // | 2 = send user ID
+	out_uint8(s, 0);  // success
+	out_uint16_be(s, 0);
+	
+	s_mark_end(s);
+	iso_send(s);
 }
 
 /* Send a CJrq message (ASN.1 PER) */
@@ -265,35 +250,20 @@ mcs_send_cjrq(uint16 chanid)
 }
 
 /* Expect a CJcf message (ASN.1 PER) */
-static BOOL
-mcs_recv_cjcf(void)
+static void
+mcs_send_cjcf(uint16 userid, uint16 chanid)
 {
-	uint8 opcode, result;
 	STREAM s;
 
-	s = iso_recv();
-	if (s == NULL)
-		return False;
-
-	in_uint8(s, opcode);
-	if ((opcode >> 2) != MCS_CJCF)
-	{
-		error("expected CJcf, got %d\n", opcode);
-		return False;
-	}
-
-	in_uint8(s, result);
-	if (result != 0)
-	{
-		error("CJrq: %d\n", result);
-		return False;
-	}
+	s = iso_init(5);
 
-	in_uint8s(s, 4);	/* mcs_userid, req_chanid */
-	if (opcode & 2)
-		in_uint8s(s, 2);	/* join_chanid */
+	out_uint8(s, (MCS_CJCF << 2));
+	out_uint8(s, 0); // success
+	out_uint16_be(s, g_mcs_userid);
+	out_uint16_be(s, chanid);
 
-	return s_check_end(s);
+	s_mark_end(s);
+	iso_send(s);
 }
 
 /* Initialise an MCS transport data packet */
@@ -338,7 +308,7 @@ mcs_send(STREAM s)
 STREAM
 mcs_recv(uint16 * channel)
 {
-	uint8 opcode, appid, length;
+	uint8 opcode, appid, length, userid;
 	STREAM s;
 
 	s = iso_recv();
@@ -349,7 +319,7 @@ mcs_recv(uint16 * channel)
 	appid = opcode >> 2;
 
 	switch (appid) {
-	case MCS_SDIN:
+	case MCS_SDRQ:
 		in_uint8s(s, 2);	/* userid */
 		in_uint16_be(s, *channel);
 		in_uint8s(s, 1);	/* flags */
@@ -364,6 +334,18 @@ mcs_recv(uint16 * channel)
 		// Erect Domain (ignore)
 		printf("Received EDrq\n");
 		return NULL;
+	case MCS_AURQ:
+		// Attach User Request, respond with AUcf (Attach User Confirm)
+		printf("Received AUrq, sending AUcf\n");
+		mcs_send_aucf(0);
+		return NULL;
+	case MCS_CJRQ:
+		// Channel Join Request, respond with CJcf (Channel Join Confirm);
+		in_uint16_be(s, userid);
+		in_uint16_be(s, *channel);
+		printf("Received CJrq for channel %hu, sending CJcf\n", *channel);
+		mcs_send_cjcf(userid, *channel);
+		return NULL;
 	default:
 		error("expected data, got %d\n", opcode);
 		return NULL;