Constify.
[obnox/wireshark/wip.git] / wiretap / netscaler.c
1 /* netscaler.c
2  *
3  * $Id$
4  *
5  * Wiretap Library
6  * Copyright (c) 2006 by Ravi Kondamuru <Ravi.Kondamuru@citrix.com>
7  *
8  * This program is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU General Public License
10  * as published by the Free Software Foundation; either version 2
11  * of the License, or (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
21  */
22
23 #ifdef HAVE_CONFIG_H
24 #include "config.h"
25 #endif
26 #include <errno.h>
27 #include <string.h>
28 #include "wtap-int.h"
29 #include "file_wrappers.h"
30 #include "buffer.h"
31 #include "netscaler.h"
32
33 /* Defines imported from netscaler code: nstypes.h */
34
35 #define ns_min(a, b)    ((a<b)?a:b)
36
37 /* Defines imported from netscaler code: nsperfrc.h */
38
39 #define NSPR_SIGSTR_V10 "NetScaler Performance Data"
40 #define NSPR_SIGSTR_V20 "NetScaler V20 Performance Data"
41 #define NSPR_SIGSTR     NSPR_SIGSTR_V20
42 /* Defined but not used */
43 #define NSPR_SIGSTR_V21 "NetScaler V21 Performance Data"
44 #define NSPR_SIGSTR_V22 "NetScaler V22 Performance Data"
45
46 #define NSPR_PAGESIZE   8192
47
48 /* The different record types
49 ** NOTE: The Record Type is two byte fields and unused space is recognized by
50 ** either bytes being zero, therefore no record should any byte value as
51 ** zero.
52 **
53 ** New Performance Record Type is only one byte.
54 */
55 #define NSPR_UNUSEDSPACE_V10    0x0000  /* rest of the page is unused */
56 #define NSPR_UNUSEDSPACE_V20    0x00    /* rest of the page is unused */
57 #define NSPR_SIGNATURE_V10      0x0101  /* signature */
58 #define NSPR_SIGNATURE_V20      0x01    /* signature */
59 #define NSPR_ABSTIME_V10        0x0107  /* data capture time in secs from 1970*/
60 #define NSPR_ABSTIME_V20        0x07    /* data capture time in secs from 1970*/
61 #define NSPR_RELTIME_V10        0x0108  /* relative time in ms from last time */
62 #define NSPR_RELTIME_V20        0x08    /* relative time in ms from last time */
63 #define NSPR_RELTIMEHR_V10      0x0109  /* high resolution relative time */
64 #define NSPR_RELTIMEHR_V20      0x09    /* high resolution relative time */
65 #define NSPR_SYSTARTIME_V10     0x010A  /* system start time */
66 #define NSPR_SYSTARTIME_V20     0x0A    /* system start time */
67 #define NSPR_RELTIME2B_V10      0x010B  /* relative time in ms from last time */
68 #define NSPR_RELTIME2B_V20      0x0B    /* relative time in ms from last time */
69
70
71 /* The high resolution relative time format.
72 ** The MS 2 bits of the high resoltion time is defined as follows:
73 ** 00 : time value is in second
74 ** 01 : time value is in mili second
75 ** 10 : time value is in micro second
76 ** 11 : time value is in nano second
77 */
78 #define NSPR_HRTIME_MASKTM      0x3FFFFFFF /* mask to get time value */
79 #define NSPR_HRTIME_MASKFMT     0xC0000000 /* time value format mask */
80 #define NSPR_HRTIME_SEC         0x00000000 /* time value in second */
81 #define NSPR_HRTIME_MSEC        0x40000000 /* time value in mili second */
82 #define NSPR_HRTIME_USEC        0x80000000 /* time value in micro second */
83 #define NSPR_HRTIME_NSEC        0xC0000000 /* time value in nano second */
84
85
86 typedef  struct nspr_header_v10
87 {
88         guint16 ph_RecordType;  /* Record Type */
89         guint16 ph_RecordSize;  /* Record Size including header */
90 } nspr_header_v10_t;
91 #define nspr_header_v10_s       sizeof(nspr_header_v10_t)
92
93 /* This is V20 short header (2 bytes long) to be included where needed */
94 #define NSPR_HEADER_V20(prefix) \
95     guint8 prefix##_RecordType;    /* Record Type */ \
96         guint8 prefix##_RecordSize     /* Record Size including header */       \
97                                         /* end of declaration */
98
99 /* This is new long header (3 bytes long) to be included where needed */
100 #define NSPR_HEADER3B_V20(prefix) \
101         guint8  prefix##_RecordType;    /* Record Type */ \
102         guint8  prefix##_RecordSizeLow; /* Record Size including header */ \
103         guint8  prefix##_RecordSizeHigh /* Record Size including header */ \
104                                         /* end of declaration */
105 #define NSPR_HEADER3B_V21 NSPR_HEADER3B_V20
106 #define NSPR_HEADER3B_V22 NSPR_HEADER3B_V20
107
108 typedef  struct nspr_hd_v20
109 {
110         NSPR_HEADER3B_V20(phd);         /* long performance header */
111
112 } nspr_hd_v20_t;
113 #define nspr_hd_v20_s   sizeof(nspr_hd_v20_t)
114
115
116 /*
117 ** How to know if header size is short or long?
118 ** The short header size can be 0-127 bytes long. If MS Bit of ph_RecordSize
119 ** is set then record size has 2 bytes
120 */
121 #define NSPR_V20RECORDSIZE_2BYTES       0x80
122
123 /* Performance Data Header with device number */
124 typedef  struct nspr_headerdev_v10
125 {
126         guint16 ph_RecordType;  /* Record Type */
127         guint16 ph_RecordSize;  /* Record Size including header */
128         guint32 ph_DevNo;       /* Network Device (NIC/CONN) number */
129 } nspr_headerdev_v10_t;
130 #define nspr_headerdev_v10_s    sizeof(nspr_headerdev_v10_t)
131
132 typedef struct nspr_hd_v10
133 {
134         nspr_header_v10_t phd;  /* performance header */
135 } nspr_hd_v10_t;
136 #define nspr_hd_v10_s   sizeof(nspr_hd_v10_t)
137
138 typedef struct nspr_hdev_v10
139 {
140         nspr_headerdev_v10_t phd;       /* performance header */
141 } nspr_hdev_v10_t;
142 #define nspr_hdev_v10_s sizeof(nspr_hdev_v10_t)
143
144 /* if structure has defined phd as first field, it can use following names */
145 #define nsprRecordType          phd.ph_RecordType
146 #define nsprRecordSize          phd.ph_RecordSize
147 #define nsprReserved            phd.ph_Reserved
148 #define nsprRecordTypeOrg       phd.ph_Reserved
149 #define nsprDevNo               phd.ph_DevNo
150
151 /* NSPR_SIGNATURE_V10 structure */
152 #define NSPR_SIGSIZE_V10        56      /* signature value size in bytes */
153
154
155 typedef struct nspr_signature_v10
156 {
157         nspr_header_v10_t phd;  /* performance header */
158         guint8  sig_EndianType; /* Endian Type for the data */
159         guint8  sig_Reserved0;
160         guint16 sig_Reserved1;
161         gchar   sig_Signature[NSPR_SIGSIZE_V10];        /* Signature value */
162 } nspr_signature_v10_t;
163 #define nspr_signature_v10_s    sizeof(nspr_signature_v10_t)
164
165 /* NSPR_SIGNATURE_V20 structure */
166 typedef struct nspr_signature_v20
167 {
168         NSPR_HEADER_V20(sig);   /* short performance header */
169         guint8  sig_EndianType; /* Endian Type for the data */
170         gchar   sig_Signature[1]; /* Signature value */
171 } nspr_signature_v20_t;
172 #define nspr_signature_v20_s    (sizeof(nspr_signature_v20_t) -1)
173
174 /* NSPR_ABSTIME_V10 and NSPR_SYSTARTIME_V10 structure */
175 typedef struct nspr_abstime_v10
176 {
177         nspr_header_v10_t phd;  /* performance header */
178         guint32 abs_RelTime;    /* relative time is ms from last time */
179         guint32 abs_Time;       /* absolute time in seconds from 1970 */
180 } nspr_abstime_v10_t;
181 #define nspr_abstime_v10_s      sizeof(nspr_abstime_v10_t)
182
183
184 /* NSPR_ABSTIME_V20 and NSPR_SYSTARTIME_V20 structure */
185 typedef struct nspr_abstime_v20
186 {
187         NSPR_HEADER_V20(abs);   /* short performance header */
188         guint16 abs_RelTime;    /* relative time is ms from last time */
189         guint32 abs_Time;       /* absolute time in seconds from 1970 */
190 } nspr_abstime_v20_t;
191 #define nspr_abstime_v20_s      sizeof(nspr_abstime_v20_t)
192
193
194
195 /* full packet trace structure */
196 typedef struct  nspr_pktracefull_v10
197 {
198         nspr_headerdev_v10_t phd;       /* performance header */
199         guint32 fp_RelTimeHr;   /* High resolution relative time */
200         guint8  fp_Data[1];     /* packet data starts here */
201 } nspr_pktracefull_v10_t;
202 #define nspr_pktracefull_v10_s  (nspr_hdev_v10_s + 4)
203
204 /* new full packet trace structure v20 */
205 typedef struct  nspr_pktracefull_v20
206 {
207         NSPR_HEADER3B_V20(fp);  /* long performance header */
208         guint8  fp_DevNo;       /* Network Device (NIC) number */
209         guint32 fp_RelTimeHr;   /* High resolution relative time */
210         guint8  fp_Data[4];     /* packet data starts here */
211 } nspr_pktracefull_v20_t;
212 #define nspr_pktracefull_v20_s  (sizeof(nspr_pktracefull_v20_t) - 4)
213
214 /* new full packet trace structure v21 */
215 typedef struct  nspr_pktracefull_v21
216 {
217         NSPR_HEADER3B_V21(fp);  /* long performance header */
218         guint8  fp_DevNo;       /* Network Device (NIC) number */
219         guint32 fp_RelTimeHr;   /* High resolution relative time */
220         guint32 fp_PcbDevNo;    /* PCB devno */
221         guint32 fp_lPcbDevNo;   /* link PCB devno */
222         guint8  fp_Data[4];     /* packet data starts here */
223 } nspr_pktracefull_v21_t;
224 #define nspr_pktracefull_v21_s  (sizeof(nspr_pktracefull_v21_t) - 4)
225
226 /* new full packet trace structure v22 */
227 typedef struct  nspr_pktracefull_v22
228 {
229         NSPR_HEADER3B_V22(fp);  /* long performance header */
230         guint8  fp_DevNo;       /* Network Device (NIC) number */
231         guint32 fp_RelTimeHr;   /* High resolution relative time */
232         guint32 fp_PcbDevNo;    /* PCB devno */
233         guint32 fp_lPcbDevNo;   /* link PCB devno */
234         guint16 fp_VlanTag;
235         guint8  fp_Data[4];     /* packet data starts here */
236 } nspr_pktracefull_v22_t;
237 #define nspr_pktracefull_v22_s  (sizeof(nspr_pktracefull_v22_t) - 4)
238
239 typedef struct  nspr_pktracefull_v23
240 {
241         NSPR_HEADER3B_V22(fp);  /* long performance header */
242         guint8  fp_DevNo;       /* Network Device (NIC) number */
243         guint32 fp_AbsTimeLowHdr;       /* High resolution low time */
244         guint32 fp_AbsTimeHighHdr; /* Higher value of the absolute time */
245         guint32 fp_PcbDevNo;    /* PCB devno */
246         guint32 fp_lPcbDevNo;   /* link PCB devno */
247         guint16 fp_VlanTag; /* vlan tag */
248         guint16 fp_Coreid; /* coreid of the packet */
249         guint8  fp_Data[4];     /* packet data starts here */
250 } nspr_pktracefull_v23_t;
251 #define nspr_pktracefull_v23_s  (sizeof(nspr_pktracefull_v23_t) - 4)
252
253 /* partial packet trace structure */
254 typedef struct  nspr_pktracepart_v10
255 {
256         nspr_headerdev_v10_t phd;       /* performance header */
257         guint32 pp_RelTimeHr;   /* High resolution relative time */
258         guint16 pp_PktSizeOrg;  /* Original packet size */
259         guint16 pp_PktOffset;   /* starting offset in packet */
260         guint8  pp_Data[1];     /* packet data starts here */
261 } nspr_pktracepart_v10_t;
262 #define nspr_pktracepart_v10_s  (nspr_pktracefull_v10_s + 4)
263
264 /* new partial packet trace structure */
265 typedef struct  nspr_pktracepart_v20
266 {
267         NSPR_HEADER3B_V20(pp);  /* long performance header */
268         guint8  pp_DevNo;       /* Network Device (NIC) number */
269         guint32 pp_RelTimeHr;   /* High resolution relative time */
270         guint16 pp_PktSizeOrg;  /* Original packet size */
271         guint16 pp_PktOffset;   /* starting offset in packet */
272         guint8  pp_Data[4];     /* packet data starts here */
273 } nspr_pktracepart_v20_t;
274 #define nspr_pktracepart_v20_s  (sizeof(nspr_pktracepart_v20_t) -4)
275
276 /* new partial packet trace structure */
277 typedef struct  nspr_pktracepart_v21
278 {
279         NSPR_HEADER3B_V21(pp);  /* long performance header */
280         guint8  pp_DevNo;       /* Network Device (NIC) number */
281         guint32 pp_RelTimeHr;   /* High resolution relative time */
282         guint16 pp_PktSizeOrg;  /* Original packet size */
283         guint16 pp_PktOffset;   /* starting offset in packet */
284         guint32 pp_PcbDevNo;    /* PCB devno */
285         guint32 pp_lPcbDevNo;   /* link PCB devno */
286         guint8  pp_Data[4];     /* packet data starts here */
287 } nspr_pktracepart_v21_t;
288 #define nspr_pktracepart_v21_s  (sizeof(nspr_pktracepart_v21_t) -4)
289
290 /* new partial packet trace structure v22 */
291 typedef struct  nspr_pktracepart_v22
292 {
293         NSPR_HEADER3B_V22(pp);  /* long performance header */
294         guint8  pp_DevNo;       /* Network Device (NIC) number */
295         guint32 pp_RelTimeHr;   /* High resolution relative time */
296         guint16 pp_PktSizeOrg;  /* Original packet size */
297         guint16 pp_PktOffset;   /* starting offset in packet */
298         guint32 pp_PcbDevNo;    /* PCB devno */
299         guint32 pp_lPcbDevNo;   /* link PCB devno */
300         guint16 pp_VlanTag;     /* Vlan Tag */
301         guint8  pp_Data[4];     /* packet data starts here */
302 } nspr_pktracepart_v22_t;
303 #define nspr_pktracepart_v22_s  (sizeof(nspr_pktracepart_v22_t) -4)
304
305 typedef struct  nspr_pktracepart_v23
306 {
307         NSPR_HEADER3B_V22(pp);  /* long performance header */
308         guint8  pp_DevNo;       /* Network Device (NIC) number */
309         guint32 pp_AbsTimeLowHdr;       /* High resolution low time */
310         guint32 pp_AbsTimeHighHdr; /* Higher value of the absolute time */
311         guint16 pp_PktSizeOrg;  /* Original packet size */
312         guint16 pp_PktOffset;   /* starting offset in packet */
313         guint32 pp_PcbDevNo;    /* PCB devno */
314         guint32 pp_lPcbDevNo;   /* link PCB devno */
315         guint16 pp_VlanTag;     /* vlan tag */
316         guint16 pp_Coreid; /* Coreid of the packet */
317         guint8  pp_Data[4];     /* packet data starts here */
318 } nspr_pktracepart_v23_t;
319 #define nspr_pktracepart_v23_s  (sizeof(nspr_pktracepart_v23_t) -4)
320
321 #define myoffsetof(type,fieldname) (&(((type*)0)->fieldname))
322
323 #define __TNO(enumprefix,structprefix,structname,hdrname,structfieldname) \
324         guint8 enumprefix##_##hdrname##_offset = (guint8)GPOINTER_TO_INT(myoffsetof(nspr_##structname##_t,structprefix##_##structfieldname));
325
326 #define __TNL(enumprefix,structprefix,structname,hdrname,structfieldname) \
327         guint8 enumprefix##_##hdrname##_len = (guint8)sizeof(((nspr_##structname##_t*)0)->structprefix##_##structfieldname);
328         
329 #define __TNV1O(enumprefix,structprefix,structname,hdrname,structfieldname) \
330         guint8 enumprefix##_##hdrname##_offset = (guint8)GPOINTER_TO_INT(myoffsetof(nspr_##structname##_t,structfieldname));
331
332 #define __TNV1L(enumprefix,structprefix,structname,hdrname,structfieldname) \
333         guint8 enumprefix##_##hdrname##_len = (guint8)sizeof(((nspr_##structname##_t*)0)->structfieldname);
334
335 #define TRACE_V10_REC_LEN_OFF(enumprefix,structprefix,structname) \
336                 __TNV1O(enumprefix,structprefix,structname,dir,phd.ph_RecordType)\
337                 __TNV1L(enumprefix,structprefix,structname,dir,phd.ph_RecordType)\
338                 __TNV1O(enumprefix,structprefix,structname,nicno,phd.ph_DevNo)\
339                 __TNV1L(enumprefix,structprefix,structname,nicno,phd.ph_DevNo)\
340                 __TNO(enumprefix,structprefix,structname,eth,Data)
341         
342 #define TRACE_V20_REC_LEN_OFF(enumprefix,structprefix,structname) \
343                 __TNO(enumprefix,structprefix,structname,dir,RecordType)\
344                 __TNL(enumprefix,structprefix,structname,dir,RecordType)\
345                 __TNO(enumprefix,structprefix,structname,nicno,DevNo)\
346                 __TNL(enumprefix,structprefix,structname,nicno,DevNo)\
347                 __TNO(enumprefix,structprefix,structname,eth,Data)
348         
349 #define TRACE_V21_REC_LEN_OFF(enumprefix,structprefix,structname) \
350                 TRACE_V20_REC_LEN_OFF(enumprefix,structprefix,structname)\
351                 __TNO(enumprefix,structprefix,structname,pcb,PcbDevNo)\
352                 __TNO(enumprefix,structprefix,structname,l_pcb,lPcbDevNo)
353
354 #define TRACE_V22_REC_LEN_OFF(enumprefix,structprefix,structname) \
355                 TRACE_V21_REC_LEN_OFF(enumprefix,structprefix,structname)\
356                 __TNO(enumprefix,structprefix,structname,vlantag,VlanTag)
357
358 #define TRACE_V23_REC_LEN_OFF(enumprefix,structprefix,structname) \
359                 TRACE_V22_REC_LEN_OFF(enumprefix,structprefix,structname)\
360                 __TNO(enumprefix,structprefix,structname,coreid,Coreid)
361
362         
363         TRACE_V10_REC_LEN_OFF(v10_part,pp,pktracepart_v10)
364         TRACE_V10_REC_LEN_OFF(v10_full,fp,pktracefull_v10)
365         TRACE_V20_REC_LEN_OFF(v20_part,pp,pktracepart_v20)
366         TRACE_V20_REC_LEN_OFF(v20_full,fp,pktracefull_v20)
367         TRACE_V21_REC_LEN_OFF(v21_part,pp,pktracepart_v21)
368         TRACE_V21_REC_LEN_OFF(v21_full,fp,pktracefull_v21)
369         TRACE_V22_REC_LEN_OFF(v22_part,pp,pktracepart_v22)
370         TRACE_V22_REC_LEN_OFF(v22_full,fp,pktracefull_v22)
371         TRACE_V23_REC_LEN_OFF(v23_part,pp,pktracepart_v23)
372         TRACE_V23_REC_LEN_OFF(v23_full,fp,pktracefull_v23)
373         
374 #undef __TNV1O
375 #undef __TNV1L
376 #undef __TNO
377 #undef __TNL
378
379
380 #define ns_setabstime(nstrace, AbsoluteTime, RelativeTimems)    \
381         do { \
382                 (nstrace)->nspm_curtime = AbsoluteTime; \
383                 (nstrace)->nspm_curtimemsec += RelativeTimems; \
384                 (nstrace)->nspm_curtimelastmsec = nstrace->nspm_curtimemsec; \
385         } while(0)
386
387
388 #define ns_setrelativetime(nstrace, RelativeTimems)     \
389         do { \
390                 guint32 rsec; \
391                 (nstrace)->nspm_curtimemsec += RelativeTimems; \
392                 rsec = (guint32)((nstrace)->nspm_curtimemsec - (nstrace)->nspm_curtimelastmsec)/1000; \
393                 (nstrace)->nspm_curtime += rsec; \
394                 (nstrace)->nspm_curtimelastmsec += rsec * 1000; \
395         } while (0)
396
397
398 typedef struct {
399         gchar *pnstrace_buf;
400         gint32 nstrace_buf_offset;
401         gint32 nstrace_buflen;
402         /* Performance Monitor Time variables */
403         guint32 nspm_curtime;           /* current time since 1970 */
404         guint64 nspm_curtimemsec;       /* current time in mili second */
405         guint64 nspm_curtimelastmsec;   /* nspm_curtime last update time in milisec */
406         guint64 nsg_creltime;
407         guint64 file_size;
408 } nstrace_t;
409
410 guint32 nspm_signature_isv10(gchar *sigp);
411 guint32 nspm_signature_isv20(gchar *sigp);
412 guint32 nspm_signature_version(wtap*, gchar*, gint32);
413 gboolean nstrace_read(wtap *wth, int *err, gchar **err_info,
414                  gint64 *data_offset);
415 gboolean nstrace_read_v10(wtap *wth, int *err, gchar **err_info,
416                  gint64 *data_offset);
417 gboolean nstrace_read_v20(wtap *wth, int *err, gchar **err_info,
418                  gint64 *data_offset);
419 gboolean nstrace_seek_read(wtap *wth, gint64 seek_off, 
420                       union wtap_pseudo_header *pseudo_header, 
421                       guchar *pd, int length,
422                       int *err, gchar **err_info _U_);
423 void nstrace_close(wtap *wth);
424 void nstrace_sequential_close(wtap *wth);
425
426 gboolean nstrace_set_start_time_v10(wtap *wth);
427 gboolean nstrace_set_start_time_v20(wtap *wth);
428 gboolean nstrace_set_start_time(wtap *wth);
429 guint64 ns_hrtime2nsec(guint32 tm);
430
431 static gboolean nstrace_dump(wtap_dumper *wdh, const struct wtap_pkthdr *phdr, 
432         const union wtap_pseudo_header *pseudo_header, const guchar *pd, int *err);
433 gboolean nstrace_add_signature(wtap_dumper *wdh);
434 gboolean nstrace_add_abstime(wtap_dumper *wdh, const struct wtap_pkthdr *phdr, const guchar *pd);
435
436
437 #define GET_READ_PAGE_SIZE(remaining_file_size) ((gint32)((remaining_file_size>NSPR_PAGESIZE)?NSPR_PAGESIZE:remaining_file_size))
438
439
440 guint64 ns_hrtime2nsec(guint32 tm)
441 {
442         guint32 val = tm & NSPR_HRTIME_MASKTM;
443         switch(tm & NSPR_HRTIME_MASKFMT)
444         {
445         case NSPR_HRTIME_SEC:   return (guint64)val*1000000000;
446         case NSPR_HRTIME_MSEC:  return (guint64)val*1000000;
447         case NSPR_HRTIME_USEC:  return (guint32)val*1000;
448         case NSPR_HRTIME_NSEC:  return val;
449         }
450         return tm;
451 }
452
453
454 /*
455 ** Netscaler trace format open routines
456 */
457 int nstrace_open(wtap *wth, int *err, gchar **err_info)
458 {
459         gchar *nstrace_buf;
460         gint64 file_size;
461         gint32 page_size;
462         nstrace_t *nstrace;
463
464         errno = WTAP_ERR_CANT_READ;
465
466         if ((file_size = wtap_file_size(wth, err)) == -1)       
467                 return 0;
468   
469         nstrace_buf = g_malloc(NSPR_PAGESIZE);
470         page_size = GET_READ_PAGE_SIZE(file_size);
471
472         switch ((wth->file_type = nspm_signature_version(wth, nstrace_buf, page_size)))
473         {
474         case WTAP_FILE_NETSCALER_1_0:
475                 wth->file_encap = WTAP_ENCAP_NSTRACE_1_0;
476                 break;
477         
478         case WTAP_FILE_NETSCALER_2_0:
479                 wth->file_encap = WTAP_ENCAP_NSTRACE_2_0;
480                 break;
481
482         default:
483                 *err = WTAP_ERR_UNSUPPORTED;
484                 *err_info = g_strdup_printf("nstrace: file type %d unsupported", wth->file_type);
485                 g_free(nstrace_buf);
486                 return 0;
487         }
488
489         if ((file_seek(wth->fh, 0, SEEK_SET, err)) == -1)
490         {
491                 *err = file_error(wth->fh);
492                 g_free(nstrace_buf);
493                 return 0;
494         }
495
496         if (page_size != file_read(nstrace_buf, 1, page_size, wth->fh))
497         {
498                 *err = file_error(wth->fh);
499                 g_free(nstrace_buf);
500                 return 0;
501         }
502         
503         wth->subtype_read = nstrace_read;
504         wth->subtype_seek_read = nstrace_seek_read;
505         wth->subtype_close = nstrace_close; 
506
507         nstrace = (nstrace_t *)g_malloc(sizeof(nstrace_t));
508         wth->priv = (void *)nstrace;
509         nstrace->pnstrace_buf = nstrace_buf;
510         nstrace->nstrace_buflen = page_size;
511         nstrace->nstrace_buf_offset = 0;
512         nstrace->nspm_curtime = 0;
513         nstrace->nspm_curtimemsec = 0;
514         nstrace->nspm_curtimelastmsec = 0;
515         nstrace->nsg_creltime = 0;
516         nstrace->file_size = file_size;
517
518   
519         /* Set the start time by looking for the abstime record */
520         if ((nstrace_set_start_time(wth)) == FALSE)
521         {
522                 /* Reset the read pointer to start of the file. */
523                 if ((file_seek(wth->fh, 0, SEEK_SET, err)) == -1)
524                 {
525                         *err = file_error(wth->fh);
526                         g_free(nstrace->pnstrace_buf);
527                         g_free(nstrace);
528                         return 0;
529                 }
530
531                 /* Read the first page of data */
532                 if (page_size != file_read(nstrace_buf, 1, page_size, wth->fh))
533                 {
534                         *err = file_error(wth->fh);
535                         g_free(nstrace->pnstrace_buf);
536                         g_free(nstrace);
537                         return 0;
538                 }
539         
540                 /* reset the buffer offset */
541                 nstrace->nstrace_buf_offset = 0;
542         }
543
544         wth->tsprecision = WTAP_FILE_TSPREC_NSEC;
545         wth->phdr.ts.secs = nstrace->nspm_curtime; 
546         wth->phdr.ts.nsecs = 0;
547
548         *err = 0;
549         return 1;
550 }
551
552
553 #define nspm_signature_func(ver) \
554         guint32 nspm_signature_isv##ver(gchar *sigp) {\
555                 return strncmp(sigp,NSPR_SIGSTR_V##ver,(sizeof(NSPR_SIGSTR_V##ver)-1));\
556         }
557
558 nspm_signature_func(10)
559 nspm_signature_func(20)
560
561 /*
562 ** Check signature and return the version number of the signature.
563 ** If not found, it returns 0. At the time of return from this function
564 ** we might not be at the first page. So after a call to this function, there 
565 ** has to be a file seek to return to the start of the first page.
566 */
567 guint32
568 nspm_signature_version(wtap *wth, gchar *nstrace_buf, gint32 len)
569 {
570         gchar *dp = nstrace_buf;
571
572         if (len == file_read(dp, 1, len, wth->fh)) {
573
574                 for ( ; len > (gint32)(ns_min(sizeof(NSPR_SIGSTR_V10), sizeof(NSPR_SIGSTR_V20))); dp++, len--)
575                 {
576 #define sigv10p ((nspr_signature_v10_t*)dp)
577                         if ((pletohs(&sigv10p->nsprRecordType) == NSPR_SIGNATURE_V10) &&
578                                 (pletohs(&sigv10p->nsprRecordSize) <= len) &&
579                                 ((gint32)sizeof(NSPR_SIGSTR_V10) <= len) &&
580                                 (!nspm_signature_isv10(sigv10p->sig_Signature)))
581                                 return WTAP_FILE_NETSCALER_1_0;
582 #undef  sigv10p
583     
584 #define sigv20p ((nspr_signature_v20_t*)dp)
585                         if ((sigv20p->sig_RecordType == NSPR_SIGNATURE_V20) &&
586                                 (sigv20p->sig_RecordSize <= len) &&
587                                 ((gint32)sizeof(NSPR_SIGSTR_V20) <= len) &&
588                                 (!nspm_signature_isv20(sigv20p->sig_Signature)))
589                                 return WTAP_FILE_NETSCALER_2_0;
590 #undef  sigv20p
591                 }
592         }
593
594         return 0;       /* no version found */
595 }
596
597 #define nspr_getv10recordtype(hdp) (pletohs(&hdp->nsprRecordType))
598 #define nspr_getv10recordsize(hdp) (pletohs(&hdp->nsprRecordSize))
599 #define nspr_getv20recordtype(hdp) (hdp->phd_RecordType)
600 #define nspr_getv20recordsize(hdp) \
601         (((hdp)->phd_RecordSizeLow & NSPR_V20RECORDSIZE_2BYTES)? \
602                 (((hdp)->phd_RecordSizeHigh * NSPR_V20RECORDSIZE_2BYTES)+ \
603                  ((hdp)->phd_RecordSizeLow & ~NSPR_V20RECORDSIZE_2BYTES)) : \
604                   (hdp)->phd_RecordSizeLow)
605
606
607 #define nstrace_set_start_time_ver(ver) \
608         gboolean nstrace_set_start_time_v##ver(wtap *wth) \
609         {\
610                 nstrace_t *nstrace = (nstrace_t *)wth->priv;\
611                 gchar* nstrace_buf = nstrace->pnstrace_buf;\
612                 gint32 nstrace_buf_offset = nstrace->nstrace_buf_offset;\
613                 gint32 nstrace_buflen = nstrace->nstrace_buflen;\
614                 do\
615                 {\
616                         while (nstrace_buf_offset < nstrace_buflen)\
617                         {\
618                                 nspr_hd_v##ver##_t *fp = (nspr_hd_v##ver##_t *) &nstrace_buf[nstrace_buf_offset];\
619                                 switch (nspr_getv##ver##recordtype(fp))\
620                                 {\
621                                         case NSPR_ABSTIME_V##ver:\
622                                                 ns_setabstime(nstrace, pletohl(&((nspr_abstime_v##ver##_t *) fp)->abs_Time), pletohs(&((nspr_abstime_v##ver##_t *) fp)->abs_RelTime));\
623                                                 nstrace->nstrace_buf_offset = nstrace_buf_offset + nspr_getv##ver##recordsize(fp);\
624                                                 nstrace->nstrace_buflen = nstrace_buflen;\
625                                                 return TRUE;\
626                                         case NSPR_UNUSEDSPACE_V10:\
627                                                 nstrace_buf_offset = nstrace_buflen;\
628                                                 break;\
629                                         default:\
630                                                 nstrace_buf_offset += nspr_getv##ver##recordsize(fp);\
631                                 }\
632                         }\
633                         nstrace_buf_offset = 0;\
634                         wth->data_offset += nstrace_buflen;\
635                         nstrace_buflen = GET_READ_PAGE_SIZE((nstrace->file_size - wth->data_offset));\
636                 }while((nstrace_buflen > 0) && (nstrace_buflen == (file_read(nstrace_buf, 1, nstrace_buflen, wth->fh))));\
637                 return FALSE;\
638         }
639
640 nstrace_set_start_time_ver(10)
641 nstrace_set_start_time_ver(20)
642
643 #undef nspr_getv10recordtype
644 #undef nspr_getv20recordtype
645 #undef nspr_getv10recordsize
646
647 /*
648 ** Set the start time of the trace file. We look for the first ABSTIME record. We use that
649 ** to set the start time. Apart from that we also make sure that we remember the position of
650 ** the next record after the ABSTIME record. Inorder to report correct time values, all trace
651 ** records before the ABSTIME record are ignored. 
652 */
653 gboolean nstrace_set_start_time(wtap *wth)
654 {
655         if (wth->file_type == WTAP_FILE_NETSCALER_1_0)
656                 return nstrace_set_start_time_v10(wth);
657         else if (wth->file_type == WTAP_FILE_NETSCALER_2_0)
658                 return nstrace_set_start_time_v20(wth);
659
660         return FALSE;
661 }
662
663 #define __TNO(enumprefix,structprefix,structname,hdrname,structfieldname) \
664         wth->pseudo_header.nstr.hdrname##_offset =  enumprefix##_##hdrname##_offset;
665
666 #define __TNL(enumprefix,structprefix,structname,hdrname,structfieldname) \
667         wth->pseudo_header.nstr.hdrname##_len = enumprefix##_##hdrname##_len;
668
669 #define __TNV1O(enumprefix,structprefix,structname,hdrname,structfieldname) \
670         __TNO(enumprefix,structprefix,structname,hdrname,structfieldname)
671
672 #define __TNV1L(enumprefix,structprefix,structname,hdrname,structfieldname) \
673         __TNL(enumprefix,structprefix,structname,hdrname,structfieldname)
674
675
676
677 /*
678 ** Netscaler trace format read routines.
679 */
680 gboolean nstrace_read_v10(wtap *wth, int *err, gchar **err_info, gint64 *data_offset)
681 {
682         nstrace_t *nstrace = (nstrace_t *)wth->priv;
683         guint64 nsg_creltime = nstrace->nsg_creltime;
684         gchar *nstrace_buf = nstrace->pnstrace_buf;
685         gint32 nstrace_buf_offset = nstrace->nstrace_buf_offset;
686         gint32 nstrace_buflen = nstrace->nstrace_buflen;
687         nspr_pktracefull_v10_t *fp;
688         nspr_pktracepart_v10_t *pp;
689       
690         *err = 0;
691         *err_info = NULL;
692         do
693         {
694                 while ((nstrace_buf_offset < nstrace_buflen) &&
695                         ((nstrace_buflen - nstrace_buf_offset) >= ((gint32)sizeof(fp->nsprRecordType))))
696                 {
697       
698                         fp = (nspr_pktracefull_v10_t *) &nstrace_buf[nstrace_buf_offset];
699                         pp = (nspr_pktracepart_v10_t *) fp;
700       
701                         switch (pletohs(&fp->nsprRecordType))
702                         {
703                         case NSPR_PDPKTRACEFULLTX_V10:
704                         case NSPR_PDPKTRACEFULLTXB_V10:
705                         case NSPR_PDPKTRACEFULLRX_V10:
706       
707                                 nsg_creltime += ns_hrtime2nsec(pletohl(&fp->fp_RelTimeHr));
708                                 wth->phdr.ts.secs = nstrace->nspm_curtime + (guint32) (nsg_creltime / 1000000000); 
709                                 wth->phdr.ts.nsecs = (guint32) (nsg_creltime % 1000000000);
710       
711                                 wth->phdr.len = pletohs(&fp->nsprRecordSize);
712                                 wth->phdr.caplen = wth->phdr.len;
713  
714      
715                                 TRACE_V10_REC_LEN_OFF(v10_full,fp,pktracefull_v10);
716                                 
717                                 buffer_assure_space(wth->frame_buffer, wth->phdr.caplen);
718                                 memcpy(buffer_start_ptr(wth->frame_buffer), fp, wth->phdr.caplen); 
719                                 *data_offset = wth->data_offset + nstrace_buf_offset;
720       
721                                 nstrace->nstrace_buf_offset = nstrace_buf_offset + wth->phdr.len;
722                                 nstrace->nstrace_buflen = nstrace_buflen;
723                                 nstrace->nsg_creltime = nsg_creltime;
724
725                                 return TRUE;
726       
727                         case NSPR_PDPKTRACEPARTTX_V10:
728                         case NSPR_PDPKTRACEPARTTXB_V10:
729                         case NSPR_PDPKTRACEPARTRX_V10:
730       
731                                 nsg_creltime += ns_hrtime2nsec(pletohl(&pp->pp_RelTimeHr));
732                                 wth->phdr.ts.secs = nstrace->nspm_curtime + (guint32) (nsg_creltime / 1000000000); 
733                                 wth->phdr.ts.nsecs = (guint32) (nsg_creltime % 1000000000);
734       
735                                 wth->phdr.len =  pletohs(&pp->pp_PktSizeOrg) + nspr_pktracepart_v10_s;
736                                 wth->phdr.caplen =  pletohs(&pp->nsprRecordSize);
737       
738                                 TRACE_V10_REC_LEN_OFF(v10_part,pp,pktracepart_v10);
739                                       
740                                 buffer_assure_space(wth->frame_buffer, wth->phdr.caplen);
741                                 memcpy(buffer_start_ptr(wth->frame_buffer), pp, wth->phdr.caplen); 
742                                 *data_offset = wth->data_offset + nstrace_buf_offset;
743       
744                                 nstrace->nstrace_buf_offset = nstrace_buf_offset + wth->phdr.caplen;
745                                 nstrace->nsg_creltime = nsg_creltime;
746                                 nstrace->nstrace_buflen = nstrace_buflen;
747
748                                 return TRUE;
749       
750                         case NSPR_ABSTIME_V10:
751
752                                 ns_setabstime(nstrace, pletohl(&((nspr_abstime_v10_t *) fp)->abs_Time), pletohl(&((nspr_abstime_v10_t *) fp)->abs_RelTime));
753                                 nstrace_buf_offset += pletohs(&fp->nsprRecordSize);
754                                 break;
755       
756                         case NSPR_RELTIME_V10:
757
758                                 ns_setrelativetime(nstrace, ((nspr_abstime_v10_t *) fp)->abs_RelTime);
759                                 nstrace_buf_offset += pletohs(&fp->nsprRecordSize);
760                                 break;
761       
762                         case NSPR_UNUSEDSPACE_V10:
763
764                                 nstrace_buf_offset = nstrace_buflen;
765                                 break;
766         
767                         default:
768
769                                 nstrace_buf_offset += pletohs(&fp->nsprRecordSize);
770                                 break;
771                         }
772                 }
773   
774                 nstrace_buf_offset = 0;
775                 wth->data_offset += nstrace_buflen;
776                 nstrace_buflen = GET_READ_PAGE_SIZE((nstrace->file_size - wth->data_offset));
777         }while((nstrace_buflen > 0) && (nstrace_buflen == (file_read(nstrace_buf, 1, nstrace_buflen, wth->fh))));
778   
779         return FALSE;
780 }
781
782 #define TIMEDEFV20(fp,type) \
783         do {\
784                 nsg_creltime += ns_hrtime2nsec(pletohl(&fp->type##_RelTimeHr));\
785                 wth->phdr.ts.secs = nstrace->nspm_curtime + (guint32) (nsg_creltime / 1000000000);\
786                 wth->phdr.ts.nsecs = (guint32) (nsg_creltime % 1000000000);\
787         }while(0)
788
789 #define TIMEDEFV23(fp,type) \
790         do {\
791                 /* access _AbsTimeHighHdr as a 64bit value */\
792                 nsg_creltime = (((guint64)fp->type##_AbsTimeHighHdr<<32) | (fp->type##_AbsTimeLowHdr));\
793                 wth->phdr.ts.secs = (guint32) (nsg_creltime / 1000000000);\
794                 wth->phdr.ts.nsecs = (guint32) (nsg_creltime % 1000000000);\
795         }while(0)
796
797 #define TIMEDEFV21(fp,type) TIMEDEFV20(fp,type)
798 #define TIMEDEFV22(fp,type) TIMEDEFV20(fp,type)
799
800 #define PPSIZEDEFV20(pp,ver) \
801         do {\
802                 wth->phdr.len = pletohs(&pp->pp_PktSizeOrg) + nspr_pktracepart_v##ver##_s;\
803                 wth->phdr.caplen = nspr_getv20recordsize((nspr_hd_v20_t *)pp);\
804         }while(0)
805
806 #define PPSIZEDEFV21(pp,ver) PPSIZEDEFV20(pp,ver)
807 #define PPSIZEDEFV22(pp,ver) PPSIZEDEFV20(pp,ver)
808 #define PPSIZEDEFV23(pp,ver) PPSIZEDEFV20(pp,ver)
809                 
810 #define FPSIZEDEFV20(fp,ver)\
811         do {\
812                 wth->phdr.len = nspr_getv20recordsize((nspr_hd_v20_t *)fp);\
813                 wth->phdr.caplen = wth->phdr.len;\
814         }while(0)
815
816 #define FPSIZEDEFV21(pp,ver) FPSIZEDEFV20(fp,ver)
817 #define FPSIZEDEFV22(pp,ver) FPSIZEDEFV20(fp,ver)
818 #define FPSIZEDEFV23(pp,ver) FPSIZEDEFV20(fp,ver)
819
820 #define PACKET_DESCRIBE(FPTIMEDEF,SIZEDEF,ver,enumprefix,type,structname,TYPE)\
821         do {\
822                 nspr_##structname##_t *fp= (nspr_##structname##_t*)&nstrace_buf[nstrace_buf_offset];\
823                 TIMEDEFV##ver(fp,type);\
824                 SIZEDEF##ver(fp,ver);\
825                 TRACE_V##ver##_REC_LEN_OFF(enumprefix,type,structname);\
826                 buffer_assure_space(wth->frame_buffer, wth->phdr.caplen);\
827                 memcpy(buffer_start_ptr(wth->frame_buffer), fp, wth->phdr.caplen);\
828                 *data_offset = wth->data_offset + nstrace_buf_offset;\
829                 nstrace->nstrace_buf_offset = nstrace_buf_offset + nspr_getv20recordsize((nspr_hd_v20_t *)fp);\
830                 nstrace->nstrace_buflen = nstrace_buflen;\
831                 nstrace->nsg_creltime = nsg_creltime;\
832                 wth->pseudo_header.nstr.rec_type = NSPR_HEADER_VERSION##TYPE;\
833                 return TRUE;\
834         }while(0)
835
836 gboolean nstrace_read_v20(wtap *wth, int *err, gchar **err_info, gint64 *data_offset)
837 {
838         nstrace_t *nstrace = (nstrace_t *)wth->priv;
839         guint64 nsg_creltime = nstrace->nsg_creltime;
840         gchar *nstrace_buf = nstrace->pnstrace_buf;
841         gint32 nstrace_buf_offset = nstrace->nstrace_buf_offset;
842         gint32 nstrace_buflen = nstrace->nstrace_buflen;
843         nspr_pktracefull_v20_t *fp20;
844         nspr_pktracefull_v21_t *fp21;
845                 
846         *err = 0;
847         *err_info = NULL;
848         do
849         {
850                 while ((nstrace_buf_offset < nstrace_buflen) &&
851                         ((nstrace_buflen - nstrace_buf_offset) >= ((gint32)sizeof(fp21->fp_RecordType))))
852                 {
853                         fp21 = (nspr_pktracefull_v21_t *) &nstrace_buf[nstrace_buf_offset];
854     
855                         switch (fp21->fp_RecordType)
856                         {
857
858 #define GENERATE_CASE(type,acttype) \
859                 case NSPR_PDPKTRACEFULLTX_V##type:\
860                 case NSPR_PDPKTRACEFULLTXB_V##type:\
861                 case NSPR_PDPKTRACEFULLRX_V##type:\
862                         PACKET_DESCRIBE(TIMEDEF,FPSIZEDEFV,type,v##type##_full,fp,pktracefull_v##type,acttype);
863                                 GENERATE_CASE(23,203);
864                                 GENERATE_CASE(22,202);
865                                 GENERATE_CASE(21,201);
866                                 GENERATE_CASE(20,200);
867 #undef GENERATE_CASE
868
869 #define GENERATE_CASE(type,acttype) \
870                 case NSPR_PDPKTRACEPARTTX_V##type:\
871                 case NSPR_PDPKTRACEPARTTXB_V##type:\
872                 case NSPR_PDPKTRACEPARTRX_V##type:\
873                         PACKET_DESCRIBE(TIMEDEF,PPSIZEDEFV,type,v##type##_part,pp,pktracepart_v##type,acttype);
874                                 GENERATE_CASE(23,203);
875                                 GENERATE_CASE(22,202);
876                                 GENERATE_CASE(21,201);
877                                 GENERATE_CASE(20,200);
878 #undef GENERATE_CASE
879
880                                 case NSPR_ABSTIME_V20:
881                                 {
882                                         fp20 = (nspr_pktracefull_v20_t *) &nstrace_buf[nstrace_buf_offset];
883                                         nstrace_buf_offset += nspr_getv20recordsize((nspr_hd_v20_t *)fp20);
884                                         ns_setabstime(nstrace, pletohl(&((nspr_abstime_v20_t *) fp20)->abs_Time), pletohs(&((nspr_abstime_v20_t *) fp20)->abs_RelTime));
885                                         break;
886                                 }
887       
888                                 case NSPR_RELTIME_V20:
889                                 {
890                                         fp20 = (nspr_pktracefull_v20_t *) &nstrace_buf[nstrace_buf_offset];
891                                         ns_setrelativetime(nstrace, pletohs(&((nspr_abstime_v20_t *) fp20)->abs_RelTime));
892                                         nstrace_buf_offset += nspr_getv20recordsize((nspr_hd_v20_t *)fp20);
893                                         break;
894                         }
895       
896                                 case NSPR_UNUSEDSPACE_V20:
897                                 {
898                                         if (nstrace_buf_offset >= NSPR_PAGESIZE/2)
899                                                 nstrace_buf_offset = nstrace_buflen;
900                                         else
901                                                 nstrace_buf_offset = NSPR_PAGESIZE/2;
902                                         break;
903                         }
904         
905                                 default:
906                                 {
907                                         fp20 = (nspr_pktracefull_v20_t *) &nstrace_buf[nstrace_buf_offset];
908                                         nstrace_buf_offset += nspr_getv20recordsize((nspr_hd_v20_t *)fp20);
909                                         break;
910                                 }
911                         }
912         }      
913    
914                 nstrace_buf_offset = 0;
915             wth->data_offset += nstrace_buflen;
916         nstrace_buflen = GET_READ_PAGE_SIZE((nstrace->file_size - wth->data_offset));
917         }while((nstrace_buflen > 0) && (nstrace_buflen == (file_read(nstrace_buf, 1, nstrace_buflen, wth->fh))));
918   
919         return FALSE;
920 }
921
922 #undef __TNO
923 #undef __TNL
924 #undef __TNV1O
925 #undef __TNV1L
926
927
928
929 gboolean nstrace_read(wtap *wth, int *err, gchar **err_info, gint64 *data_offset)
930 {
931
932         if (wth->file_type == WTAP_FILE_NETSCALER_1_0)
933                 return nstrace_read_v10(wth, err, err_info, data_offset);
934         else if (wth->file_type == WTAP_FILE_NETSCALER_2_0)
935                 return nstrace_read_v20(wth, err, err_info, data_offset);
936
937         return FALSE;
938 }
939
940
941 #define __TNO(enumprefix,structprefix,structname,hdrname,structfieldname) \
942         pseudo_header->nstr.hdrname##_offset = (guint8) enumprefix##_##hdrname##_offset;
943 #define __TNL(enumprefix,structprefix,structname,hdrname,structfieldname) \
944         pseudo_header->nstr.hdrname##_len = (guint8) enumprefix##_##hdrname##_len;;
945
946 #define __TNV1O(enumprefix,structprefix,structname,hdrname,structfieldname) \
947         __TNO(enumprefix,structprefix,structname,hdrname,structfieldname)
948 #define __TNV1L(enumprefix,structprefix,structname,hdrname,structfieldname) \
949         __TNL(enumprefix,structprefix,structname,hdrname,structfieldname)
950
951
952 gboolean nstrace_seek_read(wtap *wth, gint64 seek_off,
953     union wtap_pseudo_header *pseudo_header, guchar *pd, int length,
954     int *err, gchar **err_info _U_)
955 {
956         *err = 0;
957
958         if (file_seek(wth->random_fh, seek_off, SEEK_SET, err) == -1)
959                 return FALSE;
960
961         /*
962         ** Read the packet data.
963         */
964         if ((file_read(pd, 1, length, wth->random_fh)) != length)
965                 return FALSE;
966
967         if (wth->file_type == WTAP_FILE_NETSCALER_1_0)
968         {
969
970 #define GENERATE_CASE_FULL(type,acttype) \
971                 case NSPR_PDPKTRACEFULLTX_V##type:\
972                 case NSPR_PDPKTRACEFULLTXB_V##type:\
973                 case NSPR_PDPKTRACEFULLRX_V##type:\
974                         TRACE_V##type##_REC_LEN_OFF(v##type##_full,fp,pktracefull_v##type);\
975                         pseudo_header->nstr.rec_type = NSPR_HEADER_VERSION##acttype;\
976                         break;\
977
978 #define GENERATE_CASE_PART(type,acttype) \
979                 case NSPR_PDPKTRACEPARTTX_V##type:\
980                 case NSPR_PDPKTRACEPARTTXB_V##type:\
981                 case NSPR_PDPKTRACEPARTRX_V##type:\
982                         TRACE_V##type##_REC_LEN_OFF(v##type##_part,pp,pktracepart_v##type);\
983                         pseudo_header->nstr.rec_type = NSPR_HEADER_VERSION##acttype;\
984                         break;\
985
986                 switch (pletohs(&(( nspr_header_v10_t*)pd)->ph_RecordType))
987                 {
988                         GENERATE_CASE_FULL(10,100)
989                         GENERATE_CASE_PART(10,100)
990                 }
991         } else if (wth->file_type == WTAP_FILE_NETSCALER_2_0)
992         {
993                 switch ((( nspr_hd_v20_t*)pd)->phd_RecordType)
994                 {
995                         GENERATE_CASE_FULL(20,200)
996                         GENERATE_CASE_PART(20,200)
997                         GENERATE_CASE_FULL(21,201)
998                         GENERATE_CASE_PART(21,201)
999                         GENERATE_CASE_FULL(22,202)
1000                         GENERATE_CASE_PART(22,202)
1001                         GENERATE_CASE_FULL(23,203)
1002                         GENERATE_CASE_PART(23,203)
1003                 }
1004         }
1005
1006         return TRUE;
1007 }
1008
1009 #undef __TNL
1010 #undef __TNO
1011 #undef __TNV1L
1012 #undef __TNV1O
1013
1014
1015 /*
1016 ** Netscaler trace format close routines.
1017 */
1018 void nstrace_close(wtap *wth)
1019 {
1020         nstrace_t *nstrace = (nstrace_t *)wth->priv;
1021
1022         g_free(nstrace->pnstrace_buf);
1023 }
1024
1025
1026 typedef struct {
1027         guint16 page_offset;
1028         guint16 page_len;
1029         guint32 absrec_time;
1030 } nstrace_dump_t;
1031
1032 /* Returns 0 if we could write the specified encapsulation type,
1033 ** an error indication otherwise. */
1034 int nstrace_10_dump_can_write_encap(int encap)
1035 {
1036         if (encap == WTAP_ENCAP_NSTRACE_1_0)
1037                 return 0;
1038
1039         return WTAP_ERR_UNSUPPORTED_ENCAP;
1040 }
1041
1042
1043 /* Returns 0 if we could write the specified encapsulation type,
1044 ** an error indication otherwise. */
1045 int nstrace_20_dump_can_write_encap(int encap)
1046 {
1047         if (encap == WTAP_ENCAP_NSTRACE_2_0)
1048                 return 0;
1049
1050         return WTAP_ERR_UNSUPPORTED_ENCAP;
1051 }
1052
1053
1054 /* Returns TRUE on success, FALSE on failure; sets "*err" to an error code on
1055 ** failure */
1056 gboolean nstrace_dump_open(wtap_dumper *wdh, gboolean cant_seek, int *err)
1057 {
1058         nstrace_dump_t *nstrace;
1059
1060         if (cant_seek) 
1061         {
1062                 *err = WTAP_ERR_CANT_WRITE_TO_PIPE;
1063                 return FALSE;
1064         }
1065
1066         wdh->subtype_write = nstrace_dump;
1067
1068         nstrace = (nstrace_dump_t *)g_malloc(sizeof(nstrace_dump_t));
1069         wdh->priv = (void *)nstrace;
1070         nstrace->page_offset = 0;
1071         nstrace->page_len = NSPR_PAGESIZE;
1072         nstrace->absrec_time = 0;
1073
1074         return TRUE;
1075 }
1076
1077
1078 gboolean nstrace_add_signature(wtap_dumper *wdh)
1079 {
1080         nstrace_dump_t *nstrace = (nstrace_dump_t *)wdh->priv;
1081         size_t nwritten;
1082
1083         if (wdh->file_type == WTAP_FILE_NETSCALER_1_0)
1084         {
1085                 nspr_signature_v10_t sig10;
1086
1087                 /* populate the record */
1088                 sig10.phd.ph_RecordType = htoles(NSPR_SIGNATURE_V10);
1089                 sig10.phd.ph_RecordSize = htoles(nspr_signature_v10_s);
1090                 memcpy(sig10.sig_Signature, NSPR_SIGSTR_V10, NSPR_SIGSIZE_V10);
1091
1092                 /* Write the record into the file */
1093                 nwritten = fwrite(&sig10, 1, nspr_signature_v10_s, wdh->fh);
1094                 if (nwritten != nspr_signature_v10_s)
1095                         return FALSE;    
1096
1097                 /* Move forward the page offset */
1098                 nstrace->page_offset += (guint16) nwritten;
1099         
1100         } else if (wdh->file_type == WTAP_FILE_NETSCALER_2_0)
1101         {
1102                 gchar sig[nspr_signature_v20_s + sizeof(NSPR_SIGSTR_V20)];
1103                 nspr_signature_v20_t *sig20;
1104
1105                 sig20 = (nspr_signature_v20_t *)&sig;
1106                 sig20->sig_RecordType = NSPR_SIGNATURE_V20;
1107                 sig20->sig_RecordSize = nspr_signature_v20_s + sizeof(NSPR_SIGSTR_V20);
1108                 memcpy(sig20->sig_Signature, NSPR_SIGSTR_V20, sizeof(NSPR_SIGSTR_V20));
1109
1110                 /* Write the record into the file */
1111                 nwritten = fwrite(sig20, 1, sig20->sig_RecordSize, wdh->fh);
1112                 if (nwritten != sig20->sig_RecordSize) 
1113                         return FALSE;    
1114
1115                 /* Move forward the page offset */
1116                 nstrace->page_offset += (guint16) nwritten;
1117         
1118         } else  
1119         {
1120                 g_assert_not_reached();
1121                 return FALSE;
1122         }
1123
1124         return TRUE;
1125 }
1126
1127
1128 gboolean nstrace_add_abstime(wtap_dumper *wdh, const struct wtap_pkthdr *phdr,
1129      const guchar *pd)
1130 {
1131         nstrace_dump_t *nstrace = (nstrace_dump_t *)wdh->priv;
1132         size_t nwritten;
1133         guint64 nsg_creltime;
1134
1135         if (wdh->file_type == WTAP_FILE_NETSCALER_1_0)
1136         {
1137                 nspr_abstime_v10_t abs10;
1138                 nspr_pktracefull_v10_t fp10;
1139
1140                 /* populate the record */
1141                 abs10.phd.ph_RecordType = htoles(NSPR_ABSTIME_V10);
1142                 abs10.phd.ph_RecordSize = htoles(nspr_abstime_v10_s);
1143
1144                 memcpy(&fp10, pd, nspr_pktracefull_v10_s);
1145                 nsg_creltime = ns_hrtime2nsec(fp10.fp_RelTimeHr);
1146                 
1147                 abs10.abs_RelTime = 0;
1148                 abs10.abs_Time = htolel((guint32)phdr->ts.secs - (guint32)(nsg_creltime/1000000000));
1149
1150                 /* Write the record into the file */
1151                 nwritten = fwrite(&abs10, 1, nspr_abstime_v10_s, wdh->fh);
1152                 if (nwritten != nspr_abstime_v10_s) 
1153                         return FALSE;
1154
1155                 /* Move forward the page offset */
1156                 nstrace->page_offset += nspr_abstime_v10_s;
1157
1158         } else if (wdh->file_type == WTAP_FILE_NETSCALER_2_0)
1159         {
1160                 nspr_abstime_v20_t abs20;
1161                 nspr_pktracefull_v20_t fp20;
1162
1163                 abs20.abs_RecordType = NSPR_ABSTIME_V20;
1164                 abs20.abs_RecordSize = nspr_abstime_v20_s;
1165
1166                 memcpy(&fp20, pd, nspr_pktracefull_v20_s);
1167                 nsg_creltime = ns_hrtime2nsec(fp20.fp_RelTimeHr);
1168
1169                 abs20.abs_RelTime = 0;
1170                 abs20.abs_Time = htolel((guint32)phdr->ts.secs - (guint32)(nsg_creltime/1000000000));
1171
1172                 /* Write the record into the file */
1173                 nwritten = fwrite(&abs20, 1, nspr_abstime_v20_s, wdh->fh);
1174                 if (nwritten != nspr_abstime_v20_s) 
1175                         return FALSE;
1176
1177                 /* Move forward the page offset */
1178                 nstrace->page_offset += nspr_abstime_v20_s;
1179
1180         } else
1181         {
1182                 g_assert_not_reached();
1183                 return FALSE;
1184         }
1185
1186         return TRUE;
1187 }
1188
1189
1190 /* Write a record for a packet to a dump file.
1191    Returns TRUE on success, FALSE on failure. */
1192 static gboolean nstrace_dump(wtap_dumper *wdh, const struct wtap_pkthdr *phdr,
1193     const union wtap_pseudo_header *pseudo_header, const guchar *pd, int *err)
1194 {  
1195         nstrace_dump_t *nstrace = (nstrace_dump_t *)wdh->priv;
1196         size_t nwritten;
1197
1198         if (nstrace->page_offset == 0)
1199         {
1200                 /* Add the signature record and abs time record */
1201                 if (wdh->file_type == WTAP_FILE_NETSCALER_1_0)
1202                 {
1203                         if ((nstrace_add_signature(wdh) == FALSE) || (nstrace_add_abstime(wdh, phdr, pd) == FALSE))
1204                                 return FALSE;
1205                 } else if (wdh->file_type == WTAP_FILE_NETSCALER_2_0)
1206                 {
1207                         if ((nstrace_add_signature(wdh) == FALSE) || (nstrace_add_abstime(wdh, phdr, pd) == FALSE))
1208                                 return FALSE;
1209                 } else
1210                 {
1211                         g_assert_not_reached();
1212                         return FALSE;
1213                 }
1214         }
1215
1216         switch (pseudo_header->nstr.rec_type)
1217         {
1218         case NSPR_HEADER_VERSION100:
1219
1220                 if (wdh->file_type == WTAP_FILE_NETSCALER_1_0)
1221                 {
1222                         if (nstrace->page_offset + phdr->caplen >= nstrace->page_len)
1223                         {
1224                                 /* Start on the next page */
1225                                 if (fseek(wdh->fh, (nstrace->page_len - nstrace->page_offset), SEEK_CUR) == -1)
1226                                 {
1227                                         *err = errno;
1228                                         return FALSE;
1229                                 }
1230
1231                                 nstrace->page_offset = 0;
1232
1233                                 /* Possibly add signature and abstime records and increment offset */
1234                                 if (nstrace_add_signature(wdh) == FALSE)
1235                                         return FALSE;
1236                         }
1237
1238                         /* Write the actual record as is */
1239                         nwritten = fwrite(pd, 1, phdr->caplen, wdh->fh);
1240                         if (nwritten != phdr->caplen)
1241                         {
1242                                 if (nwritten == 0 && ferror(wdh->fh))
1243                                         *err = errno;
1244                                 else
1245                                         *err = WTAP_ERR_SHORT_WRITE;
1246
1247                                 return FALSE;
1248                         }
1249
1250                         nstrace->page_offset += (guint16) nwritten;
1251                 } else if (wdh->file_type == WTAP_FILE_NETSCALER_2_0)
1252                 {
1253                         *err = WTAP_ERR_UNSUPPORTED_FILE_TYPE;
1254                         return FALSE;    
1255                 }
1256
1257                 break;
1258
1259         case NSPR_HEADER_VERSION200:
1260         case NSPR_HEADER_VERSION201:
1261         case NSPR_HEADER_VERSION202:
1262         case NSPR_HEADER_VERSION203:
1263
1264                 if (wdh->file_type == WTAP_FILE_NETSCALER_1_0)
1265                 {
1266                         *err = WTAP_ERR_UNSUPPORTED_FILE_TYPE;
1267                         return FALSE;
1268                 } else if (wdh->file_type == WTAP_FILE_NETSCALER_2_0)
1269                 {
1270                         if (nstrace->page_offset + phdr->caplen >= nstrace->page_len)
1271                         {
1272                                 /* Start on the next page */
1273                                 if (fseek(wdh->fh, (nstrace->page_len - nstrace->page_offset), SEEK_CUR) == -1)
1274                                 {
1275                                         *err = errno;
1276                                         return FALSE;
1277                                 }
1278
1279                                 nstrace->page_offset = 0;
1280
1281                                 /* Possibly add signature and abstime records and increment offset */
1282                                 if (nstrace_add_signature(wdh) == FALSE)
1283                                 return FALSE;
1284                         }
1285
1286                         /* Write the actual record as is */
1287                         nwritten = fwrite(pd, 1, phdr->caplen, wdh->fh);
1288
1289                         if (nwritten != phdr->caplen)
1290                         {
1291                                 if (nwritten == 0 && ferror(wdh->fh))
1292                                         *err = errno;
1293                                 else
1294                                         *err = WTAP_ERR_SHORT_WRITE;
1295
1296                                 return FALSE;
1297                         }
1298
1299                         nstrace->page_offset += (guint16) nwritten;
1300                 }
1301
1302                 break;
1303
1304         default:
1305                 g_assert_not_reached();
1306                 return FALSE;
1307         } 
1308
1309         return TRUE;
1310 }
1311