src/util-misc.c

   1 /*
   2     Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013, 2014, 2015,
   3     2016
   4     Ben Kibbey <bjk@luxsci.net>
   5
   6     This file is part of pwmd.
   7
   8     Pwmd is free software: you can redistribute it and/or modify
   9     it under the terms of the GNU General Public License as published by
  10     the Free Software Foundation, either version 2 of the License, or
  11     (at your option) any later version.
  12
  13     Pwmd is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License
  19     along with Pwmd.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21 #ifdef HAVE_CONFIG_H
  22 #include <config.h>
  23 #endif
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <errno.h>
  28 #include <unistd.h>
  29 #include <sys/types.h>
  30 #include <sys/stat.h>
  31 #include <fcntl.h>
  32 #include <string.h>
  33 #include <ctype.h>
  34 #include <pwd.h>
  35
  36 #include "pwmd-error.h"
  37 #include <gcrypt.h>
  38 #include "util-misc.h"
  39 #include "util-string.h"
  40 #include "mutex.h"
  41 #include "cache.h"
  42 #include "mem.h"
  43
  44 extern void log_write (const char *fmt, ...);
  45
  46 int
  47 valid_filename (const char *filename)
  48 {
  49   const char *p;
  50
  51   if (!filename || !*filename)
  52     return 0;
  53
  54   if (filename[0] == '-' && filename[1] == 0)
  55     return 0;
  56
  57   for (p = filename; *p; p++)
  58     {
  59       if (*p == '/' || isspace (*p))
  60         return 0;
  61     }
  62
  63   return 1;
  64 }
  65
  66 /* If 'user' is null then lookup entry for 'uid'. Caller must free 'buf'. */
  67 struct passwd *
  68 get_pwd_struct (const char *user, uid_t uid, struct passwd *pw, char **buf,
  69                 gpg_error_t *rc)
  70 {
  71   struct passwd *result;
  72 #ifdef HAVE_GETPWNAM_R
  73   size_t len;
  74   int err = 0;
  75
  76   len = sysconf (_SC_GETPW_R_SIZE_MAX);
  77   if (len == -1)
  78     len = 16384;
  79
  80   *buf = xmalloc (len);
  81   if (!*buf)
  82     return NULL;
  83
  84   if (user)
  85     {
  86       err = getpwnam_r (user, pw, *buf, len, &result);
  87       if (!err)
  88         return result;
  89     }
  90   else
  91     {
  92       err = getpwuid_r (uid, pw, *buf, len, &result);
  93       if (!err)
  94         return result;
  95     }
  96
  97   errno = err;
  98   *rc = gpg_error_from_errno (err);
  99   return NULL;
 100 #else
 101   if (buf)
 102     *buf = NULL;
 103
 104   errno = 0;
 105   if (user)
 106     result = getpwnam (user);
 107   else
 108     result = getpwuid (uid);
 109
 110   if (!result)
 111     *rc = gpg_error_from_syserror ();
 112
 113   return result;
 114 #endif
 115 }
 116
 117 static char *
 118 get_pwd_entry(uid_t uid, int which)
 119 {
 120   gpg_error_t rc;
 121   char *buf;
 122   struct passwd pw;
 123   struct passwd *result = get_pwd_struct (NULL, uid, &pw, &buf, &rc);
 124   char *value = NULL;
 125
 126   if (!result)
 127     return NULL;
 128
 129   if (result)
 130     {
 131       if (!which)
 132         value = str_dup (result->pw_dir);
 133       else
 134         value = str_dup (result->pw_name);
 135     }
 136
 137   xfree (buf);
 138   return value;
 139 }
 140
 141 char *
 142 get_username (uid_t uid)
 143 {
 144   return get_pwd_entry (uid, 1);
 145 }
 146
 147 char *
 148 get_home_dir ()
 149 {
 150   if (home_directory)
 151     return home_directory;
 152
 153   home_directory = get_pwd_entry (getuid(), 0);
 154   return home_directory;
 155 }
 156
 157 char *
 158 expand_homedir (char *str)
 159 {
 160   char *p = str;
 161
 162   if (*p++ == '~')
 163     {
 164       char *dir = get_pwd_entry(getuid(), 0);
 165       char *s;
 166
 167       if (!dir)
 168         return NULL;
 169
 170       s = str_asprintf ("%s%s", dir, p);
 171       xfree (dir);
 172       return s;
 173     }
 174
 175   return str_dup (str);
 176 }
 177
 178 /* The 'more' parameter lets the calling command process remaining
 179  * non-option arguments. */
 180 gpg_error_t
 181 parse_options (char **line, struct argv_s * args[], void *data, int more)
 182 {
 183   char *p = *line;
 184   gpg_error_t rc = 0;
 185
 186   for (; p && *p; p++)
 187     {
 188       while (*p == ' ')
 189         p++;
 190
 191       if (!*p)
 192         break;
 193
 194       if (*p == '-' && *(p + 1) == '-')
 195         {
 196           p += 2;
 197           char opt[255] = { 0 }, value[255] =
 198           {
 199           0};
 200           char *tp;
 201           unsigned len;
 202
 203           if (!*p || *p == ' ')
 204             {
 205               if (*p)
 206                 p++;
 207
 208               break;
 209             }
 210
 211           for (tp = opt, len = 0; *p; p++, len++)
 212             {
 213               if (len + 1 == 255)
 214                 return GPG_ERR_LINE_TOO_LONG;
 215
 216               if (*p == ' ' || *p == '=')
 217                 {
 218                   *tp = 0;
 219
 220                   if (*p == '=')
 221                     {
 222                       int inquote = 0;
 223
 224                       p++;
 225
 226                       for (tp = value, len = 0; *p; p++, len++)
 227                         {
 228                           if (len + 1 == 255)
 229                             return GPG_ERR_LINE_TOO_LONG;
 230
 231                           if (*p == '\"')
 232                             {
 233                               inquote = !inquote;
 234                               continue;
 235                             }
 236                           else if (*p == ' ' && !inquote)
 237                             break;
 238
 239                           *tp++ = *p;
 240                         }
 241
 242                       *tp = 0;
 243                     }
 244
 245                   break;
 246                 }
 247
 248               *tp++ = *p;
 249             }
 250
 251           *tp = 0;
 252           int match = 0;
 253
 254           for (int i = 0; args[i]; i++)
 255             {
 256               if (!strcmp (args[i]->opt, opt))
 257                 {
 258                   log_write2 ("param: name='%s' value='%s'", opt, value);
 259                   if (args[i]->type == OPTION_TYPE_NOARG && *value)
 260                     return GPG_ERR_SYNTAX;
 261                   else if (args[i]->type == OPTION_TYPE_ARG && !*value)
 262                     return GPG_ERR_SYNTAX;
 263
 264                   if (args[i]->func)
 265                     {
 266                       rc = args[i]->func (data, value);
 267                       if (rc)
 268                         return rc;
 269                     }
 270
 271                   match = 1;
 272                   break;
 273                 }
 274             }
 275
 276           if (!match)
 277             return GPG_ERR_UNKNOWN_OPTION;
 278
 279           if (!*p)
 280             break;
 281
 282           continue;
 283         }
 284
 285       if (*(p+1) && !more)
 286         return GPG_ERR_SYNTAX;
 287
 288       break;
 289     }
 290
 291   *line = p;
 292   return rc;
 293 }
 294
 295 char *
 296 bin2hex (const unsigned char *data, size_t len)
 297 {
 298   size_t size = len * 2 + 1;
 299   char buf[size];
 300   size_t n;
 301
 302   buf[0] = 0;
 303
 304   for (n = 0; n < len; n++)
 305     {
 306       char c[3];
 307
 308       sprintf (c, "%02X", data[n]);
 309       strcat (buf, c);
 310     }
 311
 312   return str_dup (buf);
 313 }
 314
 315 static char *
 316 html_escape (const char *line, size_t len, int space)
 317 {
 318   struct string_s *string = string_new ("");
 319   size_t n;
 320   char *tmp;
 321
 322   if (!string)
 323     return NULL;
 324
 325   for (n = 0; n < len; n++)
 326     {
 327       switch (line[n])
 328         {
 329         case '<':
 330           string = string_append (string, "&lt;");
 331           break;
 332         case '>':
 333           string = string_append (string, "&gt;");
 334           break;
 335         case ' ':
 336           if (space)
 337             {
 338               string = string_append (string, "%20");
 339               break;
 340             }
 341         default:
 342           string = string_append_printf (string, "%c", line[n]);
 343           break;
 344         }
 345     }
 346
 347   tmp = string->str;
 348   string_free (string, 0);
 349   return tmp;
 350 }
 351
 352 static char *
 353 strip_texi_html (const char *line)
 354 {
 355   size_t len;
 356   const char *p;
 357   char *tmp;
 358   struct string_s *string = string_new ("<html><body>");
 359
 360   if (!string)
 361     return NULL;
 362
 363   /* The command usage text. */
 364   p = strchr (line, '\n');
 365   if (p)
 366     {
 367       len = strlen (line)-strlen(p);
 368       tmp = html_escape (line, len, 0);
 369       string = string_append_printf (string, "<b>%s</b><p>", tmp);
 370       xfree (tmp);
 371       p++;
 372     }
 373   else
 374     p = line;
 375
 376   for (; p && *p; p++)
 377     {
 378       int non_anchor = 1;
 379       int var = 0;
 380       int code = 0;
 381       int emph = 0;
 382       int cite = 0;
 383
 384       if (*p == '@' && *(p + 1) != '@')
 385         {
 386           p++;
 387
 388           if (!strncasecmp (p, "end ", 4))
 389             {
 390               p += 4;
 391               if (!strncasecmp (p, "table", 5))
 392                 {
 393                   string = string_append (string, "</p>");
 394                 }
 395               else if (!strncasecmp (p, "example", 7))
 396                 {
 397                   string = string_append (string, "</code>");
 398                 }
 399
 400               while (*p++ != '\n');
 401               p--;
 402               continue;
 403             }
 404           else if (!strncasecmp (p, "table ", 6))
 405             {
 406               while (*p++ != '\n');
 407               p--;
 408               continue;
 409             }
 410           else if (!strncasecmp (p, "example", 7))
 411             {
 412               string = string_append (string, "<code>");
 413               while (*p++ != '\n');
 414               p--;
 415               continue;
 416             }
 417           else if (!strncasecmp (p, "sp ", 3))
 418             {
 419               while (*p++ != '\n');
 420               p--;
 421               string = string_append (string, "<p>");
 422               continue;
 423             }
 424           else if (!strncasecmp (p, "item ", 5))
 425             {
 426               p += 5;
 427               string = string_append (string, "<p><b>");
 428               tmp = strchr (p, '\n');
 429               len = strlen (p) - strlen (tmp);
 430               while (len--)
 431                 string = string_append_printf (string, "%c", *p++);
 432
 433               string = string_append (string, "</b><br/>");
 434               continue;
 435             }
 436           else if (!strncasecmp (p, "pxref", 5)
 437                    || !strncasecmp (p, "npxref", 6))
 438             {
 439               int n = 0;
 440
 441               if (!strncasecmp (p, "npxref", 6))
 442                 n++;
 443
 444               p += n ? 7 : 6;
 445               non_anchor = 0;
 446               string = string_append_printf (string, "%s<a href=\"",
 447                                              n ? "" : "see ");
 448               goto close;
 449             }
 450           else if (!strncasecmp (p, "xref", 4))
 451             {
 452               p += 5;
 453               non_anchor = 0;
 454               string = string_append (string, "See <a href=\"");
 455               goto close;
 456             }
 457           else if (!strncasecmp (p, "url", 3))
 458             {
 459               p += 4;
 460               non_anchor = 0;
 461               string = string_append (string, "<a href=\"");
 462               goto close;
 463             }
 464           else if (!strncasecmp (p, "key", 3))
 465             {
 466               p += 4;
 467               goto close;
 468             }
 469           else if (!strncasecmp (p, "file", 4))
 470             {
 471               p += 5;
 472               var = 1;
 473               string = string_append (string, "<var>");
 474               goto close;
 475             }
 476           else if (!strncasecmp (p, "var", 3))
 477             {
 478               p += 4;
 479               var = 1;
 480               string = string_append (string, "<var>");
 481               goto close;
 482             }
 483           else if (!strncasecmp (p, "option", 6))
 484             {
 485               p += 7;
 486               var = 1;
 487               string = string_append (string, "<var>");
 488               goto close;
 489             }
 490           else if (!strncasecmp (p, "code", 4))
 491             {
 492               p += 5;
 493               code = 1;
 494               string = string_append (string, "<b>");
 495               goto close;
 496             }
 497           else if (!strncasecmp (p, "emph", 4))
 498             {
 499               p += 5;
 500               emph = 1;
 501               string = string_append (string, "<em>");
 502               goto close;
 503             }
 504           else if (!strncasecmp (p, "command", 7))
 505             {
 506               p += 8;
 507               emph = 1;
 508               string = string_append (string, "<em>");
 509               goto close;
 510             }
 511           else if (!strncasecmp (p, "cite", 4))
 512             {
 513               p += 5;
 514               cite = 1;
 515               string = string_append (string, "<cite>");
 516               goto close;
 517             }
 518           else if (!strncasecmp (p, "abbr", 4))
 519             {
 520               p += 5;
 521               emph = 1;
 522               string = string_append (string, "<em>");
 523               goto close;
 524             }
 525           else if (!strncasecmp (p, "*", 1))
 526             {
 527               string = string_append (string, "<br/>");
 528               continue;
 529             }
 530
 531           while (*p && *p != '{')
 532             {
 533               if (*++p == '*')
 534                 {
 535                   p++;
 536                   goto append;
 537                 }
 538             }
 539
 540         close:
 541           if (*p)
 542             {
 543               char buf [255];
 544
 545               tmp = strchr (p, '}');
 546               len = strlen (p) - strlen (tmp);
 547               tmp = buf;
 548               while (len--)
 549                 *tmp++ = *p++;
 550
 551               *tmp = 0;
 552
 553               if (*p)
 554                 p++;
 555
 556               if (!non_anchor)
 557                 {
 558                   tmp = html_escape (buf, strlen (buf), 1);
 559                   string = string_append_printf (string, "%s\">%s</a>", tmp, buf);
 560                   xfree (tmp);
 561                 }
 562               else
 563                 string = string_append (string, buf);
 564
 565               if (var)
 566                 string = string_append (string, "</var>");
 567               else if (code)
 568                 string = string_append (string, "</b>");
 569               else if (emph)
 570                 string = string_append (string, "</em>");
 571               else if (cite)
 572                 string = string_append (string, "</cite>");
 573             }
 574         }
 575       else if (*p == '@' && *(p + 1) == '@')
 576         p++;
 577
 578     append:
 579       string = string_append_printf (string, "%c", *p);
 580
 581       if (!*p)
 582         break;
 583     }
 584
 585   string = string_append (string, "</body></html>");
 586   tmp = string->str;
 587   string_free (string, 0);
 588   return tmp;
 589 }
 590
 591 static char *
 592 strip_texi (const char *str, int html)
 593 {
 594   const char *p;
 595   char *s;
 596   int c = 0;
 597
 598   if (html)
 599     return strip_texi_html (str);
 600
 601   s = str_dup (str);
 602   if (!s)
 603     return NULL;
 604
 605   for (p = str; *p; p++)
 606     {
 607       if (*p == '@' && *(p + 1) != '@')
 608         {
 609           p++;
 610
 611           if (!strncasecmp (p, "table", 5)
 612               || !strncasecmp (p, "end ", 4))
 613             {
 614               while (*p++ != '\n');
 615               p--;
 616               continue;
 617             }
 618           else if (!strncasecmp (p, "example", 7)
 619                    || !strncasecmp (p, "end ", 4))
 620             {
 621               while (*p++ != '\n');
 622               p--;
 623               continue;
 624             }
 625           else if (!strncasecmp (p, "sp ", 3))
 626             {
 627               while (*p++ != '\n');
 628               p--;
 629               continue;
 630             }
 631           else if (!strncasecmp (p, "item", 4))
 632             {
 633               p += 5;
 634               while (*p && *p != '\n')
 635                 s[c++] = *p++;
 636
 637               s[c++] = '\n';
 638               continue;
 639             }
 640           else if (!strncasecmp (p, "pxref", 5))
 641             {
 642               p += 5;
 643               strcat (s, "see ");
 644               c = strlen (s);
 645               goto close;
 646             }
 647           else if (!strncasecmp (p, "xref", 4))
 648             {
 649               p += 4;
 650               strcat (s, "See ");
 651               c = strlen (s);
 652               goto close;
 653             }
 654           else if (!strncasecmp (p, "*", 1))
 655             {
 656               s[c++] = '\n';
 657               s[c] = 0;
 658               continue;
 659             }
 660
 661           while (*p && *p != '{')
 662             {
 663               if (*++p == '*')
 664                 {
 665                   p++;
 666                   goto append;
 667                 }
 668             }
 669
 670         close:
 671           if (*p)
 672             {
 673               p++;
 674               s[c++] = '`';
 675               while (*p && *p != '}')
 676                 s[c++] = *p++;
 677
 678               if (*p)
 679                 p++;
 680
 681               s[c++] = '\'';
 682               s[c] = 0;
 683             }
 684         }
 685       else if (*p == '@' && *(p + 1) == '@')
 686         p++;
 687
 688     append:
 689       if (*p)
 690         s[c++] = *p;
 691
 692       s[c] = 0;
 693     }
 694
 695   s[c] = 0;
 696   return s;
 697 }
 698
 699
 700 char *
 701 strip_texi_and_wrap (const char *str, int html)
 702 {
 703   char *tmp = strip_texi (str, html);
 704   char *help = xmalloc (strlen (tmp) * 2 + 1);
 705   char *p, *ph;
 706   int i;
 707
 708   for (p = tmp, ph = help, i = 0; *p; p++, i++)
 709     {
 710       if (i == 78 || *p == '\n')
 711         {
 712           if (!isspace (*p))
 713             {
 714               char *t = ph;
 715               int n = 0;
 716
 717               while (!(*--t == ' '))
 718                 n++;
 719
 720               *t++ = '\n';
 721               i = -1;
 722               p -= n;
 723               while (n--)
 724                 {
 725                   *t++ = *p++;
 726                   i++;
 727                 }
 728             }
 729           else
 730             {
 731               *ph++ = '\n';
 732               i = -1;
 733               while (*p != '\n' && *p == ' ')
 734                 p++;
 735             }
 736         }
 737
 738       if (*p == '\n')
 739         continue;
 740       *ph++ = *p;
 741     }
 742
 743   *ph = 0;
 744   xfree (tmp);
 745   return help;
 746 }
 747
 748 void
 749 free_key (void *data)
 750 {
 751   xfree (data);
 752 }
 753
 754 gpg_error_t
 755 create_thread (void *(*cb) (void *), void *data,
 756                pthread_t * tid, int detached)
 757 {
 758   pthread_attr_t attr;
 759   int n;
 760
 761   pthread_attr_init (&attr);
 762
 763   if (detached)
 764     pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
 765
 766   n = pthread_create (tid, &attr, cb, data);
 767   pthread_attr_destroy (&attr);
 768   return gpg_error_from_errno (n);
 769 }
 770
 771 void
 772 release_mutex_cb (void *arg)
 773 {
 774   pthread_mutex_t *m = (pthread_mutex_t *) arg;
 775
 776   MUTEX_UNLOCK (m);
 777 }
 778
 779 void
 780 close_fd_cb (void *arg)
 781 {
 782   int fd = *(int *) arg;
 783
 784   if (fd != -1)
 785     close (fd);
 786 }
 787
 788 /* The advisory lock should be obtained before calling this function. */
 789 gpg_error_t
 790 get_checksum (const char *filename, unsigned char **r_crc, size_t * r_crclen)
 791 {
 792   int fd;
 793   unsigned char *buf;
 794   gpg_error_t rc = 0;
 795   struct stat st;
 796
 797   if (stat (filename, &st) == -1)
 798     return gpg_error_from_errno (errno);
 799
 800   fd = open (filename, O_RDONLY);
 801   if (fd == -1)
 802     return gpg_error_from_errno (errno);
 803
 804   pthread_cleanup_push (close_fd_cb, &fd);
 805   buf = xmalloc (st.st_size);
 806   if (buf)
 807     {
 808       pthread_cleanup_push (xfree, buf);
 809
 810       size_t len = read (fd, buf, st.st_size);
 811       if (len == st.st_size)
 812         {
 813           if (buf)
 814             {
 815               unsigned char *crc;
 816
 817               len = gcry_md_get_algo_dlen (GCRY_MD_CRC32);
 818               crc = xmalloc (len);
 819               if (crc)
 820                 {
 821                   pthread_cleanup_push (xfree, crc);
 822                   gcry_md_hash_buffer (GCRY_MD_CRC32, crc, buf, st.st_size);
 823                   *r_crc = crc;
 824                   *r_crclen = len;
 825                   pthread_cleanup_pop (0);
 826                 }
 827               else
 828                 rc = GPG_ERR_ENOMEM;
 829             }
 830         }
 831       else
 832         rc = GPG_ERR_TOO_SHORT;
 833
 834       pthread_cleanup_pop (1);
 835     }
 836   else
 837     rc = GPG_ERR_ENOMEM;
 838
 839   pthread_cleanup_pop (1);
 840   return rc;
 841 }
 842
 843 wchar_t *str_to_wchar (const char *str)
 844 {
 845     wchar_t *wc;
 846     mbstate_t ps;
 847     const char *p = str;
 848     size_t len;
 849
 850     memset (&ps, 0, sizeof(mbstate_t));
 851     len = mbsrtowcs (NULL, &p, 0, &ps);
 852     if (len == -1)
 853       return NULL;
 854
 855     wc = xcalloc (len+1, sizeof(wchar_t));
 856     if (!wc)
 857       return NULL;
 858
 859     p = str;
 860     memset (&ps, 0, sizeof(mbstate_t));
 861     len = mbsrtowcs (wc, &p, len, &ps);
 862     if (len == -1)
 863       {
 864         xfree (wc);
 865         return NULL;
 866       }
 867
 868     return wc;
 869 }
 870
 871 char *
 872 gnupg_escape (const char *str)
 873 {
 874   if (!str || !*str)
 875     return NULL;
 876
 877   char *buf = xmalloc (strlen(str)*4+1), *b = buf;
 878   const char *p;
 879
 880   if (!buf)
 881     return NULL;
 882
 883   for (p = str; p && *p; p++)
 884     {
 885       if (*p == ':')
 886         {
 887           *b++ = '\\';
 888           *b++ = 'x';
 889           *b++ = '3';
 890           *b++ = 'a';
 891           continue;
 892         }
 893       else if (*p == '\\')
 894         {
 895           *b++ = '\\';
 896           *b++ = 'x';
 897           *b++ = '5';
 898           *b++ = 'c';
 899           continue;
 900         }
 901
 902       *b++ = *p;
 903     }
 904
 905   *b = 0;
 906   return buf;
 907 }
 908
 909 /* From Beej's Guide to Network Programming. It's a good tutorial. */
 910 void *
 911 get_in_addr (struct sockaddr *sa)
 912 {
 913   if (sa->sa_family == AF_INET)
 914     return &(((struct sockaddr_in *) sa)->sin_addr);
 915
 916   return &(((struct sockaddr_in6 *) sa)->sin6_addr);
 917 }