Search Site ~Sockets~  C++ Sockets   Tutorial   Classes   License   Java Sockets  ~Examples~  SMTPProxy   PortProxy   ChatServer   Small MUD   ClientServer   HttpGet   DChat   WebServer   UDP   ICMP  ~Contact~ message board  email author

SlaveSocket.cpp Go to the documentation of this file. //#include <stdio.h> #include <Parse.h> #include <assert.h> #include "SlaveSocket.h" #include "InternalSocket.h" #include "SlaveHandler.h" #include "ProxyOutSocket.h" #include "ProxyBinSocket.h" #include "MasterSocket.h" SlaveSocket::SlaveSocket(ISocketHandler& h) :BaseSocket(h) ,m_state(0) { } SlaveSocket::~SlaveSocket() { } void SlaveSocket::OnRead() { TcpSocket::OnRead(); bool need_more = false; while (!need_more && !CloseAndDelete()) { size_t l = ibuf.GetLength(); switch (m_state) { case 0: // to id (2 bytes) if (l >= 2) { ibuf.Read( (char *)&m_ip, 2); m_ip = ntohs(m_ip); m_state = 2; } else need_more = true; break; case 2: // command (open, close) if (l >= 2) { ibuf.Read( (char *)&m_command, 2); m_command = ntohs(m_command); m_state = 3; } else need_more = true; break; case 3: // length (2 bytes) if (l >= 2) { ibuf.Read( (char *)&m_length, 2); m_length = ntohs(m_length); m_packet_ptr = 0; m_state = 4; } else need_more = true; break; case 4: // read data (length bytes) if (m_length) { if (l < m_length - m_packet_ptr) // not enough { ibuf.Read(m_packet + m_packet_ptr, l); m_packet_ptr += l; need_more = true; } else { ibuf.Read(m_packet + m_packet_ptr, m_length - m_packet_ptr); m_packet_ptr += m_length - m_packet_ptr; } } if (m_packet_ptr == (size_t)m_length) { mastercmd_t cmd = static_cast<mastercmd_t>(m_command); printmsg(m_ip, cmd, m_length); switch (cmd) { case M2S_OPEN: // open OpenConnection(); break; case M2S_CLOSE: // close CloseConnection(); break; case M2S_DATA: // data SendPacket(); break; /* case M2S_PROXY_HEADER: // http proxy header // m_ip - proxy session id# // m_port - 0 m_packet[m_length] = 0; { std::string header = m_packet; std::string host; port_t port = GetHostPort(header, host); unsigned short id; memcpy(&id, &m_ip, 2); ProxyOutSocket *p = new ProxyOutSocket(Handler(), id, false); // false - no connect p -> SetDeleteByHandler(); p -> Open(host, port); Handler().Add(p); p -> Send(header); } break; case M2S_PROXY_BODY: // http proxy body // m_ip - proxy session id# // m_port - 0 { unsigned short id; memcpy(&id, &m_ip, 2); TcpSocket *p = static_cast<SlaveHandler&>(Handler()).GetProxySocket(id); if (p) { p -> SendBuf(m_packet, m_length); } } break; case M2S_PROXY_CONNECT: // proxy connect // m_ip - proxy session id# // m_port - 0 m_packet[m_length] = 0; { std::string header = m_packet; std::string host; port_t port = GetHostPort(header, host); unsigned short id; memcpy(&id, &m_ip, 2); ProxyBinSocket *p = new ProxyBinSocket(Handler(), id); p -> SetDeleteByHandler(); p -> Open(host, port); Handler().Add(p); } break; */ } // switch (command) m_state = 0; } // length == packet_ptr break; default: printf("SlaveSocket: Bad state (%d)\n", m_state); assert(0); SetCloseAndDelete(); break; } } // while (!need_more) } void SlaveSocket::OpenConnection() { unsigned short id; memcpy(&id, &m_ip, 2); m_packet[m_length] = 0; printf(" payload: %s\n", m_packet); Parse pa(m_packet, ":"); std::string host = pa.getword(); port_t port = pa.getvalue(); printf(" host: %s port: %u\n", host.c_str(), port); InternalSocket *p = new InternalSocket(Handler(), this, id); p -> SetDeleteByHandler(); p -> Open(host, port); Handler().Add(p); } void SlaveSocket::CloseConnection() { unsigned short id; memcpy(&id, &m_ip, 2); InternalSocket *p = GetSock(id); if (p) p -> SetCloseAndDelete(); } void SlaveSocket::SendPacket() { unsigned short id; memcpy(&id, &m_ip, 2); InternalSocket *p = GetSock(id); if (p) p -> SendBuf(m_packet, m_length); } InternalSocket *SlaveSocket::GetSock(unsigned short id) { return static_cast<SlaveHandler&>(Handler()).GetSock(id); } port_t SlaveSocket::GetHostPort(const std::string& header,std::string& host) { size_t i = 0; port_t port = 0; while (i < header.size()) { size_t x = i; while (header[i] != 13 && header[i] != 10 && i < header.size()) { i++; } std::string line = header.substr(x, i - x); while (i < header.size() && (header[i] == 13 || header[i] == 10)) { i++; } Parse pa(line, ":"); std::string key = pa.getword(); if (!strcasecmp(key.c_str(), "host")) { Parse pa2(pa.getrest(), ":"); host = pa2.getword(); port = pa2.getvalue(); if (!port) port = 80; return port; } } return 0; }

Page, code, and content Copyright (C) 2006 by Anders Hedström

---