X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fclient%2Fmain.c;fp=src%2Fclient%2Fmain.c;h=da301be8020b6c9b161e44b7e222b5c07e515e63;hb=ee724c0360fac7840d032812b5ab677aa0d15074;hp=444ab34c9aba8880105e3acea515182dab12937f;hpb=afc4569cb98d4cc498e78ebb51767e5521ba2ea8;p=tpg%2Fopendispense2.git

diff --git a/src/client/main.c b/src/client/main.c
index 444ab34..da301be 100644
--- a/src/client/main.c
+++ b/src/client/main.c
@@ -12,7 +12,8 @@
 #include <stdio.h>
 #include <string.h>
 #include <ctype.h>	// isspace
-#include <unistd.h>	// close
+#include <pwd.h>	// getpwuids
+#include <unistd.h>	// close/getuid
 #include <limits.h>	// INT_MIN/INT_MAX
 #include "common.h"
 
@@ -637,6 +638,54 @@ int main(int argc, char *argv[])
 		close(sock);
 		return ret;
 	}
+	// Check a user's pin
+	else if(strcmp(gsTextArgs[0], "pincheck") == 0)
+	{
+		if( giTextArgc < 2 || giTextArgc > 3 ) {
+			fprintf(stderr, "Error: `dispense pincheck` takes one/two arguments\n");
+			ShowUsage();
+			return RV_ARGUMENTS;
+		}
+		struct passwd	*pwd = getpwuid( getuid() );
+		gsUserName = strdup(pwd->pw_name);
+		
+		const char *pin = gsTextArgs[1];
+		const char *user = gsUserName;
+		if( giTextArgc == 3 )
+			user = gsTextArgs[2];
+
+		
+		sock = OpenConnection(gsDispenseServer, giDispensePort);
+		if( sock < 0 )	return RV_SOCKET_ERROR;
+		ret = Authenticate(sock);
+		if(ret)	return ret;
+		
+		ret = DispenseCheckPin(sock, user, pin);
+		
+		close(sock);
+		return ret;
+	}
+	// Update 'your' pin
+	else if(strcmp(gsTextArgs[0], "pinset") == 0)
+	{
+		if( giTextArgc != 2 ) {
+			fprintf(stderr, "Error: `dispense pinset` takes one argument\n");
+			ShowUsage();
+			return RV_ARGUMENTS;
+		}
+		
+		const char *pin = gsTextArgs[1];
+		
+		sock = OpenConnection(gsDispenseServer, giDispensePort);
+		if( sock < 0 )	return RV_SOCKET_ERROR;
+		ret = Authenticate(sock);
+		if(ret)	return ret;
+
+		ret = DispenseSetPin(sock, pin);
+		
+		close(sock);
+		return ret;
+	}
 	// Item name / pattern
 	else
 	{