labcomm_thr_reader_writer.c 2.35 KB
Newer Older
Riccardo Gaiati's avatar
Riccardo Gaiati committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#include <errno.h>
#include <unistd.h>
#include "labcomm.h"
#include "ThrottleDrv/throttle_drv.h"

#define BUFFER_SIZE 2048

int labcomm_thr_reader(labcomm_reader_t *r, labcomm_reader_action_t action)
{
  int result = -EINVAL;
  struct thr_chn_t *ctx = r->context;

  switch (action)
  {
    case labcomm_reader_alloc:
    {
      r->data = malloc(BUFFER_SIZE);
      if (r->data)
      {
        r->data_size = BUFFER_SIZE;
        result = r->data_size;
      }
      else
      {
        r->data_size = 0;
        result = -ENOMEM;
      }
      r->count = 0;
      r->pos = 0;
    }
    break;

    case labcomm_reader_start:
    case labcomm_reader_continue:
    {
      if (r->pos < r->count)
      {
        result = r->count - r->pos;
      }
      else
      {
        int err;
        r->pos = 0;
        err = thr_read(ctx, r->data, r->data_size);
        if (err <= 0)
        {
          r->count = 0;
          result = -1;
        } else
        {
          r->count = err;
          result = r->count - r->pos;
        }
      }
    }
    break;

    case labcomm_reader_end:
    {
      result = 0;
    }
    break;

    case labcomm_reader_free:
    {
      free(r->data);
      r->data = 0;
      r->data_size = 0;
      r->count = 0;
      r->pos = 0;
      result = 0;
    }
    break;
  }
  return result;
}

int labcomm_thr_writer(labcomm_writer_t *w, labcomm_writer_action_t action)
{
  int result = 0;
  struct thr_chn_t *ctx = w->context;

  switch (action)
  {
    case labcomm_writer_alloc:
    {
      w->data = malloc(BUFFER_SIZE);
      if (! w->data)
      {
        result = -ENOMEM;
        w->data_size = 0;
        w->count = 0;
        w->pos = 0;
      }
      else
      {
        w->data_size = BUFFER_SIZE;
        w->count = BUFFER_SIZE;
        w->pos = 0;
      }
    } break;
    case labcomm_writer_free:
    {
      free(w->data);
      w->data = 0;
      w->data_size = 0;
      w->count = 0;
      w->pos = 0;
    }
    break;

    case labcomm_writer_start:
    {
      w->pos = 0;
    }
    break;

    case labcomm_writer_continue:
    {
      result = thr_send(ctx, w->data, w->pos);
      w->pos = 0;
    }
    break;

    case labcomm_writer_end:
    {
      result = thr_send(ctx, w->data, w->pos);
      w->pos = 0;
    }
    break;

    case labcomm_writer_available:
    {
      result = w->count - w->pos;
    }
    break;
  }
  return result;
}