【nodejs每日一讲】使用nodejs写一个代理服务器

theanarkh

发布于 2021-01-12 06:57:53

39900

代码可运行

文章被收录于专栏：原创分享原创分享

运行总次数：0

代码可运行

工具函数

function buildHeaders(headers) {
  const arr = [];
  for (const [k, v] of Object.entries(headers)) {
    arr.push(`${k}:${v}\r\n`)
  }
  return arr.join('');
}

function getHostAndPort(req) {
  let host;
  let port;
  try {
    [host, port] = new URL(req.url);
  } catch(e) {
    [host, port] = req.headers.host.split(':');
  } finally {
      if (!port) {
          port = 80;
      }
  }
  console.log(host, port);
  return [host, port];
}

1 http代理

const server = http.createServer((req, res) => {
  const [host, port] = getHostAndPort(req);
  http.get({
    port,
    host,
    path: req.url
  }, (response) => {
    response.pipe(res);
  });
});

2 websocket代理

server.on('upgrade', (req, res, head) => {
  const [host, port] = getHostAndPort(req);
  const client = net.connect({
    port,
    host,
  });
  client.on('connect', () => {
    client.write(`GET ${req.url}\r\n` + buildHeaders(req.headers) + '\r\n');
    res.pipe(client);
    client.pipe(res);
  });
  client.on('error', () => {
    res.destroy();
  })
});

3 https代理

server.on('connect', (req, client, head) => {
  const [host, port] = getHostAndPort(req);
  const socket = net.connect(port, host, () => {
    client.write('HTTP/1.1 200 Connection Established\r\n' +
                    'Proxy-agent: Node.js-Proxy\r\n' +
                    '\r\n');
    socket.write(head);
    socket.pipe(client);
    client.pipe(socket);
  });
});

最后写一个https服务器测试

var https = require('https');
var fs = require('fs');
const WebSocket = require('ws');

var options = {
  key: fs.readFileSync('./files/server-key.pem'),
  ca: [fs.readFileSync('./files/ca-cert.pem')],
  cert: fs.readFileSync('./files/server-cert.pem')
};

const server = https.createServer(options,function(req,res){
  res.end('ok');
}).listen(11111);