cmd_write.c

   1 /*@ S-nail - a mail user agent derived from Berkeley Mail.
   2  *@ User commands which save, copy, write (parts of) messages.
   3  *
   4  * Copyright (c) 2000-2004 Gunnar Ritter, Freiburg i. Br., Germany.
   5  * Copyright (c) 2012 - 2017 Steffen (Daode) Nurpmeso <steffen@sdaoden.eu>.
   6  */
   7 /*
   8  * Copyright (c) 1980, 1993
   9  *      The Regents of the University of California.  All rights reserved.
  10  *
  11  * Redistribution and use in source and binary forms, with or without
  12  * modification, are permitted provided that the following conditions
  13  * are met:
  14  * 1. Redistributions of source code must retain the above copyright
  15  *    notice, this list of conditions and the following disclaimer.
  16  * 2. Redistributions in binary form must reproduce the above copyright
  17  *    notice, this list of conditions and the following disclaimer in the
  18  *    documentation and/or other materials provided with the distribution.
  19  * 3. Neither the name of the University nor the names of its contributors
  20  *    may be used to endorse or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35 #undef n_FILE
  36 #define n_FILE cmd_write
  37
  38 #ifndef HAVE_AMALGAMATION
  39 # include "nail.h"
  40 #endif
  41
  42 /* Save/copy the indicated messages at the end of the passed file name.
  43  * If mark is true, mark the message "saved" */
  44 static int     save1(char *str, int domark, char const *cmd,
  45                   struct n_ignore const *itp, int convert, int sender_record,
  46                   int domove);
  47
  48 /* Snarf the file from the end of the command line and return a pointer to it.
  49  * If there is no file attached, return the mbox file.  Put a null in front of
  50  * the file name so that the message list processing won't see it, unless the
  51  * file name is the only thing on the line, in which case, return 0 in the
  52  * reference flag variable */
  53 static char *  snarf(char *linebuf, bool_t *flag, bool_t usembox);
  54
  55 static int
  56 save1(char *str, int domark, char const *cmd, struct n_ignore const *itp,
  57    int convert, int sender_record, int domove)
  58 {
  59    ui64_t mstats[1], tstats[2];
  60    struct stat st;
  61    int last = 0, *msgvec, *ip;
  62    struct message *mp;
  63    char *file = NULL, *cp, *cq;
  64    char const *disp = n_empty, *shell = NULL;
  65    FILE *obuf;
  66    bool_t success = FAL0, isflag;
  67    NYD_ENTER;
  68
  69    msgvec = salloc((msgCount + 2) * sizeof *msgvec);
  70    if (sender_record) {
  71       for (cp = str; *cp != '\0' && spacechar(*cp); ++cp)
  72          ;
  73       isflag = (*cp != '\0');
  74    } else {
  75       if ((file = snarf(str, &isflag, convert != SEND_TOFILE)) == NULL)
  76          goto jleave;
  77       while(spacechar(*file))
  78          ++file;
  79       if (*file == '|') {
  80          ++file;
  81          shell = ok_vlook(SHELL);
  82       }
  83    }
  84
  85    if (!isflag) {
  86       *msgvec = first(0, MMNORM);
  87       msgvec[1] = 0;
  88    } else if (getmsglist(str, msgvec, 0) < 0)
  89       goto jleave;
  90    if (*msgvec == 0) {
  91       if (pstate & (PS_HOOK_MASK | PS_ROBOT)) {
  92          success = TRU1;
  93          goto jleave;
  94       }
  95       printf(_("No messages to %s.\n"), cmd);
  96       goto jleave;
  97    }
  98
  99    if (sender_record) {
 100       if ((cp = nameof(message + *msgvec - 1, 0)) == NULL) {
 101          printf(_("Cannot determine message sender to %s.\n"), cmd);
 102          goto jleave;
 103       }
 104
 105       for (cq = cp; *cq != '\0' && *cq != '@'; cq++)
 106          ;
 107       *cq = '\0';
 108       if (ok_blook(outfolder)) {
 109          size_t sz = strlen(cp) +1;
 110          file = salloc(sz + 1);
 111          file[0] = '+';
 112          memcpy(file + 1, cp, sz);
 113       } else
 114          file = cp;
 115    }
 116
 117    /* Pipe target is special TODO hacked in later, normalize flow! */
 118    if (shell != NULL) {
 119       if ((obuf = Popen(file, "w", shell, NULL, 1)) == NULL) {
 120          int esave = errno;
 121
 122          n_perr(file, esave);
 123          errno = esave;
 124          goto jleave;
 125       }
 126       isflag = FAL0;
 127       disp = _("[Piped]");
 128       goto jsend;
 129    }
 130
 131    if ((file = expand(file)) == NULL)
 132       goto jleave;
 133
 134    obuf = ((convert == SEND_TOFILE) ? Fopen(file, "a+") : Zopen(file, "a+"));
 135    if (obuf == NULL) {
 136       obuf = ((convert == SEND_TOFILE) ? Fopen(file, "wx") : Zopen(file, "wx"));
 137       if (obuf == NULL) {
 138          n_perr(file, 0);
 139          goto jleave;
 140       }
 141       isflag = TRU1;
 142       disp = _("[New file]");
 143    } else {
 144       isflag = FAL0;
 145       disp = _("[Appended]");
 146    }
 147
 148    /* TODO RETURN check, but be aware of protocols: v15: Mailbox->lock()! */
 149    n_file_lock(fileno(obuf), FLT_WRITE, 0,0, UIZ_MAX);
 150
 151    if (!isflag && !fstat(fileno(obuf), &st) && S_ISREG(st.st_mode) &&
 152          fseek(obuf, -2L, SEEK_END) == 0) {
 153       char buf[2];
 154       int prependnl = 0;
 155
 156       switch (fread(buf, sizeof *buf, 2, obuf)) {
 157       case 2:
 158          if (buf[1] != '\n') {
 159             prependnl = 1;
 160             break;
 161          }
 162          /* FALLTHRU */
 163       case 1:
 164          if (buf[0] != '\n')
 165             prependnl = 1;
 166          break;
 167       default:
 168          if (ferror(obuf)) {
 169             n_perr(file, 0);
 170             goto jleave;
 171          }
 172          prependnl = 0;
 173       }
 174
 175       fflush(obuf);
 176       if (prependnl) {
 177          putc('\n', obuf);
 178          fflush(obuf);
 179       }
 180    }
 181
 182 jsend:
 183    success = TRU1;
 184    tstats[0] = tstats[1] = 0;
 185
 186    srelax_hold();
 187    for (ip = msgvec; *ip != 0 && UICMP(z, PTR2SIZE(ip - msgvec), <, msgCount);
 188          ++ip) {
 189       mp = message + *ip - 1;
 190       if (sendmp(mp, obuf, itp, NULL, convert, mstats) < 0) {
 191          success = FAL0;
 192          goto jferr;
 193       }
 194       srelax();
 195
 196       touch(mp);
 197       if (domark)
 198          mp->m_flag |= MSAVED;
 199       if (domove) {
 200          mp->m_flag |= MDELETED | MSAVED;
 201          last = *ip;
 202       }
 203
 204       tstats[0] += mstats[0];
 205       tstats[1] += mp->m_lines;/* TODO won't work, need target! v15!! */
 206    }
 207    srelax_rele();
 208
 209    fflush(obuf);
 210    if (ferror(obuf)) {
 211 jferr:
 212       n_perr(file, 0);
 213       if (!success)
 214          srelax_rele();
 215       success = FAL0;
 216    }
 217    if (shell != NULL) {
 218       if (!Pclose(obuf, TRU1))
 219          success = FAL0;
 220    } else if (Fclose(obuf) != 0)
 221       success = FAL0;
 222
 223    if (success) {
 224       printf("%s %s %" /*PRIu64 "/%"*/ PRIu64 " bytes\n",
 225          n_shexp_quote_cp(file, FAL0), disp,
 226          /*tstats[1], TODO v15: lines written */ tstats[0]);
 227    } else if (domark) {
 228       for (ip = msgvec; *ip != 0 &&
 229             UICMP(z, PTR2SIZE(ip - msgvec), <, msgCount); ++ip) {
 230          mp = message + *ip - 1;
 231          mp->m_flag &= ~MSAVED;
 232       }
 233    } else if (domove) {
 234       for (ip = msgvec; *ip != 0 &&
 235             UICMP(z, PTR2SIZE(ip - msgvec), <, msgCount); ++ip) {
 236          mp = message + *ip - 1;
 237          mp->m_flag &= ~(MSAVED | MDELETED);
 238       }
 239    }
 240
 241    if (domove && last && success) {
 242       setdot(message + last - 1);
 243       last = first(0, MDELETED);
 244       setdot(message + (last != 0 ? last - 1 : 0));
 245    }
 246 jleave:
 247    NYD_LEAVE;
 248    return (success == FAL0);
 249 }
 250
 251 static char *
 252 snarf(char *linebuf, bool_t *flag, bool_t usembox)
 253 {
 254    char *cp;
 255    NYD_ENTER;
 256
 257    if ((cp = laststring(linebuf, flag, TRU1)) == NULL) {
 258       if (usembox) {
 259          *flag = FAL0;
 260          cp = expand("&");
 261       } else
 262          n_err(_("No file specified\n"));
 263    }
 264    NYD_LEAVE;
 265    return cp;
 266 }
 267
 268 FL int
 269 c_save(void *v)
 270 {
 271    char *str = v;
 272    int rv;
 273    NYD_ENTER;
 274
 275    rv = save1(str, 1, "save", n_IGNORE_SAVE, SEND_MBOX, 0, 0);
 276    NYD_LEAVE;
 277    return rv;
 278 }
 279
 280 FL int
 281 c_Save(void *v)
 282 {
 283    char *str = v;
 284    int rv;
 285    NYD_ENTER;
 286
 287    rv = save1(str, 1, "save", n_IGNORE_SAVE, SEND_MBOX, 1, 0);
 288    NYD_LEAVE;
 289    return rv;
 290 }
 291
 292 FL int
 293 c_copy(void *v)
 294 {
 295    char *str = v;
 296    int rv;
 297    NYD_ENTER;
 298
 299    rv = save1(str, 0, "copy", n_IGNORE_SAVE, SEND_MBOX, 0, 0);
 300    NYD_LEAVE;
 301    return rv;
 302 }
 303
 304 FL int
 305 c_Copy(void *v)
 306 {
 307    char *str = v;
 308    int rv;
 309    NYD_ENTER;
 310
 311    rv = save1(str, 0, "copy", n_IGNORE_SAVE, SEND_MBOX, 1, 0);
 312    NYD_LEAVE;
 313    return rv;
 314 }
 315
 316 FL int
 317 c_move(void *v)
 318 {
 319    char *str = v;
 320    int rv;
 321    NYD_ENTER;
 322
 323    rv = save1(str, 0, "move", n_IGNORE_SAVE, SEND_MBOX, 0, 1);
 324    NYD_LEAVE;
 325    return rv;
 326 }
 327
 328 FL int
 329 c_Move(void *v)
 330 {
 331    char *str = v;
 332    int rv;
 333    NYD_ENTER;
 334
 335    rv = save1(str, 0, "move", n_IGNORE_SAVE, SEND_MBOX, 1, 1);
 336    NYD_LEAVE;
 337    return rv;
 338 }
 339
 340 FL int
 341 c_decrypt(void *v)
 342 {
 343    char *str = v;
 344    int rv;
 345    NYD_ENTER;
 346
 347    rv = save1(str, 0, "decrypt", n_IGNORE_SAVE, SEND_DECRYPT, 0, 0);
 348    NYD_LEAVE;
 349    return rv;
 350 }
 351
 352 FL int
 353 c_Decrypt(void *v)
 354 {
 355    char *str = v;
 356    int rv;
 357    NYD_ENTER;
 358
 359    rv = save1(str, 0, "decrypt", n_IGNORE_SAVE, SEND_DECRYPT, 1, 0);
 360    NYD_LEAVE;
 361    return rv;
 362 }
 363
 364 FL int
 365 c_write(void *v)
 366 {
 367    char *str = v;
 368    int rv;
 369    NYD_ENTER;
 370
 371    if (str == NULL || *str == '\0')
 372       str = savestr("/dev/null");
 373    rv = save1(str, 0, "write", n_IGNORE_ALL, SEND_TOFILE, 0, 0);
 374    NYD_LEAVE;
 375    return rv;
 376 }
 377
 378 /* s-it-mode */