irssi-dcc-send-limit.patch

   1 diff -ur irssi-0.7.98.4/src/irc/dcc/dcc-rec.h irssi-0.7.98.4.limited/src/irc/dcc/dcc-rec.h
   2 --- irssi-0.7.98.4/src/irc/dcc/dcc-rec.h        Mon Feb 19 04:40:06 2001
   3 +++ irssi-0.7.98.4.limited/src/irc/dcc/dcc-rec.h        Sat Jan 19 19:40:16 2002
   4 @@ -20,6 +20,12 @@
   5  time_t starttime; /* transfer start time */
   6  unsigned long transfd; /* bytes transferred */
   7
   8 +int timeout_tag;
   9 +
  10 +unsigned long skip_bytes;
  11 +unsigned long limit_starttime;
  12 +unsigned long max_speed;
  13 +
  14  unsigned int destroyed:1; /* We're about to destroy this DCC recond */
  15
  16  GHashTable *module_data;
  17 diff -ur irssi-0.7.98.4/src/irc/dcc/dcc-send.c irssi-0.7.98.4.limited/src/irc/dcc/dcc-send.c
  18 --- irssi-0.7.98.4/src/irc/dcc/dcc-send.c       Wed Feb 28 20:26:21 2001
  19 +++ irssi-0.7.98.4.limited/src/irc/dcc/dcc-send.c       Sat Jan 19 19:43:15 2002
  20 @@ -48,14 +48,83 @@
  21  {
  22         if (!IS_DCC_SEND(dcc)) return;
  23
  24 +       if (dcc->timeout_tag != -1)
  25 +               g_source_remove(dcc->timeout_tag);
  26 +
  27         if (dcc->fhandle != -1) close(dcc->fhandle);
  28  }
  29
  30 +static int sent_too_much(SEND_DCC_REC *dcc)
  31 +{
  32 +       GTimeVal gtv;
  33 +       gulong timediff, curtime;
  34 +       gulong transfd, speed;
  35 +
  36 +       /* 0 == unlimited speed */
  37 +       if (dcc->max_speed == 0) return 0;
  38 +
  39 +       /* get time difference in milliseconds */
  40 +       g_get_current_time(&gtv);
  41 +       curtime = (gtv.tv_sec * 1000) + (gtv.tv_usec / 1000);
  42 +
  43 +       transfd = (dcc->transfd - dcc->skip_bytes);
  44 +       timediff = curtime - dcc->limit_starttime + 1;
  45 +       speed = ((transfd * 1000) / timediff);
  46 +
  47 +       /* reset speed counter every 30 seconds */
  48 +       if (timediff >= 30000) {
  49 +               dcc->limit_starttime = curtime;
  50 +               dcc->skip_bytes = dcc->transfd;
  51 +       }
  52 +
  53 +       return (speed > (dcc->max_speed * 1024));
  54 +}
  55 +
  56 +static void dcc_send_data(SEND_DCC_REC *dcc);
  57 +
  58 +static void reset_dcc_send(SEND_DCC_REC *dcc)
  59 +{
  60 +       if (g_slist_find(dcc_conns, dcc) == NULL) {
  61 +               /* the DCC was closed during the wait */
  62 +               return;
  63 +       }
  64 +
  65 +       g_source_remove(dcc->timeout_tag);
  66 +       dcc->timeout_tag = -1;
  67 +
  68 +       dcc->tagwrite = g_input_add(dcc->handle, G_INPUT_WRITE,
  69 +                                   (GInputFunction) dcc_send_data, dcc);
  70 +
  71 +       return;
  72 +}
  73 +
  74  /* input function: DCC SEND - we're ready to send more data */
  75  static void dcc_send_data(SEND_DCC_REC *dcc)
  76  {
  77          char buffer[512];
  78 -       int ret;
  79 +       int ret, max_speed;
  80 +       GTimeVal gtv;
  81 +
  82 +       max_speed = settings_get_int("dcc_send_top_speed");
  83 +       if (max_speed != dcc->max_speed) {
  84 +               /* speed setting has changed, calculate speed from current position
  85 +                  instead of from the start to eliminate speed boosts/slowdowns */
  86 +
  87 +               dcc->max_speed = max_speed;
  88 +               dcc->skip_bytes = dcc->transfd;
  89 +
  90 +               g_get_current_time(&gtv);
  91 +               dcc->limit_starttime = (gtv.tv_sec * 1000) + (gtv.tv_usec / 1000);
  92 +       }
  93 +
  94 +       if (sent_too_much(dcc)) {
  95 +               /* disable calling this function for 1/10th of a second. */
  96 +               g_source_remove(dcc->tagwrite);
  97 +               dcc->tagwrite = -1;
  98 +               dcc->timeout_tag = g_timeout_add(100, (GSourceFunc)
  99 +                       reset_dcc_send, dcc);
 100 +               return;
 101 +       }
 102
 103         ret = read(dcc->fhandle, buffer, sizeof(buffer));
 104         if (ret <= 0) {
 105 @@ -116,6 +185,7 @@
 106          GIOChannel *handle;
 107         IPADDR addr;
 108         int port;
 109 +       GTimeVal gtv;
 110
 111         /* accept connection */
 112         handle = net_accept(dcc->handle, &addr, &port);
 113 @@ -129,6 +199,14 @@
 114         g_source_remove(dcc->tagconn);
 115         net_disconnect(dcc->handle);
 116
 117 +       dcc->timeout_tag = -1;
 118 +       dcc->skip_bytes = 0;
 119 +       dcc->max_speed = settings_get_int("dcc_send_top_speed");
 120 +
 121 +       /* get starttime in milliseconds */
 122 +       g_get_current_time(&gtv);
 123 +       dcc->limit_starttime = (gtv.tv_sec * 1000) + (gtv.tv_usec / 1000);
 124 +
 125         dcc->starttime = time(NULL);
 126         dcc->handle = handle;
 127         memcpy(&dcc->addr, &addr, sizeof(IPADDR));
 128 @@ -246,6 +324,7 @@
 129  {
 130          dcc_register_type("SEND");
 131         settings_add_str("dcc", "dcc_upload_path", "~");
 132 +       settings_add_int("dcc", "dcc_send_top_speed", 0);
 133
 134         signal_add("dcc destroyed", (SIGNAL_FUNC) sig_dcc_destroyed);
 135         command_bind("dcc send", NULL, (SIGNAL_FUNC) cmd_dcc_send);