1 #include <connection.h>
6 #include <sys/utsname.h>
10 extern std::vector<Module*> modules;
11 extern std::vector<ircd_module*> factory;
16 #define STATE_WAIT_FOR_ACK 2
28 connection::connection()
36 bool connection::CreateListener(char* host, int p)
38 sockaddr_in host_address;
42 struct linger linger = { 0 };
44 fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
50 memset((void*)&host_address, 0, sizeof(host_address));
52 host_address.sin_family = AF_INET;
56 host_address.sin_addr.s_addr = htonl(INADDR_ANY);
60 inet_aton(host,&addy);
61 host_address.sin_addr = addy;
64 host_address.sin_port = htons(p);
66 if (bind(fd,(sockaddr*)&host_address,sizeof(host_address))<0)
71 // make the socket non-blocking
72 flags = fcntl(fd, F_GETFL, 0);
73 fcntl(fd, F_SETFL, flags | O_NONBLOCK);
77 setsockopt(fd,SOL_SOCKET,SO_REUSEADDR,(const char*)&on,sizeof(on));
80 setsockopt(fd,SOL_SOCKET,SO_LINGER,(const char*)&linger,sizeof(linger));
85 bool connection::BeginLink(char* targethost, int port, char* password)
91 sprintf(connect,"S %s %s :%s",getservername().c_str(),password,getserverdesc().c_str());
92 this->haspassed = false;
93 return this->SendPacket(connect, targethost, port, 0);
98 // targethost: in dot notation a.b.c.d
99 void connection::TerminateLink(char* targethost)
103 // host: in dot notation a.b.c.d
104 // port: host byte order
105 bool connection::SendPacket(char *message, char* host, int port, long ourkey)
107 sockaddr_in host_address;
111 memset((void*)&host_address, 0, sizeof(host_address));
113 host_address.sin_family = AF_INET;
114 inet_aton(host,&addy);
115 host_address.sin_addr = addy;
117 host_address.sin_port = htons(port);
119 strcpy(p.data,message);
120 p.type = PT_SYN_WITH_DATA;
124 FOREACH_MOD OnPacketTransmit(p.data);
126 log(DEBUG,"main: Connection::SendPacket() sent '%s' to %s:%d",p.data,host,port);
128 // returns false if the packet could not be sent (e.g. target host down)
129 if (sendto(this->fd,&p,sizeof(p),0,(sockaddr*)&host_address,sizeof(host_address))<0)
131 log(DEBUG,"sendto() failed for Connection::SendPacket() with a packet of size %d",sizeof(p));
134 this->state = STATE_WAIT_FOR_ACK;
137 // host_address remains unchanged. we only want to receive from where we just sent the packet to.
139 // retry the packet up to 5 times
140 for (int retries = 0; retries < 5; retries++)
142 socklen_t host_address_size;
143 host_address.sin_family=AF_INET;
144 host_address_size=sizeof(host_address);
146 // wait for ack, or timeout.
147 // if reached a timeout, send again.
148 // the packet id in the ack must match that in the original packet
149 // this MUST operate in lock/step fashion!!!
160 int res = select(65535, &sfd, NULL, NULL, &tval);
163 while ((recvfrom(fd,&p2,sizeof(p2),0,(sockaddr*)&host_address,&host_address_size)<0) && (cycles < 10));
167 log(DEFAULT,"ERROR! connection::SendPacket() waited >10000 nanosecs for an ACK. Will resend up to 5 times");
173 log(DEFAULT,"ERROR! connection::SendPacket() received an ack for a packet it didnt send!");
174 this->state = STATE_CLEAR;
179 log(DEFAULT,"Successfully received ACK");
180 this->state = STATE_CLEAR;
186 log(DEFAULT,"We never received an ack. Something fishy going on, host is dead.");
187 this->state = STATE_CLEAR;
192 bool connection::SendSYN(char* host, int port)
194 sockaddr_in host_address;
198 memset((void*)&host_address, 0, sizeof(host_address));
200 host_address.sin_family = AF_INET;
201 inet_aton(host,&addy);
202 host_address.sin_addr = addy;
204 host_address.sin_port = htons(port);
206 p.type = PT_SYN_ONLY;
210 if (sendto(fd,&p,sizeof(p),0,(sockaddr*)&host_address,sizeof(host_address))<0)
218 bool connection::SendACK(char* host, int port, int reply_id)
220 sockaddr_in host_address;
224 memset((void*)&host_address, 0, sizeof(host_address));
226 host_address.sin_family = AF_INET;
227 inet_aton(host,&addy);
228 host_address.sin_addr = addy;
230 host_address.sin_port = htons(port);
232 p.type = PT_ACK_ONLY;
237 if (sendto(fd,&p,sizeof(p),0,(sockaddr*)&host_address,sizeof(host_address))<0)
245 // Generates a server key. This is pseudo-random.
246 // the server always uses the same server-key in all communications
247 // across the network. All other servers must remember the server key
248 // of servers in the network, e.g.:
250 // ServerA: key=5555555555
251 // ServerB: key=6666666666
252 // I am ServerC: key=77777777777
254 // If ServerC sees a packet from ServerA, and the key stored for ServerA
255 // is 0, then cache the key as the servers key.
256 // after this point, any packet from ServerA which does not contain its key,
257 // 555555555, will be silently dropped.
258 // This should prevent blind spoofing, as to fake a server you must know its
259 // assigned key, and to do that you must receive messages that are origintated
260 // from it or hack the running executable.
262 // During the AUTH phase (when server passwords are checked, the key in any
263 // packet MUST be 0). Only the initial SERVER/PASS packets may have a key
264 // of 0 (and any ACK responses to them).
267 long connection::GenKey()
270 return (random()*time(NULL));
273 // host: in dot notation a.b.c.d
274 // port: host byte order
275 bool connection::RecvPacket(char *message, char* host, int &prt, long &theirkey)
277 // returns false if no packet waiting for receive, e.g. EAGAIN or ECONNRESET
278 sockaddr_in host_address;
279 socklen_t host_address_size;
282 memset((void*)&host_address, 0, sizeof(host_address));
284 host_address.sin_family=AF_INET;
285 host_address_size=sizeof(host_address);
287 if (recvfrom(fd,&p,sizeof(p),0,(sockaddr*)&host_address,&host_address_size)<0)
292 log(DEBUG,"connection::RecvPacket(): received packet type %d '%s'",p.type,p.data);
294 if (p.type == PT_SYN_ONLY)
296 strcpy(message,p.data);
297 strcpy(host,inet_ntoa(host_address.sin_addr));
298 prt = ntohs(host_address.sin_port);
299 SendACK(host,this->port,p.id);
303 if (p.type == PT_ACK_ONLY)
305 strcpy(message,p.data);
306 strcpy(host,inet_ntoa(host_address.sin_addr));
307 prt = ntohs(host_address.sin_port);
308 this->state = STATE_CLEAR;
312 if (p.type == PT_SYN_WITH_DATA)
314 strcpy(message,p.data);
315 strcpy(host,inet_ntoa(host_address.sin_addr));
317 prt = ntohs(host_address.sin_port);
318 SendACK(host,this->port,p.id);
322 log(DEBUG,"connection::RecvPacket(): Invalid packet type %d (protocol error)",p.type);