libvncclient: add ARD principal and realm setters

Author: Chapoly1305

include/rfb/rfbclient.h

@@ -407,6 +407,11 @@ typedef struct _rfbClient {
 	 * Set by function SetClientAuthSchemes() */
 	uint32_t *clientAuthSchemes;
 
+	/** Optional configuration for ARD Kerberos auth. */
+	char *ardAuthRealm;
+	char *ardAuthClientPrincipal;
+	char *ardAuthServicePrincipal;
+
 	/** When the server is a repeater, this specifies the final destination */
 	char *destHost;
 	int destPort;
@@ -529,6 +534,9 @@ extern rfbClientLogProc rfbClientLog,rfbClientErr;
 extern rfbBool ConnectToRFBServer(rfbClient* client,const char *hostname, int port);
 extern rfbBool ConnectToRFBRepeater(rfbClient* client,const char *repeaterHost, int repeaterPort, const char *destHost, int destPort);
 extern void SetClientAuthSchemes(rfbClient* client,const uint32_t *authSchemes, int size);
+extern rfbBool rfbClientSetARDAuthRealm(rfbClient *client, const char *realm);
+extern rfbBool rfbClientSetARDAuthClientPrincipal(rfbClient *client, const char *principal);
+extern rfbBool rfbClientSetARDAuthServicePrincipal(rfbClient *client, const char *principal);
 extern rfbBool InitialiseRFBConnection(rfbClient* client);
 /**
  * Sends format and encoding parameters to the server. Your application can

src/libvncclient/rfbclient.c

@@ -102,6 +102,25 @@ rfbClientLogProc rfbClientErr=rfbDefaultClientLog;
 
 rfbClientProtocolExtension* rfbClientExtensions = NULL;
 
+static rfbBool
+SetOptionalClientString(char **dst, const char *value)
+{
+	char *copy = NULL;
+
+	if (!dst)
+		return FALSE;
+
+	if (value) {
+		copy = strdup(value);
+		if (!copy)
+			return FALSE;
+	}
+
+	free(*dst);
+	*dst = copy;
+	return TRUE;
+}
+
 static rfbBool
 SupportsARDAuthScheme(uint8_t authScheme)
 {
@@ -987,6 +1006,30 @@ SetClientAuthSchemes(rfbClient* client,const uint32_t *authSchemes, int size)
   }
 }
 
+rfbBool
+rfbClientSetARDAuthRealm(rfbClient *client, const char *realm)
+{
+  if (!client)
+    return FALSE;
+  return SetOptionalClientString(&client->ardAuthRealm, realm);
+}
+
+rfbBool
+rfbClientSetARDAuthClientPrincipal(rfbClient *client, const char *principal)
+{
+  if (!client)
+    return FALSE;
+  return SetOptionalClientString(&client->ardAuthClientPrincipal, principal);
+}
+
+rfbBool
+rfbClientSetARDAuthServicePrincipal(rfbClient *client, const char *principal)
+{
+  if (!client)
+    return FALSE;
+  return SetOptionalClientString(&client->ardAuthServicePrincipal, principal);
+}
+
 /*
  * InitialiseRFBConnection.
  */

src/libvncclient/vncviewer.c

@@ -396,6 +396,9 @@ rfbClient* rfbGetClient(int bitsPerSample,int samplesPerPixel,
   client->listen6Sock = RFB_INVALID_SOCKET;
   client->listen6Address = NULL;
   client->clientAuthSchemes = NULL;
+  client->ardAuthRealm = NULL;
+  client->ardAuthClientPrincipal = NULL;
+  client->ardAuthServicePrincipal = NULL;
 
 #ifdef LIBVNCSERVER_HAVE_SASL
   client->GetSASLMechanism = NULL;
@@ -588,6 +591,9 @@ void rfbClientCleanup(rfbClient* client) {
   free(client->serverHost);
   free(client->destHost);
   free(client->clientAuthSchemes);
+  free(client->ardAuthRealm);
+  free(client->ardAuthClientPrincipal);
+  free(client->ardAuthServicePrincipal);
   free(client->rcSource);
   free(client->rcMask);