diplodocus.org Git - nmh/blob - sbr/utils.c

   1
   2 /*
   3  * utils.c -- various utility routines
   4  *
   5  * This code is Copyright (c) 2006, by the authors of nmh.  See the
   6  * COPYRIGHT file in the root directory of the nmh distribution for
   7  * complete copyright information.
   8  */
   9
  10 #include <h/mh.h>
  11 #include <h/utils.h>
  12 #include <stdlib.h>
  13 #include <fcntl.h>
  14
  15 /*
  16  * We allocate space for messages (msgs array)
  17  * this number of elements at a time.
  18  */
  19 #define MAXMSGS 256
  20
  21 /*
  22  * Safely call malloc
  23  */
  24 void *
  25 mh_xmalloc(size_t size)
  26 {
  27     void *memory;
  28
  29     if (size == 0)
  30         adios(NULL, "Tried to malloc 0 bytes");
  31
  32     memory = malloc(size);
  33     if (!memory)
  34         adios(NULL, "Malloc failed");
  35
  36     return memory;
  37 }
  38
  39 /*
  40  * Safely call realloc
  41  */
  42 void *
  43 mh_xrealloc(void *ptr, size_t size)
  44 {
  45     void *memory;
  46
  47     /* Some non-POSIX realloc()s don't cope with realloc(NULL,sz) */
  48     if (!ptr)
  49         return mh_xmalloc(size);
  50
  51     if (size == 0)
  52         adios(NULL, "Tried to realloc 0bytes");
  53
  54     memory = realloc(ptr, size);
  55     if (!memory)
  56         adios(NULL, "Realloc failed");
  57
  58     return memory;
  59 }
  60
  61 /*
  62  * Return the present working directory, if the current directory does not
  63  * exist, or is too long, make / the pwd.
  64  */
  65 char *
  66 pwd(void)
  67 {
  68     register char *cp;
  69     static char curwd[PATH_MAX];
  70
  71     if (!getcwd (curwd, PATH_MAX)) {
  72         admonish (NULL, "unable to determine working directory");
  73         if (!mypath || !*mypath
  74                 || (strcpy (curwd, mypath), chdir (curwd)) == -1) {
  75             strcpy (curwd, "/");
  76             chdir (curwd);
  77         }
  78         return curwd;
  79     }
  80
  81     if ((cp = curwd + strlen (curwd) - 1) > curwd && *cp == '/')
  82         *cp = '\0';
  83
  84     return curwd;
  85 }
  86
  87 /*
  88  * add   -- If "s1" is NULL, this routine just creates a
  89  *       -- copy of "s2" into newly malloc'ed memory.
  90  *       --
  91  *       -- If "s1" is not NULL, then copy the concatenation
  92  *       -- of "s1" and "s2" (note the order) into newly
  93  *       -- malloc'ed memory.  Then free "s1".
  94  */
  95 char *
  96 add (const char *s2, char *s1)
  97 {
  98     char *cp;
  99     size_t len1 = 0, len2 = 0;
 100
 101     if (s1)
 102         len1 = strlen (s1);
 103     if (s2)
 104         len2 = strlen (s2);
 105
 106     cp = mh_xmalloc (len1 + len2 + 1);
 107
 108     /* Copy s1 and free it */
 109     if (s1) {
 110         memcpy (cp, s1, len1);
 111         free (s1);
 112     }
 113
 114     /* Copy s2 */
 115     if (s2)
 116         memcpy (cp + len1, s2, len2);
 117
 118     /* Now NULL terminate the string */
 119     cp[len1 + len2] = '\0';
 120
 121     return cp;
 122 }
 123
 124 /*
 125  * addlist
 126  *      Append an item to a comma separated list
 127  */
 128 char *
 129 addlist (char *list, const char *item)
 130 {
 131     if (list)
 132         list = add (", ", list);
 133
 134     return add (item, list);
 135 }
 136
 137 /*
 138  * folder_exists
 139  *      Check to see if a folder exists.
 140  */
 141 int folder_exists(const char *folder)
 142 {
 143     struct stat st;
 144     int exists = 0;
 145
 146     if (stat (folder, &st) == -1) {
 147         /* The folder either doesn't exist, or we hit an error.  Either way
 148          * return a failure.
 149          */
 150         exists = 0;
 151     } else {
 152         /* We can see a folder with the right name */
 153         exists = 1;
 154     }
 155
 156     return exists;
 157 }
 158
 159
 160 /*
 161  * create_folder
 162  *      Check to see if a folder exists, if not, prompt the user to create
 163  *      it.
 164  */
 165 void create_folder(char *folder, int autocreate, void (*done_callback)(int))
 166 {
 167     struct stat st;
 168     extern int errno;
 169     char *cp;
 170
 171     if (stat (folder, &st) == -1) {
 172         if (errno != ENOENT)
 173             adios (folder, "error on folder");
 174         if (autocreate == 0) {
 175             /* ask before creating folder */
 176             cp = concat ("Create folder \"", folder, "\"? ", NULL);
 177             if (!getanswer (cp))
 178                 done_callback (1);
 179             free (cp);
 180         } else if (autocreate == -1) {
 181             /* do not create, so exit */
 182             done_callback (1);
 183         }
 184         if (!makedir (folder))
 185             adios (NULL, "unable to create folder %s", folder);
 186     }
 187 }
 188
 189 /*
 190  * num_digits
 191  *      Return the number of digits in a nonnegative integer.
 192  */
 193 int
 194 num_digits (int n)
 195 {
 196     int ndigits = 0;
 197
 198     /* Sanity check */
 199     if (n < 0)
 200         adios (NULL, "oops, num_digits called with negative value");
 201
 202     if (n == 0)
 203         return 1;
 204
 205     while (n) {
 206         n /= 10;
 207         ndigits++;
 208     }
 209
 210     return ndigits;
 211 }
 212
 213 /*
 214  * Append a message arg to an array of them, resizing it if necessary.
 215  * Really a simple vector-of-(char *) maintenance routine.
 216  */
 217 void
 218 app_msgarg(struct msgs_array *msgs, char *cp)
 219 {
 220         if(msgs->size >= msgs->max) {
 221                 msgs->max += MAXMSGS;
 222                 msgs->msgs = mh_xrealloc(msgs->msgs,
 223                         msgs->max * sizeof(*msgs->msgs));
 224         }
 225         msgs->msgs[msgs->size++] = cp;
 226 }
 227
 228 /*
 229  * Append a message number to an array of them, resizing it if necessary.
 230  * Like app_msgarg, but with a vector-of-ints instead.
 231  */
 232
 233 void
 234 app_msgnum(struct msgnum_array *msgs, int msgnum)
 235 {
 236         if (msgs->size >= msgs->max) {
 237                 msgs->max += MAXMSGS;
 238                 msgs->msgnums = mh_xrealloc(msgs->msgnums,
 239                         msgs->max * sizeof(*msgs->msgnums));
 240         }
 241         msgs->msgnums[msgs->size++] = msgnum;
 242 }
 243
 244 /* Open a form or components file */
 245 int
 246 open_form(char **form, char *def)
 247 {
 248         int in;
 249         if (*form) {
 250                 if ((in = open (etcpath (*form), O_RDONLY)) == NOTOK)
 251                         adios (*form, "unable to open form file");
 252         } else {
 253                 if ((in = open (etcpath (def), O_RDONLY)) == NOTOK)
 254                         adios (def, "unable to open default components file");
 255                 *form = def;
 256         }
 257         return in;
 258 }
 259
 260
 261 /*
 262  * Finds first occurrence of str in buf.  buf is not a C string but a
 263  * byte array of length buflen.  str is a null-terminated C string.
 264  * find_str() does not modify buf but passes back a non-const char *
 265  * pointer so that the caller can modify it.
 266  */
 267 char *
 268 find_str (const char buf[], size_t buflen, const char *str) {
 269     const size_t len = strlen (str);
 270     size_t i;
 271
 272     for (i = 0; i + len <= buflen; ++i, ++buf) {
 273         if (! memcmp (buf, str, len)) return (char *) buf;
 274     }
 275
 276     return NULL;
 277 }
 278
 279
 280 /*
 281  * Finds last occurrence of str in buf.  buf is not a C string but a
 282  * byte array of length buflen.  str is a null-terminated C string.
 283  * find_str() does not modify buf but passes back a non-const char *
 284  * pointer so that the caller can modify it.
 285  */
 286 char *
 287 rfind_str (const char buf[], size_t buflen, const char *str) {
 288     const size_t len = strlen (str);
 289     size_t i;
 290
 291     for (i = 0, buf += buflen - len; i + len <= buflen; ++i, --buf) {
 292         if (! memcmp (buf, str, len)) return (char *) buf;
 293     }
 294
 295     return NULL;
 296 }
 297
 298
 299 /* POSIX doesn't have strcasestr() so emulate it. */
 300 char *
 301 nmh_strcasestr (const char *s1, const char *s2) {
 302     const size_t len = strlen (s2);
 303
 304     if (isupper ((unsigned char) s2[0])  ||  islower ((unsigned char)s2[0])) {
 305         char first[3];
 306         first[0] = (char) toupper ((unsigned char) s2[0]);
 307         first[1] = (char) tolower ((unsigned char) s2[0]);
 308         first[2] = '\0';
 309
 310         for (s1 = strpbrk (s1, first); s1; s1 = strpbrk (++s1, first)) {
 311             if (! strncasecmp (s1, s2, len)) return (char *) s1;
 312         }
 313     } else {
 314         for (s1 = strchr (s1, s2[0]); s1; s1 = strchr (++s1, s2[0])) {
 315             if (! strncasecmp (s1, s2, len)) return (char *) s1;
 316         }
 317     }
 318
 319     return NULL;
 320 }