src/servers.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
   6  *                       E-mail:
   7  *                <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *            the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 using namespace std;
  18
  19 #include "inspircd_config.h"
  20 #include "servers.h"
  21 #include "inspircd.h"
  22 #include <unistd.h>
  23 #include <fcntl.h>
  24 #include <poll.h>
  25 #include <sys/errno.h>
  26 #include <sys/ioctl.h>
  27 #include <sys/utsname.h>
  28 #include <vector>
  29 #include <string>
  30 #include <deque>
  31 #include <sstream>
  32 #include <map>
  33 #include "inspstring.h"
  34 #include "helperfuncs.h"
  35 #include "connection.h"
  36
  37 extern time_t TIME;
  38 extern int MaxConn;
  39
  40 std::deque<std::string> xsums;
  41
  42 serverrec::serverrec()
  43 {
  44         strlcpy(name,"",256);
  45         pingtime = 0;
  46         lastping = TIME;
  47         usercount_i = usercount = opercount = version = 0;
  48         hops_away = 1;
  49         signon = TIME;
  50         jupiter = false;
  51         fd = 0;
  52         sync_soon = false;
  53         strlcpy(nickserv,"",NICKMAX);
  54         connectors.clear();
  55 }
  56
  57
  58 serverrec::~serverrec()
  59 {
  60 }
  61
  62 serverrec::serverrec(char* n, long ver, bool jupe)
  63 {
  64         strlcpy(name,n,256);
  65         lastping = TIME;
  66         usercount_i = usercount = opercount = 0;
  67         version = ver;
  68         hops_away = 1;
  69         signon = TIME;
  70         jupiter = jupe;
  71         fd = 0;
  72         sync_soon = false;
  73         strlcpy(nickserv,"",NICKMAX);
  74         connectors.clear();
  75 }
  76
  77 bool serverrec::CreateListener(char* newhost, int p)
  78 {
  79         sockaddr_in host_address;
  80         int flags;
  81         in_addr addy;
  82         int on = 0;
  83         struct linger linger = { 0 };
  84
  85         this->port = p;
  86
  87         fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
  88         if (fd <= 0)
  89         {
  90                 return false;
  91         }
  92
  93         setsockopt(fd,SOL_SOCKET,SO_REUSEADDR,(const char*)&on,sizeof(on));
  94         linger.l_onoff = 1;
  95         linger.l_linger = 1;
  96         setsockopt(fd,SOL_SOCKET,SO_LINGER,(const char*)&linger,sizeof(linger));
  97
  98         // attempt to increase socket sendq and recvq as high as its possible
  99         // to get them on linux.
 100         int sendbuf = 32768;
 101         int recvbuf = 32768;
 102         setsockopt(fd,SOL_SOCKET,SO_SNDBUF,(const void *)&sendbuf,sizeof(sendbuf));
 103         setsockopt(fd,SOL_SOCKET,SO_RCVBUF,(const void *)&recvbuf,sizeof(sendbuf));
 104
 105         memset((void*)&host_address, 0, sizeof(host_address));
 106
 107         host_address.sin_family = AF_INET;
 108
 109         if (!strcmp(newhost,""))
 110         {
 111                 host_address.sin_addr.s_addr = htonl(INADDR_ANY);
 112         }
 113         else
 114         {
 115                 inet_aton(newhost,&addy);
 116                 host_address.sin_addr = addy;
 117         }
 118
 119         host_address.sin_port = htons(p);
 120
 121         if (bind(fd,(sockaddr*)&host_address,sizeof(host_address))<0)
 122         {
 123                 return false;
 124         }
 125
 126         // make the socket non-blocking
 127         flags = fcntl(fd, F_GETFL, 0);
 128         fcntl(fd, F_SETFL, flags | O_NONBLOCK);
 129
 130         this->port = p;
 131
 132         listen(this->fd, MaxConn);
 133
 134         return true;
 135 }
 136
 137
 138 bool serverrec::BeginLink(char* targethost, int newport, char* password, char* servername, int myport)
 139 {
 140         char connect[MAXBUF];
 141
 142         ircd_connector connector;
 143         ircd_connector *cn = this->FindHost(servername);
 144
 145
 146         if (cn)
 147         {
 148                 WriteOpers("CONNECT aborted: Server %s already exists",servername);
 149                 return false;
 150         }
 151
 152
 153         if (this->fd)
 154         {
 155                 if (connector.MakeOutboundConnection(targethost,newport))
 156                 {
 157                         // targethost has been turned into an ip...
 158                         // we dont want this as the server name.
 159                         connector.SetServerName(servername);
 160                         snprintf(connect,MAXBUF,"S %s %s %lu %s :%s",getservername().c_str(),password,(unsigned long)myport,GetRevision().c_str(),getserverdesc().c_str());
 161                         connector.SetState(STATE_NOAUTH_OUTBOUND);
 162                         connector.SetHostAndPort(targethost, newport);
 163                         this->connectors.push_back(connector);
 164                         // this packet isn't actually sent until the socket connects -- the STATE_NOAUTH_OUTBOUND state
 165                         // queues outbound data until the socket is polled as writeable (e.g. the connection is established)
 166                         return this->SendPacket(connect, servername);
 167                 }
 168                 else
 169                 {
 170                         connector.SetState(STATE_DISCONNECTED);
 171                         WriteOpers("Could not create outbound connection to %s:%d",targethost,newport);
 172                 }
 173         }
 174         return false;
 175 }
 176
 177
 178 bool serverrec::MeshCookie(char* targethost, int newport, unsigned long cookie, char* servername)
 179 {
 180         char connect[MAXBUF];
 181
 182         ircd_connector connector;
 183
 184         WriteOpers("Establishing meshed link to %s:%d",servername,newport);
 185
 186         if (this->fd)
 187         {
 188                 if (connector.MakeOutboundConnection(targethost,newport))
 189                 {
 190                         // targethost has been turned into an ip...
 191                         // we dont want this as the server name.
 192                         connector.SetServerName(servername);
 193                         snprintf(connect,MAXBUF,"- %lu %s :%s",cookie,getservername().c_str(),getserverdesc().c_str());
 194                         connector.SetState(STATE_NOAUTH_OUTBOUND);
 195                         connector.SetHostAndPort(targethost, newport);
 196                         connector.SetState(STATE_CONNECTED);
 197                         this->connectors.push_back(connector);
 198                         return this->SendPacket(connect, servername);
 199                 }
 200                 else
 201                 {
 202                         connector.SetState(STATE_DISCONNECTED);
 203                         WriteOpers("Could not create outbound connection to %s:%d",targethost,newport);
 204                 }
 205         }
 206         return false;
 207 }
 208
 209 bool serverrec::AddIncoming(int newfd, char* targethost, int sourceport)
 210 {
 211         ircd_connector connector;
 212
 213         // targethost has been turned into an ip...
 214         // we dont want this as the server name.
 215         connector.SetServerName(targethost);
 216         connector.SetDescriptor(newfd);
 217         connector.SetState(STATE_NOAUTH_INBOUND);
 218         int flags = fcntl(newfd, F_GETFL, 0);
 219         fcntl(newfd, F_SETFL, flags | O_NONBLOCK);
 220         int sendbuf = 32768;
 221         int recvbuf = 32768;
 222         setsockopt(newfd,SOL_SOCKET,SO_SNDBUF,(const void *)&sendbuf,sizeof(sendbuf));
 223         setsockopt(newfd,SOL_SOCKET,SO_RCVBUF,(const void *)&recvbuf,sizeof(sendbuf));
 224         connector.SetHostAndPort(targethost, sourceport);
 225         connector.SetState(STATE_NOAUTH_INBOUND);
 226         log(DEBUG,"serverrec::AddIncoming() Added connection: %s:%d",targethost,sourceport);
 227         this->connectors.push_back(connector);
 228         return true;
 229 }
 230
 231 void serverrec::TerminateLink(char* targethost)
 232 {
 233         // this locates the targethost in the serverrec::connectors vector of the class,
 234         // and terminates it by sending it an SQUIT token and closing its descriptor.
 235         // TerminateLink with a null string causes a terminate of ALL links
 236 }
 237
 238 // Returns a pointer to the connector for 'host'
 239 ircd_connector* serverrec::FindHost(std::string findhost)
 240 {
 241         for (int i = 0; i < this->connectors.size(); i++)
 242         {
 243                 if (this->connectors[i].GetServerName() == findhost)
 244                 {
 245                         return &this->connectors[i];
 246                 }
 247         }
 248         return NULL;
 249 }
 250
 251 void serverrec::FlushWriteBuffers()
 252 {
 253         for (int i = 0; i < this->connectors.size(); i++)
 254         {
 255                 // don't try and ping a NOAUTH_OUTBOUND state, its not authed yet!
 256                 if ((this->connectors[i].GetState() == STATE_NOAUTH_OUTBOUND) && (TIME > this->connectors[i].age+30))
 257                 {
 258                         // however if we reach this timer its connected timed out :)
 259                         WriteOpers("*** Connection to %s timed out",this->connectors[i].GetServerName().c_str());
 260                         DoSplit(this->connectors[i].GetServerName().c_str());
 261                         return;
 262                 }
 263                 else if ((this->connectors[i].GetState() == STATE_NOAUTH_INBOUND) && (TIME > this->connectors[i].age+30))
 264                 {
 265                         WriteOpers("*** Connection from %s timed out",this->connectors[i].GetServerName().c_str());
 266                         DoSplit(this->connectors[i].GetServerName().c_str());
 267                         return;
 268                 }
 269                 else if (this->connectors[i].GetState() != STATE_DISCONNECTED)
 270                 {
 271                         if (!this->connectors[i].CheckPing())
 272                         {
 273                                 WriteOpers("*** Lost single connection to %s: Ping timeout",this->connectors[i].GetServerName().c_str());
 274                                 this->connectors[i].CloseConnection();
 275                                 this->connectors[i].SetState(STATE_DISCONNECTED);
 276                         }
 277                 }
 278                 if (this->connectors[i].HasBufferedOutput())
 279                 {
 280                         if (!this->connectors[i].FlushWriteBuf())
 281                         {
 282                                 // if we're here the write() caused an error, we cannot proceed
 283                                 WriteOpers("*** Lost single connection to %s, link inactive and retrying: %s",this->connectors[i].GetServerName().c_str(),this->connectors[i].GetWriteError().c_str());
 284                                 this->connectors[i].CloseConnection();
 285                                 this->connectors[i].SetState(STATE_DISCONNECTED);
 286                         }
 287                 }
 288         }
 289 }
 290
 291 bool serverrec::SendPacket(char *message, const char* sendhost)
 292 {
 293         if ((!message) || (!sendhost))
 294                 return true;
 295
 296         ircd_connector* cn = this->FindHost(sendhost);
 297
 298         if (!strchr(message,'\n'))
 299         {
 300                 strlcat(message,"\n",MAXBUF);
 301         }
 302
 303         if (cn)
 304         {
 305                 log(DEBUG,"main: serverrec::SendPacket() sent '%s' to %s",message,cn->GetServerName().c_str());
 306
 307                 if (cn->GetState() == STATE_DISCONNECTED)
 308                 {
 309                         // fix: can only route one hop to avoid a loop
 310                         if (strncmp(message,"R ",2))
 311                         {
 312                                 log(DEBUG,"Not a double reroute");
 313                                 // this route is down, we must re-route the packet through an available point in the mesh.
 314                                 for (int k = 0; k < this->connectors.size(); k++)
 315                                 {
 316                                         log(DEBUG,"Check connector %d: %s",k,this->connectors[k].GetServerName().c_str());
 317                                         // search for another point in the mesh which can 'reach' where we want to go
 318                                         for (int m = 0; m < this->connectors[k].routes.size(); m++)
 319                                         {
 320                                                 if (!strcasecmp(this->connectors[k].routes[m].c_str(),sendhost))
 321                                                 {
 322                                                         log(DEBUG,"Found alternative route for packet: %s",this->connectors[k].GetServerName().c_str());
 323                                                         char buffer[MAXBUF];
 324                                                         snprintf(buffer,MAXBUF,"R %s %s",sendhost,message);
 325                                                         this->SendPacket(buffer,this->connectors[k].GetServerName().c_str());
 326                                                         return true;
 327                                                 }
 328                                         }
 329                                 }
 330                         }
 331                         char buffer[MAXBUF];
 332                         snprintf(buffer,MAXBUF,"& %s",sendhost);
 333                         WriteOpers("*** All connections to %s lost.",sendhost);
 334                         NetSendToAllExcept(sendhost,buffer);
 335                         DoSplit(sendhost);
 336                         return false;
 337                 }
 338
 339                 // returns false if the packet could not be sent (e.g. target host down)
 340                 if (!cn->AddWriteBuf(message))
 341                 {
 342                         // if we're here, there was an error pending, and the send cannot proceed
 343                         log(DEBUG,"cn->AddWriteBuf() failed for serverrec::SendPacket(): %s",cn->GetWriteError().c_str());
 344                         log(DEBUG,"Disabling connector: %s",cn->GetServerName().c_str());
 345                         cn->CloseConnection();
 346                         cn->SetState(STATE_DISCONNECTED);
 347                         WriteOpers("*** Lost single connection to %s, link inactive and retrying: %s",cn->GetServerName().c_str(),cn->GetWriteError().c_str());
 348                         // retry the packet along a new route so either arrival OR failure are gauranteed (bugfix)
 349                         return this->SendPacket(message,sendhost);
 350                 }
 351                 if (!cn->FlushWriteBuf())
 352                 {
 353                         // if we're here the write() caused an error, we cannot proceed
 354                         log(DEBUG,"cn->FlushWriteBuf() failed for serverrec::SendPacket(): %s",cn->GetWriteError().c_str());
 355                         log(DEBUG,"Disabling connector: %s",cn->GetServerName().c_str());
 356                         cn->CloseConnection();
 357                         cn->SetState(STATE_DISCONNECTED);
 358                         WriteOpers("*** Lost single connection to %s, link inactive and retrying: %s",cn->GetServerName().c_str(),cn->GetWriteError().c_str());
 359                         // retry the packet along a new route so either arrival OR failure are gauranteed
 360                         return this->SendPacket(message,sendhost);
 361                 }
 362                 return true;
 363         }
 364 }
 365
 366 bool already_have_sum(std::string sum)
 367 {
 368         for (int i = 0; i < xsums.size(); i++)
 369         {
 370                 if (xsums[i] == sum)
 371                 {
 372                         return true;
 373                 }
 374         }
 375         if (xsums.size() >= 128)
 376         {
 377                 xsums.pop_front();
 378         }
 379         xsums.push_back(sum);
 380         return false;
 381 }
 382
 383 // receives a packet from any where there is data waiting, first come, first served
 384 // fills the message and host values with the host where the data came from.
 385
 386 bool serverrec::RecvPacket(std::deque<std::string> &messages, char* recvhost,std::deque<std::string> &sums)
 387 {
 388         char data[65536];
 389         memset(data, 0, 65536);
 390         for (int i = 0; i < this->connectors.size(); i++)
 391         {
 392                 if (this->connectors[i].GetState() != STATE_DISCONNECTED)
 393                 {
 394                         // returns false if the packet could not be sent (e.g. target host down)
 395                         int rcvsize = 0;
 396
 397                         // check if theres any data on this socket
 398                         // if not, continue onwards to the next.
 399                         pollfd polls;
 400                         polls.fd = this->connectors[i].GetDescriptor();
 401                         polls.events = POLLIN;
 402                         int ret = poll(&polls,1,1);
 403                         if (ret <= 0) continue;
 404
 405                         rcvsize = recv(this->connectors[i].GetDescriptor(),data,65000,0);
 406                         data[rcvsize] = '\0';
 407                         if (rcvsize == -1)
 408                         {
 409                                 if (errno != EAGAIN)
 410                                 {
 411                                         log(DEBUG,"recv() failed for serverrec::RecvPacket(): %s",strerror(errno));
 412                                         log(DEBUG,"Disabling connector: %s",this->connectors[i].GetServerName().c_str());
 413                                         this->connectors[i].CloseConnection();
 414                                         this->connectors[i].SetState(STATE_DISCONNECTED);
 415                                 }
 416                         }
 417                         int pushed = 0;
 418                         if (rcvsize > 0)
 419                         {
 420                                 if (!this->connectors[i].AddBuffer(data))
 421                                 {
 422                                         WriteOpers("*** Read buffer for %s exceeds maximum, closing connection!",this->connectors[i].GetServerName().c_str());
 423                                         this->connectors[i].CloseConnection();
 424                                         this->connectors[i].SetState(STATE_DISCONNECTED);
 425                                 }
 426                                 if (this->connectors[i].BufferIsComplete())
 427                                 {
 428                                         this->connectors[i].ResetPing();
 429                                         while (this->connectors[i].BufferIsComplete())
 430                                         {
 431                                                 std::string text = this->connectors[i].GetBuffer();
 432                                                 if (text != "")
 433                                                 {
 434                                                         if ((text[0] == ':') && (text.find(" ") != std::string::npos))
 435                                                         {
 436                                                                 std::string orig = text;
 437                                                                 log(DEBUG,"Original: %s",text.c_str());
 438                                                                 std::string sum = text.substr(1,text.find(" ")-1);
 439                                                                 text = text.substr(text.find(" ")+1,text.length());
 440                                                                 std::string possible_token = text.substr(1,text.find(" ")-1);
 441                                                                 if (possible_token.length() > 1)
 442                                                                 {
 443                                                                         sums.push_back("*");
 444                                                                         text = orig;
 445                                                                         log(DEBUG,"Non-mesh, non-tokenized string passed up the chain");
 446                                                                 }
 447                                                                 else
 448                                                                 {
 449                                                                         log(DEBUG,"Packet sum: '%s'",sum.c_str());
 450                                                                         if ((already_have_sum(sum)) && (sum != "*"))
 451                                                                         {
 452                                                                                 // we don't accept dupes
 453                                                                                 continue;
 454                                                                         }
 455                                                                         sums.push_back(sum.c_str());
 456                                                                 }
 457                                                         }
 458                                                         else sums.push_back("*");
 459                                                         messages.push_back(text.c_str());
 460                                                         strlcpy(recvhost,this->connectors[i].GetServerName().c_str(),160);
 461                                                         log(DEBUG,"serverrec::RecvPacket() %d:%s->%s",pushed++,recvhost,text.c_str());
 462                                                 }
 463                                         }
 464                                         return true;
 465                                 }
 466                         }
 467                 }
 468         }
 469         // nothing new yet -- message and host will be undefined
 470         return false;
 471 }
 472