1 #include <sys/types.h>

     2 #include <sys/stat.h>

     3 #include "commands.h"

     4 #include "sig.h"

     5 #include "getln.h"

     6 #include "stralloc.h"

     7 #include "substdio.h"

     8 #include "alloc.h"

     9 #include "open.h"

    10 #include "prioq.h"

    11 #include "scan.h"

    12 #include "fmt.h"

    13 #include "str.h"

    14 #include "exit.h"

    15 #include "maildir.h"

    16 #include "readwrite.h"

    17 #include "timeoutread.h"

    18 #include "timeoutwrite.h"

    19 

    20 void die() { _exit(0); }

    21 

    22 int saferead(fd,buf,len) int fd; char *buf; int len;

    23 {

    24   int r;

    25   r = timeoutread(1200,fd,buf,len);

    26   if (r <= 0) die();

    27   return r;

    28 }

    29 

    30 int safewrite(fd,buf,len) int fd; char *buf; int len;

    31 {

    32   int r;

    33   r = timeoutwrite(1200,fd,buf,len);

    34   if (r <= 0) die();

    35   return r;

    36 }

    37 

    38 char ssoutbuf[1024];

    39 substdio ssout = SUBSTDIO_FDBUF(safewrite,1,ssoutbuf,sizeof ssoutbuf);

    40 

    41 char ssinbuf[128];

    42 substdio ssin = SUBSTDIO_FDBUF(saferead,0,ssinbuf,sizeof ssinbuf);

    43 

    44 void put(buf,len) char *buf; int len;

    45 {

    46   substdio_put(&ssout,buf,len);

    47 }

    48 void puts(s) char *s;

    49 {

    50   substdio_puts(&ssout,s);

    51 }

    52 void flush()

    53 {

    54   substdio_flush(&ssout);

    55 }

    56 void err(s) char *s;

    57 {

    58   puts("-ERR ");

    59   puts(s);

    60   puts("\r\n");

    61   flush();

    62 }

    63 

    64 void die_nomem() { err("out of memory"); die(); }

    65 void die_nomaildir() { err("this user has no $HOME/Maildir"); die(); }

    66 void die_scan() { err("unable to scan $HOME/Maildir"); die(); }

    67 

    68 void err_syntax() { err("syntax error"); }

    69 void err_unimpl() { err("unimplemented"); }

    70 void err_deleted() { err("already deleted"); }

    71 void err_nozero() { err("messages are counted from 1"); }

    72 void err_toobig() { err("not that many messages"); }

    73 void err_nosuch() { err("unable to open that message"); }

    74 void err_nounlink() { err("unable to unlink all deleted messages"); }

    75 

    76 void okay() { puts("+OK \r\n"); flush(); }

    77 

    78 void printfn(fn) char *fn;

    79 {

    80   fn += 4;

    81   put(fn,str_chr(fn,':'));

    82 }

    83 

    84 char strnum[FMT_ULONG];

    85 stralloc line = {0};

    86 

    87 void blast(ssfrom,limit)

    88 substdio *ssfrom;

    89 unsigned long limit;

    90 {

    91   int match;

    92   int inheaders = 1;

    93  

    94   for (;;) {

    95     if (getln(ssfrom,&line,&match,'\n') != 0) die();

    96     if (!match && !line.len) break;

    97     if (match) --line.len; /* no way to pass this info over POP */

    98     if (limit) if (!inheaders) if (!--limit) break;

    99     if (!line.len)

   100       inheaders = 0;

   101     else

   102       if (line.s[0] == '.')

   103         put(".",1);

   104     put(line.s,line.len);

   105     put("\r\n",2);

   106     if (!match) break;

   107   }

   108   put("\r\n.\r\n",5);

   109   flush();

   110 }

   111 

   112 stralloc filenames = {0};

   113 prioq pq = {0};

   114 

   115 struct message {

   116   int flagdeleted;

   117   unsigned long size;

   118   char *fn;

   119 } *m;

   120 int numm;

   121 

   122 int last = 0;

   123 

   124 void getlist()

   125 {

   126   struct prioq_elt pe;

   127   struct stat st;

   128   int i;

   129  

   130   maildir_clean(&line);

   131   if (maildir_scan(&pq,&filenames,1,1) == -1) die_scan();

   132  

   133   numm = pq.p ? pq.len : 0;

   134   m = (struct message *) alloc(numm * sizeof(struct message));

   135   if (!m) die_nomem();

   136  

   137   for (i = 0;i < numm;++i) {

   138     if (!prioq_min(&pq,&pe)) { numm = i; break; }

   139     prioq_delmin(&pq);

   140     m[i].fn = filenames.s + pe.id;

   141     m[i].flagdeleted = 0;

   142     if (stat(m[i].fn,&st) == -1)

   143       m[i].size = 0;

   144     else

   145       m[i].size = st.st_size;

   146   }

   147 }

   148 

   149 void pop3_stat()

   150 {

   151   int i;

   152   unsigned long total;

   153  

   154   total = 0;

   155   for (i = 0;i < numm;++i) if (!m[i].flagdeleted) total += m[i].size;

   156   puts("+OK ");

   157   put(strnum,fmt_uint(strnum,numm));

   158   puts(" ");

   159   put(strnum,fmt_ulong(strnum,total));

   160   puts("\r\n");

   161   flush();

   162 }

   163 

   164 void pop3_rset()

   165 {

   166   int i;

   167   for (i = 0;i < numm;++i) m[i].flagdeleted = 0;

   168   last = 0;

   169   okay();

   170 }

   171 

   172 void pop3_last()

   173 {

   174   puts("+OK ");

   175   put(strnum,fmt_uint(strnum,last));

   176   puts("\r\n");

   177   flush();

   178 }

   179 

   180 void pop3_quit()

   181 {

   182   int i;

   183   for (i = 0;i < numm;++i)

   184     if (m[i].flagdeleted) {

   185       if (unlink(m[i].fn) == -1) err_nounlink();

   186     }

   187     else

   188       if (str_start(m[i].fn,"new/")) {

   189 	if (!stralloc_copys(&line,"cur/")) die_nomem();

   190 	if (!stralloc_cats(&line,m[i].fn + 4)) die_nomem();

   191 	if (!stralloc_cats(&line,":2,")) die_nomem();

   192 	if (!stralloc_0(&line)) die_nomem();

   193 	rename(m[i].fn,line.s); /* if it fails, bummer */

   194       }

   195   okay();

   196   die();

   197 }

   198 

   199 int msgno(arg) char *arg;

   200 {

   201   unsigned long u;

   202   if (!scan_ulong(arg,&u)) { err_syntax(); return -1; }

   203   if (!u) { err_nozero(); return -1; }

   204   --u;

   205   if (u >= numm) { err_toobig(); return -1; }

   206   if (m[u].flagdeleted) { err_deleted(); return -1; }

   207   return u;

   208 }

   209 

   210 void pop3_dele(arg) char *arg;

   211 {

   212   int i;

   213   i = msgno(arg);

   214   if (i == -1) return;

   215   m[i].flagdeleted = 1;

   216   if (i + 1 > last) last = i + 1;

   217   okay();

   218 }

   219 

   220 void list(i,flaguidl)

   221 int i;

   222 int flaguidl;

   223 {

   224   put(strnum,fmt_uint(strnum,i + 1));

   225   puts(" ");

   226   if (flaguidl) printfn(m[i].fn);

   227   else put(strnum,fmt_ulong(strnum,m[i].size));

   228   puts("\r\n");

   229 }

   230 

   231 void dolisting(arg,flaguidl) char *arg; int flaguidl;

   232 {

   233   unsigned int i;

   234   if (*arg) {

   235     i = msgno(arg);

   236     if (i == -1) return;

   237     puts("+OK ");

   238     list(i,flaguidl);

   239   }

   240   else {

   241     okay();

   242     for (i = 0;i < numm;++i)

   243       if (!m[i].flagdeleted)

   244 	list(i,flaguidl);

   245     puts(".\r\n");

   246   }

   247   flush();

   248 }

   249 

   250 void pop3_uidl(arg) char *arg; { dolisting(arg,1); }

   251 void pop3_list(arg) char *arg; { dolisting(arg,0); }

   252 

   253 substdio ssmsg; char ssmsgbuf[1024];

   254 

   255 void pop3_top(arg) char *arg;

   256 {

   257   int i;

   258   unsigned long limit;

   259   int fd;

   260  

   261   i = msgno(arg);

   262   if (i == -1) return;

   263  

   264   arg += scan_ulong(arg,&limit);

   265   while (*arg == ' ') ++arg;

   266   if (scan_ulong(arg,&limit)) ++limit; else limit = 0;

   267  

   268   fd = open_read(m[i].fn);

   269   if (fd == -1) { err_nosuch(); return; }

   270   okay();

   271   substdio_fdbuf(&ssmsg,read,fd,ssmsgbuf,sizeof(ssmsgbuf));

   272   blast(&ssmsg,limit);

   273   close(fd);

   274 }

   275 

   276 struct commands pop3commands[] = {

   277   { "quit", pop3_quit, 0 }

   278 , { "stat", pop3_stat, 0 }

   279 , { "list", pop3_list, 0 }

   280 , { "uidl", pop3_uidl, 0 }

   281 , { "dele", pop3_dele, 0 }

   282 , { "retr", pop3_top, 0 }

   283 , { "rset", pop3_rset, 0 }

   284 , { "last", pop3_last, 0 }

   285 , { "top", pop3_top, 0 }

   286 , { "noop", okay, 0 }

   287 , { 0, err_unimpl, 0 }

   288 } ;

   289 

   290 void main(argc,argv)

   291 int argc;

   292 char **argv;

   293 {

   294   sig_alarmcatch(die);

   295   sig_pipeignore();

   296  

   297   if (!argv[1]) die_nomaildir();

   298   if (chdir(argv[1]) == -1) die_nomaildir();

   299  

   300   getlist();

   301 

   302   okay();

   303   commands(&ssin,pop3commands);

   304   die();

   305 }