1 #include "tcpto.h"

     2 #include "open.h"

     3 #include "lock.h"

     4 #include "seek.h"

     5 #include "now.h"

     6 #include "ip.h"

     7 #include "byte.h"

     8 #include "datetime.h"

     9 #include "readwrite.h"

    10 

    11 char tcpto_buf[1024];

    12 

    13 static int flagwasthere;

    14 static int fdlock;

    15 

    16 static int getbuf()

    17 {

    18  int r;

    19  int fd;

    20 

    21  fdlock = open_write("queue/lock/tcpto");

    22  if (fdlock == -1) return 0;

    23  fd = open_read("queue/lock/tcpto");

    24  if (fd == -1) { close(fdlock); return 0; }

    25  if (lock_ex(fdlock) == -1) { close(fdlock); close(fd); return 0; }

    26  r = read(fd,tcpto_buf,sizeof(tcpto_buf));

    27  close(fd);

    28  if (r < 0) { close(fdlock); return 0; }

    29  r >>= 4;

    30  if (!r) close(fdlock);

    31  return r;

    32 }

    33 

    34 int tcpto(ip) struct ip_address *ip;

    35 {

    36  int n;

    37  int i;

    38  char *record;

    39  datetime_sec when;

    40 

    41  flagwasthere = 0;

    42 

    43  n = getbuf();

    44  if (!n) return 0;

    45  close(fdlock);

    46 

    47  record = tcpto_buf;

    48  for (i = 0;i < n;++i)

    49   {

    50    if (byte_equal(ip->d,4,record))

    51     {

    52      flagwasthere = 1;

    53      if (record[4] >= 2)

    54       {

    55        when = (unsigned long) (unsigned char) record[11];

    56        when = (when << 8) + (unsigned long) (unsigned char) record[10];

    57        when = (when << 8) + (unsigned long) (unsigned char) record[9];

    58        when = (when << 8) + (unsigned long) (unsigned char) record[8];

    59 

    60        if (now() - when < ((60 + (getpid() & 31)) << 6))

    61 	 return 1;

    62       }

    63      return 0;

    64     }

    65    record += 16;

    66   }

    67  return 0;

    68 }

    69 

    70 void tcpto_err(ip,flagerr) struct ip_address *ip; int flagerr;

    71 {

    72  int n;

    73  int i;

    74  char *record;

    75  datetime_sec when;

    76  datetime_sec firstwhen;

    77  int firstpos;

    78  datetime_sec lastwhen;

    79 

    80  if (!flagerr)

    81    if (!flagwasthere)

    82      return; /* could have been added, but not worth the effort to check */

    83 

    84  n = getbuf();

    85  if (!n) return;

    86 

    87  record = tcpto_buf;

    88  for (i = 0;i < n;++i)

    89   {

    90    if (byte_equal(ip->d,4,record))

    91     {

    92      if (!flagerr)

    93        record[4] = 0;

    94      else

    95       {

    96        lastwhen = (unsigned long) (unsigned char) record[11];

    97        lastwhen = (lastwhen << 8) + (unsigned long) (unsigned char) record[10];

    98        lastwhen = (lastwhen << 8) + (unsigned long) (unsigned char) record[9];

    99        lastwhen = (lastwhen << 8) + (unsigned long) (unsigned char) record[8];

   100        when = now();

   101 

   102        if (record[4] && (when < 120 + lastwhen)) { close(fdlock); return; }

   103 

   104        if (++record[4] > 10) record[4] = 10;

   105        record[8] = when; when >>= 8;

   106        record[9] = when; when >>= 8;

   107        record[10] = when; when >>= 8;

   108        record[11] = when;

   109       }

   110      if (seek_set(fdlock,i << 4) == 0)

   111        if (write(fdlock,record,16) < 16)

   112          ; /*XXX*/

   113      close(fdlock);

   114      return;

   115     }

   116    record += 16;

   117   }

   118 

   119  if (!flagerr) { close(fdlock); return; }

   120 

   121  record = tcpto_buf;

   122  for (i = 0;i < n;++i)

   123   {

   124    if (!record[4]) break;

   125    record += 16;

   126   }

   127 

   128  if (i >= n)

   129   {

   130    firstpos = -1;

   131    record = tcpto_buf;

   132    for (i = 0;i < n;++i)

   133     {

   134      when = (unsigned long) (unsigned char) record[11];

   135      when = (when << 8) + (unsigned long) (unsigned char) record[10];

   136      when = (when << 8) + (unsigned long) (unsigned char) record[9];

   137      when = (when << 8) + (unsigned long) (unsigned char) record[8];

   138      when += (record[4] << 10);

   139      if ((firstpos < 0) || (when < firstwhen))

   140       {

   141        firstpos = i;

   142        firstwhen = when;

   143       }

   144      record += 16;

   145     }

   146    i = firstpos;

   147   }

   148 

   149  if (i >= 0)

   150   {

   151    record = tcpto_buf + (i << 4);

   152    byte_copy(record,4,ip->d);

   153    when = now();

   154    record[8] = when; when >>= 8;

   155    record[9] = when; when >>= 8;

   156    record[10] = when; when >>= 8;

   157    record[11] = when;

   158    record[4] = 1;

   159    if (seek_set(fdlock,i << 4) == 0)

   160      if (write(fdlock,record,16) < 16)

   161        ; /*XXX*/

   162   }

   163 

   164  close(fdlock);

   165 }