lib/eggdrop/throttle.c File Reference

#include <eggdrop/eggdrop.h>
#include <unistd.h>

Go to the source code of this file.

Data Structures
struct	throttle
Defines
#define	THROTTLE_LEVEL   SOCKBUF_LEVEL_THROTTLE
Typedefs
typedef struct throttle	throttle_t
Functions
static void	throttle_remove (throttle_t *t)
static int	throttle_on_read (void *client_data, int idx, char *data, int len)
static int	throttle_on_write (void *client_data, int idx, const char *data, int len)
static int	throttle_on_written (void *client_data, int idx, int len, int remaining)
static int	throttle_on_delete (void *client_data, int idx)
static throttle_t *	throttle_lookup (int idx)
static void	throttle_add (throttle_t *t)
static int	throttle_secondly (void *ignore)
int	throttle_on (int idx)
int	throttle_off (int idx)
int	throttle_set (int idx, int dir, int speed)
Variables
static const char	rcsid [] = "$Id: throttle.c,v 1.5 2004-10-17 05:14:06 stdarg Exp $"
static throttle_t *	throttle_head = NULL
static throttle_t *	throttle_next = NULL
static int	throttle_timer = -1
static sockbuf_filter_t	throttle_filter

---

Define Documentation

#define THROTTLE_LEVEL   SOCKBUF_LEVEL_THROTTLE

Definition at line 27 of file throttle.c.

Referenced by throttle_off(), throttle_on_read(), throttle_on_write(), throttle_on_written(), throttle_secondly(), and throttle_set().

---

Typedef Documentation

typedef struct throttle throttle_t

---

Function Documentation

static void throttle_add

(

throttle_t *

t

)

[static]

Definition at line 99 of file throttle.c.

References throttle::next, NULL, and throttle::prev.

Referenced by throttle_on().

{
  t->next = throttle_head;
  t->prev = NULL;
  if (throttle_head) throttle_head->prev = t;
  throttle_head = t;
}

static throttle_t* throttle_lookup

(

int

idx

)

[static]

Definition at line 89 of file throttle.c.

References throttle::idx, and throttle::next.

Referenced by throttle_off(), and throttle_set().

{
  throttle_t *t;

  for (t = throttle_head; t; t = t->next) {
    if (t->idx == idx) break;
  }
  return(t);
}

int throttle_off

(

int

idx

)

Definition at line 171 of file throttle.c.

References throttle::buf, throttle::len, NULL, sockbuf_detach_filter(), sockbuf_isvalid(), sockbuf_on_write(), THROTTLE_LEVEL, throttle_lookup(), throttle_remove(), throttle_timer, and timer_destroy().

Referenced by throttle_on_delete().

{
  throttle_t *t;

  t = throttle_lookup(idx);
  if (!t) return(-1);

  throttle_remove(t);
  if (sockbuf_isvalid(idx)) {
    sockbuf_detach_filter(idx, &throttle_filter, NULL);
    if (t->len) sockbuf_on_write(idx, THROTTLE_LEVEL, t->buf, t->len);
  }

  if (t->buf) free(t->buf);
  free(t);

  if (!throttle_head) {
    timer_destroy(throttle_timer);
    throttle_timer = -1;
  }
  return(0);
}

int throttle_on

(

int

idx

)

Definition at line 153 of file throttle.c.

References throttle::idx, egg_timeval::sec, sockbuf_attach_filter(), throttle_add(), throttle_secondly(), throttle_timer, timer_create_repeater, and egg_timeval::usec.

Referenced by script_net_throttle().

{
  throttle_t *t;
  egg_timeval_t howlong;

  t = calloc(1, sizeof(*t));
  t->idx = idx;
  throttle_add(t);
  sockbuf_attach_filter(idx, &throttle_filter, t);

  if (throttle_timer < 0) {
    howlong.sec = 1;
    howlong.usec = 0;
    throttle_timer = timer_create_repeater(&howlong, "bandwidth throttler", throttle_secondly);
  }
  return(0);
}

static int throttle_on_delete	(	void *	client_data,
		int	idx
	)			[static]

Definition at line 75 of file throttle.c.

References throttle_off().

{
  throttle_off(idx);
  return(0);
}

static int throttle_on_read	(	void *	client_data,
		int	idx,
		char *	data,
		int	len
	)			[static]

Definition at line 44 of file throttle.c.

References throttle::bytes_in, sockbuf_noread(), sockbuf_on_read(), throttle::speed_in, and THROTTLE_LEVEL.

{
  throttle_t *t = client_data;

  t->bytes_in += len;
  if (t->speed_in > 0 && t->bytes_in > t->speed_in) {
    sockbuf_noread(idx);
  }
  return sockbuf_on_read(idx, THROTTLE_LEVEL, data, len);
}

static int throttle_on_write	(	void *	client_data,
		int	idx,
		const char *	data,
		int	len
	)			[static]

Definition at line 55 of file throttle.c.

References throttle::buf, throttle::len, sockbuf_on_write(), throttle::speed_out, and THROTTLE_LEVEL.

{
  throttle_t *t = client_data;

  if (t->speed_out <= 0) return sockbuf_on_write(idx, THROTTLE_LEVEL, data, len);

  t->buf = realloc(t->buf, t->len + len);
  memcpy(t->buf + t->len, data, len);
  t->len += len;

  return(0);
}

static int throttle_on_written	(	void *	client_data,
		int	idx,
		int	len,
		int	remaining
	)			[static]

Definition at line 68 of file throttle.c.

References throttle::len, sockbuf_on_written(), and THROTTLE_LEVEL.

{
  throttle_t *t = client_data;

  return sockbuf_on_written(idx, THROTTLE_LEVEL, len, remaining + t->len);
}

static void throttle_remove

(

throttle_t *

t

)

[static]

Definition at line 107 of file throttle.c.

References throttle::next, and throttle::prev.

Referenced by throttle_off().

{
  /* If we get removed during throttle_secondly() we have to update
   * the next pointer to a valid one. */
  if (throttle_next == t) {
    throttle_next = t->next;
  }

  if (t->prev) t->prev->next = t->next;
  else throttle_head = t->next;

  if (t->next) t->next->prev = t->prev;
}

static int throttle_secondly

(

void *

ignore

)

[static]

Definition at line 122 of file throttle.c.

References throttle::buf, throttle::bytes_in, throttle::idx, throttle::len, throttle::next, sockbuf_on_write(), sockbuf_on_written(), sockbuf_read(), throttle::speed_in, throttle::speed_out, and THROTTLE_LEVEL.

Referenced by throttle_on().

{
  throttle_t *t;
  int avail, r, out;

  for (t = throttle_head; t; t = throttle_next) {
    throttle_next = t->next;
    t->bytes_in -= t->speed_in;
    if (t->bytes_in < 0) t->bytes_in = 0;
    else if (t->bytes_in < t->speed_in) sockbuf_read(t->idx);

    out = 0;
    while (t->len && out < t->speed_out) {
      /* How many bytes can we send? */
      avail = t->speed_out - out;
      if (avail > t->len) avail = t->len;

      r = sockbuf_on_write(t->idx, THROTTLE_LEVEL, t->buf, avail);
      if (r < 0) break;

      memmove(t->buf, t->buf+avail, t->len-avail);
      t->len -= avail;
      out += avail;

      if (r > 0) sockbuf_on_written(t->idx, THROTTLE_LEVEL, r, t->len + avail - r);

    }
  }
  return(0);
}

int throttle_set	(	int	idx,
		int	dir,
		int	speed
	)

Definition at line 194 of file throttle.c.

References throttle::buf, throttle::len, sockbuf_on_write(), sockbuf_on_written(), throttle::speed_in, throttle::speed_out, THROTTLE_LEVEL, and throttle_lookup().

Referenced by script_net_throttle(), and script_net_throttle_set().

{
  throttle_t *t;
  int r;

  t = throttle_lookup(idx);
  if (!t) return(-1);
  if (dir) {
    t->speed_out = speed;
    if (speed <= 0 && t->len) {
      r = sockbuf_on_write(idx, THROTTLE_LEVEL, t->buf, t->len);
      if (r > 0) sockbuf_on_written(idx, THROTTLE_LEVEL, r, t->len - r);
      t->len = 0;
    }
  }
  else t->speed_in = speed;
  return(0);
}

---

Variable Documentation

const char rcsid[] = "$Id: throttle.c,v 1.5 2004-10-17 05:14:06 stdarg Exp $" [static]

Definition at line 21 of file throttle.c.

sockbuf_filter_t throttle_filter [static]

Initial value:

 {
  "throttle",
  THROTTLE_LEVEL,
  NULL, NULL, NULL,
  throttle_on_read, throttle_on_write, throttle_on_written,
  NULL, throttle_on_delete
}

Definition at line 81 of file throttle.c.

throttle_t* throttle_head = NULL [static]

Definition at line 38 of file throttle.c.

throttle_t* throttle_next = NULL [static]

Definition at line 39 of file throttle.c.

int throttle_timer = -1 [static]

Definition at line 40 of file throttle.c.

Referenced by throttle_off(), and throttle_on().