Merge branch 'master' of lain.church:emil/probotic

This commit is contained in:
Emil 2023-08-04 02:20:02 -06:00
commit 40da5fbcda
2 changed files with 45 additions and 4 deletions

View File

@ -17,6 +17,7 @@ DECL char * raw(char const * const sql);
DECL char * remind(char * who); DECL char * remind(char * who);
DECL char * slurp(char const * fn); DECL char * slurp(char const * fn);
DECL int is_admin(char const * user); DECL int is_admin(char const * user);
DECL void parse_admins(char * admin_string);
DECL int parse_pair(char const * buf, size_t const len); DECL int parse_pair(char const * buf, size_t const len);
DECL void creds_free_password(void); DECL void creds_free_password(void);
DECL void creds_free_rest(void); DECL void creds_free_rest(void);

View File

@ -26,7 +26,7 @@
#include "help.h" #include "help.h"
#include "parse.h" #include "parse.h"
#define PARAMS_COUNT 5 #define PARAMS_COUNT 6
enum cred_param_ids_e enum cred_param_ids_e
{ {
@ -34,7 +34,8 @@ enum cred_param_ids_e
PASSWORD, PASSWORD,
CHANNEL, CHANNEL,
SERVER, SERVER,
PORT PORT,
ADMINS
}; };
char const * cred_names[] = char const * cred_names[] =
@ -43,7 +44,8 @@ char const * cred_names[] =
"password", "password",
"channel", "channel",
"server", "server",
"port" "port",
"admins"
}; };
size_t const cred_names_len[] = size_t const cred_names_len[] =
@ -52,7 +54,8 @@ size_t const cred_names_len[] =
8, 8,
7, 7,
6, 6,
4 4,
6
}; };
creds_t creds = {0}; creds_t creds = {0};
@ -189,6 +192,10 @@ parse_pair(char const * buf, size_t len)
case CHANNEL: creds.channel = strndup(buf,x); break; case CHANNEL: creds.channel = strndup(buf,x); break;
case SERVER: creds.server = strndup(buf,x); break; case SERVER: creds.server = strndup(buf,x); break;
case PORT: creds.port = atoi(buf); break; case PORT: creds.port = atoi(buf); break;
case ADMINS:
parse_admins(strndup(buf,x));
break;
} }
if (x + 2 < len) if (x + 2 < len)
{ buf += x + 1; } { buf += x + 1; }
@ -203,6 +210,39 @@ parse_pair(char const * buf, size_t len)
return 0; return 0;
} }
/* WARNING: admin_string WILL be changed */
DECL void
parse_admins(char * admin_string)
{
#define ADMIN_LIST_INIT_SIZE 8
/* prealloc with 8 */
size_t current_array_size = ADMIN_LIST_INIT_SIZE;
size_t i = 0;
char * token = NULL;
admins = calloc(ADMIN_LIST_INIT_SIZE, sizeof(char *));
while ((token = strtok(admin_string, " ")))
{
if (++i > current_array_size)
{
/* double the space */
current_array_size *= 2;
admins = realloc(admins, current_array_size);
}
admins[i - 1] = strdup(token);
}
/* set up array end marker for proper clean-up later */
if (i + 1 > current_array_size)
{
admins = realloc(admins, current_array_size + 1);
}
admins[i] = NULL;
}
DECL int DECL int
is_admin(char const * user) is_admin(char const * user)
{ {