source: zwrite.c @ 7dcef03

Last change on this file since 7dcef03 was 7dcef03, checked in by Anders Kaseorg <andersk@mit.edu>, 8 years ago
Use the Glib slice allocator for fixed-size objects The slice allocator, available since GLib 2.10, is more space-efficient than [g_]malloc. Since BarnOwl is obviously at the leading edge of space-efficient technology, this seems like a natural fit. Use it for every fixed-size object except owl_viewwin_search_data (which would need an extra destroy_cbdata function to g_slice_free it). Signed-off-by: Anders Kaseorg <andersk@mit.edu>
  • Property mode set to 100644
File size: 9.4 KB
RevLine 
[7d4fbcd]1#include "owl.h"
2
[d953ede]3CALLER_OWN owl_zwrite *owl_zwrite_new_from_line(const char *line)
[987cf3f]4{
[7dcef03]5  owl_zwrite *z = g_slice_new(owl_zwrite);
[ef4074b]6  if (owl_zwrite_create_from_line(z, line) != 0) {
[7dcef03]7    g_slice_free(owl_zwrite, z);
[987cf3f]8    return NULL;
9  }
10  return z;
11}
12
[6329cd5]13CALLER_OWN owl_zwrite *owl_zwrite_new(int argc, const char *const *argv)
14{
[7dcef03]15  owl_zwrite *z = g_slice_new(owl_zwrite);
[ef4074b]16  if (owl_zwrite_create(z, argc, argv) != 0) {
[7dcef03]17    g_slice_free(owl_zwrite, z);
[6329cd5]18    return NULL;
19  }
20  return z;
21}
22
[d427f08]23G_GNUC_WARN_UNUSED_RESULT int owl_zwrite_create_from_line(owl_zwrite *z, const char *line)
[ce7db4d]24{
[6329cd5]25  int argc;
[65b2173]26  char **argv;
[6329cd5]27  int ret;
[7d4fbcd]28
[6329cd5]29  /* parse the command line for options */
30  argv = owl_parseline(line, &argc);
31  if (argc < 0) {
32    owl_function_error("Unbalanced quotes in zwrite");
33    return -1;
34  }
35  ret = owl_zwrite_create(z, argc, strs(argv));
36  g_strfreev(argv);
37  return ret;
38}
39
40G_GNUC_WARN_UNUSED_RESULT int owl_zwrite_create(owl_zwrite *z, int argc, const char *const *argv)
41{
42  int badargs = 0;
43  char *msg = NULL;
[7d4fbcd]44 
[69f89c7]45  /* start with null entries */
[987cf3f]46  z->cmd=NULL;
[ce7db4d]47  z->realm=NULL;
48  z->class=NULL;
49  z->inst=NULL;
50  z->opcode=NULL;
51  z->zsig=NULL;
52  z->message=NULL;
[7d4fbcd]53  z->cc=0;
54  z->noping=0;
[12294d2]55  z->recips = g_ptr_array_new();
[6329cd5]56  z->zwriteline = owl_argv_quote(argc, argv);
[7d4fbcd]57
[6329cd5]58  if (argc && *(argv[0])!='-') {
59    z->cmd=g_strdup(argv[0]);
60    argc--;
61    argv++;
[e1c4636]62  }
[6329cd5]63  while (argc) {
64    if (!strcmp(argv[0], "-c")) {
65      if (argc<2) {
[7d4fbcd]66        badargs=1;
67        break;
68      }
[6329cd5]69      z->class=owl_validate_utf8(argv[1]);
70      argv+=2;
71      argc-=2;
72    } else if (!strcmp(argv[0], "-i")) {
73      if (argc<2) {
[7d4fbcd]74        badargs=1;
75        break;
76      }
[6329cd5]77      z->inst=owl_validate_utf8(argv[1]);
78      argv+=2;
79      argc-=2;
80    } else if (!strcmp(argv[0], "-r")) {
81      if (argc<2) {
[7d4fbcd]82        badargs=1;
83        break;
84      }
[6329cd5]85      z->realm=owl_validate_utf8(argv[1]);
86      argv+=2;
87      argc-=2;
88    } else if (!strcmp(argv[0], "-s")) {
89      if (argc<2) {
[ce7db4d]90        badargs=1;
91        break;
92      }
[6329cd5]93      z->zsig=owl_validate_utf8(argv[1]);
94      argv+=2;
95      argc-=2;
96    } else if (!strcmp(argv[0], "-O")) {
97      if (argc<2) {
[7d4fbcd]98        badargs=1;
99        break;
100      }
[6329cd5]101      z->opcode=owl_validate_utf8(argv[1]);
102      argv+=2;
103      argc-=2;
104    } else if (!strcmp(argv[0], "-m")) {
105      if (argc<2) {
[ce7db4d]106        badargs=1;
107        break;
108      }
[69f89c7]109      /* we must already have users or a class or an instance */
[12294d2]110      if (z->recips->len < 1 && (!z->class) && (!z->inst)) {
[ce7db4d]111        badargs=1;
112        break;
113      }
114
115      /* Once we have -m, gobble up everything else on the line */
[6329cd5]116      argv++;
117      argc--;
118      msg = g_strjoinv(" ", (char**)argv);
[ce7db4d]119      break;
[6329cd5]120    } else if (!strcmp(argv[0], "-C")) {
[7d4fbcd]121      z->cc=1;
[6329cd5]122      argv++;
123      argc--;
124    } else if (!strcmp(argv[0], "-n")) {
[7d4fbcd]125      z->noping=1;
[6329cd5]126      argv++;
127      argc--;
[7d4fbcd]128    } else {
129      /* anything unattached is a recipient */
[6329cd5]130      g_ptr_array_add(z->recips, owl_validate_utf8(argv[0]));
131      argv++;
132      argc--;
[7d4fbcd]133    }
134  }
135
136  if (badargs) {
[ef4074b]137    owl_zwrite_cleanup(z);
[7d4fbcd]138    return(-1);
139  }
140
[1fe100c]141  if (z->class == NULL &&
142      z->inst == NULL &&
[12294d2]143      z->recips->len == 0) {
[1fe100c]144    owl_function_error("You must specify a recipient for zwrite");
[ef4074b]145    owl_zwrite_cleanup(z);
[1fe100c]146    return(-1);
147  }
148
[ce7db4d]149  /* now deal with defaults */
[d4927a7]150  if (z->class==NULL) z->class=g_strdup("message");
151  if (z->inst==NULL) z->inst=g_strdup("personal");
152  if (z->realm==NULL) z->realm=g_strdup("");
153  if (z->opcode==NULL) z->opcode=g_strdup("");
[ce7db4d]154  /* z->message is allowed to stay NULL */
[a52d13a]155
156  if(msg) {
157    owl_zwrite_set_message(z, msg);
[ddbbcffa]158    g_free(msg);
[a52d13a]159  }
160
[3f3ee61]161  return(0);
162}
163
164void owl_zwrite_populate_zsig(owl_zwrite *z)
165{
[ce7db4d]166  /* get a zsig, if not given */
[de3f641]167  if (z->zsig != NULL)
168    return;
[ce7db4d]169
[de3f641]170  z->zsig = owl_perlconfig_execute(owl_global_get_zsigfunc(&g));
[7d4fbcd]171}
172
[a352029b]173void owl_zwrite_send_ping(const owl_zwrite *z)
[ce7db4d]174{
[12294d2]175  int i;
[44a61ac]176  char *to;
[7d4fbcd]177
178  if (z->noping) return;
179 
[3ef779b]180  if (strcasecmp(z->class, "message")) {
[7d4fbcd]181    return;
182  }
183
[12294d2]184  for (i = 0; i < z->recips->len; i++) {
[3f52e14]185    to = owl_zwrite_get_recip_n_with_realm(z, i);
[9d206c6]186    if (owl_zwrite_recip_is_personal(to))
187      send_ping(to, z->class, z->inst);
[ddbbcffa]188    g_free(to);
[7d4fbcd]189  }
190
191}
192
[7bfc613]193/* Set the message with no post-processing*/
194void owl_zwrite_set_message_raw(owl_zwrite *z, const char *msg)
195{
[3b8a563]196  g_free(z->message);
[7bfc613]197  z->message = owl_validate_utf8(msg);
198}
199
[e19eb97]200void owl_zwrite_set_message(owl_zwrite *z, const char *msg)
[ce7db4d]201{
[12294d2]202  int i;
[3f52e14]203  GString *message;
[a52d13a]204  char *tmp = NULL, *tmp2;
[db2dd3d]205
[3b8a563]206  g_free(z->message);
[db2dd3d]207
[919cbf2]208  if (z->cc && owl_zwrite_is_personal(z)) {
[3f52e14]209    message = g_string_new("CC: ");
[12294d2]210    for (i = 0; i < z->recips->len; i++) {
[3f52e14]211      tmp = owl_zwrite_get_recip_n_with_realm(z, i);
212      g_string_append_printf(message, "%s ", tmp);
[ddbbcffa]213      g_free(tmp);
[3538bc8]214      tmp = NULL;
[db2dd3d]215    }
[4b17a6c]216    tmp = owl_validate_utf8(msg);
[a52d13a]217    tmp2 = owl_text_expand_tabs(tmp);
[3f52e14]218    g_string_append_printf(message, "\n%s", tmp2);
219    z->message = g_string_free(message, false);
[ddbbcffa]220    g_free(tmp);
221    g_free(tmp2);
[db2dd3d]222  } else {
[a52d13a]223    tmp=owl_validate_utf8(msg);
224    z->message=owl_text_expand_tabs(tmp);
[ddbbcffa]225    g_free(tmp);
[db2dd3d]226  }
[ce7db4d]227}
228
[a352029b]229const char *owl_zwrite_get_message(const owl_zwrite *z)
[ce7db4d]230{
231  if (z->message) return(z->message);
232  return("");
233}
234
[a352029b]235int owl_zwrite_is_message_set(const owl_zwrite *z)
[ce7db4d]236{
237  if (z->message) return(1);
238  return(0);
239}
240
[a352029b]241int owl_zwrite_send_message(const owl_zwrite *z)
[ce7db4d]242{
[12294d2]243  int i, ret = 0;
[823671c]244  char *to = NULL;
[7d4fbcd]245
[ce7db4d]246  if (z->message==NULL) return(-1);
247
[12294d2]248  if (z->recips->len > 0) {
249    for (i = 0; i < z->recips->len; i++) {
[3f52e14]250      to = owl_zwrite_get_recip_n_with_realm(z, i);
[0743696]251      ret = send_zephyr(z->opcode, z->zsig, z->class, z->inst, to, z->message);
252      /* Abort on the first error, to match the zwrite binary. */
253      if (ret != 0)
254        break;
[ddbbcffa]255      g_free(to);
[823671c]256      to = NULL;
[7d4fbcd]257    }
258  } else {
[3472845]259    to = g_strdup_printf( "@%s", z->realm);
[0743696]260    ret = send_zephyr(z->opcode, z->zsig, z->class, z->inst, to, z->message);
[ce7db4d]261  }
[ddbbcffa]262  g_free(to);
[0743696]263  return ret;
[ce7db4d]264}
265
[e19eb97]266int owl_zwrite_create_and_send_from_line(const char *cmd, const char *msg)
[ce7db4d]267{
268  owl_zwrite z;
269  int rv;
[ef4074b]270  rv = owl_zwrite_create_from_line(&z, cmd);
271  if (rv != 0) return rv;
[ce7db4d]272  if (!owl_zwrite_is_message_set(&z)) {
273    owl_zwrite_set_message(&z, msg);
[7d4fbcd]274  }
[3f3ee61]275  owl_zwrite_populate_zsig(&z);
[ce7db4d]276  owl_zwrite_send_message(&z);
[c230bc1]277  owl_zwrite_cleanup(&z);
[ce7db4d]278  return(0);
[7d4fbcd]279}
280
[a352029b]281const char *owl_zwrite_get_class(const owl_zwrite *z)
[ce7db4d]282{
[7d4fbcd]283  return(z->class);
284}
285
[a352029b]286const char *owl_zwrite_get_instance(const owl_zwrite *z)
[ce7db4d]287{
[7d4fbcd]288  return(z->inst);
289}
290
[a352029b]291const char *owl_zwrite_get_opcode(const owl_zwrite *z)
[ce7db4d]292{
[4b464a4]293  return(z->opcode);
294}
295
[e19eb97]296void owl_zwrite_set_opcode(owl_zwrite *z, const char *opcode)
[9ceee9d]297{
[3b8a563]298  g_free(z->opcode);
[4b17a6c]299  z->opcode=owl_validate_utf8(opcode);
[9ceee9d]300}
301
[a352029b]302const char *owl_zwrite_get_realm(const owl_zwrite *z)
[ce7db4d]303{
[7d4fbcd]304  return(z->realm);
305}
306
[a352029b]307const char *owl_zwrite_get_zsig(const owl_zwrite *z)
[ce7db4d]308{
309  if (z->zsig) return(z->zsig);
310  return("");
[56330ff]311}
312
[24ccc01]313void owl_zwrite_set_zsig(owl_zwrite *z, const char *zsig)
314{
[3b8a563]315  g_free(z->zsig);
[d4927a7]316  z->zsig = g_strdup(zsig);
[24ccc01]317}
318
[a352029b]319int owl_zwrite_get_numrecips(const owl_zwrite *z)
[ce7db4d]320{
[12294d2]321  return z->recips->len;
[7d4fbcd]322}
323
[a352029b]324const char *owl_zwrite_get_recip_n(const owl_zwrite *z, int n)
[ce7db4d]325{
[12294d2]326  return z->recips->pdata[n];
[7d4fbcd]327}
328
[3f52e14]329/* Caller must free the result. */
[6829afc]330CALLER_OWN char *owl_zwrite_get_recip_n_with_realm(const owl_zwrite *z, int n)
[3f52e14]331{
332  if (z->realm[0]) {
333    return g_strdup_printf("%s@%s", owl_zwrite_get_recip_n(z, n), z->realm);
334  } else {
335    return g_strdup(owl_zwrite_get_recip_n(z, n));
336  }
337}
338
[bff1f22]339bool owl_zwrite_recip_is_personal(const char *recipient)
340{
341  return recipient[0] && recipient[0] != '@';
342}
343
[d6b2ae8]344bool owl_zwrite_is_personal(const owl_zwrite *z)
[ce7db4d]345{
[7d4fbcd]346  /* return true if at least one of the recipients is personal */
[12294d2]347  int i;
[7d4fbcd]348
[d6b2ae8]349  for (i = 0; i < z->recips->len; i++)
350    if (owl_zwrite_recip_is_personal(z->recips->pdata[i]))
351      return true;
352  return false;
[7d4fbcd]353}
[3f3ee61]354
[987cf3f]355void owl_zwrite_delete(owl_zwrite *z)
356{
357  owl_zwrite_cleanup(z);
[7dcef03]358  g_slice_free(owl_zwrite, z);
[987cf3f]359}
360
[c230bc1]361void owl_zwrite_cleanup(owl_zwrite *z)
[ce7db4d]362{
[3cdd6d2]363  owl_ptr_array_free(z->recips, g_free);
[3b8a563]364  g_free(z->cmd);
365  g_free(z->zwriteline);
366  g_free(z->class);
367  g_free(z->inst);
368  g_free(z->opcode);
369  g_free(z->realm);
370  g_free(z->message);
371  g_free(z->zsig);
[7d4fbcd]372}
[719119de]373
374/*
375 * Returns a zwrite line suitable for replying, specifically the
376 * message field is stripped out. Result should be freed with
[ddbbcffa]377 * g_free.
[a5b5d00]378 *
379 * If not a CC, only the recip_index'th user will be replied to.
[719119de]380 */
[6829afc]381CALLER_OWN char *owl_zwrite_get_replyline(const owl_zwrite *z, int recip_index)
[719119de]382{
383  /* Match ordering in zwrite help. */
384  GString *buf = g_string_new("");
385  int i;
386
387  /* Disturbingly, it is apparently possible to z->cmd to be null if
388   * owl_zwrite_create_from_line got something starting with -. And we
389   * can't kill it because this is exported to perl. */
390  owl_string_append_quoted_arg(buf, z->cmd ? z->cmd : "zwrite");
391  if (z->noping) {
392    g_string_append(buf, " -n");
393  }
394  if (z->cc) {
395    g_string_append(buf, " -C");
396  }
397  if (strcmp(z->class, "message")) {
398    g_string_append(buf, " -c ");
399    owl_string_append_quoted_arg(buf, z->class);
400  }
401  if (strcmp(z->inst, "personal")) {
402    g_string_append(buf, " -i ");
403    owl_string_append_quoted_arg(buf, z->inst);
404  }
405  if (z->realm && z->realm[0] != '\0') {
406    g_string_append(buf, " -r ");
407    owl_string_append_quoted_arg(buf, z->realm);
408  }
409  if (z->opcode && z->opcode[0] != '\0') {
410    g_string_append(buf, " -O ");
411    owl_string_append_quoted_arg(buf, z->opcode);
412  }
[a5b5d00]413  if (z->cc) {
[12294d2]414    for (i = 0; i < z->recips->len; i++) {
[a5b5d00]415      g_string_append_c(buf, ' ');
[12294d2]416      owl_string_append_quoted_arg(buf, z->recips->pdata[i]);
[a5b5d00]417    }
[12294d2]418  } else if (recip_index < z->recips->len) {
[719119de]419    g_string_append_c(buf, ' ');
[12294d2]420    owl_string_append_quoted_arg(buf, z->recips->pdata[recip_index]);
[719119de]421  }
422
423  return g_string_free(buf, false);
424}
Note: See TracBrowser for help on using the repository browser.