utils/utils.c

   1 /*
   2  * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
   3  * Copyright (C) 2013 John Crispin <blogic@openwrt.org>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU Lesser General Public License version 2.1
   7  * as published by the Free Software Foundation
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  */
  14
  15 #include <libubox/avl.h>
  16 #include <libubox/avl-cmp.h>
  17 #include "utils.h"
  18 #include <asm-generic/setup.h>
  19 #include <regex.h>
  20 #include <unistd.h>
  21 #include <sys/types.h>
  22 #include <sys/stat.h>
  23 #include <fcntl.h>
  24
  25 void
  26 __blobmsg_list_init(struct blobmsg_list *list, int offset, int len, blobmsg_list_cmp cmp)
  27 {
  28         avl_init(&list->avl, avl_strcmp, false, NULL);
  29         list->node_offset = offset;
  30         list->node_len = len;
  31         list->cmp = cmp;
  32 }
  33
  34 int
  35 blobmsg_list_fill(struct blobmsg_list *list, void *data, int len, bool array)
  36 {
  37         struct avl_tree *tree = &list->avl;
  38         struct blobmsg_list_node *node;
  39         struct blob_attr *cur;
  40         void *ptr;
  41         int count = 0;
  42         int rem = len;
  43
  44         __blob_for_each_attr(cur, data, rem) {
  45                 if (!blobmsg_check_attr(cur, !array))
  46                         continue;
  47
  48                 ptr = calloc(1, list->node_len);
  49                 if (!ptr)
  50                         return -1;
  51
  52                 node = (void *) ((char *)ptr + list->node_offset);
  53                 if (array)
  54                         node->avl.key = blobmsg_data(cur);
  55                 else
  56                         node->avl.key = blobmsg_name(cur);
  57                 node->data = cur;
  58                 if (avl_insert(tree, &node->avl)) {
  59                         free(ptr);
  60                         continue;
  61                 }
  62
  63                 count++;
  64         }
  65
  66         return count;
  67 }
  68
  69 void
  70 blobmsg_list_move(struct blobmsg_list *list, struct blobmsg_list *src)
  71 {
  72         struct blobmsg_list_node *node, *tmp;
  73         void *ptr;
  74
  75         avl_remove_all_elements(&src->avl, node, avl, tmp) {
  76                 if (avl_insert(&list->avl, &node->avl)) {
  77                         ptr = ((char *) node - list->node_offset);
  78                         free(ptr);
  79                 }
  80         }
  81 }
  82
  83 void
  84 blobmsg_list_free(struct blobmsg_list *list)
  85 {
  86         struct blobmsg_list_node *node, *tmp;
  87         void *ptr;
  88
  89         avl_remove_all_elements(&list->avl, node, avl, tmp) {
  90                 ptr = ((char *) node - list->node_offset);
  91                 free(ptr);
  92         }
  93 }
  94
  95 bool
  96 blobmsg_list_equal(struct blobmsg_list *l1, struct blobmsg_list *l2)
  97 {
  98         struct blobmsg_list_node *n1, *n2;
  99         int count = l1->avl.count;
 100
 101         if (count != l2->avl.count)
 102                 return false;
 103
 104         n1 = avl_first_element(&l1->avl, n1, avl);
 105         n2 = avl_first_element(&l2->avl, n2, avl);
 106
 107         while (count-- > 0) {
 108                 int len;
 109
 110                 len = blob_len(n1->data);
 111                 if (len != blob_len(n2->data))
 112                         return false;
 113
 114                 if (memcmp(n1->data, n2->data, len) != 0)
 115                         return false;
 116
 117                 if (l1->cmp && !l1->cmp(n1, n2))
 118                         return false;
 119
 120                 if (!count)
 121                         break;
 122
 123                 n1 = avl_next_element(n1, avl);
 124                 n2 = avl_next_element(n2, avl);
 125         }
 126
 127         return true;
 128 }
 129
 130 char* get_cmdline_val(const char* name, char* out, int len)
 131 {
 132         char pattern[CMDLINE_SIZE + 1];
 133         char line[CMDLINE_SIZE + 1];
 134         char *res = NULL, *tty;
 135         int r, fd;
 136         regex_t pat_cmdline;
 137         regmatch_t matches[2];
 138
 139         fd = open("/proc/cmdline", O_RDONLY);
 140         if (fd < 0)
 141                 return NULL;
 142
 143         r = read(fd, line, CMDLINE_SIZE);
 144         if ( r <= 0 ) {
 145                 close(fd);
 146                 return NULL;
 147         }
 148         line[r] = '\0';
 149         close(fd);
 150
 151         sprintf( pattern, "%s=([^ \n]*)", name);
 152         regcomp(&pat_cmdline, pattern, REG_EXTENDED);
 153         if (!regexec(&pat_cmdline, line, 2, matches, 0)) {
 154                 line[matches[1].rm_eo] = '\0';
 155                 tty = (line + matches[1].rm_so);
 156                 strncpy(out, tty, len);
 157                 tty[len-1] = '\0';
 158                 res = out;
 159         }
 160
 161         regfree(&pat_cmdline);
 162
 163         return res;
 164 }