aim_conn.c

   1
   2 /*
   3  *  aim_conn.c
   4  *
   5  * Does all this gloriously nifty connection handling stuff...
   6  *
   7  */
   8
   9 #include <faim/aim.h>
  10
  11 void aim_connrst(struct aim_session_t *sess)
  12 {
  13   int i;
  14   for (i = 0; i < AIM_CONN_MAX; i++)
  15     {
  16       sess->conns[i].fd = -1;
  17       sess->conns[i].type = -1;
  18       sess->conns[i].status = 0;
  19       sess->conns[i].seqnum = 0;
  20       sess->conns[i].lastactivity = 0;
  21       sess->conns[i].forcedlatency = 0;
  22       aim_clearhandlers(&(sess->conns[i]));
  23       sess->conns[i].handlerlist = NULL;
  24     }
  25
  26 }
  27
  28 struct aim_conn_t *aim_conn_getnext(struct aim_session_t *sess)
  29 {
  30   int i;
  31   for (i=0;i<AIM_CONN_MAX;i++)
  32     if (sess->conns[i].fd == -1)
  33       return &(sess->conns[i]);
  34   return NULL;
  35 }
  36
  37 void aim_conn_close(struct aim_conn_t *deadconn)
  38 {
  39   if (deadconn->fd >= 3)
  40     close(deadconn->fd);
  41   deadconn->fd = -1;
  42   deadconn->type = -1;
  43   deadconn->seqnum = 0;
  44   deadconn->lastactivity = 0;
  45   deadconn->forcedlatency = 0;
  46   aim_clearhandlers(deadconn);
  47   deadconn->handlerlist = NULL;
  48   if (deadconn->priv)
  49     free(deadconn->priv);
  50   deadconn->priv = NULL;
  51 }
  52
  53 struct aim_conn_t *aim_getconn_type(struct aim_session_t *sess,
  54                                     int type)
  55 {
  56   int i;
  57   for (i=0; i<AIM_CONN_MAX; i++)
  58     if (sess->conns[i].type == type)
  59       return &(sess->conns[i]);
  60   return NULL;
  61 }
  62
  63 /*
  64  * aim_newconn(type, dest)
  65  *
  66  * Opens a new connection to the specified dest host of type type.
  67  *
  68  * TODO: fix for proxies
  69  * FIXME: Return errors in a more sane way.
  70  *
  71  */
  72 struct aim_conn_t *aim_newconn(struct aim_session_t *sess,
  73                                int type, char *dest)
  74 {
  75   struct aim_conn_t *connstruct;
  76   int ret;
  77   struct sockaddr_in sa;
  78   struct hostent *hp;
  79   u_short port = FAIM_LOGIN_PORT;
  80   char *host = NULL;
  81   int i=0;
  82
  83   if (!dest || ((connstruct=aim_conn_getnext(sess))==NULL))
  84     return NULL;
  85
  86   connstruct->type = type;
  87
  88   /*
  89    * As of 23 Jul 1999, AOL now sends the port number, preceded by a
  90    * colon, in the BOS redirect.  This fatally breaks all previous
  91    * libfaims.  Bad, bad AOL.
  92    *
  93    * We put this here to catch every case.
  94    *
  95    */
  96
  97   for(i=0;i<strlen(dest);i++)
  98     {
  99       if (dest[i] == ':') {
 100         port = atoi(&(dest[i+1]));
 101         break;
 102       }
 103     }
 104   host = (char *)malloc(i+1);
 105   strncpy(host, dest, i);
 106   host[i] = '\0';
 107
 108   hp = gethostbyname2(host, AF_INET);
 109   free(host);
 110
 111   if (hp == NULL)
 112     {
 113       connstruct->status = (h_errno | AIM_CONN_STATUS_RESOLVERR);
 114       return connstruct;
 115     }
 116
 117   memset(&sa.sin_zero, 0, 8);
 118   sa.sin_port = htons(port);
 119   memcpy(&sa.sin_addr, hp->h_addr, hp->h_length);
 120   sa.sin_family = hp->h_addrtype;
 121
 122   connstruct->fd = socket(hp->h_addrtype, SOCK_STREAM, 0);
 123   ret = connect(connstruct->fd, (struct sockaddr *)&sa, sizeof(struct sockaddr_in));
 124   if( ret < 0)
 125     {
 126       connstruct->fd = -1;
 127       connstruct->status = (errno | AIM_CONN_STATUS_CONNERR);
 128       return connstruct;
 129     }
 130
 131   return connstruct;
 132 }
 133
 134 int aim_conngetmaxfd(struct aim_session_t *sess)
 135 {
 136   int i,j;
 137   j=0;
 138   for (i=0;i<AIM_CONN_MAX;i++)
 139     if(sess->conns[i].fd > j)
 140       j = sess->conns[i].fd;
 141   return j;
 142 }
 143
 144 int aim_countconn(struct aim_session_t *sess)
 145 {
 146   int i,cnt;
 147   cnt = 0;
 148   for (i=0;i<AIM_CONN_MAX;i++)
 149     if (sess->conns[i].fd > -1)
 150       cnt++;
 151   return cnt;
 152 }
 153
 154 /*
 155  * aim_select(timeout)
 156  *
 157  * Waits for a socket with data or for timeout, whichever comes first.
 158  * See select(2).
 159  *
 160  * Return codes in *status:
 161  *   -1  error in select() (NULL returned)
 162  *    0  no events pending (NULL returned)
 163  *    1  outgoing data pending (NULL returned)
 164  *    2  incoming data pending (connection with pending data returned)
 165  *
 166  */
 167 struct aim_conn_t *aim_select(struct aim_session_t *sess,
 168                               struct timeval *timeout, int *status)
 169 {
 170   fd_set fds;
 171   int i;
 172
 173   if (aim_countconn(sess) <= 0)
 174     return 0;
 175
 176   /*
 177    * If we have data waiting to be sent, return immediatly
 178    */
 179   if (sess->queue_outgoing != NULL) {
 180     *status = 1;
 181     return NULL;
 182   }
 183
 184   FD_ZERO(&fds);
 185
 186   for(i=0;i<AIM_CONN_MAX;i++)
 187     if (sess->conns[i].fd>-1)
 188       FD_SET(sess->conns[i].fd, &fds);
 189
 190   if ((i = select(aim_conngetmaxfd(sess)+1, &fds, NULL, NULL, timeout))>=1) {
 191     int j;
 192     for (j=0;j<AIM_CONN_MAX;j++) {
 193         if (sess->conns[j].fd > -1) {
 194             if ((FD_ISSET(sess->conns[j].fd, &fds))) {
 195               *status = 2;
 196               return &(sess->conns[j]);  /* return the first waiting struct */
 197             }
 198           }
 199         }
 200     /* should never get here */
 201   }
 202
 203   *status = i; /* may be 0 or -1 */
 204   return NULL;  /* no waiting or error, return */
 205 }
 206
 207 int aim_conn_isready(struct aim_conn_t *conn)
 208 {
 209   if (conn)
 210     return (conn->status & 0x0001);
 211   else
 212     return -1;
 213 }
 214
 215 int aim_conn_setstatus(struct aim_conn_t *conn, int status)
 216 {
 217   if (conn)
 218     return (conn->status ^= status);
 219   else
 220     return -1;
 221 }
 222
 223 int aim_conn_setlatency(struct aim_conn_t *conn, int newval)
 224 {
 225   if (!conn)
 226     return -1;
 227
 228   conn->forcedlatency = newval;
 229   conn->lastactivity = 0; /* reset this just to make sure */
 230
 231   return 0;
 232 }
 233
 234 void aim_session_init(struct aim_session_t *sess)
 235 {
 236   int i;
 237
 238   if (!sess)
 239     return;
 240
 241   memset(sess->logininfo.screen_name, 0x00, MAXSNLEN);
 242   sess->logininfo.BOSIP = NULL;
 243   memset(sess->logininfo.cookie, 0x00, AIM_COOKIELEN);
 244   sess->logininfo.email = NULL;
 245   sess->logininfo.regstatus = 0x00;
 246
 247   for (i = 0; i < AIM_CONN_MAX; i++)
 248     {
 249       sess->conns[i].fd = -1;
 250       sess->conns[i].type = -1;
 251       sess->conns[i].status = 0;
 252       sess->conns[i].seqnum = 0;
 253       sess->conns[i].lastactivity = 0;
 254       sess->conns[i].forcedlatency = 0;
 255       sess->conns[i].handlerlist = NULL;
 256       sess->conns[i].priv = NULL;
 257     }
 258
 259   sess->queue_outgoing = NULL;
 260   sess->queue_incoming = NULL;
 261   sess->pendingjoin = NULL;
 262   sess->outstanding_snacs = NULL;
 263   sess->snac_nextid = 0x00000001;
 264
 265   return;
 266 }