X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fcokebank_basic%2Fbank.c;h=56db422105bbe0d59a091f555ed84cea7715677e;hb=f02c7f90fe0e654d5587fa88c8dd8297ef150b54;hp=d78f90f6b542bdd17158431c9fd13d8827052cc8;hpb=229e4bcc0d38421e8baf266d2bf585ad3e88a327;p=tpg%2Fopendispense2.git diff --git a/src/cokebank_basic/bank.c b/src/cokebank_basic/bank.c index d78f90f..56db422 100644 --- a/src/cokebank_basic/bank.c +++ b/src/cokebank_basic/bank.c @@ -13,18 +13,13 @@ #include #include #include +#include #include "common.h" -enum { - FLAG_TYPEMASK = 0x03, - USER_TYPE_NORMAL = 0x00, - USER_TYPE_COKE = 0x01, - USER_TYPE_WHEEL = 0x02, - USER_TYPE_GOD = 0x03 -}; +#define USE_UNIX_GROUPS 1 // === PROTOTYPES === -static int GetUnixID(const char *Name); +static int GetUnixID(const char *Username); // === GLOBALS === tUser *gaBank_Users; @@ -32,13 +27,29 @@ tUser *gaBank_Users; FILE *gBank_File; // === CODE === -int Bank_GetUserByUnixID(int UnixID) +static int Bank_int_WriteEntry(int ID) { - int i; + if( ID < 0 || ID >= giBank_NumUsers ) { + return -1; + } + + // Commit to file + fseek(gBank_File, ID*sizeof(gaBank_Users[0]), SEEK_SET); + fwrite(&gaBank_Users[ID], sizeof(gaBank_Users[0]), 1, gBank_File); + + return 0; +} + +int Bank_GetUserByName(const char *Username) +{ + int i, uid; + + uid = GetUnixID(Username); + // Expensive search :( for( i = 0; i < giBank_NumUsers; i ++ ) { - if( gaBank_Users[i].UnixID == UnixID ) + if( gaBank_Users[i].UnixID == uid ) return i; } @@ -53,51 +64,110 @@ int Bank_GetUserBalance(int ID) return gaBank_Users[ID].Balance; } -int Bank_AlterUserBalance(int ID, int Delta) +int Bank_GetUserFlags(int ID) { - // Sanity if( ID < 0 || ID >= giBank_NumUsers ) return -1; - // Update - gaBank_Users[ID].Balance += Delta; + // root + if( gaBank_Users[ID].UnixID == 0 ) { + gaBank_Users[ID].Flags |= USER_FLAG_WHEEL|USER_FLAG_COKE; + } - // Commit - fseek(gBank_File, ID*sizeof(gaBank_Users[0]), SEEK_SET); - fwrite(&gaBank_Users[ID], sizeof(gaBank_Users[0]), 1, gBank_File); + #if USE_UNIX_GROUPS + // TODO: Implement checking the PAM groups and status instead, then + // fall back on the database. (and update if there is a difference) + if( gaBank_Users[ID].UnixID > 0 ) + { + struct passwd *pwd; + struct group *grp; + int i; + + // Get username + pwd = getpwuid( gaBank_Users[ID].UnixID ); + + // Check for additions to the "coke" group + grp = getgrnam("coke"); + if( grp ) { + for( i = 0; grp->gr_mem[i]; i ++ ) + { + if( strcmp(grp->gr_mem[i], pwd->pw_name) == 0 ) { + gaBank_Users[ID].Flags |= USER_FLAG_COKE; + break ; + } + } + } + + // Check for additions to the "wheel" group + grp = getgrnam("wheel"); + if( grp ) { + for( i = 0; grp->gr_mem[i]; i ++ ) + { + if( strcmp(grp->gr_mem[i], pwd->pw_name) == 0 ) { + gaBank_Users[ID].Flags |= USER_FLAG_WHEEL; + break ; + } + } + } + } + #endif + + return gaBank_Users[ID].Flags; +} + +int Bank_SetUserFlags(int ID, int Mask, int Value) +{ + // Sanity + if( ID < 0 || ID >= giBank_NumUsers ) + return -1; + + // Silently ignore changes to root and meta accounts + if( gaBank_Users[ID].UnixID <= 0 ) return 0; + + gaBank_Users[ID].Flags &= ~Mask; + gaBank_Users[ID].Flags |= Value; + + Bank_int_WriteEntry(ID); return 0; } -int Bank_SetUserBalance(int ID, int Value) +int Bank_AlterUserBalance(int ID, int Delta) { // Sanity if( ID < 0 || ID >= giBank_NumUsers ) return -1; // Update - gaBank_Users[ID].Balance = Value; - - // Commit - fseek(gBank_File, ID*sizeof(gaBank_Users[0]), SEEK_SET); - fwrite(&gaBank_Users[ID], sizeof(gaBank_Users[0]), 1, gBank_File); + gaBank_Users[ID].Balance += Delta; + + Bank_int_WriteEntry(ID); return 0; } int Bank_GetMinAllowedBalance(int ID) { + int flags; if( ID < 0 || ID >= giBank_NumUsers ) - return -1; + return 0; - switch( gaBank_Users[ID].Flags & FLAG_TYPEMASK ) - { - case USER_TYPE_NORMAL: return 0; - case USER_TYPE_COKE: return -2000; - case USER_TYPE_WHEEL: return -10000; - case USER_TYPE_GOD: return INT_MIN; - default: return 0; - } + flags = Bank_GetUserFlags(ID); + + // Internal accounts have no limit + if( (flags & USER_FLAG_INTERNAL) ) + return INT_MIN; + + // Wheel is allowed to go to -$100 + if( (flags & USER_FLAG_WHEEL) ) + return -10000; + + // Coke is allowed to go to -$20 + if( (flags & USER_FLAG_COKE) ) + return -2000; + + // For everyone else, no negative + return 0; } /** @@ -118,12 +188,20 @@ int Bank_AddUser(const char *Username) gaBank_Users[giBank_NumUsers].Balance = 0; gaBank_Users[giBank_NumUsers].Flags = 0; - // Commit to file - fseek(gBank_File, giBank_NumUsers*sizeof(gaBank_Users[0]), SEEK_SET); - fwrite(&gaBank_Users[giBank_NumUsers], sizeof(gaBank_Users[0]), 1, gBank_File); + if( strcmp(Username, COKEBANK_DEBT_ACCT) == 0 ) { + gaBank_Users[giBank_NumUsers].Flags = USER_FLAG_INTERNAL; + } + else if( strcmp(Username, COKEBANK_SALES_ACCT) == 0 ) { + gaBank_Users[giBank_NumUsers].Flags = USER_FLAG_INTERNAL; + } + else if( strcmp(Username, "root") == 0 ) { + gaBank_Users[giBank_NumUsers].Flags = USER_FLAG_WHEEL|USER_FLAG_COKE; + } // Increment count giBank_NumUsers ++; + + Bank_int_WriteEntry(giBank_NumUsers - 1); return 0; } @@ -140,10 +218,10 @@ char *Bank_GetUserName(int ID) return NULL; if( gaBank_Users[ID].UnixID == -1 ) - return strdup(">sales"); + return strdup(COKEBANK_SALES_ACCT); if( gaBank_Users[ID].UnixID == -2 ) - return strdup(">liability"); + return strdup(COKEBANK_DEBT_ACCT); pwd = getpwuid(gaBank_Users[ID].UnixID); if( !pwd ) return NULL; @@ -155,10 +233,10 @@ static int GetUnixID(const char *Username) { int uid; - if( strcmp(Username, ">sales") == 0 ) { // Pseudo account that sales are made into + if( strcmp(Username, COKEBANK_SALES_ACCT) == 0 ) { // Pseudo account that sales are made into uid = -1; } - else if( strcmp(Username, ">liability") == 0 ) { // Pseudo acount that money is added from + else if( strcmp(Username, COKEBANK_DEBT_ACCT) == 0 ) { // Pseudo acount that money is added from uid = -2; } else {