root/lang/cplusplus/friends_framework/trunk/test/proxy.cpp

#include <errno.h>
#include <fcntl.h>
#include "friends_framework.h"

static CoopThreadMgr mgr(new CoopSocketMgr(), new CoopTimeoutMgr(600), 10);
static sockaddr_in server_addr, my_addr;

struct ProxyConn {
  CoopRunnableSocket<ProxyConn> clientSock;
  CoopRunnableSocket<ProxyConn> serverSock;
  CoopRunnableTimeout<ProxyConn> timeout;
  CoopBuffer dataToServer;
  CoopBuffer dataToClient;
  ProxyConn(int clientFd, sockaddr *, socklen_t)
  : clientSock(&mgr, this, &ProxyConn::OnClientRead, &ProxyConn::OnClientWrite),
    serverSock(&mgr, this, &ProxyConn::OnServerRead,
               &ProxyConn::OnServerConnect),
    timeout(&mgr, this, &ProxyConn::OnTimeout) {
    clientSock.SetSock(clientFd);
  }
  ProxyConn *Init();
  void ResetTimeout() {
    timeout.SetTimeout(10 * mgr.TicksPerSecond());
  }
  void OnTimeout(CoopTimeout *) {
    delete this;
  }
  void OnServerConnect(CoopSocket *) {
    ResetTimeout();
    clientSock.SetWaitForRead();
    serverSock.SetWaitForRead();
    serverSock.ClearWaitForWrite();
    serverSock.writef = &ProxyConn::OnServerWrite;
  }
  void OnClientRead(CoopSocket *) {
    ResetTimeout();
    if (! CoopTransfer::OnRead(&clientSock, &serverSock, dataToServer))
      delete this;
  }
  void OnServerWrite(CoopSocket *) {
    ResetTimeout();
    if (! CoopTransfer::OnWrite(&clientSock, &serverSock, dataToServer))
      delete this;
  }
  void OnServerRead(CoopSocket *) {
    ResetTimeout();
    if (! CoopTransfer::OnRead(&serverSock, &clientSock, dataToClient))
      delete this;
  }
  void OnClientWrite(CoopSocket *) {
    ResetTimeout();
    if (! CoopTransfer::OnWrite(&serverSock, &clientSock, dataToClient))
      delete this;
  }
};

ProxyConn *ProxyConn::Init()
{
  serverSock.SetSock(::socket(AF_INET, SOCK_STREAM, 0));
  assert(serverSock.GetSock() != -1);
  if (::connect(serverSock.GetSock(), reinterpret_cast<sockaddr*>(&server_addr),
                sizeof(server_addr)) == -1) {
    switch (errno) {
    case EAGAIN:
    case EINPROGRESS:
      serverSock.SetWaitForWrite();
      break;
    default:
      perror("failed to connect to server");
      delete this;
      return NULL;
      }
  } else {
    OnServerConnect(&serverSock);
  }
  ResetTimeout();
  return this;
}

int main(void)
{
  server_addr.sin_family = AF_INET;
  server_addr.sin_addr.s_addr = htonl(0x7f000001);
  server_addr.sin_port = htons(80);
  
  my_addr.sin_family = AF_INET;
  my_addr.sin_addr.s_addr = htonl(0x7f000001);
  my_addr.sin_port = htons(9999);
  
  if ((new CoopListener<ProxyConn>(&mgr))->Init(reinterpret_cast<sockaddr*>(&my_addr), sizeof(my_addr)) == NULL)
    exit(1);
  
  mgr.Run();
  
  return 0;
}