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                         return this->SendPacket(connect, servername);
 165                 }
 166                 else
 167                 {
 168                         connector.SetState(STATE_DISCONNECTED);
 169                         WriteOpers("Could not create outbound connection to %s:%d",targethost,newport);
 170                 }
 171         }
 172         return false;
 173 }
 174
 175
 176 bool serverrec::MeshCookie(char* targethost, int newport, unsigned long cookie, char* servername)
 177 {
 178         char connect[MAXBUF];
 179
 180         ircd_connector connector;
 181
 182         WriteOpers("Establishing meshed link to %s:%d",servername,newport);
 183
 184         if (this->fd)
 185         {
 186                 if (connector.MakeOutboundConnection(targethost,newport))
 187                 {
 188                         // targethost has been turned into an ip...
 189                         // we dont want this as the server name.
 190                         connector.SetServerName(servername);
 191                         snprintf(connect,MAXBUF,"- %lu %s :%s",cookie,getservername().c_str(),getserverdesc().c_str());
 192                         connector.SetState(STATE_NOAUTH_OUTBOUND);
 193                         connector.SetHostAndPort(targethost, newport);
 194                         connector.SetState(STATE_CONNECTED);
 195                         this->connectors.push_back(connector);
 196                         return this->SendPacket(connect, servername);
 197                 }
 198                 else
 199                 {
 200                         connector.SetState(STATE_DISCONNECTED);
 201                         WriteOpers("Could not create outbound connection to %s:%d",targethost,newport);
 202                 }
 203         }
 204         return false;
 205 }
 206
 207 bool serverrec::AddIncoming(int newfd, char* targethost, int sourceport)
 208 {
 209         ircd_connector connector;
 210
 211         // targethost has been turned into an ip...
 212         // we dont want this as the server name.
 213         connector.SetServerName(targethost);
 214         connector.SetDescriptor(newfd);
 215         connector.SetState(STATE_NOAUTH_INBOUND);
 216         int flags = fcntl(newfd, F_GETFL, 0);
 217         fcntl(newfd, F_SETFL, flags | O_NONBLOCK);
 218         int sendbuf = 32768;
 219         int recvbuf = 32768;
 220         setsockopt(newfd,SOL_SOCKET,SO_SNDBUF,(const void *)&sendbuf,sizeof(sendbuf));
 221         setsockopt(newfd,SOL_SOCKET,SO_RCVBUF,(const void *)&recvbuf,sizeof(sendbuf));
 222         connector.SetHostAndPort(targethost, sourceport);
 223         connector.SetState(STATE_NOAUTH_INBOUND);
 224         log(DEBUG,"serverrec::AddIncoming() Added connection: %s:%d",targethost,sourceport);
 225         this->connectors.push_back(connector);
 226         return true;
 227 }
 228
 229 void serverrec::TerminateLink(char* targethost)
 230 {
 231         // this locates the targethost in the serverrec::connectors vector of the class,
 232         // and terminates it by sending it an SQUIT token and closing its descriptor.
 233         // TerminateLink with a null string causes a terminate of ALL links
 234 }
 235
 236 // Returns a pointer to the connector for 'host'
 237 ircd_connector* serverrec::FindHost(std::string findhost)
 238 {
 239         for (int i = 0; i < this->connectors.size(); i++)
 240         {
 241                 if (this->connectors[i].GetServerName() == findhost)
 242                 {
 243                         return &this->connectors[i];
 244                 }
 245         }
 246         return NULL;
 247 }
 248
 249 void serverrec::FlushWriteBuffers()
 250 {
 251         for (int i = 0; i < this->connectors.size(); i++)
 252         {
 253                 if (this->connectors[i].GetState() != STATE_DISCONNECTED)
 254                 {
 255                         if (!this->connectors[i].CheckPing())
 256                         {
 257                                 WriteOpers("*** Lost single connection to %s: Ping timeout",this->connectors[i].GetServerName().c_str());
 258                                 this->connectors[i].CloseConnection();
 259                                 this->connectors[i].SetState(STATE_DISCONNECTED);
 260                         }
 261                 }
 262                 if (this->connectors[i].HasBufferedOutput())
 263                 {
 264                         if (!this->connectors[i].FlushWriteBuf())
 265                         {
 266                                 // if we're here the write() caused an error, we cannot proceed
 267                                 WriteOpers("*** Lost single connection to %s, link inactive and retrying: %s",this->connectors[i].GetServerName().c_str(),this->connectors[i].GetWriteError().c_str());
 268                                 this->connectors[i].CloseConnection();
 269                                 this->connectors[i].SetState(STATE_DISCONNECTED);
 270                         }
 271                 }
 272         }
 273 }
 274
 275 bool serverrec::SendPacket(char *message, const char* sendhost)
 276 {
 277         if ((!message) || (!sendhost))
 278                 return true;
 279
 280         ircd_connector* cn = this->FindHost(sendhost);
 281
 282         if (!strchr(message,'\n'))
 283         {
 284                 strlcat(message,"\n",MAXBUF);
 285         }
 286
 287         if (cn)
 288         {
 289                 log(DEBUG,"main: serverrec::SendPacket() sent '%s' to %s",message,cn->GetServerName().c_str());
 290
 291                 if (cn->GetState() == STATE_DISCONNECTED)
 292                 {
 293                         // fix: can only route one hop to avoid a loop
 294                         if (strncmp(message,"R ",2))
 295                         {
 296                                 log(DEBUG,"Not a double reroute");
 297                                 // this route is down, we must re-route the packet through an available point in the mesh.
 298                                 for (int k = 0; k < this->connectors.size(); k++)
 299                                 {
 300                                         log(DEBUG,"Check connector %d: %s",k,this->connectors[k].GetServerName().c_str());
 301                                         // search for another point in the mesh which can 'reach' where we want to go
 302                                         for (int m = 0; m < this->connectors[k].routes.size(); m++)
 303                                         {
 304                                                 if (!strcasecmp(this->connectors[k].routes[m].c_str(),sendhost))
 305                                                 {
 306                                                         log(DEBUG,"Found alternative route for packet: %s",this->connectors[k].GetServerName().c_str());
 307                                                         char buffer[MAXBUF];
 308                                                         snprintf(buffer,MAXBUF,"R %s %s",sendhost,message);
 309                                                         this->SendPacket(buffer,this->connectors[k].GetServerName().c_str());
 310                                                         return true;
 311                                                 }
 312                                         }
 313                                 }
 314                         }
 315                         char buffer[MAXBUF];
 316                         snprintf(buffer,MAXBUF,"& %s",sendhost);
 317                         WriteOpers("*** All connections to %s lost.",sendhost);
 318                         NetSendToAllExcept(sendhost,buffer);
 319                         DoSplit(sendhost);
 320                         return false;
 321                 }
 322
 323                 // returns false if the packet could not be sent (e.g. target host down)
 324                 if (!cn->AddWriteBuf(message))
 325                 {
 326                         // if we're here, there was an error pending, and the send cannot proceed
 327                         log(DEBUG,"cn->AddWriteBuf() failed for serverrec::SendPacket(): %s",cn->GetWriteError().c_str());
 328                         log(DEBUG,"Disabling connector: %s",cn->GetServerName().c_str());
 329                         cn->CloseConnection();
 330                         cn->SetState(STATE_DISCONNECTED);
 331                         WriteOpers("*** Lost single connection to %s, link inactive and retrying: %s",cn->GetServerName().c_str(),cn->GetWriteError().c_str());
 332                         // retry the packet along a new route so either arrival OR failure are gauranteed (bugfix)
 333                         return this->SendPacket(message,sendhost);
 334                 }
 335                 if (!cn->FlushWriteBuf())
 336                 {
 337                         // if we're here the write() caused an error, we cannot proceed
 338                         log(DEBUG,"cn->FlushWriteBuf() failed for serverrec::SendPacket(): %s",cn->GetWriteError().c_str());
 339                         log(DEBUG,"Disabling connector: %s",cn->GetServerName().c_str());
 340                         cn->CloseConnection();
 341                         cn->SetState(STATE_DISCONNECTED);
 342                         WriteOpers("*** Lost single connection to %s, link inactive and retrying: %s",cn->GetServerName().c_str(),cn->GetWriteError().c_str());
 343                         // retry the packet along a new route so either arrival OR failure are gauranteed
 344                         return this->SendPacket(message,sendhost);
 345                 }
 346                 return true;
 347         }
 348 }
 349
 350 bool already_have_sum(std::string sum)
 351 {
 352         for (int i = 0; i < xsums.size(); i++)
 353         {
 354                 if (xsums[i] == sum)
 355                 {
 356                         return true;
 357                 }
 358         }
 359         if (xsums.size() >= 128)
 360         {
 361                 xsums.pop_front();
 362         }
 363         xsums.push_back(sum);
 364         return false;
 365 }
 366
 367 // receives a packet from any where there is data waiting, first come, first served
 368 // fills the message and host values with the host where the data came from.
 369
 370 bool serverrec::RecvPacket(std::deque<std::string> &messages, char* recvhost,std::deque<std::string> &sums)
 371 {
 372         char data[65536];
 373         memset(data, 0, 65536);
 374         for (int i = 0; i < this->connectors.size(); i++)
 375         {
 376                 if (this->connectors[i].GetState() != STATE_DISCONNECTED)
 377                 {
 378                         // returns false if the packet could not be sent (e.g. target host down)
 379                         int rcvsize = 0;
 380
 381                         // check if theres any data on this socket
 382                         // if not, continue onwards to the next.
 383                         pollfd polls;
 384                         polls.fd = this->connectors[i].GetDescriptor();
 385                         polls.events = POLLIN;
 386                         int ret = poll(&polls,1,1);
 387                         if (ret <= 0) continue;
 388
 389                         rcvsize = recv(this->connectors[i].GetDescriptor(),data,65000,0);
 390                         data[rcvsize] = '\0';
 391                         if (rcvsize == -1)
 392                         {
 393                                 if (errno != EAGAIN)
 394                                 {
 395                                         log(DEBUG,"recv() failed for serverrec::RecvPacket(): %s",strerror(errno));
 396                                         log(DEBUG,"Disabling connector: %s",this->connectors[i].GetServerName().c_str());
 397                                         this->connectors[i].CloseConnection();
 398                                         this->connectors[i].SetState(STATE_DISCONNECTED);
 399                                 }
 400                         }
 401                         int pushed = 0;
 402                         if (rcvsize > 0)
 403                         {
 404                                 if (!this->connectors[i].AddBuffer(data))
 405                                 {
 406                                         WriteOpers("*** Read buffer for %s exceeds maximum, closing connection!",this->connectors[i].GetServerName().c_str());
 407                                         this->connectors[i].CloseConnection();
 408                                         this->connectors[i].SetState(STATE_DISCONNECTED);
 409                                 }
 410                                 if (this->connectors[i].BufferIsComplete())
 411                                 {
 412                                         this->connectors[i].ResetPing();
 413                                         while (this->connectors[i].BufferIsComplete())
 414                                         {
 415                                                 std::string text = this->connectors[i].GetBuffer();
 416                                                 if (text != "")
 417                                                 {
 418                                                         if ((text[0] == ':') && (text.find(" ") != std::string::npos))
 419                                                         {
 420                                                                 std::string orig = text;
 421                                                                 log(DEBUG,"Original: %s",text.c_str());
 422                                                                 std::string sum = text.substr(1,text.find(" ")-1);
 423                                                                 text = text.substr(text.find(" ")+1,text.length());
 424                                                                 std::string possible_token = text.substr(1,text.find(" ")-1);
 425                                                                 if (possible_token.length() > 1)
 426                                                                 {
 427                                                                         sums.push_back("*");
 428                                                                         text = orig;
 429                                                                         log(DEBUG,"Non-mesh, non-tokenized string passed up the chain");
 430                                                                 }
 431                                                                 else
 432                                                                 {
 433                                                                         log(DEBUG,"Packet sum: '%s'",sum.c_str());
 434                                                                         if ((already_have_sum(sum)) && (sum != "*"))
 435                                                                         {
 436                                                                                 // we don't accept dupes
 437                                                                                 continue;
 438                                                                         }
 439                                                                         sums.push_back(sum.c_str());
 440                                                                 }
 441                                                         }
 442                                                         else sums.push_back("*");
 443                                                         messages.push_back(text.c_str());
 444                                                         strlcpy(recvhost,this->connectors[i].GetServerName().c_str(),160);
 445                                                         log(DEBUG,"serverrec::RecvPacket() %d:%s->%s",pushed++,recvhost,text.c_str());
 446                                                 }
 447                                         }
 448                                         return true;
 449                                 }
 450                         }
 451                 }
 452         }
 453         // nothing new yet -- message and host will be undefined
 454         return false;
 455 }
 456