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 (isspace((gint) *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 (toupper(data[pos]) == 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         (isspace((int) (c)) || ispunct((int) (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 ? (toupper(data[pos]) == 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                                 g_free(dir);
 438                                 return -1;
 439                         }
 440                 }
 441                 g_free(dir);
 442
 443                 if (*p++ == '\0')
 444                         break;
 445         }
 446
 447         return 0;
 448 }
 449
 450 /* convert ~/ to $HOME */
 451 char *convert_home(const char *path)
 452 {
 453         return *path == '~' && (*(path+1) == '/' || *(path+1) == '\0') ?
 454                 g_strconcat(g_get_home_dir(), path+1, NULL) :
 455                 g_strdup(path);
 456 }
 457
 458 int g_istr_equal(gconstpointer v, gconstpointer v2)
 459 {
 460         return g_strcasecmp((const char *) v, (const char *) v2) == 0;
 461 }
 462
 463 int g_istr_cmp(gconstpointer v, gconstpointer v2)
 464 {
 465         return g_strcasecmp((const char *) v, (const char *) v2);
 466 }
 467
 468 /* a char* hash function from ASU */
 469 unsigned int g_istr_hash(gconstpointer v)
 470 {
 471         const char *s = (const char *) v;
 472         unsigned int h = 0, g;
 473
 474         while (*s != '\0') {
 475                 h = (h << 4) + toupper(*s);
 476                 if ((g = h & 0xf0000000UL)) {
 477                         h = h ^ (g >> 24);
 478                         h = h ^ g;
 479                 }
 480                 s++;
 481         }
 482
 483         return h /* % M */;
 484 }
 485
 486 /* Find `mask' from `data', you can use * and ? wildcards. */
 487 int match_wildcards(const char *cmask, const char *data)
 488 {
 489         char *mask, *newmask, *p1, *p2;
 490         int ret;
 491
 492         newmask = mask = g_strdup(cmask);
 493         for (; *mask != '\0' && *data != '\0'; mask++) {
 494                 if (*mask != '*') {
 495                         if (*mask != '?' && toupper(*mask) != toupper(*data))
 496                                 break;
 497
 498                         data++;
 499                         continue;
 500                 }
 501
 502                 while (*mask == '?' || *mask == '*') mask++;
 503                 if (*mask == '\0') {
 504                         data += strlen(data);
 505                         break;
 506                 }
 507
 508                 p1 = strchr(mask, '*');
 509                 p2 = strchr(mask, '?');
 510                 if (p1 == NULL || (p2 < p1 && p2 != NULL)) p1 = p2;
 511
 512                 if (p1 != NULL) *p1 = '\0';
 513
 514                 data = stristr(data, mask);
 515                 if (data == NULL) break;
 516
 517                 data += strlen(mask);
 518                 mask += strlen(mask)-1;
 519
 520                 if (p1 != NULL) *p1 = p1 == p2 ? '?' : '*';
 521         }
 522
 523         while (*mask == '*') mask++;
 524
 525         ret = data != NULL && *data == '\0' && *mask == '\0';
 526         g_free(newmask);
 527
 528         return ret;
 529 }
 530
 531 /* Return TRUE if all characters in `str' are numbers.
 532    Stop when `end_char' is found from string. */
 533 int is_numeric(const char *str, char end_char)
 534 {
 535         g_return_val_if_fail(str != NULL, FALSE);
 536
 537         if (*str == '\0' || *str == end_char)
 538                 return FALSE;
 539
 540         while (*str != '\0' && *str != end_char) {
 541                 if (!isdigit(*str)) return FALSE;
 542                 str++;
 543         }
 544
 545         return TRUE;
 546 }
 547
 548 /* replace all `from' chars in string to `to' chars. returns `str' */
 549 char *replace_chars(char *str, char from, char to)
 550 {
 551         char *p;
 552
 553         for (p = str; *p != '\0'; p++) {
 554                 if (*p == from) *p = to;
 555         }
 556         return str;
 557 }
 558
 559 int octal2dec(int octal)
 560 {
 561         int dec, n;
 562
 563         dec = 0; n = 1;
 564         while (octal != 0) {
 565                 dec += n*(octal%10);
 566                 octal /= 10; n *= 8;
 567         }
 568
 569         return dec;
 570 }
 571
 572 int dec2octal(int decimal)
 573 {
 574         int octal, pos;
 575
 576         octal = 0; pos = 0;
 577         while (decimal > 0) {
 578                 octal += (decimal & 7)*(pos == 0 ? 1 : pos);
 579                 decimal /= 8;
 580                 pos += 10;
 581         }
 582
 583         return octal;
 584 }
 585
 586 /* convert all low-ascii (<32) to ^<A..> combinations */
 587 char *show_lowascii(const char *channel)
 588 {
 589         char *str, *p;
 590
 591         str = p = g_malloc(strlen(channel)*2+1);
 592         while (*channel != '\0') {
 593                 if ((unsigned char) *channel >= 32)
 594                         *p++ = *channel;
 595                 else {
 596                         *p++ = '^';
 597                         *p++ = *channel + 'A'-1;
 598                 }
 599                 channel++;
 600         }
 601         *p = '\0';
 602
 603         return str;
 604 }
 605
 606 /* Get time in human readable form with localtime() + asctime() */
 607 char *my_asctime(time_t t)
 608 {
 609         struct tm *tm;
 610         char *str;
 611         int len;
 612
 613         tm = localtime(&t);
 614         str = g_strdup(asctime(tm));
 615
 616         len = strlen(str);
 617         if (len > 0) str[len-1] = '\0';
 618         return str;
 619 }
 620
 621 /* Returns number of columns needed to print items.
 622    save_column_widths is filled with length of each column. */
 623 int get_max_column_count(GSList *items, COLUMN_LEN_FUNC len_func,
 624                          int max_width, int max_columns,
 625                          int item_extra, int item_min_size,
 626                          int **save_column_widths, int *rows)
 627 {
 628         GSList *tmp;
 629         int **columns, *columns_width, *columns_rows;
 630         int item_pos, items_count;
 631         int ret, len, max_len, n, col;
 632
 633         items_count = g_slist_length(items);
 634         if (items_count == 0) {
 635                 *save_column_widths = NULL;
 636                 *rows = 0;
 637                 return 0;
 638         }
 639
 640         len = max_width/(item_extra+item_min_size);
 641         if (len <= 0) len = 1;
 642         if (max_columns <= 0 || len < max_columns)
 643                 max_columns = len;
 644
 645         columns = g_new0(int *, max_columns);
 646         columns_width = g_new0(int, max_columns);
 647         columns_rows = g_new0(int, max_columns);
 648
 649         for (n = 1; n < max_columns; n++) {
 650                 columns[n] = g_new0(int, n+1);
 651                 columns_rows[n] = items_count <= n+1 ? 1 :
 652                         (items_count+n)/(n+1);
 653         }
 654
 655         /* for each possible column count, save the column widths and
 656            find the biggest column count that fits to screen. */
 657         item_pos = 0; max_len = 0;
 658         for (tmp = items; tmp != NULL; tmp = tmp->next) {
 659                 len = item_extra+len_func(tmp->data);
 660                 if (max_len < len)
 661                         max_len = len;
 662
 663                 for (n = 1; n < max_columns; n++) {
 664                         if (columns_width[n] > max_width)
 665                                 continue; /* too wide */
 666
 667                         col = item_pos/columns_rows[n];
 668                         if (columns[n][col] < len) {
 669                                 columns_width[n] += len-columns[n][col];
 670                                 columns[n][col] = len;
 671                         }
 672                 }
 673
 674                 item_pos++;
 675         }
 676
 677         for (n = max_columns-1; n >= 1; n--) {
 678                 if (columns_width[n] <= max_width &&
 679                     columns[n][n] > 0)
 680                         break;
 681         }
 682         ret = n+1;
 683
 684         *save_column_widths = g_new(int, ret);
 685         if (ret == 1) {
 686                 **save_column_widths = max_len;
 687                 *rows = 1;
 688         } else {
 689                 memcpy(*save_column_widths, columns[ret-1], sizeof(int)*ret);
 690                 *rows = columns_rows[ret-1];
 691         }
 692
 693         for (n = 1; n < max_columns; n++)
 694                 g_free(columns[n]);
 695         g_free(columns_width);
 696         g_free(columns_rows);
 697         g_free(columns);
 698
 699         return ret;
 700 }
 701
 702 /* Return a column sorted copy of a list. */
 703 GSList *columns_sort_list(GSList *list, int rows)
 704 {
 705         GSList *tmp, *sorted;
 706         int row, skip;
 707
 708         if (list == NULL || rows == 0)
 709                 return list;
 710
 711         sorted = NULL;
 712
 713         for (row = 0; row < rows; row++) {
 714                 tmp = g_slist_nth(list, row);
 715                 skip = 1;
 716                 for (; tmp != NULL; tmp = tmp->next) {
 717                         if (--skip == 0) {
 718                                 skip = rows;
 719                                 sorted = g_slist_append(sorted, tmp->data);
 720                         }
 721                 }
 722         }
 723
 724         g_return_val_if_fail(g_slist_length(sorted) ==
 725                              g_slist_length(list), sorted);
 726         return sorted;
 727 }