316 lines
9.8 KiB
TypeScript
316 lines
9.8 KiB
TypeScript
|
import { tap, expect } from '@push.rocks/tapbundle';
|
||
|
import * as plugins from '../plugins.js';
|
||
|
import * as net from 'net';
|
||
|
import { startTestServer, stopTestServer, TEST_PORT, sendEmailWithRawSocket } from '../server.loader.js';
|
||
|
|
||
|
let testServer: any;
|
||
|
|
||
|
tap.test('setup - start test server', async () => {
|
||
|
testServer = await startTestServer();
|
||
|
await plugins.smartdelay.delayFor(1000);
|
||
|
});
|
||
|
|
||
|
tap.test('Extremely Long Headers - should handle single extremely long header', async (tools) => {
|
||
|
const done = tools.defer();
|
||
|
|
||
|
const socket = net.createConnection({
|
||
|
host: 'localhost',
|
||
|
port: TEST_PORT,
|
||
|
timeout: 30000
|
||
|
});
|
||
|
|
||
|
let dataBuffer = '';
|
||
|
|
||
|
socket.on('data', (data) => {
|
||
|
dataBuffer += data.toString();
|
||
|
console.log('Server response:', data.toString());
|
||
|
|
||
|
if (dataBuffer.includes('220 ')) {
|
||
|
// Send EHLO
|
||
|
socket.write('EHLO testclient\r\n');
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('EHLO')) {
|
||
|
// Send MAIL FROM
|
||
|
socket.write('MAIL FROM:<sender@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('sender accepted')) {
|
||
|
// Send RCPT TO
|
||
|
socket.write('RCPT TO:<recipient@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('recipient accepted')) {
|
||
|
// Send DATA
|
||
|
socket.write('DATA\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('354 ')) {
|
||
|
// Send email with extremely long header
|
||
|
const longValue = 'X'.repeat(3000);
|
||
|
const emailContent = [
|
||
|
`Subject: Test Email`,
|
||
|
`From: sender@example.com`,
|
||
|
`To: recipient@example.com`,
|
||
|
`X-Long-Header: ${longValue}`,
|
||
|
'',
|
||
|
'This email has an extremely long header.',
|
||
|
'.',
|
||
|
''
|
||
|
].join('\r\n');
|
||
|
|
||
|
socket.write(emailContent);
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('Message accepted') ||
|
||
|
dataBuffer.includes('552 ') ||
|
||
|
dataBuffer.includes('554 ') ||
|
||
|
dataBuffer.includes('500 ')) {
|
||
|
// Either accepted or gracefully rejected
|
||
|
const accepted = dataBuffer.includes('250 ');
|
||
|
console.log(`Long header test ${accepted ? 'accepted' : 'rejected'}`);
|
||
|
|
||
|
socket.write('QUIT\r\n');
|
||
|
socket.end();
|
||
|
done.resolve();
|
||
|
}
|
||
|
});
|
||
|
|
||
|
socket.on('error', (err) => {
|
||
|
console.error('Socket error:', err);
|
||
|
done.reject(err);
|
||
|
});
|
||
|
|
||
|
socket.on('timeout', () => {
|
||
|
console.error('Socket timeout');
|
||
|
socket.destroy();
|
||
|
done.reject(new Error('Socket timeout'));
|
||
|
});
|
||
|
|
||
|
await done.promise;
|
||
|
});
|
||
|
|
||
|
tap.test('Extremely Long Headers - should handle multi-line header with many segments', async (tools) => {
|
||
|
const done = tools.defer();
|
||
|
|
||
|
const socket = net.createConnection({
|
||
|
host: 'localhost',
|
||
|
port: TEST_PORT,
|
||
|
timeout: 30000
|
||
|
});
|
||
|
|
||
|
let dataBuffer = '';
|
||
|
|
||
|
socket.on('data', (data) => {
|
||
|
dataBuffer += data.toString();
|
||
|
console.log('Server response:', data.toString());
|
||
|
|
||
|
if (dataBuffer.includes('220 ')) {
|
||
|
socket.write('EHLO testclient\r\n');
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('EHLO')) {
|
||
|
socket.write('MAIL FROM:<sender@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('sender accepted')) {
|
||
|
socket.write('RCPT TO:<recipient@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('recipient accepted')) {
|
||
|
socket.write('DATA\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('354 ')) {
|
||
|
// Create multi-line header with 50 segments
|
||
|
const segments = [];
|
||
|
for (let i = 0; i < 50; i++) {
|
||
|
segments.push(` Segment ${i}: ${' '.repeat(60)}value`);
|
||
|
}
|
||
|
|
||
|
const emailContent = [
|
||
|
`Subject: Test Email`,
|
||
|
`From: sender@example.com`,
|
||
|
`To: recipient@example.com`,
|
||
|
`X-Multi-Line: Initial value`,
|
||
|
...segments,
|
||
|
'',
|
||
|
'This email has a multi-line header with many segments.',
|
||
|
'.',
|
||
|
''
|
||
|
].join('\r\n');
|
||
|
|
||
|
socket.write(emailContent);
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('Message accepted') ||
|
||
|
dataBuffer.includes('552 ') ||
|
||
|
dataBuffer.includes('554 ') ||
|
||
|
dataBuffer.includes('500 ')) {
|
||
|
const accepted = dataBuffer.includes('250 ');
|
||
|
console.log(`Multi-line header test ${accepted ? 'accepted' : 'rejected'}`);
|
||
|
|
||
|
socket.write('QUIT\r\n');
|
||
|
socket.end();
|
||
|
done.resolve();
|
||
|
}
|
||
|
});
|
||
|
|
||
|
socket.on('error', (err) => {
|
||
|
console.error('Socket error:', err);
|
||
|
done.reject(err);
|
||
|
});
|
||
|
|
||
|
socket.on('timeout', () => {
|
||
|
console.error('Socket timeout');
|
||
|
socket.destroy();
|
||
|
done.reject(new Error('Socket timeout'));
|
||
|
});
|
||
|
|
||
|
await done.promise;
|
||
|
});
|
||
|
|
||
|
tap.test('Extremely Long Headers - should handle multiple long headers in one email', async (tools) => {
|
||
|
const done = tools.defer();
|
||
|
|
||
|
const socket = net.createConnection({
|
||
|
host: 'localhost',
|
||
|
port: TEST_PORT,
|
||
|
timeout: 30000
|
||
|
});
|
||
|
|
||
|
let dataBuffer = '';
|
||
|
|
||
|
socket.on('data', (data) => {
|
||
|
dataBuffer += data.toString();
|
||
|
console.log('Server response:', data.toString());
|
||
|
|
||
|
if (dataBuffer.includes('220 ')) {
|
||
|
socket.write('EHLO testclient\r\n');
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('EHLO')) {
|
||
|
socket.write('MAIL FROM:<sender@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('sender accepted')) {
|
||
|
socket.write('RCPT TO:<recipient@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('recipient accepted')) {
|
||
|
socket.write('DATA\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('354 ')) {
|
||
|
// Create multiple long headers
|
||
|
const header1 = 'A'.repeat(1000);
|
||
|
const header2 = 'B'.repeat(1500);
|
||
|
const header3 = 'C'.repeat(2000);
|
||
|
|
||
|
const emailContent = [
|
||
|
`Subject: Test Email with Multiple Long Headers`,
|
||
|
`From: sender@example.com`,
|
||
|
`To: recipient@example.com`,
|
||
|
`X-Long-Header-1: ${header1}`,
|
||
|
`X-Long-Header-2: ${header2}`,
|
||
|
`X-Long-Header-3: ${header3}`,
|
||
|
'',
|
||
|
'This email has multiple long headers.',
|
||
|
'.',
|
||
|
''
|
||
|
].join('\r\n');
|
||
|
|
||
|
const totalHeaderSize = header1.length + header2.length + header3.length;
|
||
|
console.log(`Total header size: ${totalHeaderSize} bytes`);
|
||
|
|
||
|
socket.write(emailContent);
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('Message accepted') ||
|
||
|
dataBuffer.includes('552 ') ||
|
||
|
dataBuffer.includes('554 ') ||
|
||
|
dataBuffer.includes('500 ')) {
|
||
|
const accepted = dataBuffer.includes('250 ');
|
||
|
console.log(`Multiple long headers test ${accepted ? 'accepted' : 'rejected'}`);
|
||
|
|
||
|
socket.write('QUIT\r\n');
|
||
|
socket.end();
|
||
|
done.resolve();
|
||
|
}
|
||
|
});
|
||
|
|
||
|
socket.on('error', (err) => {
|
||
|
console.error('Socket error:', err);
|
||
|
done.reject(err);
|
||
|
});
|
||
|
|
||
|
socket.on('timeout', () => {
|
||
|
console.error('Socket timeout');
|
||
|
socket.destroy();
|
||
|
done.reject(new Error('Socket timeout'));
|
||
|
});
|
||
|
|
||
|
await done.promise;
|
||
|
});
|
||
|
|
||
|
tap.test('Extremely Long Headers - should handle header with exactly RFC limit', async (tools) => {
|
||
|
const done = tools.defer();
|
||
|
|
||
|
const socket = net.createConnection({
|
||
|
host: 'localhost',
|
||
|
port: TEST_PORT,
|
||
|
timeout: 30000
|
||
|
});
|
||
|
|
||
|
let dataBuffer = '';
|
||
|
|
||
|
socket.on('data', (data) => {
|
||
|
dataBuffer += data.toString();
|
||
|
console.log('Server response:', data.toString());
|
||
|
|
||
|
if (dataBuffer.includes('220 ')) {
|
||
|
socket.write('EHLO testclient\r\n');
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('EHLO')) {
|
||
|
socket.write('MAIL FROM:<sender@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('sender accepted')) {
|
||
|
socket.write('RCPT TO:<recipient@example.com>\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('recipient accepted')) {
|
||
|
socket.write('DATA\r\n');
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('354 ')) {
|
||
|
// Create header line exactly at RFC 5322 limit (998 chars excluding CRLF)
|
||
|
// Header name and colon take some space
|
||
|
const headerName = 'X-RFC-Limit';
|
||
|
const colonSpace = ': ';
|
||
|
const remainingSpace = 998 - headerName.length - colonSpace.length;
|
||
|
const headerValue = 'X'.repeat(remainingSpace);
|
||
|
|
||
|
const emailContent = [
|
||
|
`Subject: Test Email`,
|
||
|
`From: sender@example.com`,
|
||
|
`To: recipient@example.com`,
|
||
|
`${headerName}${colonSpace}${headerValue}`,
|
||
|
'',
|
||
|
'This email has a header at exactly the RFC limit.',
|
||
|
'.',
|
||
|
''
|
||
|
].join('\r\n');
|
||
|
|
||
|
socket.write(emailContent);
|
||
|
dataBuffer = '';
|
||
|
} else if (dataBuffer.includes('250 ') && dataBuffer.includes('Message accepted') ||
|
||
|
dataBuffer.includes('552 ') ||
|
||
|
dataBuffer.includes('554 ') ||
|
||
|
dataBuffer.includes('500 ')) {
|
||
|
const accepted = dataBuffer.includes('250 ');
|
||
|
console.log(`RFC limit header test ${accepted ? 'accepted' : 'rejected'}`);
|
||
|
|
||
|
socket.write('QUIT\r\n');
|
||
|
socket.end();
|
||
|
done.resolve();
|
||
|
}
|
||
|
});
|
||
|
|
||
|
socket.on('error', (err) => {
|
||
|
console.error('Socket error:', err);
|
||
|
done.reject(err);
|
||
|
});
|
||
|
|
||
|
socket.on('timeout', () => {
|
||
|
console.error('Socket timeout');
|
||
|
socket.destroy();
|
||
|
done.reject(new Error('Socket timeout'));
|
||
|
});
|
||
|
|
||
|
await done.promise;
|
||
|
});
|
||
|
|
||
|
tap.test('cleanup - stop test server', async () => {
|
||
|
await stopTestServer(testServer);
|
||
|
});
|
||
|
|
||
|
tap.start();
|