雑記２

const http = require('http')
const https = require('https')
const fs = require('fs')
const url = require('url')
const net = require('net')
const tls = require('tls')
const pem = require('pem')

const ConnectionEstablishedMsg = new Buffer('HTTP/1.1 200 Connection Established\r\n\r\n')
const cache = { }

const key = fs.readFileSync('../keys/fake-ca.key.pem')
const cert = fs.readFileSync('../keys/fake-ca.crt.pem')

const httpsServer = https.createServer({
	key,
	cert,
	SNICallback: (serverName, cb) => {
		if (cache[serverName]) {
			cb(null, cache[serverName])

			return
		}

		pem.createCertificate({
			country: 'JP',
			state: 'Tokyo',
			locality: 'Shibuya',
			organization: 'preflight.cc',
			commonName: serverName,
			altNames: [ serverName ],
			serviceKey: key,
			serviceCertificate: cert,
			serial: Date.now(),
			days: 300
		}, (err, generated) => {
			if (err) {
				cb(err)

				return
			}

			const ctx = tls.createSecureContext({
				key: generated.clientKey,
				cert: generated.certificate
			})

			cache[serverName] = ctx

			cb(null, ctx)
		})
	}
}, (req, res) => {
	const fullUrl = `https://${req.headers.host}${req.url}`

	const parsedUrl = url.parse(fullUrl)

	const pReq = https.request({
		method: req.method,
		host: parsedUrl.hostname,
		port: parsedUrl.port || 443,
		path: parsedUrl.path,
		headers: req.headers
	}, pRes => {
		res.writeHead(pRes.statusCode, pRes.headers)
		pRes.pipe(res)
	})

	req.pipe(pReq)
})

httpsServer.listen()

const server = http.createServer((req, res) => {
	const parsedUrl = url.parse(req.url)

	const pReq = http.request({
		method: req.method,
		host: parsedUrl.hostname,
		port: parsedUrl.port || 80,
		path: parsedUrl.path,
		headers: req.headers
	}, (pRes) => {
		res.writeHead(pRes.statusCode, pRes.headers)
		pRes.pipe(res)
	})

	req.pipe(pReq)
})

server.on('connect', (req, sock, head) => {
	const addr = httpsServer.address()

	const pSock = net.connect(addr.port, addr.address, () => {
		sock.write(ConnectionEstablishedMsg)
		pSock.write(head)

		sock.pipe(pSock)
		pSock.pipe(sock)
	})
})

server.listen(8081)