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

MasterSocket.cpp Go to the documentation of this file. //#include <stdio.h> #include <assert.h> #include "MasterSocket.h" #include "MHandler.h" #include "ProxyInSocket.h" MasterSocket::MasterSocket(ISocketHandler& h) :BaseSocket(h) ,m_state(0) { } MasterSocket::~MasterSocket() { } void MasterSocket::Init() { EnableSSL(); } void MasterSocket::InitSSLServer() { InitializeContext("master", "comb.pem", "", SSLv23_method()); } void MasterSocket::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); if (m_length > PACKETSIZE) { printf("Incoming length (%d) larger than local buffer (%d) - terminating connection\n", m_length, PACKETSIZE); SetCloseAndDelete(); } m_packet_ptr = 0; m_state = 4; } else need_more = true; break; case 4: // read data (length bytes) if (m_length) { if (m_packet_ptr + l > PACKETSIZE) { printf("Trying to read beyond local buffer size (%d + %d > %d) - terminating connection\n", m_packet_ptr, l, PACKETSIZE); SetCloseAndDelete(); break; } 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_length == (short)m_packet_ptr) { slavecmd_t cmd = static_cast<slavecmd_t>(m_command); printmsg(m_ip, cmd, m_length); switch (cmd) { case S2M_CONNECT: // connect break; case S2M_DISCONNECTED: // disconnected CloseExtConn(); break; case S2M_DATA: // data SendToExt(); break; /* case S2M_PROXY_HEADER: // proxy header { unsigned short id; memcpy(&id, &m_ip, 2); ProxyInSocket *p = static_cast<MHandler&>(Handler()).GetProxySocket(id); if (p) p -> SendBuf(m_packet, m_packet_ptr); } break; case S2M_PROXY_DATA: // proxy data { unsigned short id; memcpy(&id, &m_ip, 2); ProxyInSocket *p = static_cast<MHandler&>(Handler()).GetProxySocket(id); if (p) p -> SendBuf(m_packet, m_packet_ptr); } break; case S2M_PROXY_CONNECT: // proxy connect { unsigned short id; memcpy(&id, &m_ip, 2); ProxyInSocket *p = static_cast<MHandler&>(Handler()).GetProxySocket(id); if (p) { p -> Send("HTTP/1.0 200 Connection Established\r\n\r\n"); } } break; case S2M_PROXY_CONNECT_FAILED: // proxy connect failed { unsigned short id; memcpy(&id, &m_ip, 2); ProxyInSocket *p = static_cast<MHandler&>(Handler()).GetProxySocket(id); if (p) { p -> Send("HTTP/1.0 404 Failed\r\n\r\n"); } } break; case S2M_PROXY_CLOSE: // close { unsigned short id; memcpy(&id, &m_ip, 2); ProxyInSocket *p = static_cast<MHandler&>(Handler()).GetProxySocket(id); if (p) p -> SetCloseAndDelete(); } break; */ } m_state = 0; } break; default: printf("MasterSocket: Bad state (%d)\n", m_state); assert(0); SetCloseAndDelete(); break; } } // while (!need_more) } void MasterSocket::SendToExt() { unsigned short id; memcpy(&id, &m_ip, 2); static_cast<MHandler&>(Handler()).SendToExt(id, m_packet, m_length); } void MasterSocket::CloseExtConn() { unsigned short id; memcpy(&id, &m_ip, 2); static_cast<MHandler&>(Handler()).CloseExtConn(id); } void MasterSocket::OnSSLAccept() { TcpSocket::OnSSLAccept(); } void MasterSocket::OnAccept() { TcpSocket::OnAccept(); printf("Incoming connection from: %s:%u\n", GetRemoteAddress().c_str(), GetRemotePort()); // check valid ip if (GetRemoteAddress() != static_cast<MHandler&>(Handler()).GetValidIP()) { printf("Refusing IP: %s\n", GetRemoteAddress().c_str()); SetCloseAndDelete(); } } void MasterSocket::OnDelete() { printf("Lost connection to: %s:%u\n", GetRemoteAddress().c_str(), GetRemotePort()); }

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

---