aboutsummaryrefslogtreecommitdiffstats
path: root/plugin_pop3.c
blob: a8c72f89bb2e01fd3ffe1c5cb66da38a93a6ad70 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
#                                               -*- Autoconf -*-
# Process this file with autoconf to produce a configure script.


# $Id$
# $URL$


# LCD4Linux autoconf script
#
# Copyright (C) 1999, 2000, 2001, 2002, 2003 Michael Reinelt <michael@reinelt.co.at>
# Copyright (C) 2004, 2005, 2006, 2007 The LCD4Linux Team <lcd4linux-devel@users.sourceforge.net>
#
# This file is part of LCD4Linux.
#
# LCD4Linux is free software; you can redistribute it and/or mo
/* $Id$
 * $URL$
 *
 * Plugin to check POP3 mail accounts
 *
 * Copyright (C) 2004 Javi Garcia Dominguez (aka Stolz) <javi@gsmlandia.com>
 * Based on code from  pop3check (C) 1999 http://sourceforge.net/projects/pop3check
 *     Simon Liddington <squidly@users.sourceforge.net> is the pop3check current maintainer.
 *     The pop3check original author is Steven Radack <steve@lconn.net>.
 *
 * Copyright (C) 2004 The LCD4Linux Team <lcd4linux-devel@users.sourceforge.net>
 *
 * This file is a pluging for LCD4Linux.
 *
 * LCD4Linux is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * LCD4Linux is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */


#include "config.h"
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include "debug.h"
#include "plugin.h"
#include "cfg.h"

/*added */
#include <sys/socket.h>
#include <sys/types.h>
/*#include <netinet/in.h> */
#include <netdb.h>
#include <unistd.h>
/*#include <pwd.h> */
#include <stdio.h>

#ifdef WITH_DMALLOC
#include <dmalloc.h>
#endif

/*POP 3 */
#define POPERR           "-ERR"
#define LOCKEDERR        "-ERR account is locked by another session or for maintenance, try again."
#define BUFSIZE          8192
#define POP3PORT         110
#define MAX_NUM_ACCOUNTS 3


struct check {
    int id;
    char *username;
    char *password;
    char *server;
    int port;
    int messages;
    struct check *next;
};


/************************ PROTOTYPES  ********************************/
/* list */
static struct check *check_node_alloc(void);
static void check_node_add(struct check **head, struct check *new_check);
static void check_destroy(struct check **head);

/* pop3 */
static void pop3_check_messages(struct check *hi, int verbose);
static void pop3_recv_crlf_terminated(int sockfd, char *buf, int size);

/* socket  */
static int tcp_connect(struct check *hi);


/************************ GLOBAL ***********************************/
static char Section[] = "Plugin:POP3";
static struct check *head = NULL;
/********************************************************************/


/************************ LIST  ***********************************/

static struct check *check_node_alloc(void)
{
    struct check *new_check;
    new_check = (struct check *) calloc(1, sizeof(struct check));
    if (new_check == NULL) {
	error("[POP3] out of memory\n");
    }
    return new_check;
}

static void check_node_add(struct check **head, struct check *new_check)
{
    new_check->next = *head;
    *head = new_check;
}

static void check_destroy(struct check **head)
{
    struct check *iter;
    while (*head) {
	iter = (*head)->next;
	free((*head)->username);
	free((*head)->password);
	free((*head)->server);
	free(*head);
	*head = iter;
    }
    *head = NULL;
}

/************************ POP3  ********************************/
static void pop3_check_messages(struct check *hi, int verbose)
{
    char buf[BUFSIZE];
    int sockfd;

    if ((sockfd = tcp_connect(hi)) < 0) {
	hi->messages = -1;
	return;
    }

    pop3_recv_crlf_terminated(sockfd, buf, sizeof(buf));	/* server greeting */
    if (verbose)
	info("[POP3] %s -> %s\n", hi->server, buf);

    snprintf(buf, sizeof(buf), "USER %s\r\n", hi->username);
    write(sockfd, buf, strlen(buf));
    buf[strlen(buf) - 1] = '\0';
    if (verbose)
	info("[POP3] %s <- %s\n", hi->server, buf);
    pop3_recv_crlf_terminated(sockfd, buf, sizeof(buf));	/* response from USER command */
    if (verbose)
	info("[POP3] %s -> %s\n", hi->server, buf);

    snprintf(buf, sizeof(buf), "PASS %s\r\n", hi->password);
    write(sockfd, buf, strlen(buf));
    if (verbose)
	info("[POP3] %s <- PASS ???\n", hi->server);
    pop3_recv_crlf_terminated(sockfd, buf, sizeof(buf));	/* response from PASS command */
    if (verbose)
	info("[POP3] %s -> %s\n", hi->server, buf);

    if (strncmp(buf, LOCKEDERR, strlen(LOCKEDERR)) == 0) {
	hi->messages = -2;
	close(sockfd);
	return;
    }
    if (strncmp(buf, POPERR, strlen(POPERR)) == 0) {
	error("[POP3] error logging into %s\n", hi->server);
	error("[POP3] server responded: %s\n", buf);
	hi->messages = -1;
	close(sockfd);
	return;
    }

    snprintf(buf, sizeof(buf), "STAT\r\n");
    write(sockfd, buf, strlen(buf));
    if (verbose)
	info("[POP3] %s <- STAT\n", hi->server);
    pop3_recv_crlf_terminated(sockfd, buf, sizeof(buf));	/* response from PASS command */
    if (verbose)
	info("[POP3] %s -> %s\n", hi->server, buf);

    strtok(buf, " ");
    hi->messages = atoi(strtok(NULL, " "));

    snprintf(buf, sizeof(buf), "QUIT\r\n");
    write(sockfd, buf, strlen(buf));
    if (verbose)
	info("[POP3] %s <- QUIT\n", hi->server);
    pop3_recv_crlf_terminated(sockfd, buf, sizeof(buf));	/* response from QUIT command */
    if (verbose)
	info("[POP3] %s -> %s\n", hi->server, buf);

    close(sockfd);
}

static void pop3_recv_crlf_terminated(int sockfd, char *buf, int size)
{
    /* receive one line server responses terminated with CRLF */
    char *pos;
    int bytes = 0;
    memset(buf, 0, size);
    while ((pos = strstr(buf, "\r\n")) == NULL)
	bytes += read(sockfd, buf + bytes, size - bytes);
    *pos = '\0';
}

/************************ SOCKET  ********************************/
static int tcp_connect(struct check *hi)
{
    struct sockaddr_in addr;
    struct hostent *he = gethostbyname(hi->server);
    int sockfd;

    if (hi == NULL)
	return -1;

    if (!he) {
	error("[POP3] Failed to lookup %s\n", hi->server);
	return (-1);
    }

    memset((char *) &addr, 0, sizeof(struct sockaddr_in));
    addr.sin_family = AF_INET;
    memcpy(&(addr.sin_addr.s_addr), he->h_addr, he->h_length);
    addr.sin_port = htons(hi->port);

    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
	perror("socket()");
	return (-1);
    }

    if (connect(sockfd, (struct sockaddr *) &addr, sizeof(struct sockaddr)) < 0) {
	perror("connect()");
	close(sockfd);
	return (-1);
    }

    return (sockfd);
}


static int getConfig(void)
{
    struct check *node = NULL;
    int i, n = 0;
    char *user = (char *) calloc(1, sizeof("user") + sizeof(int));
    char *password = (char *) calloc(1, sizeof("password") + sizeof(int));
    char *server = (char *) calloc(1, sizeof("server") + sizeof(int));
    char *port = (char *) calloc(1, sizeof("port") + sizeof(int));

    for (i = 1; i <= MAX_NUM_ACCOUNTS; i++) {
	char *x;
	sprintf(user, "user%d", i);
	sprintf(password, "password%d", i);
	sprintf(server, "server%d", i);
	sprintf(port, "port%d", i);

	x = cfg_get(Section, server, "");
	if (*x == '\0') {
	    info("[POP3] No '%s.%s' entry from %s, disabling POP3 account #%d", Section, server, cfg_source(), i);
	    free(x);
	} else {
	    node = check_node_alloc();
	    node->id = i;
	    node->server = x;
	    node->messages = 0;
	    node->next = NULL;

	    x = cfg_get(Section, user, "");
	    if (*x == '\0') {
		info("[POP3] No '%s.%s' entry from %s, disabling POP3 account #%d", Section, user, cfg_source(), i);
		free(x);
	    } else {
		node->username = x;
		x = cfg_get(Section, password, "");
		if (*x == '\0') {
		    info("[POP3] No '%s.%s' entry from %s, disabling POP3 account #%d", Section, password, cfg_source(),
			 i);
		    free(x);
		} else {
		    node->password = x;
		    if (cfg_number(Section, port, POP3PORT, 1, 65536, &node->port) < 1) {
			info("[POP3] No '%s.%s' entry from %s, %d will be used for account #%d", Section, port,
			     cfg_source(), POP3PORT, i);
		    }
		    check_node_add(&head, node);
		    n++;
		}
	    }
	}
    }
    return (n);
}


static int configure_pop3(void)
{
    static int configured = 0;
    int n;

    if (configured != 0)
	return configured;

    n = getConfig();
    /* by now, head should point to a list of all our accounts */
    if (head) {
	info("[POP3] %d POP3 accounts have been succesfully defined", n);
	configured = 1;
    } else {
	configured = -1;
    }
    return configured;
}


static void my_POP3check(RESULT * result, RESULT * check)
{
    double param = R2N(check);
    struct check *node = NULL;
    double value;

    if (configure_pop3() < 0) {
	value = -1;
	SetResult(&result, R_NUMBER, &value);
	return;
    }

    for (node = head; node; node = node->next) {
	if (node->id == param)
	    break;
    }
    if (node == NULL) {		/*Inexistent account */
	value = -1;
    } else {
	pop3_check_messages(node, 0);
	value = (double) node->messages;
    }
    SetResult(&result, R_NUMBER, &value);
}


int plugin_init_pop3(void)
{
    AddFunction("POP3check", 1, my_POP3check);
    return 0;
}

void plugin_exit_pop3(void)
{
    check_destroy(&head);
}