source: util.c @ 1672650

release-1.6release-1.7release-1.8release-1.9
Last change on this file since 1672650 was 1672650, checked in by Anders Kaseorg <andersk@mit.edu>, 11 years ago
Rename atokenize_free to atokenize_delete. Signed-off-by: Anders Kaseorg <andersk@mit.edu> Reviewed-by: Nelson Elhage <nelhage@mit.edu>
  • Property mode set to 100644
File size: 17.2 KB
Line 
1#include "owl.h"
2#include <stdlib.h>
3#include <string.h>
4#include <unistd.h>
5#include <ctype.h>
6#include <pwd.h>
7#include <sys/stat.h>
8#include <sys/types.h>
9
10void sepbar(const char *in)
11{
12  WINDOW *sepwin;
13  const owl_messagelist *ml;
14  const owl_view *v;
15  int x, y, i;
16  const char *foo, *appendtosepbar;
17
18  sepwin=owl_global_get_curs_sepwin(&g);
19  ml=owl_global_get_msglist(&g);
20  v=owl_global_get_current_view(&g);
21
22  werase(sepwin);
23  wattron(sepwin, A_REVERSE);
24  if (owl_global_is_fancylines(&g)) {
25    whline(sepwin, ACS_HLINE, owl_global_get_cols(&g));
26  } else {
27    whline(sepwin, '-', owl_global_get_cols(&g));
28  }
29
30  if (owl_global_is_sepbar_disable(&g)) {
31    getyx(sepwin, y, x);
32    wmove(sepwin, y, owl_global_get_cols(&g)-1);
33    return;
34  }
35
36  wmove(sepwin, 0, 2);
37
38  if (owl_messagelist_get_size(ml) == 0)
39    waddstr(sepwin, " (-/-) ");
40  else
41    wprintw(sepwin, " (%i/%i/%i) ", owl_global_get_curmsg(&g) + 1,
42            owl_view_get_size(v),
43            owl_messagelist_get_size(ml));
44
45  foo=owl_view_get_filtname(v);
46  if (strcmp(foo, owl_global_get_view_home(&g)))
47      wattroff(sepwin, A_REVERSE);
48  wprintw(sepwin, " %s ", owl_view_get_filtname(v));
49  if (strcmp(foo, owl_global_get_view_home(&g)))
50      wattron(sepwin, A_REVERSE);
51
52  if (owl_mainwin_is_curmsg_truncated(owl_global_get_mainwin(&g))) {
53    getyx(sepwin, y, x);
54    wmove(sepwin, y, x+2);
55    wattron(sepwin, A_BOLD);
56    waddstr(sepwin, " <truncated> ");
57    wattroff(sepwin, A_BOLD);
58  }
59
60  i=owl_mainwin_get_last_msg(owl_global_get_mainwin(&g));
61  if ((i != -1) &&
62      (i < owl_view_get_size(v)-1)) {
63    getyx(sepwin, y, x);
64    wmove(sepwin, y, x+2);
65    wattron(sepwin, A_BOLD);
66    waddstr(sepwin, " <more> ");
67    wattroff(sepwin, A_BOLD);
68  }
69
70  if (owl_global_get_rightshift(&g)>0) {
71    getyx(sepwin, y, x);
72    wmove(sepwin, y, x+2);
73    wprintw(sepwin, " right: %i ", owl_global_get_rightshift(&g));
74  }
75
76  if (owl_global_is_zaway(&g) || owl_global_is_aaway(&g)) {
77    getyx(sepwin, y, x);
78    wmove(sepwin, y, x+2);
79    wattron(sepwin, A_BOLD);
80    wattroff(sepwin, A_REVERSE);
81    if (owl_global_is_zaway(&g) && owl_global_is_aaway(&g)) {
82      waddstr(sepwin, " AWAY ");
83    } else if (owl_global_is_zaway(&g)) {
84      waddstr(sepwin, " Z-AWAY ");
85    } else if (owl_global_is_aaway(&g)) {
86      waddstr(sepwin, " A-AWAY ");
87    }
88    wattron(sepwin, A_REVERSE);
89    wattroff(sepwin, A_BOLD);
90  }
91
92  if (owl_global_get_curmsg_vert_offset(&g)) {
93    getyx(sepwin, y, x);
94    wmove(sepwin, y, x+2);
95    wattron(sepwin, A_BOLD);
96    wattroff(sepwin, A_REVERSE);
97    waddstr(sepwin, " SCROLL ");
98    wattron(sepwin, A_REVERSE);
99    wattroff(sepwin, A_BOLD);
100  }
101 
102  if (in) {
103    getyx(sepwin, y, x);
104    wmove(sepwin, y, x+2);
105    waddstr(sepwin, in);
106  }
107
108  appendtosepbar = owl_global_get_appendtosepbar(&g);
109  if (appendtosepbar && *appendtosepbar) {
110    getyx(sepwin, y, x);
111    wmove(sepwin, y, x+2);
112    waddstr(sepwin, " ");
113    waddstr(sepwin, owl_global_get_appendtosepbar(&g));
114    waddstr(sepwin, " ");
115  }
116
117  getyx(sepwin, y, x);
118  wmove(sepwin, y, owl_global_get_cols(&g)-1);
119   
120  wattroff(sepwin, A_BOLD);
121  wattroff(sepwin, A_REVERSE);
122  update_panels();
123}
124
125char **atokenize(const char *buffer, const char *sep, int *i)
126{
127  /* each element of return must be freed by user */
128  char **args;
129  char *workbuff, *foo;
130  int done=0, first=1, count=0;
131
132  workbuff=owl_malloc(strlen(buffer)+1);
133  memcpy(workbuff, buffer, strlen(buffer)+1);
134
135  args=NULL;
136  while (!done) {
137    if (first) {
138      first=0;
139      foo=strtok(workbuff, sep);
140    } else {
141      foo=strtok(NULL, sep);
142    }
143    if (foo==NULL) {
144      done=1;
145    } else {
146      args=owl_realloc(args, sizeof(char *) * (count+1));
147      args[count] = owl_strdup(foo);
148      count++;
149    }
150  }
151  *i=count;
152  owl_free(workbuff);
153  return(args);
154}
155
156const char *skiptokens(const char *buff, int n) {
157  /* skips n tokens and returns where that would be. */
158  char quote = 0;
159  while (*buff && n>0) {
160      while (*buff == ' ') buff++;
161      while (*buff && (quote || *buff != ' ')) {
162        if(quote) {
163          if(*buff == quote) quote = 0;
164        } else if(*buff == '"' || *buff == '\'') {
165          quote = *buff;
166        }
167        buff++;
168      }
169      while (*buff == ' ') buff++;
170      n--;
171  }
172  return buff;
173}
174
175/* Return a "nice" version of the path.  Tilde expansion is done, and
176 * duplicate slashes are removed.  Caller must free the return.
177 */
178char *owl_util_makepath(const char *in)
179{
180  int i, j, x;
181  char *out, user[MAXPATHLEN];
182  struct passwd *pw;
183
184  out=owl_malloc(MAXPATHLEN+1);
185  out[0]='\0';
186  j=strlen(in);
187  x=0;
188  for (i=0; i<j; i++) {
189    if (in[i]=='~') {
190      if ( (i==(j-1)) ||          /* last character */
191           (in[i+1]=='/') ) {     /* ~/ */
192        /* use my homedir */
193        pw=getpwuid(getuid());
194        if (!pw) {
195          out[x]=in[i];
196        } else {
197          out[x]='\0';
198          strcat(out, pw->pw_dir);
199          x+=strlen(pw->pw_dir);
200        }
201      } else {
202        /* another user homedir */
203        int a, b;
204        b=0;
205        for (a=i+1; i<j; a++) {
206          if (in[a]==' ' || in[a]=='/') {
207            break;
208          } else {
209            user[b]=in[a];
210            i++;
211            b++;
212          }
213        }
214        user[b]='\0';
215        pw=getpwnam(user);
216        if (!pw) {
217          out[x]=in[i];
218        } else {
219          out[x]='\0';
220          strcat(out, pw->pw_dir);
221          x+=strlen(pw->pw_dir);
222        }
223      }
224    } else if (in[i]=='/') {
225      /* check for a double / */
226      if (i<(j-1) && (in[i+1]=='/')) {
227        /* do nothing */
228      } else {
229        out[x]=in[i];
230        x++;
231      }
232    } else {
233      out[x]=in[i];
234      x++;
235    }
236  }
237  out[x]='\0';
238  return(out);
239}
240
241void atokenize_delete(char **tok, int nels)
242{
243  int i;
244  for (i=0; i<nels; i++) {
245    owl_free(tok[i]);
246  }
247  owl_free(tok);
248}
249
250
251void owl_parsefree(char **argv, int argc)
252{
253  int i;
254
255  if (!argv) return;
256 
257  for (i=0; i<argc; i++) {
258    if (argv[i]) owl_free(argv[i]);
259  }
260  owl_free(argv);
261}
262
263char **owl_parseline(const char *line, int *argc)
264{
265  /* break a command line up into argv, argc.  The caller must free
266     the returned values.  If there is an error argc will be set to
267     -1, argv will be NULL and the caller does not need to free
268     anything */
269
270  char **argv;
271  int i, len, between=1;
272  char *curarg;
273  char quote;
274
275  argv=owl_malloc(sizeof(char *));
276  len=strlen(line);
277  curarg=owl_malloc(len+10);
278  strcpy(curarg, "");
279  quote='\0';
280  *argc=0;
281  for (i=0; i<len+1; i++) {
282    /* find the first real character */
283    if (between) {
284      if (line[i]==' ' || line[i]=='\t' || line[i]=='\0') {
285        continue;
286      } else {
287        between=0;
288        i--;
289        continue;
290      }
291    }
292
293    /* deal with a quote character */
294    if (line[i]=='"' || line[i]=="'"[0]) {
295      /* if this type of quote is open, close it */
296      if (quote==line[i]) {
297        quote='\0';
298        continue;
299      }
300
301      /* if no quoting is open then open with this */
302      if (quote=='\0') {
303        quote=line[i];
304        continue;
305      }
306
307      /* if another type of quote is open then treat this as a literal */
308      curarg[strlen(curarg)+1]='\0';
309      curarg[strlen(curarg)]=line[i];
310      continue;
311    }
312
313    /* if it's not a space or end of command, then use it */
314    if (line[i]!=' ' && line[i]!='\t' && line[i]!='\n' && line[i]!='\0') {
315      curarg[strlen(curarg)+1]='\0';
316      curarg[strlen(curarg)]=line[i];
317      continue;
318    }
319
320    /* otherwise, if we're not in quotes, add the whole argument */
321    if (quote=='\0') {
322      /* add the argument */
323      argv=owl_realloc(argv, sizeof(char *)*((*argc)+1));
324      argv[*argc] = owl_strdup(curarg);
325      *argc=*argc+1;
326      strcpy(curarg, "");
327      between=1;
328      continue;
329    }
330
331    /* if it is a space and we're in quotes, then use it */
332    curarg[strlen(curarg)+1]='\0';
333    curarg[strlen(curarg)]=line[i];
334  }
335
336  owl_free(curarg);
337
338  /* check for unbalanced quotes */
339  if (quote!='\0') {
340    owl_parsefree(argv, *argc);
341    *argc=-1;
342    return(NULL);
343  }
344
345  return(argv);
346}
347
348/* caller must free the return */
349char *owl_util_minutes_to_timestr(int in)
350{
351  int days, hours;
352  long run;
353  char *out;
354
355  run=in;
356
357  days=run/1440;
358  run-=days*1440;
359  hours=run/60;
360  run-=hours*60;
361
362  if (days>0) {
363    out=owl_sprintf("%i d %2.2i:%2.2li", days, hours, run);
364  } else {
365    out=owl_sprintf("    %2.2i:%2.2li", hours, run);
366  }
367  return(out);
368}
369
370/* hooks for doing memory allocation et. al. in owl */
371
372void *owl_malloc(size_t size)
373{
374  return(g_malloc(size));
375}
376
377void owl_free(void *ptr)
378{
379  g_free(ptr);
380}
381
382char *owl_strdup(const char *s1)
383{
384  return(g_strdup(s1));
385}
386
387void *owl_realloc(void *ptr, size_t size)
388{
389  return(g_realloc(ptr, size));
390}
391
392/* allocates memory and returns the string or null.
393 * caller must free the string.
394 */
395char *owl_sprintf(const char *fmt, ...)
396{
397  va_list ap;
398  char *ret = NULL;
399  va_start(ap, fmt);
400  ret = g_strdup_vprintf(fmt, ap);
401  va_end(ap);
402  return ret;
403}
404
405/* These are in order of their value in owl.h */
406static const struct {
407  int number;
408  const char *name;
409} color_map[] = {
410  {OWL_COLOR_INVALID, "invalid"},
411  {OWL_COLOR_DEFAULT, "default"},
412  {OWL_COLOR_BLACK, "black"},
413  {OWL_COLOR_RED, "red"},
414  {OWL_COLOR_GREEN, "green"},
415  {OWL_COLOR_YELLOW,"yellow"},
416  {OWL_COLOR_BLUE, "blue"},
417  {OWL_COLOR_MAGENTA, "magenta"},
418  {OWL_COLOR_CYAN, "cyan"},
419  {OWL_COLOR_WHITE, "white"},
420};
421
422/* Return the owl color associated with the named color.  Return -1
423 * if the named color is not available
424 */
425int owl_util_string_to_color(const char *color)
426{
427  int c, i;
428  char *p;
429
430  for (i = 0; i < (sizeof(color_map)/sizeof(color_map[0])); i++)
431    if (strcasecmp(color, color_map[i].name) == 0)
432      return color_map[i].number;
433
434  c = strtol(color, &p, 10);
435  if (p != color && c >= -1 && c < COLORS) {
436    return(c);
437  }
438  return(OWL_COLOR_INVALID);
439}
440
441/* Return a string name of the given owl color */
442const char *owl_util_color_to_string(int color)
443{
444  if (color >= OWL_COLOR_INVALID && color <= OWL_COLOR_WHITE)
445    return color_map[color - OWL_COLOR_INVALID].name;
446  return("Unknown color");
447}
448
449/* Get the default tty name.  Caller must free the return */
450char *owl_util_get_default_tty(void)
451{
452  const char *tmp;
453  char *out;
454
455  if (getenv("DISPLAY")) {
456    out=owl_strdup(getenv("DISPLAY"));
457  } else if ((tmp=ttyname(fileno(stdout)))!=NULL) {
458    out=owl_strdup(tmp);
459    if (!strncmp(out, "/dev/", 5)) {
460      owl_free(out);
461      out=owl_strdup(tmp+5);
462    }
463  } else {
464    out=owl_strdup("unknown");
465  }
466  return(out);
467}
468
469/* strip leading and trailing new lines.  Caller must free the
470 * return.
471 */
472char *owl_util_stripnewlines(const char *in)
473{
474 
475  char  *tmp, *ptr1, *ptr2, *out;
476
477  ptr1=tmp=owl_strdup(in);
478  while (ptr1[0]=='\n') {
479    ptr1++;
480  }
481  ptr2=ptr1+strlen(ptr1)-1;
482  while (ptr2>ptr1 && ptr2[0]=='\n') {
483    ptr2[0]='\0';
484    ptr2--;
485  }
486
487  out=owl_strdup(ptr1);
488  owl_free(tmp);
489  return(out);
490}
491
492/* Delete all lines matching "line" from the named file.  If no such
493 * line is found the file is left intact.  If backup==1 then leave a
494 * backup file containing the original contents.  The match is
495 * case-insensitive.
496 *
497 * Returns the number of lines removed
498 */
499int owl_util_file_deleteline(const char *filename, const char *line, int backup)
500{
501  char *backupfile, *newfile, *buf = NULL;
502  FILE *old, *new;
503  struct stat st;
504  int numremoved = 0;
505
506  if ((old = fopen(filename, "r")) == NULL) {
507    owl_function_error("Cannot open %s (for reading): %s",
508                       filename, strerror(errno));
509    return 0;
510  }
511
512  if (fstat(fileno(old), &st) != 0) {
513    owl_function_error("Cannot stat %s: %s", filename, strerror(errno));
514    return 0;
515  }
516
517  newfile = owl_sprintf("%s.new", filename);
518  if ((new = fopen(newfile, "w")) == NULL) {
519    owl_function_error("Cannot open %s (for writing): %s",
520                       filename, strerror(errno));
521    free(newfile);
522    fclose(old);
523    return 0;
524  }
525
526  if (fchmod(fileno(new), st.st_mode & 0777) != 0) {
527    owl_function_error("Cannot set permissions on %s: %s",
528                       filename, strerror(errno));
529    unlink(newfile);
530    fclose(new);
531    free(newfile);
532    fclose(old);
533    return 0;
534  }
535
536  while (owl_getline_chomp(&buf, old))
537    if (strcasecmp(buf, line) != 0)
538      fprintf(new, "%s\n", buf);
539    else
540      numremoved++;
541  owl_free(buf);
542
543  fclose(new);
544  fclose(old);
545
546  if (backup) {
547    backupfile = owl_sprintf("%s.backup", filename);
548    unlink(backupfile);
549    if (link(filename, backupfile) != 0) {
550      owl_function_error("Cannot link %s: %s", backupfile, strerror(errno));
551      owl_free(backupfile);
552      unlink(newfile);
553      owl_free(newfile);
554      return 0;
555    }
556    owl_free(backupfile);
557  }
558
559  if (rename(newfile, filename) != 0) {
560    owl_function_error("Cannot move %s to %s: %s",
561                       newfile, filename, strerror(errno));
562    numremoved = 0;
563  }
564
565  unlink(newfile);
566  owl_free(newfile);
567
568  return numremoved;
569}
570
571int owl_util_max(int a, int b)
572{
573  if (a>b) return(a);
574  return(b);
575}
576
577int owl_util_min(int a, int b)
578{
579  if (a<b) return(a);
580  return(b);
581}
582
583/* Return the base class or instance from a zephyr class, by removing
584   leading `un' or trailing `.d'.
585   The caller is responsible for freeing the allocated string.
586*/
587char * owl_util_baseclass(const char * class)
588{
589  char *start, *end;
590
591  while(!strncmp(class, "un", 2)) {
592    class += 2;
593  }
594
595  start = owl_strdup(class);
596  end = start + strlen(start) - 1;
597  while(end > start && *end == 'd' && *(end-1) == '.') {
598    end -= 2;
599  }
600  *(end + 1) = 0;
601
602  return start;
603}
604
605const char * owl_get_datadir(void)
606{
607  const char * datadir = getenv("BARNOWL_DATA_DIR");
608  if(datadir != NULL)
609    return datadir;
610  return DATADIR;
611}
612
613/* Strips format characters from a valid utf-8 string. Returns the
614   empty string if 'in' does not validate. */
615char * owl_strip_format_chars(const char *in)
616{
617  char *r;
618  if (g_utf8_validate(in, -1, NULL)) {
619    const char *s, *p;
620    r = owl_malloc(strlen(in)+1);
621    r[0] = '\0';
622    s = in;
623    p = strchr(s, OWL_FMTEXT_UC_STARTBYTE_UTF8);
624    while(p) {
625      /* If it's a format character, copy up to it, and skip all
626         immediately following format characters. */
627      if (owl_fmtext_is_format_char(g_utf8_get_char(p))) {
628        strncat(r, s, p-s);
629        p = g_utf8_next_char(p);
630        while (owl_fmtext_is_format_char(g_utf8_get_char(p))) {
631          p = g_utf8_next_char(p);
632        }
633        s = p;
634        p = strchr(s, OWL_FMTEXT_UC_STARTBYTE_UTF8);
635      }
636      else {
637        p = strchr(p+1, OWL_FMTEXT_UC_STARTBYTE_UTF8);
638      }
639    }
640    if (s) strcat(r,s);
641  }
642  else {
643    r = owl_strdup("");
644  }
645  return r;
646}
647
648/* If in is not UTF-8, convert from ISO-8859-1. We may want to allow
649 * the caller to specify an alternative in the future. We also strip
650 * out characters in Unicode Plane 16, as we use that plane internally
651 * for formatting.
652 */
653char * owl_validate_or_convert(const char *in)
654{
655  if (g_utf8_validate(in, -1, NULL)) {
656    return owl_strip_format_chars(in);
657  }
658  else {
659    return g_convert(in, -1,
660                     "UTF-8", "ISO-8859-1",
661                     NULL, NULL, NULL);
662  }
663}
664/*
665 * Validate 'in' as UTF-8, and either return a copy of it, or an empty
666 * string if it is invalid utf-8.
667 */
668char * owl_validate_utf8(const char *in)
669{
670  char *out;
671  if (g_utf8_validate(in, -1, NULL)) {
672    out = owl_strdup(in);
673  } else {
674    out = owl_strdup("");
675  }
676  return out;
677}
678
679/* This is based on _extract() and _isCJ() from perl's Text::WrapI18N */
680int owl_util_can_break_after(gunichar c)
681{
682 
683  if (c == ' ') return 1;
684  if (c >= 0x3000 && c <= 0x312f) {
685    /* CJK punctuations, Hiragana, Katakana, Bopomofo */
686    if (c == 0x300a || c == 0x300c || c == 0x300e ||
687        c == 0x3010 || c == 0x3014 || c == 0x3016 ||
688        c == 0x3018 || c == 0x301a)
689      return 0;
690    return 1;
691  }
692  if (c >= 0x31a0 && c <= 0x31bf) {return 1;}  /* Bopomofo */
693  if (c >= 0x31f0 && c <= 0x31ff) {return 1;}  /* Katakana extension */
694  if (c >= 0x3400 && c <= 0x9fff) {return 1;}  /* Han Ideogram */
695  if (c >= 0xf900 && c <= 0xfaff) {return 1;}  /* Han Ideogram */
696  if (c >= 0x20000 && c <= 0x2ffff) {return 1;}  /* Han Ideogram */
697  return 0;
698}
699
700char *owl_escape_highbit(const char *str)
701{
702  GString *out = g_string_new("");
703  unsigned char c;
704  while((c = (*str++))) {
705    if(c == '\\') {
706      g_string_append(out, "\\\\");
707    } else if(c & 0x80) {
708      g_string_append_printf(out, "\\x%02x", (int)c);
709    } else {
710      g_string_append_c(out, c);
711    }
712  }
713  return g_string_free(out, 0);
714}
715
716/* innards of owl_getline{,_chomp} below */
717static int owl_getline_internal(char **s, FILE *fp, int newline)
718{
719  int size = 0;
720  int target = 0;
721  int count = 0;
722  int c;
723
724  while (1) {
725    c = getc(fp);
726    if ((target + 1) > size) {
727      size += BUFSIZ;
728      *s = owl_realloc(*s, size);
729    }
730    if (c == EOF)
731      break;
732    count++;
733    if (c != '\n' || newline)
734        (*s)[target++] = c;
735    if (c == '\n')
736      break;
737  }
738  (*s)[target] = 0;
739
740  return count;
741}
742
743/* Read a line from fp, allocating memory to hold it, returning the number of
744 * byte read.  *s should either be NULL or a pointer to memory allocated with
745 * owl_malloc; it will be owl_realloc'd as appropriate.  The caller must
746 * eventually free it.  (This is roughly the interface of getline in the gnu
747 * libc).
748 *
749 * The final newline will be included if it's there.
750 */
751int owl_getline(char **s, FILE *fp)
752{
753  return owl_getline_internal(s, fp, 1);
754}
755
756/* As above, but omitting the final newline */
757int owl_getline_chomp(char **s, FILE *fp)
758{
759  return owl_getline_internal(s, fp, 0);
760}
761
762/* Read the rest of the input available in fp into a string. */
763char *owl_slurp(FILE *fp)
764{
765  char *buf = NULL;
766  char *p;
767  int size = 0;
768  int count;
769
770  while (1) {
771    buf = owl_realloc(buf, size + BUFSIZ);
772    p = &buf[size];
773    size += BUFSIZ;
774
775    if ((count = fread(p, 1, BUFSIZ, fp)) < BUFSIZ)
776      break;
777  }
778  p[count] = 0;
779
780  return buf;
781}
Note: See TracBrowser for help on using the repository browser.