transaction.js

"use strict";
// An SMTP Transaction

var config = require('./config');
var Header = require('./mailheader').Header;
var body   = require('./mailbody');
var utils  = require('haraka-utils');
var util   = require('util');
var MessageStream = require('./messagestream');

var MAX_HEADER_LINES = config.get('max_header_lines') || 1000;

function Transaction () {
    this.uuid = null;
    this.mail_from = null;
    this.rcpt_to = [];
    this.header_lines = [];
    this.data_lines = [];
    this.attachment_start_hooks = [];
    this.banner = null;
    this.body_filters = [];
    this.data_bytes = 0;
    this.header_pos = 0;
    this.found_hb_sep = false;
    this.body = null;
    this.parse_body = false;
    this.notes = {};
    this.header = new Header();
    this.message_stream = null;
    this.discard_data = false;
    this.resetting = false;
    this.rcpt_count = {
        accept:   0,
        tempfail: 0,
        reject:   0,
    };
    this.data_post_start = null;
    this.data_post_delay = 0;
    this.encoding = 'utf8';
}

exports.Transaction = Transaction;

exports.createTransaction = function (uuid) {
    var t = new Transaction();
    t.uuid = uuid || utils.uuid();
    // Initialize MessageStream here to pass in the UUID
    t.message_stream = new MessageStream(
        config.get('smtp.ini'), t.uuid, t.header.header_list);
    return t;
};

Transaction.prototype.ensure_body = function () {
    var self = this;
    if (this.body) {
        return;
    }

    this.body = new body.Body(this.header);
    this.attachment_start_hooks.forEach(function (h) {
        self.body.on('attachment_start', h);
    });
    if (this.banner) {
        this.body.set_banner(this.banner);
    }
    this.body_filters.forEach(function (o) {
        self.body.add_filter(function (ct, enc, buf) {
            if ((util.isRegExp(o.ct_match) &&
                 o.ct_match.test(ct.toLowerCase())) ||
                    ct.toLowerCase()
                        .indexOf(String(o.ct_match)
                            .toLowerCase()) === 0) {
                return o.filter(ct, enc, buf);
            }
        });
    });
};

Transaction.prototype.add_data = function (line) {
    if (typeof line === 'string') { // This shouldn't ever really happen...
        line = new Buffer(line, this.encoding);
    }
    // check if this is the end of headers line
    if (this.header_pos === 0 &&
        (line[0] === 0x0A || (line[0] === 0x0D && line[1] === 0x0A)) ) {
        this.header.parse(this.header_lines);
        this.header_pos = this.header_lines.length;
        this.found_hb_sep = true;
        if (this.parse_body) {
            this.ensure_body();
        }
    }
    else if (this.header_pos === 0) {
        // Build up headers
        if (this.header_lines.length < MAX_HEADER_LINES) {
            if (line[0] === 0x2E) line = line.slice(1); // Strip leading "."
            this.header_lines.push(
                line.toString(this.encoding).replace(/\r\n$/, '\n'));
        }
    }
    else if (this.header_pos && this.parse_body) {
        if (line[0] === 0x2E) line = line.slice(1); // Strip leading "."
        var new_line = this.body.parse_more(
            line.toString(this.encoding).replace(/\r\n$/, '\n'));

        if (!new_line.length) {
            return; // buffering for banners
        }

        new_line = new_line.replace(/^\./gm, '..').replace(/\r?\n/gm, '\r\n');
        line = new Buffer(new_line, this.encoding);
    }

    if (!this.discard_data) this.message_stream.add_line(line);
};

Transaction.prototype.end_data = function (cb) {
    if (!this.found_hb_sep && this.header_lines.length) {
        // Headers not parsed yet - must be a busted email
        // Strategy: Find the first line that doesn't look like a header.
        // Treat anything before that as headers, anything after as body.
        var header_pos = 0;
        for (var i = 0; i < this.header_lines.length; i++) {
            // Anything that doesn't match a header or continuation
            if (!/^(?:([^\s:]*):\s*([\s\S]*)$|[ \t])/.test(this.header_lines[i])) {
                break;
            }
            header_pos = i;
        }
        var body_lines = this.header_lines.splice(header_pos + 1);
        this.header.parse(this.header_lines);
        this.header_pos = header_pos;
        if (this.parse_body) {
            this.ensure_body();
            for (var j = 0; j < body_lines.length; j++) {
                this.body.parse_more(body_lines[j]);
            }
        }
    }
    if (this.header_pos && this.parse_body) {
        var data = this.body.parse_end();
        if (data.length) {
            data = data.toString(this.encoding)
                .replace(/^\./gm, '..')
                .replace(/\r?\n/gm, '\r\n');
            var line = new Buffer(data, this.encoding);

            this.body.force_end();

            if (!this.discard_data) this.message_stream.add_line(line);
        }
    }

    if (!this.discard_data) {
        this.message_stream.add_line_end(cb);
    }
    else {
        cb();
    }
};

Transaction.prototype.add_header = function (key, value) {
    this.header.add_end(key, value);
    if (this.header_pos > 0) this.reset_headers();
};

Transaction.prototype.add_leading_header = function (key, value) {
    this.header.add(key, value);
    if (this.header_pos > 0) this.reset_headers();
};

Transaction.prototype.reset_headers = function () {
    var header_lines = this.header.lines();
    this.header_pos = header_lines.length;
};

Transaction.prototype.remove_header = function (key) {
    this.header.remove(key);
    if (this.header_pos > 0) this.reset_headers();
};

Transaction.prototype.attachment_hooks = function (start, data, end) {
    this.parse_body = 1;
    this.attachment_start_hooks.push(start);
};

Transaction.prototype.set_banner = function (text, html) {
    // throw "transaction.set_banner is currently non-functional";
    this.parse_body = true;
    if (!html) {
        html = text.replace(/\n/g, '<br/>\n');
    }
    this.banner = [text, html];
};

Transaction.prototype.add_body_filter = function (ct_match, filter) {
    this.parse_body = true;
    this.body_filters.push({'ct_match': ct_match, 'filter': filter});
};