*/
#include <stdio.h>
#include <stdlib.h>
+#include <string.h>
#include <limits.h>
+#include <pwd.h>
#include "common.h"
-enum {
- FLAG_TYPEMASK = 0x03,
- USER_TYPE_NORMAL = 0x00,
- USER_TYPE_COKE = 0x01,
- USER_TYPE_WHEEL = 0x02,
- USER_TYPE_GOD = 0x03
-};
+// === PROTOTYPES ===
+static int GetUnixID(const char *Username);
// === GLOBALS ===
tUser *gaBank_Users;
FILE *gBank_File;
// === CODE ===
-int Bank_GetUserByUnixID(int UnixID)
+int Bank_GetUserByName(const char *Username)
{
- int i;
+ 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;
}
return gaBank_Users[ID].Balance;
}
+int Bank_GetUserFlags(int ID)
+{
+ if( ID < 0 || ID >= giBank_NumUsers )
+ return INT_MIN;
+
+ return gaBank_Users[ID].Flags;
+}
+
int Bank_AlterUserBalance(int ID, int Delta)
{
// Sanity
int Bank_GetMinAllowedBalance(int ID)
{
if( ID < 0 || ID >= giBank_NumUsers )
- return -1;
+ return 0;
- switch( gaBank_Users[ID].Flags & FLAG_TYPEMASK )
+// printf("gaBank_Users[%i].Flags = 0x%x\n", ID, gaBank_Users[ID].Flags);
+
+ switch( gaBank_Users[ID].Flags & USER_FLAG_TYPEMASK )
{
case USER_TYPE_NORMAL: return 0;
case USER_TYPE_COKE: return -2000;
}
}
-int Bank_GetUserUnixID(int ID)
-{
- if( ID < 0 || ID >= giBank_NumUsers )
- return -1;
-
- return gaBank_Users[ID].UnixID;
-}
-
/**
* \brief Create a new user in our database
*/
-int Bank_AddUser(int UnixID)
+int Bank_AddUser(const char *Username)
{
void *tmp;
+ int uid = GetUnixID(Username);
// Can has moar space plz?
tmp = realloc(gaBank_Users, (giBank_NumUsers+1)*sizeof(gaBank_Users[0]));
gaBank_Users = tmp;
// Crete new user
- gaBank_Users[giBank_NumUsers].UnixID = UnixID;
+ gaBank_Users[giBank_NumUsers].UnixID = uid;
gaBank_Users[giBank_NumUsers].Balance = 0;
gaBank_Users[giBank_NumUsers].Flags = 0;
+ if( strcmp(Username, COKEBANK_DEBT_ACCT) == 0 ) {
+ gaBank_Users[giBank_NumUsers].Flags = USER_TYPE_GOD; // No minium
+ }
+ else if( strcmp(Username, "root") == 0 ) {
+ gaBank_Users[giBank_NumUsers].Flags = USER_TYPE_GOD; // No minium
+ }
+
// 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);
return 0;
}
+
+// ---
+// Unix user dependent code
+// TODO: Modify to keep its own list of usernames
+// ---
+char *Bank_GetUserName(int ID)
+{
+ struct passwd *pwd;
+
+ if( ID < 0 || ID >= giBank_NumUsers )
+ return NULL;
+
+ if( gaBank_Users[ID].UnixID == -1 )
+ return strdup(COKEBANK_SALES_ACCT);
+
+ if( gaBank_Users[ID].UnixID == -2 )
+ return strdup(COKEBANK_DEBT_ACCT);
+
+ pwd = getpwuid(gaBank_Users[ID].UnixID);
+ if( !pwd ) return NULL;
+
+ return strdup(pwd->pw_name);
+}
+
+static int GetUnixID(const char *Username)
+{
+ int uid;
+
+ if( strcmp(Username, COKEBANK_SALES_ACCT) == 0 ) { // Pseudo account that sales are made into
+ uid = -1;
+ }
+ else if( strcmp(Username, COKEBANK_DEBT_ACCT) == 0 ) { // Pseudo acount that money is added from
+ uid = -2;
+ }
+ else {
+ struct passwd *pwd;
+ // Get user ID
+ pwd = getpwnam(Username);
+ if( !pwd ) return -1;
+ uid = pwd->pw_uid;
+ }
+ return uid;
+}