Subversion Repositories shark

Rev

Rev 907 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
551 giacomo 1
/*
2
 * Project: S.Ha.R.K.
3
 *
4
 * Coordinators:
5
 *   Giorgio Buttazzo    <giorgio@sssup.it>
6
 *   Paolo Gai           <pj@gandalf.sssup.it>
7
 *
8
 * Authors     :
9
 *   Giacomo Guidi       <giacomo@gandalf.sssup.it>
10
 *   (see the web pages for full authors list)
11
 *
12
 * ReTiS Lab (Scuola Superiore S.Anna - Pisa - Italy)
13
 *
14
 * http://www.sssup.it
15
 * http://retis.sssup.it
16
 * http://shark.sssup.it
17
 */
18
 
19
/*
20
 * This program is free software; you can redistribute it and/or modify
21
 * it under the terms of the GNU General Public License as published by
22
 * the Free Software Foundation; either version 2 of the License, or
23
 * (at your option) any later version.
24
 *
25
 * This program is distributed in the hope that it will be useful,
26
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
27
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
28
 * GNU General Public License for more details.
29
 *
30
 * You should have received a copy of the GNU General Public License
31
 * along with this program; if not, write to the Free Software
32
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
33
 */
34
 
35
#include <ll/sys/types.h>
36
#include <stdlib.h>
37
#include <string.h>
38
#include <tracer.h>
39
#include <unistd.h>
40
 
41
#include <FTrace_udp.h>
42
#include <FTrace_OSD.h>
43
#include <FTrace_chunk.h>
44
 
45
#include <kernel/kern.h>
46
 
47
#include <drivers/udpip.h>
48
 
907 mauro 49
//#define TRACER_UDP_DEBUG
551 giacomo 50
 
51
#define UDP_MAXSIZE 1000
52
#define TRACER_PORT 20000
53
 
778 giacomo 54
#define DELAY_USEC 5000
551 giacomo 55
 
56
BYTE pkt[UDP_MAXSIZE];
57
int total_pkt_size = 0;
58
 
59
int TracerUDPInit = 0;
60
UDP_ADDR target,local;
61
char local_ip[20],target_ip[20];
62
int socket;
63
 
64
PID udp_task;
65
 
552 giacomo 66
int sending_flag = 0;
67
 
551 giacomo 68
struct FTrace_udp_header {
69
  char id[12];
70
  unsigned int pkt_number;
71
  unsigned int number_of_events;
72
  unsigned int total_size;
73
};
74
 
75
//Init UDP, if flag = 1 init the network driver
76
int FTrace_OSD_init_udp(int flag, char *l_ip, char *t_ip) {
77
 
78
  struct net_model m = net_base;
79
 
80
  strcpy(local_ip,l_ip);
81
  strcpy(target_ip,t_ip);
82
 
83
  if (flag) {
84
 
85
    net_setudpip(m, local_ip, "255.255.255.255");
86
 
87
    if (net_init(&m) != 1) {
88
      return -1;
89
    }
90
 
91
  }
92
 
93
  ip_str2addr(local_ip,&(local.s_addr));
94
  local.s_port = TRACER_PORT;
95
  socket = udp_bind(&local, NULL);
96
 
97
  ip_str2addr(target_ip,&(target.s_addr));
98
  target.s_port = TRACER_PORT;
99
 
100
  sleep(1);
101
 
102
  TracerUDPInit = 1;
103
 
104
  return 0;
105
 
106
}
107
 
108
int send_udp_event(void *p, int size) {
109
 
110
  static BYTE *current = pkt;
111
  static int events_number = 0;
112
  static int packet_number = 0;
113
 
114
  struct FTrace_udp_header head = {"TRACER-V1.0",0,0,0};
115
  struct FTrace_udp_header *phead = (struct FTrace_udp_header *)(pkt);
116
 
117
  if ((total_pkt_size + size < UDP_MAXSIZE) && p != NULL ) {
118
 
119
    if (total_pkt_size == 0) {
120
 
121
      memcpy(pkt,&head,sizeof(struct FTrace_udp_header));
122
 
123
      current = pkt;
124
      current += sizeof(struct FTrace_udp_header);
125
 
126
      total_pkt_size += sizeof(struct FTrace_udp_header);
127
      phead->total_size = total_pkt_size;
128
 
129
      packet_number++;
130
      phead->pkt_number = packet_number;
131
 
132
    }
133
 
134
    events_number++;
135
    phead->number_of_events = events_number;
136
 
137
    total_pkt_size += size;
138
    phead->total_size = total_pkt_size;
139
 
140
    //Copy the event
141
    memcpy(current,p,size);
142
    current += size;
143
 
144
    return 0;
145
 
146
  } else {
147
 
148
    #ifdef TRACER_UDP_DEBUG
778 giacomo 149
      cprintf(".");
551 giacomo 150
    #endif
151
 
152
    if (TracerUDPInit == 1) {
153
      udp_sendto(socket, pkt, total_pkt_size, &target);
778 giacomo 154
      usleep(DELAY_USEC);
551 giacomo 155
    }
156
 
157
    events_number = 0;
158
    total_pkt_size = 0;
159
    current = pkt;
160
 
161
    if (p != NULL) send_udp_event(p,size);
162
 
163
    return 0;    
164
 
165
  }
166
 
167
  return -1;
168
 
169
}
170
 
171
//Sender Task
172
TASK udp_sender(void *arg)
173
{
174
  FTrace_Chunk_Ptr c = (FTrace_Chunk_Ptr)(arg);
175
 
176
  DWORD total_size = (DWORD)(c->size) + (DWORD)(c->emergency_size);
177
  DWORD start = (DWORD)(c->osd + FTRACE_OSD_CHUNK_HEAD);
178
  DWORD current = start;
179
  DWORD count = 0;  
180
 
181
  while (current+16 <= start+total_size) {
182
 
561 giacomo 183
        if (*(WORD *)(current) != 0) {
551 giacomo 184
                send_udp_event((void *)(current), 16);
185
                count++;
186
        }
187
 
188
        current += 16;
189
 
190
  }
191
 
192
  // Flush the buffer out
193
  send_udp_event(NULL, 0);
194
 
552 giacomo 195
  sending_flag = 0;
196
 
551 giacomo 197
  #ifdef TRACER_UDP_DEBUG
778 giacomo 198
    cprintf("\nTotal Chunk Event Sent: %d\n",(int)count);
551 giacomo 199
  #endif  
200
 
201
  return 0;  
202
 
203
}
204
 
205
int FTrace_OSD_create_udp_task(FTrace_Chunk_Ptr c)
206
{
207
 
208
  NRT_TASK_MODEL st;
209
 
552 giacomo 210
  sending_flag = 1;
211
 
551 giacomo 212
  nrt_task_default_model(st);
213
  nrt_task_def_arg(st,(void *)c);
214
 
215
  udp_task = task_create("UDP_Sender",udp_sender,&st,NULL);
216
  if (udp_task == NIL) {
217
    cprintf("Error creating UDP Sender\n");
927 pj 218
    exit(1);
551 giacomo 219
  }
220
 
221
  task_activate(udp_task);
222
 
223
  return 0;
224
 
225
}