static int split_fields(char *part, int *n, struct xlstr **strs) {
int i, res;
char *c, *fld;
if(part==NULL || *part=='\0' || strs==NULL)
return -1;
*n = 0;
*strs = 0;
c = part;
while (*c) {
res = get_next_part(&c, &fld, FLD_DELIM, 1);
if (res < 0) return res;
(*n)++;
}
*strs = pkg_malloc( (*n)*sizeof(**strs));
if (*strs==NULL) {
ERR(MODULE_NAME": split_fields: not enough pkg memory\n");
return E_OUT_OF_MEM;
}
memset(*strs, 0, (*n)*sizeof(**strs));
i = 0;
c = part;
while (*c) {
res = get_next_part(&c, &(*strs)[i].s, FLD_DELIM, 0);
if (res < 0) return res;
trim_apostr(&(*strs)[i].s);
i++;
}
return 0;
}
/* Opens the file with the given NAME.
Returns the new file if successful or a null pointer
otherwise.
Fails if no file named NAME exists,
or if an internal memory allocation fails. */
struct inode *
filesys_open (const char *name)
{
struct dir *search_dir = get_cwd(name);
if (search_dir == NULL) {
return NULL;
}
struct inode *inode = NULL;
if (name[0] == '/') {
inode = dir_get_inode(search_dir);
}
char *part = malloc(NAME_MAX + 1);
if (part == NULL) {
return false;
}
memset(part, 0, NAME_MAX + 1);
int retrieved_next_part;
for (retrieved_next_part = get_next_part(part, &name); retrieved_next_part > 0;
retrieved_next_part = get_next_part(part, &name)) {
if (dir_lookup (search_dir, part, &inode)) {
if (!inode_is_dir(inode)) {
break;
} else {
dir_close(search_dir);
search_dir = dir_open(inode);
if (search_dir == NULL) {
free(part);
return false;
}
}
} else {
inode = NULL;
break;
}
}
if (inode != NULL && inode_is_dir(inode) && get_next_part(part, &name) == 0) {
inode = inode_reopen(inode);
}
dir_close(search_dir);
if (inode != NULL && get_next_part(part, &name) != 0) {
inode_close(inode);
inode = NULL;
}
free(part);
return inode;
}
uint8_t internal_printf(int fd, char *fmt, va_list ap, uint32_t *ret)
{
char str[PRINTF_BUFFER_SIZE_MAX];
uint32_t str_index;
t_buffer buffer;
str_index = 0;
while (*fmt != 0)
{
ft_memset(buffer.data, 0, INTERNAL_DATA_SIZE_MAX);
buffer.size = 0;
fmt += get_next_part(&buffer, fmt, ap);
ft_memcpy(str + str_index, buffer.data, buffer.size);
str_index += buffer.size;
if (str_index + INTERNAL_DATA_SIZE_MAX >= PRINTF_BUFFER_SIZE_MAX)
{
write(fd, str, str_index);
*ret += str_index;
str_index = 0;
}
}
if (str_index)
{
write(fd, str, str_index);
*ret += str_index;
}
return (1);
}
/* Creates a file named NAME with the given INITIAL_SIZE.
Returns true if successful, false otherwise.
Fails if a file named NAME already exists,
or if internal memory allocation fails. */
bool
filesys_create (const char *name, off_t initial_size, uint32_t is_dir)
{
block_sector_t inode_sector = (block_sector_t) -1;
struct inode *inode = NULL;
struct dir *search_dir = get_cwd(name);
if (search_dir == NULL) {
return false;
}
char *part = malloc(NAME_MAX + 1);
if (part == NULL) {
return false;
}
memset(part, 0, NAME_MAX + 1);
int retrieved_next_part;
for (retrieved_next_part = get_next_part(part, &name); retrieved_next_part > 0;
retrieved_next_part = get_next_part(part, &name)) {
if (dir_lookup (search_dir, part, &inode)) {
if (!inode_is_dir(inode)) {
break;
} else {
dir_close(search_dir);
search_dir = dir_open(inode);
if (search_dir == NULL) {
free(part);
return false;
}
}
} else {
inode = NULL;
break;
}
}
if (inode != NULL || get_next_part(part, &name) != 0) {
if (inode != NULL && !inode_is_dir(inode)) {
inode_close(inode);
}
dir_close(search_dir);
free(part);
return false;
}
bool success = false;
if (is_dir) {
block_sector_t parent_sector = inode_get_inumber(dir_get_inode(search_dir));
success = (search_dir != NULL
&& free_map_allocate (1, &inode_sector)
&& dir_create (inode_sector, initial_size, parent_sector)
&& dir_add (search_dir, part, inode_sector));
} else {
success = (search_dir != NULL
&& free_map_allocate (1, &inode_sector)
&& inode_create (inode_sector, initial_size, is_dir)
&& dir_add (search_dir, part, inode_sector));
}
if (!success)
free_map_release (inode_sector, 1);
dir_close (search_dir);
free(part);
return success;
}
/* Deletes the file named NAME.
Returns true if successful, false on failure.
Fails if no file named NAME exists,
or if an internal memory allocation fails. */
bool
filesys_remove (const char *name)
{
bool success = false;
struct inode *inode = NULL;
struct dir *search_dir = get_cwd(name);
struct dir *parent_dir = NULL;
if (search_dir == NULL) {
return false;
}
parent_dir = dir_reopen(search_dir);
char *part = malloc(NAME_MAX + 1);
if (part == NULL) {
return false;
}
memset(part, 0, NAME_MAX + 1);
int retrieved_next_part;
for (retrieved_next_part = get_next_part(part, &name); retrieved_next_part > 0;
retrieved_next_part = get_next_part(part, &name)) {
if (dir_lookup (search_dir, part, &inode)) {
dir_close(parent_dir);
parent_dir = dir_reopen(search_dir);
if (parent_dir == NULL) {
dir_close(search_dir);
free(part);
return false;
}
if (!inode_is_dir(inode)) {
break;
} else {
dir_close(search_dir);
search_dir = dir_open(inode);
if (search_dir == NULL) {
dir_close(parent_dir);
free(part);
return false;
}
}
} else {
inode = NULL;
break;
}
}
if (inode == NULL || get_next_part(part, &name) != 0) {
if (inode != NULL && !inode_is_dir(inode)) {
inode_close(inode);
}
dir_close(parent_dir);
dir_close(search_dir);
free(part);
return false;
}
if (parent_dir == NULL || search_dir == NULL) {
dir_close(search_dir);
dir_close(parent_dir);
free(part);
return false;
}
if (inode_is_dir(inode)) {
char buffer[NAME_MAX + 1];
if (inode_is_open(inode) || dir_readdir(search_dir, buffer)) {
success = false;
} else {
inode_close(inode);
success = dir_remove (parent_dir, part);
}
} else {
inode_close(inode);
success = dir_remove(parent_dir, part);
}
dir_close(search_dir);
dir_close(parent_dir);
free(part);
return success;
}
static int parse_ops(char* act_s, struct dbops_action** action, int has_name) {
int res = 0, i;
char *c, *s, *part;
static int query_no = 0;
s = act_s;
*action = pkg_malloc(sizeof(**action));
if (!*action) return E_OUT_OF_MEM;
memset(*action, 0, sizeof(**action));
(*action)->query_no = query_no++;
eat_spaces(s);
c = s;
eat_alphanum(c);
if (has_name) {
char *c2;
c2 = c;
eat_spaces(c2);
if (c != s && *c2 == '=') {
*c = '\0';
if (find_action_by_name(s, -1) != NULL) {
ERR(MODULE_NAME": parse_ops: duplicate query name: %s\n", s);
return E_CFG;
}
(*action)->query_name = s;
s = c2+1;
eat_spaces(s);
c = s;
eat_alphanum(c);
}
else {
ERR(MODULE_NAME": parse_ops: query_no: %d, valid query name not found in '%s'\n%s\n%s\n", (*action)->query_no, s, c, c2);
return E_CFG;
}
}
if (c[0] == ':' && c[1] == '/' && c[2] == '/') { /* database part is optional */
for (c=s; *c!=':'; c++) {
*c = tolower(*c); /* _type_://user:host/database_name/ */
}
(*action)->db_url = s;
s = c+1;
while (*s == '/') s++;
res = get_next_part(&s, &part, PART_DELIM, 1); /* type://_user:host_/database_name/ */
if (res < 0) return res;
res = get_next_part(&s, &part, PART_DELIM, 0); /* type://user:host/_database_name_/ */
if (res < 0) return res;
}
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
for (c = part; *c && *c != PART_DELIM; c++) {
if (*c == ' ') {
(*action)->is_raw_query = 1;
*c = '\0';
break;
}
}
if (strcasecmp(part, "select") == 0)
(*action)->operation = OPEN_QUERY_OPS;
else if (strcasecmp(part, "insert") == 0)
(*action)->operation = INSERT_OPS;
else if (strcasecmp(part, "update") == 0)
(*action)->operation = UPDATE_OPS;
else if (strcasecmp(part, "replace") == 0)
(*action)->operation = REPLACE_OPS;
else if (strcasecmp(part, "delete") == 0)
(*action)->operation = DELETE_OPS;
else {
if ((*action)->is_raw_query) *c = ' ';
ERR(MODULE_NAME": parse_ops: query: %s(%d), unknown type of query '%s'\n", (*action)->query_name, (*action)->query_no, part);
return E_CFG;
}
if ((*action)->is_raw_query) {
*c = ' ';
(*action)->raw.s = part;
(*action)->table.s = part;
}
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
if (!(*action)->is_raw_query) {
if (!*part) {
ERR(MODULE_NAME": parse_ops: query: %s(%d), table not specified near '%s' in '%s'\n", (*action)->query_name, (*action)->query_no, s, act_s);
return E_CFG;
}
trim_apostr(&part);
(*action)->table.s = part;
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
switch ((*action)->operation) {
case OPEN_QUERY_OPS:
case UPDATE_OPS:
case REPLACE_OPS:
case INSERT_OPS:
res = split_fields(part, &(*action)->field_count, &(*action)->fields);
if (res < 0) return res;
if ((*action)->field_count == 0) {
ERR(MODULE_NAME": parse_ops: query: %s(%d), no field specified near '%s' ?n '%s'\n", (*action)->query_name, (*action)->query_no, part, act_s);
return E_CFG;
}
break;
case DELETE_OPS:
res = split_fields(part, &(*action)->where_count, &(*action)->wheres);
if (res < 0) return res;
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
res = split_fields(part, &(*action)->op_count, &(*action)->ops);
if (res < 0) return res;
break;
default:;
}
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
switch ((*action)->operation) {
case OPEN_QUERY_OPS:
case UPDATE_OPS:
res = split_fields(part, &(*action)->where_count, &(*action)->wheres);
if (res < 0) return res;
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
res = split_fields(part, &(*action)->op_count, &(*action)->ops);
if (res < 0) return res;
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
switch ((*action)->operation) {
case OPEN_QUERY_OPS:
if (*part) {
(*action)->order.s = part;
}
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
break;
default:;
}
break;
default:
;
}
}
/* values */
res = split_fields(part, &(*action)->value_count, &(*action)->values);
if (res < 0) return res;
if ((*action)->value_count) {
(*action)->value_types = (int*)pkg_malloc(sizeof(int) * (*action)->value_count);
if ((*action)->value_types == NULL) {
ERR(MODULE_NAME": No memory left\n");
return -1;
}
for (i=0; i<(*action)->value_count; i++) {
(*action)->value_types[i] = DB_CSTR; // DB_NONE; /* let decide db driver itself, FIXME: until jjanak changes then default type is string */
res = get_type(&(*action)->values[i].s, &(*action)->value_types[i]);
if (res < 0) return res;
}
}
/* extra options */
res = get_next_part(&s, &part, PART_DELIM, 0);
if (res < 0) return res;
(*action)->extra_ops_count = 0;
c = part;
while (*c) {
char *fld;
res = get_next_part(&c, &fld, FLD_DELIM, 1);
if (res < 0) return res;
(*action)->extra_ops_count++;
}
if ((*action)->extra_ops_count > 0) {
(*action)->extra_ops = pkg_malloc( (*action)->extra_ops_count*sizeof(*(*action)->extra_ops));
if (!(*action)->extra_ops) {
ERR(MODULE_NAME": parse_ops: not enough pkg memory\n");
return E_OUT_OF_MEM;
}
memset((*action)->extra_ops, 0, (*action)->extra_ops_count*sizeof(*(*action)->extra_ops));
i = 0;
c = part;
while (*c) {
char *fld;
res = get_next_part(&c, &fld, FLD_DELIM, 0);
if (res < 0) return res;
/* name=[i|s]:value */
(*action)->extra_ops[i].name = fld;
eat_alphanum(fld);
if (*fld != '=') {
ERR(MODULE_NAME": parse_ops: query: %s(%d), bad extra parameter format in '%s'\n", (*action)->query_name, (*action)->query_no, (*action)->extra_ops[i].name);
return E_CFG;
}
*fld = '\0';
fld++;
while (*fld==' ' || *fld=='\t') fld++;
(*action)->extra_ops[i].type = DB_NONE;
res = get_type(&fld, &(*action)->extra_ops[i].type);
if (res < 0) return res;
trim_apostr(&fld);
(*action)->extra_ops[i].value = fld;
DEBUG(MODULE_NAME": extra_ops #%d, name='%s', type=%d, val='%s'\n", i, (*action)->extra_ops[i].name, (*action)->extra_ops[i].type, (*action)->extra_ops[i].value);
i++;
}
}
if (*s) {
ERR(MODULE_NAME": parse_ops: query: %s(%d), too many parameters/parts, remaining '%s' in '%s'\n", (*action)->query_name, (*action)->query_no, s, act_s);
return E_CFG;
}
if ((*action)->is_raw_query) {
DEBUG(MODULE_NAME": query: %s(%d) oper:%d database:'%s' query:'%s' value#:%d extra_ops#:%d\n", (*action)->query_name, (*action)->query_no, (*action)->operation, (*action)->db_url, (*action)->raw.s, (*action)->value_count, (*action)->extra_ops_count);
}
else {
/* check num of fields */
if ((((*action)->operation==OPEN_QUERY_OPS)?0:(*action)->field_count)+(*action)->where_count != (*action)->value_count) {
ERR(MODULE_NAME": parse_ops: query: %s(%d), number of values does not correspond to number of fields (%d+%d!=%d) in '%s'\n", (*action)->query_name, (*action)->query_no, ((*action)->operation==OPEN_QUERY_OPS)?0:(*action)->field_count, (*action)->where_count, (*action)->value_count, act_s);
return E_CFG;
}
DEBUG(MODULE_NAME": query_no:%d oper:%d database:'%s' table:'%s' 'field#:'%d' where#:'%d' order:'%s' value#:%d extra_ops#:%d\n", (*action)->query_no, (*action)->operation, (*action)->db_url, (*action)->table.s, (*action)->field_count, (*action)->where_count, (*action)->order.s, (*action)->value_count, (*action)->extra_ops_count);
}
return 0;
}
/* timer_id=route_no,interval_ms[,"slow"|"fast"[,"enable"]] */
static int declare_timer(modparam_t type, char* param) {
int n;
unsigned int route_no, interval, enabled, flags;
struct timer_action *pa;
char *p, *save_p, c, *timer_name;
str s;
timer_name = 0;
save_p = p = param;
eat_alphanum(p);
if (*p != '=' || p == save_p) goto err;
*p = '\0';
timer_name = save_p;
p++;
if (find_action_by_name(pkg_timer_actions, timer_name, -1) != NULL) {
ERR(MODULE_NAME": declare_timer: timer '%s' already exists\n", timer_name);
return E_CFG;
}
save_p = p;
if (!get_next_part(&p, &s, ',')) goto err;
c = s.s[s.len];
s.s[s.len] = '\0';
n = route_get(&main_rt, s.s);
s.s[s.len] = c;
if (n == -1) goto err;
route_no = n;
save_p = p;
if (!get_next_part(&p, &s, ',')) goto err;
if (str2int(&s, &interval) < 0) goto err;
save_p = p;
flags = 0;
if (get_next_part(&p, &s, ',')) {
if (s.len == 4 && strncasecmp(s.s, "FAST", 4)==0)
flags = F_TIMER_FAST;
else if (s.len == 4 && strncasecmp(s.s, "SLOW", 4)==0)
;
else goto err;
}
save_p = p;
enabled = 0;
if (get_next_part(&p, &s, ',')) {
if (s.len == 6 && strncasecmp(s.s, "ENABLE", 6)==0)
enabled = 1;
else goto err;
}
pa = pkg_malloc(sizeof(*pa)); /* cannot use shmmem here! */
if (!pa) {
ERR(MODULE_NAME": cannot allocate timer data\n");
return E_OUT_OF_MEM;
}
memset(pa, 0, sizeof(*pa));
pa->timer_name = timer_name;
pa->route_no = route_no;
pa->interval = interval;
pa->enable_on_start = enabled;
pa->flags = flags;
pa->next = pkg_timer_actions;
pkg_timer_actions = pa;
return 0;
err:
ERR(MODULE_NAME": declare_timer: timer_name: '%s', error near '%s'\n", timer_name, save_p);
return E_CFG;
}