appl/popper/maildir.c

   1 /*
   2  * Copyright (c) 1998 Kungliga Tekniska Högskolan
   3  * (Royal Institute of Technology, Stockholm, Sweden).
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  *
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  *
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * 3. Neither the name of the Institute nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #include <popper.h>
  35 #include <dirent.h>
  36 RCSID("$Id$");
  37
  38 static void
  39 make_path(POP *p, MsgInfoList *mp, int new, char *buf, size_t len)
  40 {
  41     snprintf(buf, len, "%s/%s%s%s", p->drop_name,
  42              new ? "new" : "cur", mp ? "/" : "", mp ? mp->name : "");
  43 }
  44
  45 static int
  46 scan_file(POP *p, MsgInfoList *mp)
  47 {
  48     char path[MAXPATHLEN];
  49     FILE *f;
  50     char buf[1024];
  51     int eoh = 0;
  52
  53     make_path(p, mp, mp->flags & NEW_FLAG, path, sizeof(path));
  54     f = fopen(path, "r");
  55
  56     if(f == NULL) {
  57 #ifdef DEBUG
  58         if(p->debug)
  59             pop_log(p, POP_DEBUG,
  60                     "Failed to open message file `%s': %s",
  61                     path, strerror(errno));
  62 #endif
  63         return pop_msg (p, POP_FAILURE,
  64                         "Failed to open message file `%s'", path);
  65     }
  66     while(fgets(buf, sizeof(buf), f)) {
  67         if(buf[strlen(buf) - 1] == '\n')
  68             mp->lines++;
  69         mp->length += strlen(buf);
  70         if(eoh)
  71             continue;
  72         if(strcmp(buf, "\n") == 0)
  73             eoh = 1;
  74         parse_header(mp, buf);
  75     }
  76     fclose(f);
  77     return add_missing_headers(p, mp);
  78 }
  79
  80 static int
  81 scan_dir(POP *p, int new)
  82 {
  83     char tmp[MAXPATHLEN];
  84     DIR *dir;
  85     struct dirent *dent;
  86     MsgInfoList *mp = p->mlp;
  87     int n_mp = p->msg_count;
  88     int e;
  89
  90     make_path(p, NULL, new, tmp, sizeof(tmp));
  91     mkdir(tmp, 0700);
  92     dir = opendir(tmp);
  93     while((dent = readdir(dir)) != NULL) {
  94         if(strcmp(dent->d_name, ".") == 0 || strcmp(dent->d_name, "..") == 0)
  95             continue;
  96         mp = realloc(mp, (n_mp + 1) * sizeof(*mp));
  97         if(mp == NULL) {
  98             p->msg_count = 0;
  99             return pop_msg (p, POP_FAILURE,
 100                             "Can't build message list for '%s': Out of memory",
 101                             p->user);
 102         }
 103         memset(mp + n_mp, 0, sizeof(*mp));
 104         mp[n_mp].name = strdup(dent->d_name);
 105         if(mp[n_mp].name == NULL) {
 106             p->msg_count = 0;
 107             return pop_msg (p, POP_FAILURE,
 108                             "Can't build message list for '%s': Out of memory",
 109                             p->user);
 110         }
 111         mp[n_mp].number = n_mp + 1;
 112         mp[n_mp].flags = 0;
 113         if(new)
 114             mp[n_mp].flags |= NEW_FLAG;
 115         e = scan_file(p, &mp[n_mp]);
 116         if(e != POP_SUCCESS)
 117             return e;
 118         p->drop_size += mp[n_mp].length;
 119         n_mp++;
 120     }
 121     closedir(dir);
 122     p->mlp = mp;
 123     p->msg_count = n_mp;
 124     return POP_SUCCESS;
 125 }
 126
 127 int
 128 pop_maildir_info(POP *p)
 129 {
 130     int e;
 131
 132     p->temp_drop[0] = '\0';
 133     p->mlp = NULL;
 134     p->msg_count = 0;
 135
 136     e = scan_dir(p, 0);
 137     if(e != POP_SUCCESS) return e;
 138
 139     e = scan_dir(p, 1);
 140     if(e != POP_SUCCESS) return e;
 141     return POP_SUCCESS;
 142 }
 143
 144 int
 145 pop_maildir_update(POP *p)
 146 {
 147     int i;
 148     char tmp1[MAXPATHLEN], tmp2[MAXPATHLEN];
 149     for(i = 0; i < p->msg_count; i++) {
 150         make_path(p, &p->mlp[i], p->mlp[i].flags & NEW_FLAG,
 151                   tmp1, sizeof(tmp1));
 152         if(p->mlp[i].flags & DEL_FLAG) {
 153 #ifdef DEBUG
 154             if(p->debug)
 155                 pop_log(p, POP_DEBUG, "Removing `%s'", tmp1);
 156 #endif
 157             if(unlink(tmp1) < 0) {
 158 #ifdef DEBUG
 159                 if(p->debug)
 160                     pop_log(p, POP_DEBUG, "Failed to remove `%s': %s",
 161                             tmp1, strerror(errno));
 162 #endif
 163                 /* return failure? */
 164             }
 165         } else if((p->mlp[i].flags & NEW_FLAG) &&
 166                   (p->mlp[i].flags & RETR_FLAG)) {
 167             make_path(p, &p->mlp[i], 0, tmp2, sizeof(tmp2));
 168 #ifdef DEBUG
 169             if(p->debug)
 170                 pop_log(p, POP_DEBUG, "Linking `%s' to `%s'", tmp1, tmp2);
 171 #endif
 172             if(link(tmp1, tmp2) == 0) {
 173 #ifdef DEBUG
 174                 if(p->debug)
 175                     pop_log(p, POP_DEBUG, "Removing `%s'", tmp1);
 176 #endif
 177                 if(unlink(tmp1) < 0) {
 178 #ifdef DEBUG
 179                     if(p->debug)
 180                         pop_log(p, POP_DEBUG, "Failed to remove `%s'", tmp1);
 181 #endif
 182                     /* return failure? */
 183                 }
 184             } else {
 185                 if(errno == EXDEV) {
 186 #ifdef DEBUG
 187                     if(p->debug)
 188                         pop_log(p, POP_DEBUG, "Trying to rename `%s' to `%s'",
 189                                 tmp1, tmp2);
 190 #endif
 191                     if(rename(tmp1, tmp2) < 0) {
 192 #ifdef DEBUG
 193                     if(p->debug)
 194                         pop_log(p, POP_DEBUG, "Failed to rename `%s' to `%s'",
 195                                 tmp1, tmp2);
 196 #endif
 197                     }
 198                 }
 199             }
 200         }
 201     }
 202     return(pop_quit(p));
 203 }
 204
 205 int
 206 pop_maildir_open(POP *p, MsgInfoList *mp)
 207 {
 208     char tmp[MAXPATHLEN];
 209     make_path(p, mp, mp->flags & NEW_FLAG, tmp, sizeof(tmp));
 210     if(p->drop)
 211         fclose(p->drop);
 212     p->drop = fopen(tmp, "r");
 213     if(p->drop == NULL)
 214         return pop_msg(p, POP_FAILURE, "Failed to open message file");
 215     return POP_SUCCESS;
 216 }