apps/irssi/src/core/misc.c

   1 /*
   2  misc.c : irssi
   3
   4     Copyright (C) 1999 Timo Sirainen
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program; if not, write to the Free Software
  18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 #include "module.h"
  22 #include "misc.h"
  23 #include "pidwait.h"
  24
  25 #include <errno.h>
  26 #ifdef HAVE_REGEX_H
  27 #  include <regex.h>
  28 #endif
  29
  30 typedef struct {
  31         int condition;
  32         GInputFunction function;
  33         void *data;
  34 } IRSSI_INPUT_REC;
  35
  36 static int irssi_io_invoke(GIOChannel *source, GIOCondition condition,
  37                            void *data)
  38 {
  39         IRSSI_INPUT_REC *rec = data;
  40         int icond = 0;
  41
  42         if (condition & (G_IO_ERR | G_IO_HUP | G_IO_NVAL)) {
  43                 /* error, we have to call the function.. */
  44                 if (rec->condition & G_IO_IN)
  45                         icond |= G_INPUT_READ;
  46                 else
  47                         icond |= G_INPUT_WRITE;
  48         }
  49
  50         if (condition & (G_IO_IN | G_IO_PRI))
  51                 icond |= G_INPUT_READ;
  52         if (condition & G_IO_OUT)
  53                 icond |= G_INPUT_WRITE;
  54
  55         if (rec->condition & icond)
  56                 rec->function(rec->data, source, icond);
  57
  58         return TRUE;
  59 }
  60
  61 int g_input_add_full(GIOChannel *source, int priority, int condition,
  62                      GInputFunction function, void *data)
  63 {
  64         IRSSI_INPUT_REC *rec;
  65         unsigned int result;
  66         GIOCondition cond;
  67
  68         rec = g_new(IRSSI_INPUT_REC, 1);
  69         rec->condition = condition;
  70         rec->function = function;
  71         rec->data = data;
  72
  73         cond = (GIOCondition) (G_IO_ERR|G_IO_HUP|G_IO_NVAL);
  74         if (condition & G_INPUT_READ)
  75                 cond |= G_IO_IN|G_IO_PRI;
  76         if (condition & G_INPUT_WRITE)
  77                 cond |= G_IO_OUT;
  78
  79         result = g_io_add_watch_full(source, priority, cond,
  80                                      irssi_io_invoke, rec, g_free);
  81
  82         return result;
  83 }
  84
  85 int g_input_add(GIOChannel *source, int condition,
  86                 GInputFunction function, void *data)
  87 {
  88         return g_input_add_full(source, G_PRIORITY_DEFAULT, condition,
  89                                 function, data);
  90 }
  91
  92 int g_timeval_cmp(const GTimeVal *tv1, const GTimeVal *tv2)
  93 {
  94         if (tv1->tv_sec < tv2->tv_sec)
  95                 return -1;
  96         if (tv1->tv_sec > tv2->tv_sec)
  97                 return 1;
  98
  99         return tv1->tv_usec < tv2->tv_usec ? -1 :
 100                 tv1->tv_usec > tv2->tv_usec ? 1 : 0;
 101 }
 102
 103 long get_timeval_diff(const GTimeVal *tv1, const GTimeVal *tv2)
 104 {
 105         long secs, usecs;
 106
 107         secs = tv1->tv_sec - tv2->tv_sec;
 108         usecs = tv1->tv_usec - tv2->tv_usec;
 109         if (usecs < 0) {
 110                 usecs += 1000000;
 111                 secs--;
 112         }
 113         usecs = usecs/1000 + secs * 1000;
 114
 115         return usecs;
 116 }
 117
 118 int find_substr(const char *list, const char *item)
 119 {
 120         const char *ptr;
 121
 122         g_return_val_if_fail(list != NULL, FALSE);
 123         g_return_val_if_fail(item != NULL, FALSE);
 124
 125         if (*item == '\0')
 126                 return FALSE;
 127
 128         for (;;) {
 129                 while (i_isspace(*list)) list++;
 130                 if (*list == '\0') break;
 131
 132                 ptr = strchr(list, ' ');
 133                 if (ptr == NULL) ptr = list+strlen(list);
 134
 135                 if (g_strncasecmp(list, item, ptr-list) == 0 &&
 136                     item[ptr-list] == '\0')
 137                         return TRUE;
 138
 139                 list = ptr;
 140         }
 141
 142         return FALSE;
 143 }
 144
 145 int strarray_length(char **array)
 146 {
 147         int len;
 148
 149         g_return_val_if_fail(array != NULL, 0);
 150
 151         len = 0;
 152         while (*array) {
 153                 len++;
 154                 array++;
 155         }
 156         return len;
 157 }
 158
 159 int strarray_find(char **array, const char *item)
 160 {
 161         char **tmp;
 162         int index;
 163
 164         g_return_val_if_fail(array != NULL, 0);
 165         g_return_val_if_fail(item != NULL, 0);
 166
 167         index = 0;
 168         for (tmp = array; *tmp != NULL; tmp++, index++) {
 169                 if (g_strcasecmp(*tmp, item) == 0)
 170                         return index;
 171         }
 172
 173         return -1;
 174 }
 175
 176 int execute(const char *cmd)
 177 {
 178         char **args;
 179 #ifndef WIN32
 180         int pid;
 181 #endif
 182
 183         g_return_val_if_fail(cmd != NULL, -1);
 184
 185 #ifndef WIN32
 186         pid = fork();
 187         if (pid == -1) return FALSE;
 188         if (pid != 0) {
 189                 pidwait_add(pid);
 190                 return pid;
 191         }
 192
 193         args = g_strsplit(cmd, " ", -1);
 194         execvp(args[0], args);
 195         g_strfreev(args);
 196
 197         _exit(99);
 198         return -1;
 199 #else
 200         args = g_strsplit(cmd, " ", -1);
 201         _spawnvp(_P_DETACH, args[0], args);
 202         g_strfreev(args);
 203         return 0;
 204 #endif
 205 }
 206
 207 GSList *gslist_find_string(GSList *list, const char *key)
 208 {
 209         for (list = list; list != NULL; list = list->next)
 210                 if (strcmp(list->data, key) == 0) return list;
 211
 212         return NULL;
 213 }
 214
 215 GSList *gslist_find_icase_string(GSList *list, const char *key)
 216 {
 217         for (list = list; list != NULL; list = list->next)
 218                 if (g_strcasecmp(list->data, key) == 0) return list;
 219
 220         return NULL;
 221 }
 222
 223 void *gslist_foreach_find(GSList *list, FOREACH_FIND_FUNC func, const void *data)
 224 {
 225         void *ret;
 226
 227         while (list != NULL) {
 228                 ret = func(list->data, (void *) data);
 229                 if (ret != NULL) return ret;
 230
 231                 list = list->next;
 232         }
 233
 234         return NULL;
 235 }
 236
 237 /* `list' contains pointer to structure with a char* to string. */
 238 char *gslistptr_to_string(GSList *list, int offset, const char *delimiter)
 239 {
 240         GString *str;
 241         char **data, *ret;
 242
 243         str = g_string_new(NULL);
 244         while (list != NULL) {
 245                 data = G_STRUCT_MEMBER_P(list->data, offset);
 246
 247                 if (str->len != 0) g_string_append(str, delimiter);
 248                 g_string_append(str, *data);
 249                 list = list->next;
 250         }
 251
 252         ret = str->str;
 253         g_string_free(str, FALSE);
 254         return ret;
 255 }
 256
 257 /* `list' contains char* */
 258 char *gslist_to_string(GSList *list, const char *delimiter)
 259 {
 260         GString *str;
 261         char *ret;
 262
 263         str = g_string_new(NULL);
 264         while (list != NULL) {
 265                 if (str->len != 0) g_string_append(str, delimiter);
 266                 g_string_append(str, list->data);
 267
 268                 list = list->next;
 269         }
 270
 271         ret = str->str;
 272         g_string_free(str, FALSE);
 273         return ret;
 274 }
 275
 276 void hash_save_key(char *key, void *value, GSList **list)
 277 {
 278         *list = g_slist_append(*list, key);
 279 }
 280
 281 /* save all keys in hash table to linked list - you shouldn't remove any
 282    items while using this list, use g_slist_free() after you're done with it */
 283 GSList *hashtable_get_keys(GHashTable *hash)
 284 {
 285         GSList *list;
 286
 287         list = NULL;
 288         g_hash_table_foreach(hash, (GHFunc) hash_save_key, &list);
 289         return list;
 290 }
 291
 292 GList *glist_find_string(GList *list, const char *key)
 293 {
 294         for (list = list; list != NULL; list = list->next)
 295                 if (strcmp(list->data, key) == 0) return list;
 296
 297         return NULL;
 298 }
 299
 300 GList *glist_find_icase_string(GList *list, const char *key)
 301 {
 302         for (list = list; list != NULL; list = list->next)
 303                 if (g_strcasecmp(list->data, key) == 0) return list;
 304
 305         return NULL;
 306 }
 307
 308 char *stristr(const char *data, const char *key)
 309 {
 310         const char *max;
 311         int keylen, datalen, pos;
 312
 313         keylen = strlen(key);
 314         datalen = strlen(data);
 315
 316         if (keylen > datalen)
 317                 return NULL;
 318         if (keylen == 0)
 319                 return (char *) data;
 320
 321         max = data+datalen-keylen;
 322         pos = 0;
 323         while (data <= max) {
 324                 if (key[pos] == '\0')
 325                         return (char *) data;
 326
 327                 if (i_toupper(data[pos]) == i_toupper(key[pos]))
 328                         pos++;
 329                 else {
 330                         data++;
 331                         pos = 0;
 332                 }
 333         }
 334
 335         return NULL;
 336 }
 337
 338 #define isbound(c) \
 339         ((unsigned char) (c) < 128 && \
 340         (i_isspace(c) || i_ispunct(c)))
 341
 342 char *strstr_full_case(const char *data, const char *key, int icase)
 343 {
 344         const char *start, *max;
 345         int keylen, datalen, pos, match;
 346
 347         keylen = strlen(key);
 348         datalen = strlen(data);
 349
 350         if (keylen > datalen)
 351                 return NULL;
 352         if (keylen == 0)
 353                 return (char *) data;
 354
 355         max = data+datalen-keylen;
 356         start = data; pos = 0;
 357         while (data <= max) {
 358                 if (key[pos] == '\0') {
 359                         if (data[pos] != '\0' && !isbound(data[pos])) {
 360                                 data++;
 361                                 pos = 0;
 362                                 continue;
 363                         }
 364                         return (char *) data;
 365                 }
 366
 367                 match = icase ? (i_toupper(data[pos]) == i_toupper(key[pos])) :
 368                                  data[pos] == key[pos];
 369
 370                 if (match && (pos != 0 || data == start || isbound(data[-1])))
 371                         pos++;
 372                 else {
 373                         data++;
 374                         pos = 0;
 375                 }
 376         }
 377
 378         return NULL;
 379 }
 380
 381 char *strstr_full(const char *data, const char *key)
 382 {
 383         return strstr_full_case(data, key, FALSE);
 384 }
 385
 386 char *stristr_full(const char *data, const char *key)
 387 {
 388         return strstr_full_case(data, key, TRUE);
 389 }
 390
 391 int regexp_match(const char *str, const char *regexp)
 392 {
 393 #ifdef HAVE_REGEX_H
 394         regex_t preg;
 395         int ret;
 396
 397         if (regcomp(&preg, regexp, REG_EXTENDED|REG_ICASE|REG_NOSUB) != 0)
 398                 return 0;
 399
 400         ret = regexec(&preg, str, 0, NULL, 0);
 401         regfree(&preg);
 402
 403         return ret == 0;
 404 #else
 405         return FALSE;
 406 #endif
 407 }
 408
 409 /* Create the directory and all it's parent directories */
 410 int mkpath(const char *path, int mode)
 411 {
 412         struct stat statbuf;
 413         const char *p;
 414         char *dir;
 415
 416         g_return_val_if_fail(path != NULL, -1);
 417
 418         p = g_path_skip_root((char *) path);
 419         if (p == NULL) {
 420                 /* not a full path, maybe not what we wanted
 421                    but continue anyway.. */
 422                 p = path;
 423         }
 424         for (;;) {
 425                 if (*p != G_DIR_SEPARATOR && *p != '\0') {
 426                         p++;
 427                         continue;
 428                 }
 429
 430                 dir = g_strndup(path, (int) (p-path));
 431                 if (stat(dir, &statbuf) != 0) {
 432 #ifndef WIN32
 433                         if (mkdir(dir, mode) == -1)
 434 #else
 435                         if (_mkdir(dir) == -1)
 436 #endif
 437                         {
 438                                 g_free(dir);
 439                                 return -1;
 440                         }
 441                 }
 442                 g_free(dir);
 443
 444                 if (*p++ == '\0')
 445                         break;
 446         }
 447
 448         return 0;
 449 }
 450
 451 /* convert ~/ to $HOME */
 452 char *convert_home(const char *path)
 453 {
 454         return *path == '~' && (*(path+1) == '/' || *(path+1) == '\0') ?
 455                 g_strconcat(g_get_home_dir(), path+1, NULL) :
 456                 g_strdup(path);
 457 }
 458
 459 int g_istr_equal(gconstpointer v, gconstpointer v2)
 460 {
 461         return g_strcasecmp((const char *) v, (const char *) v2) == 0;
 462 }
 463
 464 int g_istr_cmp(gconstpointer v, gconstpointer v2)
 465 {
 466         return g_strcasecmp((const char *) v, (const char *) v2);
 467 }
 468
 469 /* a char* hash function from ASU */
 470 unsigned int g_istr_hash(gconstpointer v)
 471 {
 472         const char *s = (const char *) v;
 473         unsigned int h = 0, g;
 474
 475         while (*s != '\0') {
 476                 h = (h << 4) + i_toupper(*s);
 477                 if ((g = h & 0xf0000000UL)) {
 478                         h = h ^ (g >> 24);
 479                         h = h ^ g;
 480                 }
 481                 s++;
 482         }
 483
 484         return h /* % M */;
 485 }
 486
 487 /* Find `mask' from `data', you can use * and ? wildcards. */
 488 int match_wildcards(const char *cmask, const char *data)
 489 {
 490         char *mask, *newmask, *p1, *p2;
 491         int ret;
 492
 493         newmask = mask = g_strdup(cmask);
 494         for (; *mask != '\0' && *data != '\0'; mask++) {
 495                 if (*mask != '*') {
 496                         if (*mask != '?' && i_toupper(*mask) != i_toupper(*data))
 497                                 break;
 498
 499                         data++;
 500                         continue;
 501                 }
 502
 503                 while (*mask == '?' || *mask == '*') mask++;
 504                 if (*mask == '\0') {
 505                         data += strlen(data);
 506                         break;
 507                 }
 508
 509                 p1 = strchr(mask, '*');
 510                 p2 = strchr(mask, '?');
 511                 if (p1 == NULL || (p2 < p1 && p2 != NULL)) p1 = p2;
 512
 513                 if (p1 != NULL) *p1 = '\0';
 514
 515                 data = stristr(data, mask);
 516                 if (data == NULL) break;
 517
 518                 data += strlen(mask);
 519                 mask += strlen(mask)-1;
 520
 521                 if (p1 != NULL) *p1 = p1 == p2 ? '?' : '*';
 522         }
 523
 524         while (*mask == '*') mask++;
 525
 526         ret = data != NULL && *data == '\0' && *mask == '\0';
 527         g_free(newmask);
 528
 529         return ret;
 530 }
 531
 532 /* Return TRUE if all characters in `str' are numbers.
 533    Stop when `end_char' is found from string. */
 534 int is_numeric(const char *str, char end_char)
 535 {
 536         g_return_val_if_fail(str != NULL, FALSE);
 537
 538         if (*str == '\0' || *str == end_char)
 539                 return FALSE;
 540
 541         while (*str != '\0' && *str != end_char) {
 542                 if (!i_isdigit(*str)) return FALSE;
 543                 str++;
 544         }
 545
 546         return TRUE;
 547 }
 548
 549 /* replace all `from' chars in string to `to' chars. returns `str' */
 550 char *replace_chars(char *str, char from, char to)
 551 {
 552         char *p;
 553
 554         for (p = str; *p != '\0'; p++) {
 555                 if (*p == from) *p = to;
 556         }
 557         return str;
 558 }
 559
 560 int octal2dec(int octal)
 561 {
 562         int dec, n;
 563
 564         dec = 0; n = 1;
 565         while (octal != 0) {
 566                 dec += n*(octal%10);
 567                 octal /= 10; n *= 8;
 568         }
 569
 570         return dec;
 571 }
 572
 573 int dec2octal(int decimal)
 574 {
 575         int octal, pos;
 576
 577         octal = 0; pos = 0;
 578         while (decimal > 0) {
 579                 octal += (decimal & 7)*(pos == 0 ? 1 : pos);
 580                 decimal /= 8;
 581                 pos += 10;
 582         }
 583
 584         return octal;
 585 }
 586
 587 /* convert all low-ascii (<32) to ^<A..> combinations */
 588 char *show_lowascii(const char *channel)
 589 {
 590         char *str, *p;
 591
 592         str = p = g_malloc(strlen(channel)*2+1);
 593         while (*channel != '\0') {
 594                 if ((unsigned char) *channel >= 32)
 595                         *p++ = *channel;
 596                 else {
 597                         *p++ = '^';
 598                         *p++ = *channel + 'A'-1;
 599                 }
 600                 channel++;
 601         }
 602         *p = '\0';
 603
 604         return str;
 605 }
 606
 607 /* Get time in human readable form with localtime() + asctime() */
 608 char *my_asctime(time_t t)
 609 {
 610         struct tm *tm;
 611         char *str;
 612         int len;
 613
 614         tm = localtime(&t);
 615         str = g_strdup(asctime(tm));
 616
 617         len = strlen(str);
 618         if (len > 0) str[len-1] = '\0';
 619         return str;
 620 }
 621
 622 /* Returns number of columns needed to print items.
 623    save_column_widths is filled with length of each column. */
 624 int get_max_column_count(GSList *items, COLUMN_LEN_FUNC len_func,
 625                          int max_width, int max_columns,
 626                          int item_extra, int item_min_size,
 627                          int **save_column_widths, int *rows)
 628 {
 629         GSList *tmp;
 630         int **columns, *columns_width, *columns_rows;
 631         int item_pos, items_count;
 632         int ret, len, max_len, n, col;
 633
 634         items_count = g_slist_length(items);
 635         if (items_count == 0) {
 636                 *save_column_widths = NULL;
 637                 *rows = 0;
 638                 return 0;
 639         }
 640
 641         len = max_width/(item_extra+item_min_size);
 642         if (len <= 0) len = 1;
 643         if (max_columns <= 0 || len < max_columns)
 644                 max_columns = len;
 645
 646         columns = g_new0(int *, max_columns);
 647         columns_width = g_new0(int, max_columns);
 648         columns_rows = g_new0(int, max_columns);
 649
 650         for (n = 1; n < max_columns; n++) {
 651                 columns[n] = g_new0(int, n+1);
 652                 columns_rows[n] = items_count <= n+1 ? 1 :
 653                         (items_count+n)/(n+1);
 654         }
 655
 656         /* for each possible column count, save the column widths and
 657            find the biggest column count that fits to screen. */
 658         item_pos = 0; max_len = 0;
 659         for (tmp = items; tmp != NULL; tmp = tmp->next) {
 660                 len = item_extra+len_func(tmp->data);
 661                 if (max_len < len)
 662                         max_len = len;
 663
 664                 for (n = 1; n < max_columns; n++) {
 665                         if (columns_width[n] > max_width)
 666                                 continue; /* too wide */
 667
 668                         col = item_pos/columns_rows[n];
 669                         if (columns[n][col] < len) {
 670                                 columns_width[n] += len-columns[n][col];
 671                                 columns[n][col] = len;
 672                         }
 673                 }
 674
 675                 item_pos++;
 676         }
 677
 678         for (n = max_columns-1; n >= 1; n--) {
 679                 if (columns_width[n] <= max_width &&
 680                     columns[n][n] > 0)
 681                         break;
 682         }
 683         ret = n+1;
 684
 685         *save_column_widths = g_new(int, ret);
 686         if (ret == 1) {
 687                 **save_column_widths = max_len;
 688                 *rows = 1;
 689         } else {
 690                 memcpy(*save_column_widths, columns[ret-1], sizeof(int)*ret);
 691                 *rows = columns_rows[ret-1];
 692         }
 693
 694         for (n = 1; n < max_columns; n++)
 695                 g_free(columns[n]);
 696         g_free(columns_width);
 697         g_free(columns_rows);
 698         g_free(columns);
 699
 700         return ret;
 701 }
 702
 703 /* Return a column sorted copy of a list. */
 704 GSList *columns_sort_list(GSList *list, int rows)
 705 {
 706         GSList *tmp, *sorted;
 707         int row, skip;
 708
 709         if (list == NULL || rows == 0)
 710                 return list;
 711
 712         sorted = NULL;
 713
 714         for (row = 0; row < rows; row++) {
 715                 tmp = g_slist_nth(list, row);
 716                 skip = 1;
 717                 for (; tmp != NULL; tmp = tmp->next) {
 718                         if (--skip == 0) {
 719                                 skip = rows;
 720                                 sorted = g_slist_append(sorted, tmp->data);
 721                         }
 722                 }
 723         }
 724
 725         g_return_val_if_fail(g_slist_length(sorted) ==
 726                              g_slist_length(list), sorted);
 727         return sorted;
 728 }
 729
 730 /* Expand escape string, the first character in data should be the
 731    one after '\'. Returns the expanded character or -1 if error. */
 732 int expand_escape(const char **data)
 733 {
 734         char digit[4];
 735
 736         switch (**data) {
 737         case 't':
 738                 return '\t';
 739         case 'r':
 740                 return '\r';
 741         case 'n':
 742                 return '\n';
 743         case 'e':
 744                 return 27; /* ESC */
 745
 746         case 'x':
 747                 /* hex digit */
 748                 if (!i_isxdigit((*data)[1]) || !i_isxdigit((*data)[2]))
 749                         return -1;
 750
 751                 digit[0] = (*data)[1];
 752                 digit[1] = (*data)[2];
 753                 digit[2] = '\0';
 754                 *data += 2;
 755                 return strtol(digit, NULL, 16);
 756         case 'c':
 757                 /* control character (\cA = ^A) */
 758                 (*data)++;
 759                 return i_toupper(**data) - 64;
 760         default:
 761                 if (!i_isdigit(**data))
 762                         return -1;
 763
 764                 /* octal */
 765                 digit[0] = (*data)[0];
 766                 digit[1] = (*data)[1];
 767                 digit[2] = (*data)[2];
 768                 digit[3] = '\0';
 769                 *data += 2;
 770                 return strtol(digit, NULL, 8);
 771         }
 772 }