qp.c

一个C语言写的快速贝叶斯垃圾邮件过滤工具

/* $Id: qp.c,v 1.18 2005/09/04 14:49:43 relson Exp $ */

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

NAME:
   qp.c -- decode quoted printable text

AUTHOR:
   David Relson <[e-mail]>

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

#include "common.h"

#include "qp.h"

/* Local Variables */

static byte qp_xlate2045[256];
static byte qp_xlate2047[256];

static int hex_to_bin(byte c) {
    switch (c) {
	case '0': return 0;
	case '1': return 1;
	case '2': return 2;
	case '3': return 3;
	case '4': return 4;
	case '5': return 5;
	case '6': return 6;
	case '7': return 7;
	case '8': return 8;
	case '9': return 9;
	case 'a': case 'A': return 10;
	case 'b': case 'B': return 11;
	case 'c': case 'C': return 12;
	case 'd': case 'D': return 13;
	case 'e': case 'E': return 14;
	case 'f': case 'F': return 15;
	default: return -1;
    }
}

/* Function Definitions  */

uint qp_decode(word_t *word, qp_mode mode)
{
    uint size = word->leng;
    byte *s = word->text;	/* src */
    byte *d = word->text;	/* dst */
    byte *e = s + size;		/* end */

    while (s < e)
    {
	byte ch = *s++;
	int x, y;
	switch (ch) {
	    case '=':
		if (mode == RFC2045 && s + 1 <= e && s[0] == '\n') {
		    /* continuation line, trailing = */
		    s++;
		    continue;
		}
		if (s + 2 <= e && 
			(y = hex_to_bin(s[0])) >= 0 && (x = hex_to_bin(s[1])) >= 0) {
		    /* encoded character */
		    ch = (byte) (y << 4 | x);
		    s += 2;
		}
		break;
	    case '_':
		if (mode == RFC2047)
		    ch = ' ';
		break;
	}
	*d++ = ch;
    }
    /* do not stuff NUL byte here:
     * if there was one, it has been copied! */
    return d - word->text;
}

/* rfc2047 - QP    [!->@-~]+
*/

static void qp_init(void)
{
    uint i;
    static bool first = true;

    if (!first)
	return;
    first = false;

    for (i = 33; i <= 126; i ++) {
	qp_xlate2045[i] = qp_xlate2047[i] = (byte) i;
    }

    qp_xlate2045['\t'] = '\t';	/* HTAB */
    qp_xlate2045[' ']  = ' ';	/* SP */
    qp_xlate2045['=']  = 0;	/* illegal */

    qp_xlate2047['_']  = ' ';	/* SP */
    qp_xlate2047['=']  = 0;	/* illegal */
    qp_xlate2047['?']  = 0;	/* illegal */

    return;
}

bool qp_validate(const word_t *word, qp_mode mode)
{
    uint i;
    const byte *qp_xlate = mode == RFC2047 ? qp_xlate2047 : qp_xlate2045;

    qp_init();

    for (i = 0; i < word->leng; i += 1) {
	byte b = word->text[i];
	byte v = qp_xlate[b];
	if (v == 0)
	    switch (b) {
		case '=': /* allowed in encoded words,
			     but must be encoded */
		    break;
		default:
		    return false;
	    }
    }

    return true;
}