perf/util/strlist.c

0001 // SPDX-License-Identifier: GPL-2.0-only
0002 /*
0003  * (c) 2009 Arnaldo Carvalho de Melo <acme@redhat.com>
0004  */
0005
0006 #include "strlist.h"
0007 #include <errno.h>
0008 #include <stdio.h>
0009 #include <stdlib.h>
0010 #include <string.h>
0011 #include <unistd.h>
0012 #include <linux/zalloc.h>
0013
0014 static
0015 struct rb_node *strlist__node_new(struct rblist *rblist, const void *entry)
0016 {
0017     const char *s = entry;
0018     struct rb_node *rc = NULL;
0019     struct strlist *strlist = container_of(rblist, struct strlist, rblist);
0020     struct str_node *snode = malloc(sizeof(*snode));
0021
0022     if (snode != NULL) {
0023         if (strlist->dupstr) {
0024             s = strdup(s);
0025             if (s == NULL)
0026                 goto out_delete;
0027         }
0028         snode->s = s;
0029         rc = &snode->rb_node;
0030     }
0031
0032     return rc;
0033
0034 out_delete:
0035     free(snode);
0036     return NULL;
0037 }
0038
0039 static void str_node__delete(struct str_node *snode, bool dupstr)
0040 {
0041     if (dupstr)
0042         zfree((char **)&snode->s);
0043     free(snode);
0044 }
0045
0046 static
0047 void strlist__node_delete(struct rblist *rblist, struct rb_node *rb_node)
0048 {
0049     struct strlist *slist = container_of(rblist, struct strlist, rblist);
0050     struct str_node *snode = container_of(rb_node, struct str_node, rb_node);
0051
0052     str_node__delete(snode, slist->dupstr);
0053 }
0054
0055 static int strlist__node_cmp(struct rb_node *rb_node, const void *entry)
0056 {
0057     const char *str = entry;
0058     struct str_node *snode = container_of(rb_node, struct str_node, rb_node);
0059
0060     return strcmp(snode->s, str);
0061 }
0062
0063 int strlist__add(struct strlist *slist, const char *new_entry)
0064 {
0065     return rblist__add_node(&slist->rblist, new_entry);
0066 }
0067
0068 int strlist__load(struct strlist *slist, const char *filename)
0069 {
0070     char entry[1024];
0071     int err;
0072     FILE *fp = fopen(filename, "r");
0073
0074     if (fp == NULL)
0075         return -errno;
0076
0077     while (fgets(entry, sizeof(entry), fp) != NULL) {
0078         const size_t len = strlen(entry);
0079
0080         if (len == 0)
0081             continue;
0082         entry[len - 1] = '\0';
0083
0084         err = strlist__add(slist, entry);
0085         if (err != 0)
0086             goto out;
0087     }
0088
0089     err = 0;
0090 out:
0091     fclose(fp);
0092     return err;
0093 }
0094
0095 void strlist__remove(struct strlist *slist, struct str_node *snode)
0096 {
0097     rblist__remove_node(&slist->rblist, &snode->rb_node);
0098 }
0099
0100 struct str_node *strlist__find(struct strlist *slist, const char *entry)
0101 {
0102     struct str_node *snode = NULL;
0103     struct rb_node *rb_node = rblist__find(&slist->rblist, entry);
0104
0105     if (rb_node)
0106         snode = container_of(rb_node, struct str_node, rb_node);
0107
0108     return snode;
0109 }
0110
0111 static int strlist__parse_list_entry(struct strlist *slist, const char *s,
0112                      const char *subst_dir)
0113 {
0114     int err;
0115     char *subst = NULL;
0116
0117     if (strncmp(s, "file://", 7) == 0)
0118         return strlist__load(slist, s + 7);
0119
0120     if (subst_dir) {
0121         err = -ENOMEM;
0122         if (asprintf(&subst, "%s/%s", subst_dir, s) < 0)
0123             goto out;
0124
0125         if (access(subst, F_OK) == 0) {
0126             err = strlist__load(slist, subst);
0127             goto out;
0128         }
0129
0130         if (slist->file_only) {
0131             err = -ENOENT;
0132             goto out;
0133         }
0134     }
0135
0136     err = strlist__add(slist, s);
0137 out:
0138     free(subst);
0139     return err;
0140 }
0141
0142 static int strlist__parse_list(struct strlist *slist, const char *s, const char *subst_dir)
0143 {
0144     char *sep;
0145     int err;
0146
0147     while ((sep = strchr(s, ',')) != NULL) {
0148         *sep = '\0';
0149         err = strlist__parse_list_entry(slist, s, subst_dir);
0150         *sep = ',';
0151         if (err != 0)
0152             return err;
0153         s = sep + 1;
0154     }
0155
0156     return *s ? strlist__parse_list_entry(slist, s, subst_dir) : 0;
0157 }
0158
0159 struct strlist *strlist__new(const char *list, const struct strlist_config *config)
0160 {
0161     struct strlist *slist = malloc(sizeof(*slist));
0162
0163     if (slist != NULL) {
0164         bool dupstr = true;
0165         bool file_only = false;
0166         const char *dirname = NULL;
0167
0168         if (config) {
0169             dupstr = !config->dont_dupstr;
0170             dirname = config->dirname;
0171             file_only = config->file_only;
0172         }
0173
0174         rblist__init(&slist->rblist);
0175         slist->rblist.node_cmp    = strlist__node_cmp;
0176         slist->rblist.node_new    = strlist__node_new;
0177         slist->rblist.node_delete = strlist__node_delete;
0178
0179         slist->dupstr    = dupstr;
0180         slist->file_only = file_only;
0181
0182         if (list && strlist__parse_list(slist, list, dirname) != 0)
0183             goto out_error;
0184     }
0185
0186     return slist;
0187 out_error:
0188     free(slist);
0189     return NULL;
0190 }
0191
0192 void strlist__delete(struct strlist *slist)
0193 {
0194     if (slist != NULL)
0195         rblist__delete(&slist->rblist);
0196 }
0197
0198 struct str_node *strlist__entry(const struct strlist *slist, unsigned int idx)
0199 {
0200     struct str_node *snode = NULL;
0201     struct rb_node *rb_node;
0202
0203     rb_node = rblist__entry(&slist->rblist, idx);
0204     if (rb_node)
0205         snode = container_of(rb_node, struct str_node, rb_node);
0206
0207     return snode;
0208 }