X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fclient%2Fmain.c;h=f30ce43b7f5d7f5f08e30557bb5925fb9e674f48;hb=2b718e476dbff288ddf4e948024bcb6c92cc3281;hp=1f38f7ed9ebd13b972eaf6520794ece034fb7426;hpb=1e05819dd64ab9d8232f85645cfb4abb0bad406e;p=tpg%2Fopendispense2.git diff --git a/src/client/main.c b/src/client/main.c index 1f38f7e..f30ce43 100644 --- a/src/client/main.c +++ b/src/client/main.c @@ -15,6 +15,7 @@ #include #include #include +#include #include // close #include // gethostbyname @@ -22,255 +23,540 @@ #include #include #include +#include // SHA1 + +#define USE_NCURSES_INTERFACE 0 +#define DEBUG_TRACE_SERVER 0 +#define USE_AUTOAUTH 1 // === TYPES === typedef struct sItem { - char *Ident; + char *Type; + int ID; char *Desc; int Price; } tItem; // === PROTOTYPES === + int main(int argc, char *argv[]); +void ShowUsage(void); +// --- GUI --- int ShowNCursesUI(void); +void ShowItemAt(int Row, int Col, int Width, int Index); void PrintAlign(int Row, int Col, int Width, const char *Left, char Pad1, const char *Mid, char Pad2, const char *Right, ...); - - int sendf(int Socket, const char *Format, ...); +// --- Coke Server Communication --- int OpenConnection(const char *Host, int Port); -void Authenticate(int Socket); + int Authenticate(int Socket); +void PopulateItemList(int Socket); + int Dispense_ItemInfo(int Socket, const char *Type, int ID); + int DispenseItem(int Socket, const char *Type, int ID); + int Dispense_AlterBalance(int Socket, const char *Username, int Ammount, const char *Reason); + int Dispense_SetBalance(int Socket, const char *Username, int Balance, const char *Reason); + int Dispense_Give(int Socket, const char *Username, int Ammount, const char *Reason); + int Dispense_Donate(int Socket, int Ammount, const char *Reason); + int Dispense_EnumUsers(int Socket); + int Dispense_ShowUser(int Socket, const char *Username); +void _PrintUserLine(const char *Line); + int Dispense_AddUser(int Socket, const char *Username); + int Dispense_SetUserType(int Socket, const char *Username, const char *TypeString); +// --- Helpers --- +char *ReadLine(int Socket); + int sendf(int Socket, const char *Format, ...); char *trim(char *string); int RunRegex(regex_t *regex, const char *string, int nMatches, regmatch_t *matches, const char *errorMessage); void CompileRegex(regex_t *regex, const char *pattern, int flags); // === GLOBALS === -char *gsDispenseServer = "localhost"; +char *gsDispenseServer = "heathred"; int giDispensePort = 11020; + tItem *gaItems; int giNumItems; -regex_t gArrayRegex; -regex_t gItemRegex; +regex_t gArrayRegex, gItemRegex, gSaltRegex, gUserInfoRegex, gUserItemIdentRegex; + int gbIsAuthenticated = 0; + +char *gsItemPattern; //!< Item pattern +char *gsEffectiveUser; //!< '-u' Dispense as another user + int gbUseNCurses = 0; //!< '-G' Use the NCurses GUI? + int gbDryRun = 0; //!< '-n' Read-only + int giMinimumBalance = INT_MIN; //!< '-m' Minumum balance for `dispense acct` + int giMaximumBalance = INT_MAX; //!< '-M' Maximum balance for `dispense acct` // === CODE === int main(int argc, char *argv[]) { int sock; - int i, responseCode, len; + int i; char buffer[BUFSIZ]; // -- Create regular expressions // > Code Type Count ... CompileRegex(&gArrayRegex, "^([0-9]{3})\\s+([A-Za-z]+)\\s+([0-9]+)", REG_EXTENDED); // // > Code Type Ident Price Desc - CompileRegex(&gItemRegex, "^([0-9]{3})\\s+(.+?)\\s+(.+?)\\s+([0-9]+)\\s+(.+)$", REG_EXTENDED); - - // Connect to server - sock = OpenConnection(gsDispenseServer, giDispensePort); - if( sock < 0 ) return -1; + CompileRegex(&gItemRegex, "^([0-9]{3})\\s+([A-Za-z]+)\\s+([A-Za-z]+):([0-9]+)\\s+([0-9]+)\\s+(.+)$", REG_EXTENDED); + // > Code 'SALT' salt + CompileRegex(&gSaltRegex, "^([0-9]{3})\\s+([A-Za-z]+)\\s+(.+)$", REG_EXTENDED); + // > Code 'User' Username Balance Flags + CompileRegex(&gUserInfoRegex, "^([0-9]{3})\\s+([A-Za-z]+)\\s+([^ ]+)\\s+(-?[0-9]+)\\s+(.+)$", REG_EXTENDED); + // > Item Ident + CompileRegex(&gUserItemIdentRegex, "^([A-Za-z]+):([0-9]+)$", REG_EXTENDED); - // Determine what to do - if( argc > 1 ) + // Parse Arguments + for( i = 1; i < argc; i ++ ) { - if( strcmp(argv[1], "acct") == 0 ) + char *arg = argv[i]; + + if( arg[0] == '-' ) + { + switch(arg[1]) + { + case 'h': + case '?': + ShowUsage(); + return 0; + + case 'm': // Minimum balance + if( i + 1 >= argc ) { + fprintf(stderr, "%s: -m takes an argument\n", argv[0]); + ShowUsage(); + } + giMinimumBalance = atoi(argv[++i]); + break; + case 'M': // Maximum balance + if( i + 1 >= argc ) { + fprintf(stderr, "%s: -M takes an argument\n", argv[0]); + ShowUsage(); + } + giMaximumBalance = atoi(argv[++i]); + break; + + case 'u': // Override User + if( i + 1 >= argc ) { + fprintf(stderr, "%s: -u takes an argument\n", argv[0]); + ShowUsage(); + } + gsEffectiveUser = argv[++i]; + break; + + case 'H': // Override remote host + if( i + 1 >= argc ) { + fprintf(stderr, "%s: -H takes an argument\n", argv[0]); + ShowUsage(); + } + gsDispenseServer = argv[++i]; + break; + case 'P': // Override remote port + if( i + 1 >= argc ) { + fprintf(stderr, "%s: -P takes an argument\n", argv[0]); + ShowUsage(); + } + giDispensePort = atoi(argv[++i]); + break; + + case 'G': // Use GUI + gbUseNCurses = 1; + break; + case 'n': // Dry Run / read-only + gbDryRun = 1; + break; + } + + continue; + } + + // + // `dispense acct` + // - + if( strcmp(arg, "acct") == 0 ) + { + // Connect to server + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + + // List accounts? + if( i + 1 == argc ) { + Dispense_EnumUsers(sock); + return 0; + } + + // argv[i+1]: Username + + // Alter account? + if( i + 2 < argc ) + { + if( i + 3 >= argc ) { + fprintf(stderr, "Error: `dispense acct' needs a reason\n"); + exit(1); + } + + // Authentication required + if( Authenticate(sock) ) + return -1; + + // argv[i+1]: Username + // argv[i+2]: Ammount + // argv[i+3]: Reason + + if( argv[i+2][0] == '=' ) { + // Set balance + if( argv[i+2][1] != '0' && atoi(&argv[i+2][1]) == 0 ) { + fprintf(stderr, "Error: Invalid balance to be set\n"); + exit(1); + } + + Dispense_SetBalance(sock, argv[i+1], atoi(argv[i+2]+1), argv[i+3]); + } + else { + // Alter balance + Dispense_AlterBalance(sock, argv[i+1], atoi(argv[i+2]), argv[i+3]); + } + } + + // Show user information + Dispense_ShowUser(sock, argv[i+1]); + + close(sock); + return 0; + } + // + // `dispense give` + // - "Here, have some money." + if( strcmp(arg, "give") == 0 ) + { + if( i + 3 >= argc ) { + fprintf(stderr, "`dispense give` takes three arguments\n"); + ShowUsage(); + return -1; + } + + // argv[i+1]: Destination + // argv[i+2]: Ammount + // argv[i+3]: Reason + + // Connect to server + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + + // Authenticate + if( Authenticate(sock) ) + return -1; + + Dispense_Give(sock, argv[i+1], atoi(argv[i+2]), argv[i+3]); + return 0; + } + // + // `dispense user` + // - User administration (Admin Only) + if( strcmp(arg, "user") == 0 ) + { + // Check argument count + if( i + 1 >= argc ) { + fprintf(stderr, "Error: `dispense user` requires arguments\n"); + ShowUsage(); + exit(1); + } + + // Connect to server + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + + // Attempt authentication + if( Authenticate(sock) ) + return -1; + + // Add new user? + if( strcmp(argv[i+1], "add") == 0 ) + { + if( i + 2 >= argc ) { + fprintf(stderr, "Error: `dispense user add` requires an argument\n"); + ShowUsage(); + exit(1); + } + + Dispense_AddUser(sock, argv[i+2]); + } + // Update a user + else if( strcmp(argv[i+1], "type") == 0 ) + { + if( i + 3 >= argc ) { + fprintf(stderr, "Error: `dispense user type` requires two arguments\n"); + ShowUsage(); + exit(1); + } + + Dispense_SetUserType(sock, argv[i+2], argv[i+3]); + } + else + { + fprintf(stderr, "Error: Unknown sub-command for `dispense user`\n"); + ShowUsage(); + exit(1); + } + return 0; + } + + // Donation! + if( strcmp(arg, "donate") == 0 ) { - // Alter account - // List accounts + // Check argument count + if( i + 2 >= argc ) { + fprintf(stderr, "Error: `dispense donate` requires two arguments\n"); + ShowUsage(); + exit(1); + } + + // Connect to server + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + + // Attempt authentication + if( Authenticate(sock) ) + return -1; + + // Do donation + Dispense_Donate(sock, atoi(argv[i+1]), argv[i+1]); + return 0; } + + else { + // Item name / pattern + gsItemPattern = arg; + break; + } } - - // Ask server for stock list - send(sock, "ENUM_ITEMS\n", 11, 0); - len = recv(sock, buffer, BUFSIZ-1, 0); - buffer[len] = '\0'; - - trim(buffer); - printf("Output: %s\n", buffer); + // Connect to server + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + + // Get items + PopulateItemList(sock); - responseCode = atoi(buffer); - if( responseCode != 201 ) - { - fprintf(stderr, "Unknown response from dispense server (Response Code %i)\n", responseCode); - return -1; - } + // Disconnect from server + close(sock); - // Get item list + if( gsItemPattern ) { - char *itemType, *itemStart; - int count; - regmatch_t matches[4]; - - // Expected format: 201 Items ... - RunRegex(&gArrayRegex, buffer, 4, matches, "Malformed server response"); - - itemType = &buffer[ matches[2].rm_so ]; buffer[ matches[2].rm_eo ] = '\0'; - count = atoi( &buffer[ matches[3].rm_so ] ); - - // Check array type - if( strcmp(itemType, "Items") != 0 ) { - // What the?! - fprintf(stderr, "Unexpected array type, expected 'Items', got '%s'\n", - itemType); - return -1; + regmatch_t matches[3]; + // Door (hard coded) + if( strcmp(gsItemPattern, "door") == 0 ) + { + // Connect, Authenticate, dispense and close + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + Authenticate(sock); + DispenseItem(sock, "door", 0); + close(sock); + return 0; } - - itemStart = &buffer[ matches[3].rm_eo ]; - - gaItems = malloc( count * sizeof(tItem) ); - - for( giNumItems = 0; giNumItems < count && itemStart; giNumItems ++ ) + // Item id (:) + else if( RunRegex(&gUserItemIdentRegex, gsItemPattern, 3, matches, NULL) == 0 ) { - char *next = strchr( ++itemStart, ' ' ); - if( next ) *next = '\0'; - gaItems[giNumItems].Ident = strdup(itemStart); - itemStart = next; + char *ident; + int id; + + // Get and finish ident + ident = gsItemPattern + matches[1].rm_so; + gsItemPattern[matches[1].rm_eo] = '\0'; + // Get ID + id = atoi( gsItemPattern + matches[2].rm_so ); + + // Connect, Authenticate, dispense and close + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + + Dispense_ItemInfo(sock, ident, id); + + Authenticate(sock); + DispenseItem(sock, ident, id); + close(sock); + return 0; + } + // Item number (6 = coke) + else if( strcmp(gsItemPattern, "0") == 0 || atoi(gsItemPattern) > 0 ) + { + i = atoi(gsItemPattern); + } + // Item prefix + else + { + int j; + int best = -1; + for( i = 0; i < giNumItems; i ++ ) + { + // Prefix match (with case-insensitive match) + for( j = 0; gsItemPattern[j]; j ++ ) + { + if( gaItems[i].Desc[j] == gsItemPattern[j] ) + continue; + if( tolower(gaItems[i].Desc[j]) == tolower(gsItemPattern[j]) ) + continue; + break; + } + // Check if the prefix matched + if( gsItemPattern[j] != '\0' ) + continue; + + // Prefect match + if( gaItems[i].Desc[j] == '\0' ) { + best = i; + break; + } + + // Only one match allowed + if( best == -1 ) { + best = i; + } + else { + // TODO: Allow ambiguous matches? + } + } + + if( best == -1 ) + { + fprintf(stderr, "No item matches the passed string\n"); + return 1; + } + + i = best; } } - - // Get item information - for( i = 0; i < giNumItems; i ++ ) + else if( gbUseNCurses ) { - regmatch_t matches[6]; - - // Print item Ident - printf("%2i %s\t", i, gaItems[i].Ident); - - // Get item info - sendf(sock, "ITEM_INFO %s\n", gaItems[i].Ident); - len = recv(sock, buffer, BUFSIZ-1, 0); - buffer[len] = '\0'; - trim(buffer); - - responseCode = atoi(buffer); - if( responseCode != 202 ) { - fprintf(stderr, "Unknown response from dispense server (Response Code %i)\n", responseCode); - return -1; - } - - RunRegex(&gItemRegex, buffer, 6, matches, "Malformed server response"); - - buffer[ matches[3].rm_eo ] = '\0'; - - gaItems[i].Price = atoi( buffer + matches[4].rm_so ); - gaItems[i].Desc = strdup( buffer + matches[5].rm_so ); - - printf("%3i %s\n", gaItems[i].Price, gaItems[i].Desc); + i = ShowNCursesUI(); } - - Authenticate(sock); - - // and choose what to dispense - // TODO: ncurses interface (with separation between item classes) - // - Hmm... that would require standardising the item ID to be : - // Oh, why not :) - - #if 1 - i = ShowNCursesUI(); - #else - - for(;;) + else { - char *buf; - - fgets(buffer, BUFSIZ, stdin); - - buf = trim(buffer); - - if( buf[0] == 'q' ) break; - - i = atoi(buf); - - printf("buf = '%s', atoi(buf) = %i\n", buf, i); - - if( i != 0 || buf[0] == '0' ) + // Very basic dispense interface + for( i = 0; i < giNumItems; i ++ ) { + printf("%2i %s:%i\t%3i %s\n", i, gaItems[i].Type, gaItems[i].ID, + gaItems[i].Price, gaItems[i].Desc); + } + printf(" q Quit\n"); + for(;;) { - printf("i = %i\n", i); + char *buf; + + i = -1; + + fgets(buffer, BUFSIZ, stdin); + + buf = trim(buffer); + + if( buf[0] == 'q' ) break; - if( i < 0 || i >= giNumItems ) { - printf("Bad item (should be between 0 and %i)\n", giNumItems); - continue; + i = atoi(buf); + + if( i != 0 || buf[0] == '0' ) + { + if( i < 0 || i >= giNumItems ) { + printf("Bad item %i (should be between 0 and %i)\n", i, giNumItems); + continue; + } + break; } - break; } } - #endif + + // Check for a valid item ID if( i >= 0 ) - { - // Dispense! - sendf(sock, "DISPENSE %s\n", gaItems[i].Ident); - - len = recv(sock, buffer, BUFSIZ-1, 0); - buffer[len] = '\0'; - trim(buffer); + { + // Connect, Authenticate, dispense and close + sock = OpenConnection(gsDispenseServer, giDispensePort); + if( sock < 0 ) return -1; + + Dispense_ItemInfo(sock, gaItems[i].Type, gaItems[i].ID); - responseCode = atoi(buffer); - switch( responseCode ) - { - case 200: - printf("Dispense OK\n"); - break; - case 401: - printf("Not authenticated\n"); - break; - case 402: - printf("Insufficient balance\n"); - break; - case 406: - printf("Bad item name, bug report\n"); - break; - case 500: - printf("Item failed to dispense, is the slot empty?\n"); - break; - default: - printf("Unknown response code %i\n", responseCode); - break; - } + Authenticate(sock); + DispenseItem(sock, gaItems[i].Type, gaItems[i].ID); + close(sock); } - close(sock); - return 0; } -void ShowItemAt(int Row, int Col, int Width, int Index) +void ShowUsage(void) { - int _x, _y, times; - - move( Row, Col ); - - if( Index < 0 || Index >= giNumItems ) { - printw("%02i OOR", Index); - return ; - } - printw("%02i %s", Index, gaItems[Index].Desc); - - getyx(stdscr, _y, _x); - // Assumes max 4 digit prices - times = Width - 4 - (_x - Col); // TODO: Better handling for large prices - while(times--) addch(' '); - printw("%4i", gaItems[Index].Price); + printf( + "Usage:\n" + " == Everyone ==\n" + " dispense\n" + " Show interactive list\n" + " dispense \n" + " Dispense named item\n" + " dispense give \"\"\n" + " Give money to another user\n" + " dispense donate \"\"\n" + " Donate to the club\n" + " == Coke members == \n" + " dispense acct []\n" + " Show user balances\n" + " dispense acct [+-] \"\"\n" + " Alter a account value\n" + " == Dispense administrators ==\n" + " dispense acct = \"\"\n" + " Set an account balance\n" + " dispense user add \n" + " Create new coke account (Admins only)\n" + " dispense user type \n" + " Alter a user's flags\n" + " is a comma-separated list of user, coke, admin or disabled\n" + " Flags are removed by preceding the name with '-' or '!'\n" + "\n" + "General Options:\n" + " -u \n" + " Set a different user (Coke members only)\n" + " -h / -?\n" + " Show help text\n" + " -G\n" + " Use alternate GUI\n" + " -m \n" + " -M \n" + " Set the Maximum/Minimum balances shown in `dispense acct`\n" + ); } +// ------------------- +// --- NCurses GUI --- +// ------------------- /** + * \brief Render the NCurses UI */ int ShowNCursesUI(void) { + // TODO: ncurses interface (with separation between item classes) + // - Hmm... that would require standardising the item ID to be : + // Oh, why not :) int ch; int i, times; int xBase, yBase; - const int displayMinWidth = 34; + const int displayMinWidth = 40; const int displayMinItems = 8; char *titleString = "Dispense"; int itemCount = displayMinItems; int itemBase = 0; + int currentItem = 0; + int ret = -2; // -2: Used for marking "no return yet" - int height = itemCount + 3; - int width = displayMinWidth; + int height, width; // Enter curses mode initscr(); raw(); noecho(); + // Get item count + // - 6: randomly chosen (Need at least 3) + itemCount = LINES - 6; + if( itemCount > giNumItems ) + itemCount = giNumItems; + + // Get dimensions + height = itemCount + 3; + width = displayMinWidth; + + // Get positions xBase = COLS/2 - width/2; yBase = LINES/2 - height/2; @@ -283,23 +569,29 @@ int ShowNCursesUI(void) for( i = 0; i < itemCount; i ++ ) { move( yBase + 1 + i, xBase ); - addch('|'); - addch(' '); + + if( currentItem == itemBase + i ) { + printw("| -> "); + } + else { + printw("| "); + } // Check for ... row + // - Oh god, magic numbers! if( i == 0 && itemBase > 0 ) { printw(" ..."); - times = width - 1 - 8; + times = width-1 - 8 - 3; while(times--) addch(' '); } else if( i == itemCount - 1 && itemBase < giNumItems - itemCount ) { printw(" ..."); - times = width - 1 - 8; + times = width-1 - 8 - 3; while(times--) addch(' '); } // Show an item else { - ShowItemAt( yBase + 1 + i, xBase + 2, width - 4, itemBase + i); + ShowItemAt( yBase + 1 + i, xBase + 5, width - 7, itemBase + i); addch(' '); } @@ -340,11 +632,19 @@ int ShowNCursesUI(void) switch(ch) { case 'B': - if( itemBase < giNumItems - (itemCount) ) + //if( itemBase < giNumItems - (itemCount) ) + // itemBase ++; + if( currentItem < giNumItems - 1 ) + currentItem ++; + if( itemBase + itemCount - 1 <= currentItem && itemBase + itemCount < giNumItems ) itemBase ++; break; case 'A': - if( itemBase > 0 ) + //if( itemBase > 0 ) + // itemBase --; + if( currentItem > 0 ) + currentItem --; + if( itemBase + 1 > currentItem && itemBase > 0 ) itemBase --; break; } @@ -354,7 +654,18 @@ int ShowNCursesUI(void) } } else { - break; + switch(ch) + { + case '\n': + ret = currentItem; + break; + case 'q': + ret = -1; // -1: Return with no dispense + break; + } + + // Check if the return value was changed + if( ret != -2 ) break; } } @@ -362,10 +673,52 @@ int ShowNCursesUI(void) // Leave endwin(); - return -1; + return ret; +} + +/** + * \brief Show item \a Index at (\a Col, \a Row) + * \note Part of the NCurses UI + */ +void ShowItemAt(int Row, int Col, int Width, int Index) +{ + int _x, _y, times; + char *name; + int price; + + move( Row, Col ); + + if( Index < 0 || Index >= giNumItems ) { + name = "OOR"; + price = 0; + } + else { + name = gaItems[Index].Desc; + price = gaItems[Index].Price; + } + + printw("%02i %s", Index, name); + + getyx(stdscr, _y, _x); + // Assumes max 4 digit prices + times = Width - 4 - (_x - Col); // TODO: Better handling for large prices + while(times--) addch(' '); + printw("%4i", price); } -void PrintAlign(int Row, int Col, int Width, const char *Left, char Pad1, const char *Mid, char Pad2, const char *Right, ...) +/** + * \brief Print a three-part string at the specified position (formatted) + * \note NCurses UI Helper + * + * Prints \a Left on the left of the area, \a Right on the righthand side + * and \a Mid in the middle of the area. These are padded with \a Pad1 + * between \a Left and \a Mid, and \a Pad2 between \a Mid and \a Right. + * + * ::printf style format codes are allowed in \a Left, \a Mid and \a Right, + * and the arguments to these are read in that order. + */ +void PrintAlign(int Row, int Col, int Width, const char *Left, char Pad1, + const char *Mid, char Pad2, const char *Right, ...) { int lLen, mLen, rLen; int times; @@ -414,26 +767,9 @@ void PrintAlign(int Row, int Col, int Width, const char *Left, char Pad1, const } } -// === HELPERS === -int sendf(int Socket, const char *Format, ...) -{ - va_list args; - int len; - - va_start(args, Format); - len = vsnprintf(NULL, 0, Format, args); - va_end(args); - - { - char buf[len+1]; - va_start(args, Format); - vsnprintf(buf, len+1, Format, args); - va_end(args); - - return send(Socket, buf, len, 0); - } -} - +// --------------------- +// --- Coke Protocol --- +// --------------------- int OpenConnection(const char *Host, int Port) { struct hostent *host; @@ -459,16 +795,24 @@ int OpenConnection(const char *Host, int Port) return -1; } - #if USE_AUTOAUTH + if( geteuid() == 0 ) { + int i; struct sockaddr_in localAddr; memset(&localAddr, 0, sizeof(localAddr)); localAddr.sin_family = AF_INET; // IPv4 - localAddr.sin_port = 1023; // IPv4 - // Attempt to bind to low port for autoauth - bind(sock, &localAddr, sizeof(localAddr)); + + // Loop through all the top ports until one is avaliable + for( i = 1001; i < 1024; i ++) + { + localAddr.sin_port = htons(i); // IPv4 + // Attempt to bind to low port for autoauth + if( bind(sock, (struct sockaddr*)&localAddr, sizeof(localAddr)) == 0 ) + break; + } + if( i == 1024 ) + printf("Warning: AUTOAUTH unavaliable\n"); } - #endif if( connect(sock, (struct sockaddr *) &serverAddr, sizeof(serverAddr)) < 0 ) { fprintf(stderr, "Failed to connect to server\n"); @@ -478,11 +822,20 @@ int OpenConnection(const char *Host, int Port) return sock; } -void Authenticate(int Socket) +/** + * \brief Authenticate with the server + * \return Boolean Failure + */ +int Authenticate(int Socket) { struct passwd *pwd; - char buf[512]; + char *buf; int responseCode; + char salt[32]; + int i; + regmatch_t matches[4]; + + if( gbIsAuthenticated ) return 0; // Get user name pwd = getpwuid( getuid() ); @@ -491,27 +844,808 @@ void Authenticate(int Socket) sendf(Socket, "AUTOAUTH %s\n", pwd->pw_name); // Check if it worked - recv(Socket, buf, 511, 0); - trim(buf); + buf = ReadLine(Socket); responseCode = atoi(buf); switch( responseCode ) { - case 200: // Authenticated, return :) - return ; - case 401: // Untrusted, attempt password authentication + case 200: // Autoauth succeeded, return + free(buf); break; - case 404: // Bad Username - fprintf(stderr, "Bad Username '%s'\n", pwd->pw_name); - exit(-1); - default: - fprintf(stderr, "Unkown response code %i from server\n", responseCode); - printf("%s\n", buf); - exit(-1); - } - printf("%s\n", buf); -} + case 401: // Untrusted, attempt password authentication + free(buf); + + sendf(Socket, "USER %s\n", pwd->pw_name); + printf("Using username %s\n", pwd->pw_name); + + buf = ReadLine(Socket); + + // TODO: Get Salt + // Expected format: 100 SALT ... + // OR : 100 User Set + RunRegex(&gSaltRegex, buf, 4, matches, "Malformed server response"); + responseCode = atoi(buf); + if( responseCode != 100 ) { + fprintf(stderr, "Unknown repsonse code %i from server\n%s\n", responseCode, buf); + free(buf); + return -1; // ERROR + } + + // Check for salt + if( memcmp( buf+matches[2].rm_so, "SALT", matches[2].rm_eo - matches[2].rm_so) == 0) { + // Store it for later + memcpy( salt, buf + matches[3].rm_so, matches[3].rm_eo - matches[3].rm_so ); + salt[ matches[3].rm_eo - matches[3].rm_so ] = 0; + } + free(buf); + + // Give three attempts + for( i = 0; i < 3; i ++ ) + { + int ofs = strlen(pwd->pw_name)+strlen(salt); + char tmpBuf[42]; + char tmp[ofs+20]; + char *pass = getpass("Password: "); + uint8_t h[20]; + + // Create hash string + // + strcpy(tmp, pwd->pw_name); + strcat(tmp, salt); + SHA1( (unsigned char*)pass, strlen(pass), h ); + memcpy(tmp+ofs, h, 20); + + // Hash all that + SHA1( (unsigned char*)tmp, ofs+20, h ); + sprintf(tmpBuf, "%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x", + h[ 0], h[ 1], h[ 2], h[ 3], h[ 4], h[ 5], h[ 6], h[ 7], h[ 8], h[ 9], + h[10], h[11], h[12], h[13], h[14], h[15], h[16], h[17], h[18], h[19] + ); + + // Send password + sendf(Socket, "PASS %s\n", tmpBuf); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + // Auth OK? + if( responseCode == 200 ) break; + // Bad username/password + if( responseCode == 401 ) continue; + + fprintf(stderr, "Unknown repsonse code %i from server\n%s\n", responseCode, buf); + free(buf); + return -1; + } + free(buf); + if( i == 3 ) + return 2; // 2 = Bad Password + break; + + case 404: // Bad Username + fprintf(stderr, "Bad Username '%s'\n", pwd->pw_name); + free(buf); + return 1; + + default: + fprintf(stderr, "Unkown response code %i from server\n", responseCode); + printf("%s\n", buf); + free(buf); + return -1; + } + + // Set effective user + if( gsEffectiveUser ) { + sendf(Socket, "SETEUSER %s\n", gsEffectiveUser); + + buf = ReadLine(Socket); + responseCode = atoi(buf); + + switch(responseCode) + { + case 200: + printf("Running as '%s' by '%s'\n", gsEffectiveUser, pwd->pw_name); + break; + + case 403: + printf("Only coke members can use `dispense -u`\n"); + free(buf); + return -1; + + case 404: + printf("Invalid user selected\n"); + free(buf); + return -1; + + default: + fprintf(stderr, "Unkown response code %i from server\n", responseCode); + printf("%s\n", buf); + free(buf); + exit(-1); + } + + free(buf); + } + + gbIsAuthenticated = 1; + + return 0; +} + + +/** + * \brief Fill the item information structure + * \return Boolean Failure + */ +void PopulateItemList(int Socket) +{ + char *buf; + int responseCode; + + char *itemType, *itemStart; + int count, i; + regmatch_t matches[4]; + + // Ask server for stock list + send(Socket, "ENUM_ITEMS\n", 11, 0); + buf = ReadLine(Socket); + + //printf("Output: %s\n", buf); + + responseCode = atoi(buf); + if( responseCode != 201 ) { + fprintf(stderr, "Unknown response from dispense server (Response Code %i)\n", responseCode); + exit(-1); + } + + // - Get item list - + + // Expected format: + // 201 Items + // 202 Item + RunRegex(&gArrayRegex, buf, 4, matches, "Malformed server response"); + + itemType = &buf[ matches[2].rm_so ]; buf[ matches[2].rm_eo ] = '\0'; + count = atoi( &buf[ matches[3].rm_so ] ); + + // Check array type + if( strcmp(itemType, "Items") != 0 ) { + // What the?! + fprintf(stderr, "Unexpected array type, expected 'Items', got '%s'\n", + itemType); + exit(-1); + } + + itemStart = &buf[ matches[3].rm_eo ]; + + free(buf); + + giNumItems = count; + gaItems = malloc( giNumItems * sizeof(tItem) ); + + // Fetch item information + for( i = 0; i < giNumItems; i ++ ) + { + regmatch_t matches[7]; + + // Get item info + buf = ReadLine(Socket); + responseCode = atoi(buf); + + if( responseCode != 202 ) { + fprintf(stderr, "Unknown response from dispense server (Response Code %i)\n", responseCode); + exit(-1); + } + + RunRegex(&gItemRegex, buf, 7, matches, "Malformed server response"); + + buf[ matches[3].rm_eo ] = '\0'; + + gaItems[i].Type = strdup( buf + matches[3].rm_so ); + gaItems[i].ID = atoi( buf + matches[4].rm_so ); + gaItems[i].Price = atoi( buf + matches[5].rm_so ); + gaItems[i].Desc = strdup( buf + matches[6].rm_so ); + + free(buf); + } + + // Read end of list + buf = ReadLine(Socket); + responseCode = atoi(buf); + + if( responseCode != 200 ) { + fprintf(stderr, "Unknown response from dispense server %i\n'%s'", + responseCode, buf + ); + exit(-1); + } + + free(buf); +} + + +/** + * \brief Get information on an item + * \return Boolean Failure + */ +int Dispense_ItemInfo(int Socket, const char *Type, int ID) +{ + int responseCode; + char *buf; + regmatch_t matches[7]; + char *type, *desc; + int id, price; + + // Dispense! + sendf(Socket, "ITEM_INFO %s:%i\n", Type, ID); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + switch( responseCode ) + { + case 202: + break; + + case 406: + printf("Bad item name\n"); + free(buf); + return 1; + + default: + printf("Unknown response code %i ('%s')\n", responseCode, buf); + free(buf); + return -1; + } + + + RunRegex(&gItemRegex, buf, 7, matches, "Malformed server response"); + + buf[ matches[3].rm_eo ] = '\0'; + + type = buf + matches[3].rm_so; buf[matches[3].rm_eo] = '\0'; + id = atoi( buf + matches[4].rm_so ); + price = atoi( buf + matches[5].rm_so ); + desc = buf + matches[6].rm_so; buf[matches[6].rm_eo] = '\0'; + + printf("%8s:%-2i %2i.%02i %s\n", type, id, price/100, price%100, desc); + + free(buf); + + return 0; +} + +/** + * \brief Dispense an item + * \return Boolean Failure + */ +int DispenseItem(int Socket, const char *Type, int ID) +{ + int ret, responseCode; + char *buf; + + // Check for a dry run + if( gbDryRun ) { + printf("Dry Run - No action\n"); + return 0; + } + + // Dispense! + sendf(Socket, "DISPENSE %s:%i\n", Type, ID); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + switch( responseCode ) + { + case 200: + printf("Dispense OK\n"); + ret = 0; + break; + case 401: + printf("Not authenticated\n"); + ret = 1; + break; + case 402: + printf("Insufficient balance\n"); + ret = 1; + break; + case 406: + printf("Bad item name\n"); + ret = 1; + break; + case 500: + printf("Item failed to dispense, is the slot empty?\n"); + ret = 1; + break; + case 501: + printf("Dispense not possible (slot empty/permissions)\n"); + ret = 1; + break; + default: + printf("Unknown response code %i ('%s')\n", responseCode, buf); + ret = -2; + break; + } + + free(buf); + return ret; +} + +/** + * \brief Alter a user's balance + */ +int Dispense_AlterBalance(int Socket, const char *Username, int Ammount, const char *Reason) +{ + char *buf; + int responseCode; + + // Check for a dry run + if( gbDryRun ) { + printf("Dry Run - No action\n"); + return 0; + } + + sendf(Socket, "ADD %s %i %s\n", Username, Ammount, Reason); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + free(buf); + + switch(responseCode) + { + case 200: return 0; // OK + case 402: + fprintf(stderr, "Insufficient balance\n"); + return 1; + case 403: // Not in coke + fprintf(stderr, "You are not in coke (sucker)\n"); + return 1; + case 404: // Unknown user + fprintf(stderr, "Unknown user '%s'\n", Username); + return 2; + default: + fprintf(stderr, "Unknown response code %i\n", responseCode); + return -1; + } + + return -1; +} + +/** + * \brief Set a user's balance + * \note Only avaliable to dispense admins + */ +int Dispense_SetBalance(int Socket, const char *Username, int Balance, const char *Reason) +{ + char *buf; + int responseCode; + + // Check for a dry run + if( gbDryRun ) { + printf("Dry Run - No action\n"); + return 0; + } + + sendf(Socket, "SET %s %i %s\n", Username, Balance, Reason); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + free(buf); + + switch(responseCode) + { + case 200: return 0; // OK + case 403: // Not in coke + fprintf(stderr, "You are not an admin\n"); + return 1; + case 404: // Unknown user + fprintf(stderr, "Unknown user '%s'\n", Username); + return 2; + default: + fprintf(stderr, "Unknown response code %i\n", responseCode); + return -1; + } + + return -1; +} + +/** + * \brief Give money to another user + */ +int Dispense_Give(int Socket, const char *Username, int Ammount, const char *Reason) +{ + char *buf; + int responseCode; + + if( Ammount < 0 ) { + printf("Sorry, you can only give, you can't take.\n"); + return -1; + } + + // Fast return on zero + if( Ammount == 0 ) { + printf("Are you actually going to give any?\n"); + return 0; + } + + // Check for a dry run + if( gbDryRun ) { + printf("Dry Run - No action\n"); + return 0; + } + + sendf(Socket, "GIVE %s %i %s\n", Username, Ammount, Reason); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + free(buf); + + switch(responseCode) + { + case 200: return 0; // OK + + case 402: + fprintf(stderr, "Insufficient balance\n"); + return 1; + + case 404: // Unknown user + fprintf(stderr, "Unknown user '%s'\n", Username); + return 2; + + default: + fprintf(stderr, "Unknown response code %i\n", responseCode); + return -1; + } + + return -1; +} + + +/** + * \brief Donate money to the club + */ +int Dispense_Donate(int Socket, int Ammount, const char *Reason) +{ + char *buf; + int responseCode; + + if( Ammount < 0 ) { + printf("Sorry, you can only give, you can't take.\n"); + return -1; + } + + // Fast return on zero + if( Ammount == 0 ) { + printf("Are you actually going to give any?\n"); + return 0; + } + + // Check for a dry run + if( gbDryRun ) { + printf("Dry Run - No action\n"); + return 0; + } + + sendf(Socket, "DONATE %i %s\n", Ammount, Reason); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + free(buf); + + switch(responseCode) + { + case 200: return 0; // OK + + case 402: + fprintf(stderr, "Insufficient balance\n"); + return 1; + + default: + fprintf(stderr, "Unknown response code %i\n", responseCode); + return -1; + } + + return -1; +} + +/** + * \brief Enumerate users + */ +int Dispense_EnumUsers(int Socket) +{ + char *buf; + int responseCode; + int nUsers; + regmatch_t matches[4]; + + if( giMinimumBalance != INT_MIN ) { + if( giMaximumBalance != INT_MAX ) { + sendf(Socket, "ENUM_USERS min_balance:%i max_balance:%i\n", giMinimumBalance, giMaximumBalance); + } + else { + sendf(Socket, "ENUM_USERS min_balance:%i\n", giMinimumBalance); + } + } + else { + if( giMaximumBalance != INT_MAX ) { + sendf(Socket, "ENUM_USERS max_balance:%i\n", giMaximumBalance); + } + else { + sendf(Socket, "ENUM_USERS\n"); + } + } + buf = ReadLine(Socket); + responseCode = atoi(buf); + + switch(responseCode) + { + case 201: break; // Ok, length follows + + default: + fprintf(stderr, "Unknown response code %i\n%s\n", responseCode, buf); + free(buf); + return -1; + } + + // Get count (not actually used) + RunRegex(&gArrayRegex, buf, 4, matches, "Malformed server response"); + nUsers = atoi( buf + matches[3].rm_so ); + printf("%i users returned\n", nUsers); + + // Free string + free(buf); + + // Read returned users + do { + buf = ReadLine(Socket); + responseCode = atoi(buf); + + if( responseCode != 202 ) break; + + _PrintUserLine(buf); + free(buf); + } while(responseCode == 202); + + // Check final response + if( responseCode != 200 ) { + fprintf(stderr, "Unknown response code %i\n%s\n", responseCode, buf); + free(buf); + return -1; + } + + free(buf); + + return 0; +} + +int Dispense_ShowUser(int Socket, const char *Username) +{ + char *buf; + int responseCode, ret; + + sendf(Socket, "USER_INFO %s\n", Username); + buf = ReadLine(Socket); + + responseCode = atoi(buf); + + switch(responseCode) + { + case 202: + _PrintUserLine(buf); + ret = 0; + break; + + case 404: + printf("Unknown user '%s'\n", Username); + ret = 1; + break; + + default: + fprintf(stderr, "Unknown response code %i '%s'\n", responseCode, buf); + ret = -1; + break; + } + + free(buf); + + return ret; +} + +void _PrintUserLine(const char *Line) +{ + regmatch_t matches[6]; + int bal; + + RunRegex(&gUserInfoRegex, Line, 6, matches, "Malformed server response"); + // 3: Username + // 4: Balance + // 5: Flags + { + int usernameLen = matches[3].rm_eo - matches[3].rm_so; + char username[usernameLen + 1]; + int flagsLen = matches[5].rm_eo - matches[5].rm_so; + char flags[flagsLen + 1]; + + memcpy(username, Line + matches[3].rm_so, usernameLen); + username[usernameLen] = '\0'; + memcpy(flags, Line + matches[5].rm_so, flagsLen); + flags[flagsLen] = '\0'; + + bal = atoi(Line + matches[4].rm_so); + printf("%-15s: $%4i.%02i (%s)\n", username, bal/100, abs(bal)%100, flags); + } +} + +int Dispense_AddUser(int Socket, const char *Username) +{ + char *buf; + int responseCode, ret; + + // Check for a dry run + if( gbDryRun ) { + printf("Dry Run - No action\n"); + return 0; + } + + sendf(Socket, "USER_ADD %s\n", Username); + + buf = ReadLine(Socket); + responseCode = atoi(buf); + + switch(responseCode) + { + case 200: + printf("User '%s' added\n", Username); + ret = 0; + break; + + case 403: + printf("Only wheel can add users\n"); + ret = 1; + break; + + case 404: + printf("User '%s' already exists\n", Username); + ret = 0; + break; + + default: + fprintf(stderr, "Unknown response code %i '%s'\n", responseCode, buf); + ret = -1; + break; + } + + free(buf); + + return ret; +} + +int Dispense_SetUserType(int Socket, const char *Username, const char *TypeString) +{ + char *buf; + int responseCode, ret; + + // Check for a dry run + if( gbDryRun ) { + printf("Dry Run - No action\n"); + return 0; + } + + // TODO: Pre-validate the string + + sendf(Socket, "USER_FLAGS %s %s\n", Username, TypeString); + + buf = ReadLine(Socket); + responseCode = atoi(buf); + + switch(responseCode) + { + case 200: + printf("User '%s' updated\n", Username); + ret = 0; + break; + + case 403: + printf("Only wheel can modify users\n"); + ret = 1; + break; + + case 404: + printf("User '%s' does not exist\n", Username); + ret = 0; + break; + + case 407: + printf("Flag string is invalid\n"); + ret = 0; + break; + + default: + fprintf(stderr, "Unknown response code %i '%s'\n", responseCode, buf); + ret = -1; + break; + } + + free(buf); + + return ret; +} + +// --------------- +// --- Helpers --- +// --------------- +char *ReadLine(int Socket) +{ + static char buf[BUFSIZ]; + static int bufPos = 0; + static int bufValid = 0; + int len; + char *newline = NULL; + int retLen = 0; + char *ret = malloc(10); + + #if DEBUG_TRACE_SERVER + printf("ReadLine: "); + #endif + fflush(stdout); + + ret[0] = '\0'; + + while( !newline ) + { + if( bufValid ) { + len = bufValid; + } + else { + len = recv(Socket, buf+bufPos, BUFSIZ-1-bufPos, 0); + buf[bufPos+len] = '\0'; + } + + newline = strchr( buf+bufPos, '\n' ); + if( newline ) { + *newline = '\0'; + } + + retLen += strlen(buf+bufPos); + ret = realloc(ret, retLen + 1); + strcat( ret, buf+bufPos ); + + if( newline ) { + int newLen = newline - (buf+bufPos) + 1; + bufValid = len - newLen; + bufPos += newLen; + } + if( len + bufPos == BUFSIZ - 1 ) bufPos = 0; + } + + #if DEBUG_TRACE_SERVER + printf("%i '%s'\n", retLen, ret); + #endif + + return ret; +} + +int sendf(int Socket, const char *Format, ...) +{ + va_list args; + int len; + + va_start(args, Format); + len = vsnprintf(NULL, 0, Format, args); + va_end(args); + + { + char buf[len+1]; + va_start(args, Format); + vsnprintf(buf, len+1, Format, args); + va_end(args); + + #if DEBUG_TRACE_SERVER + printf("sendf: %s", buf); + #endif + + return send(Socket, buf, len, 0); + } +} char *trim(char *string) { @@ -536,7 +1670,7 @@ int RunRegex(regex_t *regex, const char *string, int nMatches, regmatch_t *match int ret; ret = regexec(regex, string, nMatches, matches, 0); - if( ret ) { + if( ret && errorMessage ) { size_t len = regerror(ret, regex, NULL, 0); char errorStr[len]; regerror(ret, regex, errorStr, len);