Home | History | Annotate | Download | only in util 
      1 /*
      2  * (c) 2009 Arnaldo Carvalho de Melo <acme (at) redhat.com>
      3  *
      4  * Licensed under the GPLv2.
      5  */
      6 
      7 #include "strlist.h"
      8 #include <errno.h>
      9 #include <stdio.h>
     10 #include <stdlib.h>
     11 #include <string.h>
     12 
     13 static
     14 struct rb_node *strlist__node_new(struct rblist *rblist, const void *entry)
     15 {
     16 	const char *s = entry;
     17 	struct rb_node *rc = NULL;
     18 	struct strlist *strlist = container_of(rblist, struct strlist, rblist);
     19 	struct str_node *snode = malloc(sizeof(*snode));
     20 
     21 	if (snode != NULL) {
     22 		if (strlist->dupstr) {
     23 			s = strdup(s);
     24 			if (s == NULL)
     25 				goto out_delete;
     26 		}
     27 		snode->s = s;
     28 		rc = &snode->rb_node;
     29 	}
     30 
     31 	return rc;
     32 
     33 out_delete:
     34 	free(snode);
     35 	return NULL;
     36 }
     37 
     38 static void str_node__delete(struct str_node *snode, bool dupstr)
     39 {
     40 	if (dupstr)
     41 		free((void *)snode->s);
     42 	free(snode);
     43 }
     44 
     45 static
     46 void strlist__node_delete(struct rblist *rblist, struct rb_node *rb_node)
     47 {
     48 	struct strlist *slist = container_of(rblist, struct strlist, rblist);
     49 	struct str_node *snode = container_of(rb_node, struct str_node, rb_node);
     50 
     51 	str_node__delete(snode, slist->dupstr);
     52 }
     53 
     54 static int strlist__node_cmp(struct rb_node *rb_node, const void *entry)
     55 {
     56 	const char *str = entry;
     57 	struct str_node *snode = container_of(rb_node, struct str_node, rb_node);
     58 
     59 	return strcmp(snode->s, str);
     60 }
     61 
     62 int strlist__add(struct strlist *slist, const char *new_entry)
     63 {
     64 	return rblist__add_node(&slist->rblist, new_entry);
     65 }
     66 
     67 int strlist__load(struct strlist *slist, const char *filename)
     68 {
     69 	char entry[1024];
     70 	int err;
     71 	FILE *fp = fopen(filename, "r");
     72 
     73 	if (fp == NULL)
     74 		return errno;
     75 
     76 	while (fgets(entry, sizeof(entry), fp) != NULL) {
     77 		const size_t len = strlen(entry);
     78 
     79 		if (len == 0)
     80 			continue;
     81 		entry[len - 1] = '\0';
     82 
     83 		err = strlist__add(slist, entry);
     84 		if (err != 0)
     85 			goto out;
     86 	}
     87 
     88 	err = 0;
     89 out:
     90 	fclose(fp);
     91 	return err;
     92 }
     93 
     94 void strlist__remove(struct strlist *slist, struct str_node *snode)
     95 {
     96 	rblist__remove_node(&slist->rblist, &snode->rb_node);
     97 }
     98 
     99 struct str_node *strlist__find(struct strlist *slist, const char *entry)
    100 {
    101 	struct str_node *snode = NULL;
    102 	struct rb_node *rb_node = rblist__find(&slist->rblist, entry);
    103 
    104 	if (rb_node)
    105 		snode = container_of(rb_node, struct str_node, rb_node);
    106 
    107 	return snode;
    108 }
    109 
    110 static int strlist__parse_list_entry(struct strlist *slist, const char *s)
    111 {
    112 	if (strncmp(s, "file://", 7) == 0)
    113 		return strlist__load(slist, s + 7);
    114 
    115 	return strlist__add(slist, s);
    116 }
    117 
    118 int strlist__parse_list(struct strlist *slist, const char *s)
    119 {
    120 	char *sep;
    121 	int err;
    122 
    123 	while ((sep = strchr(s, ',')) != NULL) {
    124 		*sep = '\0';
    125 		err = strlist__parse_list_entry(slist, s);
    126 		*sep = ',';
    127 		if (err != 0)
    128 			return err;
    129 		s = sep + 1;
    130 	}
    131 
    132 	return *s ? strlist__parse_list_entry(slist, s) : 0;
    133 }
    134 
    135 struct strlist *strlist__new(bool dupstr, const char *list)
    136 {
    137 	struct strlist *slist = malloc(sizeof(*slist));
    138 
    139 	if (slist != NULL) {
    140 		rblist__init(&slist->rblist);
    141 		slist->rblist.node_cmp    = strlist__node_cmp;
    142 		slist->rblist.node_new    = strlist__node_new;
    143 		slist->rblist.node_delete = strlist__node_delete;
    144 
    145 		slist->dupstr	 = dupstr;
    146 		if (list && strlist__parse_list(slist, list) != 0)
    147 			goto out_error;
    148 	}
    149 
    150 	return slist;
    151 out_error:
    152 	free(slist);
    153 	return NULL;
    154 }
    155 
    156 void strlist__delete(struct strlist *slist)
    157 {
    158 	if (slist != NULL)
    159 		rblist__delete(&slist->rblist);
    160 }
    161 
    162 struct str_node *strlist__entry(const struct strlist *slist, unsigned int idx)
    163 {
    164 	struct str_node *snode = NULL;
    165 	struct rb_node *rb_node;
    166 
    167 	rb_node = rblist__entry(&slist->rblist, idx);
    168 	if (rb_node)
    169 		snode = container_of(rb_node, struct str_node, rb_node);
    170 
    171 	return snode;
    172 }
    173