/* Copyright 1993-95 by Carl Harris, Jr. Copyright 1996 by Eric S. Raymond
 * All rights reserved.
 * For license terms, see the file COPYING in this directory.
 */

/***********************************************************************
  module:       pop3.c
  project:      fetchmail
  programmer:   Carl Harris, ceharris@mal.com
		Hacks and bug fixes by esr.
  description:  POP3 client code.

 ***********************************************************************/

#include  <config.h>
#include  <stdio.h>
 
#ifdef KERBEROS_V4
#include <krb.h>
#include <des.h>
/* <netinet/in.h> must be included before "socket.h".  */
#include <netinet/in.h>
#include <netdb.h>
#endif

#include  "socket.h"
#include  "fetchmail.h"

static int last;

int pop3_ok (socket, argbuf)
/* parse command response */
int socket;
char *argbuf;
{
  int ok;
  char buf [POPBUFSIZE+1];
  char *bufp;

  if (SockGets(socket, buf, sizeof(buf)) >= 0) {
    if (outlevel == O_VERBOSE)
      fprintf(stderr,"%s\n",buf);

    bufp = buf;
    if (*bufp == '+' || *bufp == '-')
      bufp++;
    else
      return(PS_PROTOCOL);

    while (isalpha(*bufp))
      bufp++;
    *(bufp++) = '\0';

    if (strcmp(buf,"+OK") == 0)
      ok = 0;
    else if (strcmp(buf,"-ERR") == 0)
      ok = PS_ERROR;
    else
      ok = PS_PROTOCOL;

    if (argbuf != NULL)
      strcpy(argbuf,bufp);
  }
  else 
    ok = PS_SOCKET;

  return(ok);
}

int pop3_getauth(socket, queryctl, greeting)
/* apply for connection authorization */
int socket;
struct hostrec *queryctl;
char *greeting;
{
    char buf [POPBUFSIZE+1];

    /* build MD5 digest from greeting timestamp + password */
    if (queryctl->protocol == P_APOP) 
    {
	char *start,*end;
	char *msg;

	/* find start of timestamp */
	for (start = greeting;  *start != 0 && *start != '<';  start++)
	    continue;
	if (*start == 0) {
	    fprintf(stderr,"Required APOP timestamp not found in greeting\n");
	    return(PS_AUTHFAIL);
	}

	/* find end of timestamp */
	for (end = start;  *end != 0  && *end != '>';  end++)
	    continue;
	if (*end == 0 || (end - start - 1) == 1) {
	    fprintf(stderr,"Timestamp syntax error in greeting\n");
	    return(PS_AUTHFAIL);
	}

	/* copy timestamp and password into digestion buffer */
	msg = (char *)xmalloc((end-start-1) + strlen(queryctl->password) + 1);
	*(++end) = 0;
	strcpy(msg,start);
	strcat(msg,queryctl->password);

	strcpy(queryctl->digest, MD5Digest(msg));
	free(msg);
    }

    switch (queryctl->protocol) {
    case P_KPOP:
#ifndef KERBEROS_V4
	strcat (buf, "KPOP support not compiled into this executable.\n");
	return(PS_ERROR);
#endif

    case P_POP3:
	if ((gen_transact(socket,"USER %s", queryctl->remotename)) != 0)
	    return(PS_ERROR);

	if ((gen_transact(socket, "PASS %s", queryctl->password)) != 0)
	    return(PS_ERROR);
	break;

    case P_APOP:
	if ((gen_transact(socket, "APOP %s %s",
			  queryctl->remotename, queryctl->digest)) != 0)
	    return(PS_ERROR);
	break;

    default:
	fprintf(stderr,"Undefined protocol request in POP3_auth\n");
    }

    /* we're approved */
    return(0);
}

#ifdef KERBEROS_V4
int
pop3_kerberos_auth (socket, queryctl) 
     int socket;
     struct hostrec * queryctl;
{
  char * host_primary;
  KTEXT ticket;
  MSG_DAT msg_data;
  CREDENTIALS cred;
  Key_schedule schedule;
  int rem;
  
  /* Get the primary name of the host.  */
  {
    struct hostent * hp = (gethostbyname (queryctl->servername));
    if (hp == 0)
      {
	fprintf (stderr, "MAILHOST unknown: %s\n", queryctl->servername);
	return (PS_ERROR);
      }
    host_primary = ((char *) (malloc ((strlen (hp -> h_name)) + 1)));
    strcpy (host_primary, (hp -> h_name));
  }
  
  ticket = ((KTEXT) (malloc (sizeof (KTEXT_ST))));
  rem
    = (krb_sendauth (0L, socket, ticket, "pop",
		     host_primary,
		     ((char *) (krb_realmofhost (host_primary))),
		     ((unsigned long) 0),
		     (&msg_data),
		     (&cred),
		     (schedule),
		     ((struct sockaddr_in *) 0),
		     ((struct sockaddr_in *) 0),
		     "KPOPV0.1"));
  free (ticket);
  free (host_primary);
  if (rem != KSUCCESS)
    {
      fprintf (stderr, "kerberos error: %s\n", (krb_get_err_text (rem)));
      return (PS_ERROR);
    }
  return (0);
}
#endif /* KERBEROS_V4 */

static pop3_getrange(socket, queryctl, countp)
/* get range of messages to be fetched */
int socket;
struct hostrec *queryctl;
int *countp;
{
    int ok;
    char buf [POPBUFSIZE+1];

    /* get the total message count */
    gen_send(socket, "STAT");
    ok = pop3_ok(socket, buf);
    if (ok == 0)
	sscanf(buf,"%d %*d", countp);
    else
	return(ok);

    /*
     * Newer, RFC-1725-conformant POP servers may not have the LAST command.
     */
    last = 0;
    if (*countp > 0 && !queryctl->fetchall)
    {
	char id [IDLEN+1];

	gen_send(socket,"LAST");
	ok = pop3_ok(socket, buf);
	if (ok == 0 && sscanf(buf, "%d", &last) == 0)
	    return(PS_ERROR);
    }

    return(0);
}

static int pop3_is_old(socket, queryctl, num)
int socket;
struct hostrec *queryctl;
int num;
{
    return (num <= last);
}

static int pop3_fetch(socket, number, lenp)
/* request nth message */
int socket;
int number;
int *lenp; 
{
    int ok;
    char buf [POPBUFSIZE+1];

    gen_send(socket, "RETR %d", number);
    if ((ok = pop3_ok(socket, buf)) != 0)
	return(ok);
    *lenp = atoi(buf);
    return(0);
}

static pop3_delete(socket, queryctl, number)
/* delete a given message */
int socket;
struct hostrec *queryctl;
int number;
{
    return(gen_transact(socket, "DELE %d", number));
}

const static struct method pop3 =
{
    "POP3",		/* Post Office Protocol v3 */
    110,		/* standard POP3 port */
    0,			/* this is not a tagged protocol */
    1,			/* this uses a message delimiter */
    pop3_ok,		/* parse command response */
    pop3_getauth,	/* get authorization */
    pop3_getrange,	/* query range of messages */
    pop3_is_old,	/* how do we tell a message is old? */
    pop3_fetch,		/* request given message */
    NULL,		/* no message trailer */
    pop3_delete,	/* how to delete a message */
    NULL,		/* no POP3 expunge command */
    "QUIT",		/* the POP3 exit command */
};

int doPOP3 (queryctl)
/* retrieve messages using POP3 */
struct hostrec *queryctl;
{
    if (queryctl->mailbox[0]) {
	fprintf(stderr,"Option --remote is not supported with POP3\n");
	return(PS_SYNTAX);
    }
    if (queryctl->protocol == P_KPOP)
      {
	struct method kpop_method = pop3;
	kpop_method.port = 1109;
	return(do_protocol(queryctl, &kpop_method));
      }
    return(do_protocol(queryctl, &pop3));
}