palacios/src/geekos/ring_buffer.c

   1 #include <geekos/ring_buffer.h>
   2 #include <geekos/malloc.h>
   3 #include <geekos/ktypes.h>
   4
   5
   6
   7 void init_ring_buffer(struct ring_buffer * ring, uint_t size) {
   8   ring->buf = Malloc(size);
   9   ring->size = size;
  10
  11   ring->start = 0;
  12   ring->end = 0;
  13   ring->current_len = 0;
  14 }
  15
  16
  17 struct ring_buffer * create_ring_buffer(uint_t size) {
  18   struct ring_buffer * ring = (struct ring_buffer *)Malloc(sizeof(struct ring_buffer));
  19
  20   init_ring_buffer(ring, size);
  21
  22   return ring;
  23 }
  24
  25 void free_ring_buffer(struct ring_buffer * ring) {
  26   Free(ring->buf);
  27   Free(ring);
  28 }
  29
  30
  31
  32
  33
  34 static inline uchar_t * get_buf_ptr(struct ring_buffer * ring) {
  35   return (uchar_t *)&(ring->buf + ring->start);
  36 }
  37
  38
  39 static inline int get_section_size(struct ring_buffer * ring) {
  40   return ring->size - ring->start;
  41 }
  42
  43 static inline int is_loop(struct ring_buffer * ring, uint_t len) {
  44   if ((ring->start >= ring->end) && (ring->current_len > 0)) {
  45     // end is past the end of the buffer
  46     if (get_section_size(ring) < len) {
  47       return 1;
  48     }
  49   }
  50
  51   return 0;
  52 }
  53
  54
  55 int rb_read(struct ring_buffer * ring, char * dst, uint_t len) {
  56   int read_len = 0;
  57   int ring_data_len = ring->current_len;
  58
  59   read_len = (len > ring_data_len) ? ring_data_len : len;
  60
  61   if (is_loop(ring, read_len)) {
  62     int first_len = get_section_size(ring);
  63
  64     memcpy(dst, get_buf_ptr(ring), first_len);
  65     read_len -= first_len;
  66     ring->start = 0;
  67
  68     memcpy(dst + first_len, get_buf_ptr, read_len);
  69
  70   } else {
  71     memcpy(dst, get_buf_ptr(ring), read_len);
  72   }
  73
  74   return read_len;
  75
  76 }