typedef nspace::hash_map<in_addr,string*, nspace::hash<in_addr>, InAddr_HashComp> address_cache;
typedef std::deque<command_t> command_table;
+// This table references users by file descriptor.
+// its an array to make it VERY fast, as all lookups are referenced
+// by an integer, meaning there is no need for a scan/search operation.
+userrec* fd_ref_table[65536];
+
+int statsAccept = 0, statsRefused = 0, statsUnknown = 0, statsCollisions = 0, statsDns = 0, statsDnsGood = 0, statsDnsBad = 0, statsConnects = 0, statsSent= 0, statsRecv = 0;
+
serverrec* me[32];
FILE *log_file;
std::vector<userrec*> all_opers;
+static char already_sent[65536];
+
void AddOper(userrec* user)
{
log(DEBUG,"Oper added to optimization list");
chop(tb);
if (sock != -1)
{
- write(sock,tb,bytes > 514 ? 514 : bytes);
+ int MOD_RESULT = 0;
+ FOREACH_RESULT(OnRawSocketWrite(sock,tb,bytes > 512 ? 512 : bytes));
+ if (!MOD_RESULT)
+ write(sock,tb,bytes > 512 ? 512 : bytes);
+ if (fd_ref_table[sock])
+ {
+ fd_ref_table[sock]->bytes_out += (bytes > 512 ? 512 : bytes);
+ fd_ref_table[sock]->cmds_out++;
+ }
+ statsSent += (bytes > 512 ? 512 : bytes);
}
}
chop(tb);
if (sock != -1)
{
- write(sock,tb,bytes > 514 ? 514 : bytes);
+ int MOD_RESULT = 0;
+ FOREACH_RESULT(OnRawSocketWrite(sock,tb,bytes > 512 ? 512 : bytes));
+ if (!MOD_RESULT)
+ write(sock,tb,bytes > 512 ? 512 : bytes);
+ if (fd_ref_table[sock])
+ {
+ fd_ref_table[sock]->bytes_out += (bytes > 512 ? 512 : bytes);
+ fd_ref_table[sock]->cmds_out++;
+ }
+ statsSent += (bytes > 512 ? 512 : bytes);
}
}
chop(tb);
if (sock != -1)
{
- write(sock,tb,bytes > 514 ? 514 : bytes);
+ int MOD_RESULT = 0;
+ FOREACH_RESULT(OnRawSocketWrite(sock,tb,bytes > 512 ? 512 : bytes));
+ if (!MOD_RESULT)
+ write(sock,tb,bytes > 512 ? 512 : bytes);
+ if (fd_ref_table[sock])
+ {
+ fd_ref_table[sock]->bytes_out += (bytes > 512 ? 512 : bytes);
+ fd_ref_table[sock]->cmds_out++;
+ }
+ statsSent += (bytes > 512 ? 512 : bytes);
}
}
va_end(argsPtr);
// FIX: Stops a message going to the same person more than once
- std::vector<int> already_sent;
+ bzero(&already_sent,65536);
+
+ bool sent_to_at_least_one = false;
for (int i = 0; i < MAXCHANS; i++)
{
{
char* o = (*ulist)[j];
userrec* otheruser = (userrec*)o;
- bool do_send = true;
- for (int t = 0; t < already_sent.size(); t++)
- {
- if (already_sent[t] == otheruser->fd)
- {
- do_send = false;
- break;
- }
- }
- if (do_send)
+ if (!already_sent[otheruser->fd])
{
- already_sent.push_back(otheruser->fd);
+ already_sent[otheruser->fd] = 1;
WriteFrom(otheruser->fd,u,"%s",textbuffer);
+ sent_to_at_least_one = true;
}
}
}
}
// if the user was not in any channels, no users will receive the text. Make sure the user
// receives their OWN message for WriteCommon
- if (!already_sent.size())
+ if (!sent_to_at_least_one)
{
WriteFrom(u->fd,u,"%s",textbuffer);
}
vsnprintf(textbuffer, MAXBUF, text, argsPtr);
va_end(argsPtr);
- std::vector<int> already_sent;
+ bzero(&already_sent,65536);
for (int i = 0; i < MAXCHANS; i++)
{
userrec* otheruser = (userrec*)o;
if (u != otheruser)
{
- bool do_send = true;
- for (int t = 0; t < already_sent.size(); t++)
- {
- if (already_sent[t] == otheruser->fd)
- {
- do_send = false;
- break;
- }
- }
- if (do_send)
+ if (!already_sent[otheruser->fd])
{
- already_sent.push_back(otheruser->fd);
+ already_sent[otheruser->fd] = 1;
WriteFrom(otheruser->fd,u,"%s",textbuffer);
}
}
if (iter != chanlist.end())
{
log(DEBUG,"del_channel: destroyed: %s",Ptr->name);
- if (iter->second) delete iter->second;
+ delete Ptr;
chanlist.erase(iter);
}
}
if (iter != chanlist.end())
{
log(DEBUG,"del_channel: destroyed: %s",Ptr->name);
- if (iter->second) delete iter->second;
+ delete Ptr;
chanlist.erase(iter);
}
}
Write(user->fd,"ERROR :Closing link (%s@%s) [%s]",user->ident,user->host,reason);
log(DEBUG,"closing fd %lu",(unsigned long)user->fd);
- /* bugfix, cant close() a nonblocking socket (sux!) */
if (user->registered == 7) {
FOREACH_MOD OnUserQuit(user);
WriteCommonExcept(user,"QUIT :%s",reason);
if (user->fd > -1)
{
+ FOREACH_MOD OnRawSocketClose(user->fd);
shutdown(user->fd,2);
close(user->fd);
}
if (iter != clientlist.end())
{
- log(DEBUG,"deleting user hash value %lu",(unsigned long)iter->second);
- if ((iter->second) && (user->registered == 7)) {
- if (iter->second) delete iter->second;
- }
+ log(DEBUG,"deleting user hash value %lu",(unsigned long)user);
+ if (user->fd > -1)
+ fd_ref_table[user->fd] = NULL;
+ delete user;
clientlist.erase(iter);
}
}
Write(user->fd,"ERROR :Closing link (%s@%s) [%s]",user->ident,user->host,reason);
log(DEBUG,"closing fd %lu",(unsigned long)user->fd);
- /* bugfix, cant close() a nonblocking socket (sux!) */
if (user->registered == 7) {
FOREACH_MOD OnUserQuit(user);
WriteCommonExcept(user,"QUIT :%s",reason);
if (user->fd > -1)
{
+ FOREACH_MOD OnRawSocketClose(user->fd);
shutdown(user->fd,2);
close(user->fd);
}
if (iter != clientlist.end())
{
- log(DEBUG,"deleting user hash value %lu",(unsigned long)iter->second);
- if ((iter->second) && (user->registered == 7)) {
- if (iter->second) delete iter->second;
- }
+ log(DEBUG,"deleting user hash value %lu",(unsigned long)user);
+ if (user->fd > -1)
+ fd_ref_table[user->fd] = NULL;
+ delete user;
clientlist.erase(iter);
}
}
iter = clientlist.find(tempnick);
- if (iter != clientlist.end()) return;
+ // fix by brain.
+ // as these nicknames are 'RFC impossible', we can be sure nobody is going to be
+ // using one as a registered connection. As theyre per fd, we can also safely assume
+ // that we wont have collisions. Therefore, if the nick exists in the list, its only
+ // used by a dead socket, erase the iterator so that the new client may reclaim it.
+ // this was probably the cause of 'server ignores me when i hammer it with reconnects'
+ // issue in earlier alphas/betas
+ if (iter != clientlist.end())
+ {
+ clientlist.erase(iter);
+ }
/*
* It is OK to access the value here this way since we know
}
if (clientlist.size() == MAXCLIENTS)
+ {
kill_link(clientlist[tempnick],"No more connections allowed in this class");
+ return;
+ }
+
+ // this is done as a safety check to keep the file descriptors within range of fd_ref_table.
+ // its a pretty big but for the moment valid assumption:
+ // file descriptors are handed out starting at 0, and are recycled as theyre freed.
+ // therefore if there is ever an fd over 65535, 65536 clients must be connected to the
+ // irc server at once (or the irc server otherwise initiating this many connections, files etc)
+ // which for the time being is a physical impossibility (even the largest networks dont have more
+ // than about 10,000 users on ONE server!)
+ if (socket > 65535)
+ {
+ kill_link(clientlist[tempnick],"Server is full");
+ return;
+ }
char* e = matches_exception(ip);
char reason[MAXBUF];
snprintf(reason,MAXBUF,"Z-Lined: %s",r);
kill_link(clientlist[tempnick],reason);
+ return;
}
}
+ fd_ref_table[socket] = clientlist[tempnick];
}
// this function counts all users connected, wether they are registered or NOT.
WholeMOTD = WholeMOTD + std::string(":") + std::string(ServerName) + std::string(" 376 ") + std::string(user->nick) + std::string(" :End of message of the day.\r\n");
// only one write operation
send(user->fd,WholeMOTD.c_str(),WholeMOTD.length(),0);
-
+ statsSent += WholeMOTD.length();
}
void ShowRULES(userrec *user)
/* shows the message of the day, and any other on-logon stuff */
void FullConnectUser(userrec* user)
{
+ statsConnects++;
user->registered = 7;
user->idle_lastmsg = TIME;
log(DEBUG,"ConnectUser: %s",user->nick);
FOREACH_RESULT(OnUserPreNick(user,newnick));
if (MOD_RESULT) {
+ statsCollisions++;
kill_link(user,"Nickname collision");
return;
}
if (matches_qline(newnick))
{
+ statsCollisions++;
kill_link(user,"Nickname collision");
return;
}
{
if (strchr("@!\"$%^&*(){}[]_=+;:'#~,<>/?\\|`",command[x]))
{
+ statsUnknown++;
WriteServ(user->fd,"421 %s %s :Unknown command",user->nick,command);
return;
}
/* ikky /stats counters */
if (temp)
{
- if (user)
- {
- user->bytes_in += strlen(temp);
- user->cmds_in++;
- user->bytes_out+=strlen(temp);
- user->cmds_out++;
- }
cmdlist[i].use_count++;
cmdlist[i].total_bytes+=strlen(temp);
}
}
if ((!cmd_found) && (user))
{
+ statsUnknown++;
WriteServ(user->fd,"421 %s %s :Unknown command",user->nick,command);
}
}
if (count2 == clientlist.end()) break;
+ userrec* curr = NULL;
+
if (count2->second)
- if (count2->second->fd != 0)
+ curr = count2->second;
+
+ if ((curr) && (curr->fd != 0))
{
+#ifdef _POSIX_PRIORITY_SCHEDULING
+ sched_yield();
+#endif
// assemble up to 64 sockets into an fd_set
// to implement a pooling mechanism.
//
{
if (count2 != clientlist.end())
{
+ curr = count2->second;
// we don't check the state of remote users.
- if ((count2->second->fd != -1) && (count2->second->fd != FD_MAGIC_NUMBER))
+ if ((curr->fd != -1) && (curr->fd != FD_MAGIC_NUMBER))
{
- FD_SET (count2->second->fd, &sfd);
+ FD_SET (curr->fd, &sfd);
// registration timeout -- didnt send USER/NICK/HOST in the time specified in
// their connection class.
- if ((TIME > count2->second->timeout) && (count2->second->registered != 7))
+ if ((TIME > curr->timeout) && (curr->registered != 7))
{
- log(DEBUG,"InspIRCd: registration timeout: %s",count2->second->nick);
- kill_link(count2->second,"Registration timeout");
+ log(DEBUG,"InspIRCd: registration timeout: %s",curr->nick);
+ kill_link(curr,"Registration timeout");
goto label;
}
- if ((TIME > count2->second->signon) && (count2->second->registered == 3) && (AllModulesReportReady(count2->second)))
+ if ((TIME > curr->signon) && (curr->registered == 3) && (AllModulesReportReady(curr)))
{
log(DEBUG,"signon exceed, registered=3, and modules ready, OK");
- count2->second->dns_done = true;
- FullConnectUser(count2->second);
+ curr->dns_done = true;
+ statsDnsBad++;
+ FullConnectUser(curr);
goto label;
}
- if ((count2->second->dns_done) && (count2->second->registered == 3) && (AllModulesReportReady(count2->second))) // both NICK and USER... and DNS
+ if ((curr->dns_done) && (curr->registered == 3) && (AllModulesReportReady(curr))) // both NICK and USER... and DNS
{
log(DEBUG,"dns done, registered=3, and modules ready, OK");
- FullConnectUser(count2->second);
+ FullConnectUser(curr);
goto label;
}
- if ((TIME > count2->second->nping) && (isnick(count2->second->nick)) && (count2->second->registered == 7))
+ if ((TIME > curr->nping) && (isnick(curr->nick)) && (curr->registered == 7))
{
- if ((!count2->second->lastping) && (count2->second->registered == 7))
+ if ((!curr->lastping) && (curr->registered == 7))
{
- log(DEBUG,"InspIRCd: ping timeout: %s",count2->second->nick);
- kill_link(count2->second,"Ping timeout");
+ log(DEBUG,"InspIRCd: ping timeout: %s",curr->nick);
+ kill_link(curr,"Ping timeout");
goto label;
}
- Write(count2->second->fd,"PING :%s",ServerName);
- log(DEBUG,"InspIRCd: pinging: %s",count2->second->nick);
- count2->second->lastping = 0;
- count2->second->nping = TIME+count2->second->pingmax; // was hard coded to 120
+ Write(curr->fd,"PING :%s",ServerName);
+ log(DEBUG,"InspIRCd: pinging: %s",curr->nick);
+ curr->lastping = 0;
+ curr->nping = TIME+curr->pingmax; // was hard coded to 120
}
}
count2++;
selectResult2 = select(65535, &sfd, NULL, NULL, &tval);
// now loop through all of the items in this pool if any are waiting
- //if (selectResult2 > 0)
+ if (selectResult2 > 0)
for (user_hash::iterator count2a = xcount; count2a != endingiter; count2a++)
{
#ifdef _POSIX_PRIORITY_SCHEDULING
sched_yield();
#endif
-
+ userrec* cu = count2a->second;
result = EAGAIN;
- if ((count2a->second->fd != FD_MAGIC_NUMBER) && (count2a->second->fd != -1) && (FD_ISSET (count2a->second->fd, &sfd)))
+ if ((cu->fd != FD_MAGIC_NUMBER) && (cu->fd != -1) && (FD_ISSET (cu->fd, &sfd)))
{
- result = read(count2a->second->fd, data, 65535);
+ log(DEBUG,"Data waiting on socket %d",cu->fd);
+ int MOD_RESULT = 0;
+ int result2 = 0;
+ FOREACH_RESULT(OnRawSocketRead(cu->fd,data,65535,result2));
+ if (!MOD_RESULT)
+ {
+ result = read(cu->fd, data, 65535);
+ }
+ else result = result2;
+ log(DEBUG,"Read result: %d",result);
if (result)
{
+ statsRecv += result;
// perform a check on the raw buffer as an array (not a string!) to remove
// characters 0 and 7 which are illegal in the RFC - replace them with spaces.
// hopefully this should stop even more people whining about "Unknown command: *"
}
if (result > 0)
data[result] = '\0';
- userrec* current = count2a->second;
+ userrec* current = cu;
int currfd = current->fd;
int floodlines = 0;
// add the data to the users buffer
char sanitized[MAXBUF];
// use GetBuffer to copy single lines into the sanitized string
std::string single_line = current->GetBuffer();
+ current->bytes_in += single_line.length();
+ current->cmds_in++;
if (single_line.length()>512)
{
log(DEFAULT,"Excess flood from: %s!%s@%s",current->nick,current->ident,current->host);
{
// we're gonna re-scan to check if the nick is gone, after every
// command - if it has, we're gonna bail
- bool find_again = false;
process_buffer(sanitized,current);
- // look for the user's record in case it's changed
- for (user_hash::iterator c2 = clientlist.begin(); c2 != clientlist.end(); c2++)
- {
- if (c2->second->fd == currfd)
- {
- // found again, update pointer
- current == c2->second;
- find_again = true;
- break;
- }
- }
- if (!find_again)
+ // look for the user's record in case it's changed... if theyve quit,
+ // we cant do anything more with their buffer, so bail.
+ // there used to be an ugly, slow loop here. Now we have a reference
+ // table, life is much easier (and FASTER)
+ if (!fd_ref_table[currfd])
goto label;
}
if ((result == -1) && (errno != EAGAIN) && (errno != EINTR))
{
- log(DEBUG,"killing: %s",count2a->second->nick);
- kill_link(count2a->second,strerror(errno));
+ log(DEBUG,"killing: %s",cu->nick);
+ kill_link(cu,strerror(errno));
goto label;
}
}
{
if (count2->second)
{
- log(DEBUG,"InspIRCd: Exited: %s",count2a->second->nick);
- kill_link(count2a->second,"Client exited");
+ log(DEBUG,"InspIRCd: Exited: %s",cu->nick);
+ kill_link(cu,"Client exited");
// must bail here? kill_link removes the hash, corrupting the iterator
log(DEBUG,"Bailing from client exit");
goto label;
{
WriteOpers("*** WARNING: Accept failed on port %lu (%s)",(unsigned long)ports[count],target);
log(DEBUG,"InspIRCd: accept failed: %lu",(unsigned long)ports[count]);
+ statsRefused++;
}
else
{
+ FOREACH_MOD OnRawSocketAccept(incomingSockfd, resolved, ports[count]);
+ statsAccept++;
AddClient(incomingSockfd, resolved, ports[count], false, inet_ntoa (client.sin_addr));
log(DEBUG,"InspIRCd: adding client on port %lu fd=%lu",(unsigned long)ports[count],(unsigned long)incomingSockfd);
}
if (0) {};
#ifdef _POSIX_PRIORITY_SCHEDULING
sched_yield();
+ sched_yield();
#endif
}
/* not reached */