darwin-x86-and-leopard-20060309
[openafs.git] / src / xstat / xstat_fs_test.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  * 
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 /*
11  * Description:
12  *      Test of the xstat_fs module.
13  *
14  *------------------------------------------------------------------------*/
15
16 #include <afsconfig.h>
17 #include <afs/param.h>
18
19 RCSID
20     ("$Header$");
21
22 #include "xstat_fs.h"           /*Interface for xstat_fs module */
23 #include <cmd.h>                /*Command line interpreter */
24 #include <time.h>
25 #ifdef HAVE_STRING_H
26 #include <string.h>
27 #else
28 #ifdef HAVE_STRINGS_H
29 #include <strings.h>
30 #endif
31 #endif
32
33 /*
34  * External routines that don't have explicit include file definitions.
35  */
36 extern struct hostent *hostutil_GetHostByName();
37
38 /*
39  * Command line parameter indices.
40  *      P_FS_NAMES : List of FileServer names.
41  *      P_COLL_IDS : List of collection IDs to pick up.
42  *      P_ONESHOT  : Are we gathering exactly one round of data?
43  *      P_DEBUG    : Enable debugging output?
44  */
45 #define P_FS_NAMES      0
46 #define P_COLL_IDS      1
47 #define P_ONESHOT       2
48 #define P_FREQUENCY     3
49 #define P_PERIOD        4
50 #define P_DEBUG         5
51
52 /*
53  * Private globals.
54  */
55 static int debugging_on = 0;    /*Are we debugging? */
56 static int one_shot = 0;        /*Single round of data collection? */
57
58 static char *opNames[] = {
59     "FetchData",
60     "FetchACL",
61     "FetchStatus",
62     "StoreData",
63     "StoreACL",
64     "StoreStatus",
65     "RemoveFile",
66     "CreateFile",
67     "Rename",
68     "Symlink",
69     "Link",
70     "MakeDir",
71     "RemoveDir",
72     "SetLock",
73     "ExtendLock",
74     "ReleaseLock",
75     "GetStatistics",
76     "GiveUpCallbacks",
77     "GetVolumeInfo",
78     "GetVolumeStatus",
79     "SetVolumeStatus",
80     "GetRootVolume",
81     "CheckToken",
82     "GetTime",
83     "NGetVolumeInfo",
84     "BulkStatus",
85     "XStatsVersion",
86     "GetXStats",
87     "GetCapabilities"
88 };
89
90 static char *xferOpNames[] = {
91     "FetchData",
92     "StoreData"
93 };
94
95
96 /*------------------------------------------------------------------------
97  * PrintCallInfo
98  *
99  * Description:
100  *      Print out the AFS_XSTATSCOLL_CALL_INFO collection we just
101  *      received.
102  *
103  * Arguments:
104  *      None.
105  *
106  * Returns:
107  *      Nothing.
108  *
109  * Environment:
110  *      All the info we need is nestled into xstat_fs_Results.
111  *
112  * Side Effects:
113  *      As advertised.
114  *------------------------------------------------------------------------*/
115
116 void
117 PrintCallInfo()
118 {                               /*PrintCallInfo */
119
120     static char rn[] = "PrintCallInfo"; /*Routine name */
121     register int i;             /*Loop variable */
122     int numInt32s;              /*# int32words returned */
123     afs_int32 *currInt32;       /*Ptr to current afs_int32 value */
124     char *printableTime;        /*Ptr to printable time string */
125
126     /*
127      * Just print out the results of the particular probe.
128      */
129     numInt32s = xstat_fs_Results.data.AFS_CollData_len;
130     currInt32 = (afs_int32 *) (xstat_fs_Results.data.AFS_CollData_val);
131     printableTime = ctime((time_t *) & (xstat_fs_Results.probeTime));
132     printableTime[strlen(printableTime) - 1] = '\0';
133
134     printf("AFS_XSTATSCOLL_CALL_INFO (coll %d) for FS %s\n[Probe %d, %s]\n\n",
135            xstat_fs_Results.collectionNumber,
136            xstat_fs_Results.connP->hostName, xstat_fs_Results.probeNum,
137            printableTime);
138
139     if (debugging_on)
140         printf("\n[%d entries returned at 0x%x]\n\n", numInt32s, currInt32);
141
142     for (i = 0; i < numInt32s; i++)
143         printf("%d ", *currInt32++);
144     fprintf(stderr, "\n");
145
146 }                               /*PrintCallInfo */
147
148
149 /*------------------------------------------------------------------------
150  * PrintOverallPerfInfo
151  *
152  * Description:
153  *      Print out overall performance numbers.
154  *
155  * Arguments:
156  *      a_ovP : Ptr to the overall performance numbers.
157  *
158  * Returns:
159  *      Nothing.
160  *
161  * Environment:
162  *      Nothing interesting.
163  *
164  * Side Effects:
165  *      As advertised.
166  *------------------------------------------------------------------------*/
167
168 void
169 PrintOverallPerfInfo(struct afs_PerfStats *a_ovP)
170 {
171     printf("\t%10d numPerfCalls\n\n", a_ovP->numPerfCalls);
172
173     /*
174      * Vnode cache section.
175      */
176     printf("\t%10d vcache_L_Entries\n", a_ovP->vcache_L_Entries);
177     printf("\t%10d vcache_L_Allocs\n", a_ovP->vcache_L_Allocs);
178     printf("\t%10d vcache_L_Gets\n", a_ovP->vcache_L_Gets);
179     printf("\t%10d vcache_L_Reads\n", a_ovP->vcache_L_Reads);
180     printf("\t%10d vcache_L_Writes\n\n", a_ovP->vcache_L_Writes);
181
182     printf("\t%10d vcache_S_Entries\n", a_ovP->vcache_S_Entries);
183     printf("\t%10d vcache_S_Allocs\n", a_ovP->vcache_S_Allocs);
184     printf("\t%10d vcache_S_Gets\n", a_ovP->vcache_S_Gets);
185     printf("\t%10d vcache_S_Reads\n", a_ovP->vcache_S_Reads);
186     printf("\t%10d vcache_S_Writes\n\n", a_ovP->vcache_S_Writes);
187
188     printf("\t%10d vcache_H_Entries\n", a_ovP->vcache_H_Entries);
189     printf("\t%10d vcache_H_Gets\n", a_ovP->vcache_H_Gets);
190     printf("\t%10d vcache_H_Replacements\n\n", a_ovP->vcache_H_Replacements);
191
192     /*
193      * Directory package section.
194      */
195     printf("\t%10d dir_Buffers\n", a_ovP->dir_Buffers);
196     printf("\t%10d dir_Calls\n", a_ovP->dir_Calls);
197     printf("\t%10d dir_IOs\n\n", a_ovP->dir_IOs);
198
199     /*
200      * Rx section.
201      */
202     printf("\t%10d rx_packetRequests\n", a_ovP->rx_packetRequests);
203     printf("\t%10d rx_noPackets_RcvClass\n", a_ovP->rx_noPackets_RcvClass);
204     printf("\t%10d rx_noPackets_SendClass\n", a_ovP->rx_noPackets_SendClass);
205     printf("\t%10d rx_noPackets_SpecialClass\n",
206            a_ovP->rx_noPackets_SpecialClass);
207     printf("\t%10d rx_socketGreedy\n", a_ovP->rx_socketGreedy);
208     printf("\t%10d rx_bogusPacketOnRead\n", a_ovP->rx_bogusPacketOnRead);
209     printf("\t%10d rx_bogusHost\n", a_ovP->rx_bogusHost);
210     printf("\t%10d rx_noPacketOnRead\n", a_ovP->rx_noPacketOnRead);
211     printf("\t%10d rx_noPacketBuffersOnRead\n",
212            a_ovP->rx_noPacketBuffersOnRead);
213     printf("\t%10d rx_selects\n", a_ovP->rx_selects);
214     printf("\t%10d rx_sendSelects\n", a_ovP->rx_sendSelects);
215     printf("\t%10d rx_packetsRead_RcvClass\n",
216            a_ovP->rx_packetsRead_RcvClass);
217     printf("\t%10d rx_packetsRead_SendClass\n",
218            a_ovP->rx_packetsRead_SendClass);
219     printf("\t%10d rx_packetsRead_SpecialClass\n",
220            a_ovP->rx_packetsRead_SpecialClass);
221     printf("\t%10d rx_dataPacketsRead\n", a_ovP->rx_dataPacketsRead);
222     printf("\t%10d rx_ackPacketsRead\n", a_ovP->rx_ackPacketsRead);
223     printf("\t%10d rx_dupPacketsRead\n", a_ovP->rx_dupPacketsRead);
224     printf("\t%10d rx_spuriousPacketsRead\n", a_ovP->rx_spuriousPacketsRead);
225     printf("\t%10d rx_packetsSent_RcvClass\n",
226            a_ovP->rx_packetsSent_RcvClass);
227     printf("\t%10d rx_packetsSent_SendClass\n",
228            a_ovP->rx_packetsSent_SendClass);
229     printf("\t%10d rx_packetsSent_SpecialClass\n",
230            a_ovP->rx_packetsSent_SpecialClass);
231     printf("\t%10d rx_ackPacketsSent\n", a_ovP->rx_ackPacketsSent);
232     printf("\t%10d rx_pingPacketsSent\n", a_ovP->rx_pingPacketsSent);
233     printf("\t%10d rx_abortPacketsSent\n", a_ovP->rx_abortPacketsSent);
234     printf("\t%10d rx_busyPacketsSent\n", a_ovP->rx_busyPacketsSent);
235     printf("\t%10d rx_dataPacketsSent\n", a_ovP->rx_dataPacketsSent);
236     printf("\t%10d rx_dataPacketsReSent\n", a_ovP->rx_dataPacketsReSent);
237     printf("\t%10d rx_dataPacketsPushed\n", a_ovP->rx_dataPacketsPushed);
238     printf("\t%10d rx_ignoreAckedPacket\n", a_ovP->rx_ignoreAckedPacket);
239     printf("\t%10d rx_totalRtt_Sec\n", a_ovP->rx_totalRtt_Sec);
240     printf("\t%10d rx_totalRtt_Usec\n", a_ovP->rx_totalRtt_Usec);
241     printf("\t%10d rx_minRtt_Sec\n", a_ovP->rx_minRtt_Sec);
242     printf("\t%10d rx_minRtt_Usec\n", a_ovP->rx_minRtt_Usec);
243     printf("\t%10d rx_maxRtt_Sec\n", a_ovP->rx_maxRtt_Sec);
244     printf("\t%10d rx_maxRtt_Usec\n", a_ovP->rx_maxRtt_Usec);
245     printf("\t%10d rx_nRttSamples\n", a_ovP->rx_nRttSamples);
246     printf("\t%10d rx_nServerConns\n", a_ovP->rx_nServerConns);
247     printf("\t%10d rx_nClientConns\n", a_ovP->rx_nClientConns);
248     printf("\t%10d rx_nPeerStructs\n", a_ovP->rx_nPeerStructs);
249     printf("\t%10d rx_nCallStructs\n", a_ovP->rx_nCallStructs);
250     printf("\t%10d rx_nFreeCallStructs\n", a_ovP->rx_nFreeCallStructs);
251     printf("\t%10d rx_nBusies\n\n", a_ovP->rx_nBusies);
252
253     printf("\t%10d fs_nBusies\n\n", a_ovP->fs_nBusies);
254     /*
255      * Host module fields.
256      */
257     printf("\t%10d host_NumHostEntries\n", a_ovP->host_NumHostEntries);
258     printf("\t%10d host_HostBlocks\n", a_ovP->host_HostBlocks);
259     printf("\t%10d host_NonDeletedHosts\n", a_ovP->host_NonDeletedHosts);
260     printf("\t%10d host_HostsInSameNetOrSubnet\n",
261            a_ovP->host_HostsInSameNetOrSubnet);
262     printf("\t%10d host_HostsInDiffSubnet\n", a_ovP->host_HostsInDiffSubnet);
263     printf("\t%10d host_HostsInDiffNetwork\n",
264            a_ovP->host_HostsInDiffNetwork);
265     printf("\t%10d host_NumClients\n", a_ovP->host_NumClients);
266     printf("\t%10d host_ClientBlocks\n\n", a_ovP->host_ClientBlocks);
267
268     printf("\t%10d sysname_ID\n", a_ovP->sysname_ID);
269 }
270
271
272 /*------------------------------------------------------------------------
273  * PrintOpTiming
274  *
275  * Description:
276  *      Print out the contents of an RPC op timing structure.
277  *
278  * Arguments:
279  *      a_opIdx   : Index of the AFS operation we're printing number on.
280  *      a_opTimeP : Ptr to the op timing structure to print.
281  *
282  * Returns:
283  *      Nothing.
284  *
285  * Environment:
286  *      Nothing interesting.
287  *
288  * Side Effects:
289  *      As advertised.
290  *------------------------------------------------------------------------*/
291
292 void
293 PrintOpTiming(int a_opIdx, struct fs_stats_opTimingData *a_opTimeP)
294 {
295     double fSumTime, avg;
296
297     fSumTime =
298         ((double)(a_opTimeP->sumTime.tv_sec)) +
299         (((double)(a_opTimeP->sumTime.tv_usec)) / ((double)(1000000)));
300 /*    printf("Double sum time is %f\n", fSumTime);*/
301     avg = fSumTime / ((double)(a_opTimeP->numSuccesses));
302
303     printf
304         ("%15s: %d ops (%d OK); sum=%d.%06d, sqr=%d.%06d, min=%d.%06d, max=%d.%06d\n",
305          opNames[a_opIdx], a_opTimeP->numOps, a_opTimeP->numSuccesses,
306          a_opTimeP->sumTime.tv_sec, a_opTimeP->sumTime.tv_usec,
307          a_opTimeP->sqrTime.tv_sec, a_opTimeP->sqrTime.tv_usec,
308          a_opTimeP->minTime.tv_sec, a_opTimeP->minTime.tv_usec,
309          a_opTimeP->maxTime.tv_sec, a_opTimeP->maxTime.tv_usec);
310 }
311
312
313 /*------------------------------------------------------------------------
314  * PrintXferTiming
315  *
316  * Description:
317  *      Print out the contents of a data transfer structure.
318  *
319  * Arguments:
320  *      a_opIdx : Index of the AFS operation we're printing number on.
321  *      a_xferP : Ptr to the data transfer structure to print.
322  *
323  * Returns:
324  *      Nothing.
325  *
326  * Environment:
327  *      Nothing interesting.
328  *
329  * Side Effects:
330  *      As advertised.
331  *------------------------------------------------------------------------*/
332
333 void
334 PrintXferTiming(int a_opIdx, struct fs_stats_xferData *a_xferP)
335 {
336     double fSumTime, avg;
337
338     fSumTime =
339         ((double)(a_xferP->sumTime.tv_sec)) +
340         ((double)(a_xferP->sumTime.tv_usec)) / ((double)(1000000));
341
342     avg = fSumTime / ((double)(a_xferP->numSuccesses));
343
344     printf
345         ("%s: %d xfers (%d OK), time sum=%d.%06d, sqr=%d.%06d, min=%d.%06d, max=%d.%06d\n",
346          xferOpNames[a_opIdx], a_xferP->numXfers, a_xferP->numSuccesses,
347          a_xferP->sumTime.tv_sec, a_xferP->sumTime.tv_usec,
348          a_xferP->sqrTime.tv_sec, a_xferP->sqrTime.tv_usec,
349          a_xferP->minTime.tv_sec, a_xferP->minTime.tv_usec,
350          a_xferP->maxTime.tv_sec, a_xferP->maxTime.tv_usec);
351     printf("\t[bytes: sum=%lu, min=%d, max=%d]\n", a_xferP->sumBytes,
352            a_xferP->minBytes, a_xferP->maxBytes);
353     printf
354         ("\t[buckets: 0: %d, 1: %d, 2: %d, 3: %d, 4: %d, 5: %d, 6: %d, 7: %d, 8: %d]\n",
355          a_xferP->count[0], a_xferP->count[1], a_xferP->count[2],
356          a_xferP->count[3], a_xferP->count[4], a_xferP->count[5],
357          a_xferP->count[6], a_xferP->count[7], a_xferP->count[8]);
358 }
359
360
361 /*------------------------------------------------------------------------
362  * PrintDetailedPerfInfo
363  *
364  * Description:
365  *      Print out a set of detailed performance numbers.
366  *
367  * Arguments:
368  *      a_detP : Ptr to detailed perf numbers to print.
369  *
370  * Returns:
371  *      Nothing.
372  *
373  * Environment:
374  *      Nothing interesting.
375  *
376  * Side Effects:
377  *      As advertised.
378  *------------------------------------------------------------------------*/
379
380 void
381 PrintDetailedPerfInfo(struct fs_stats_DetailedStats *a_detP)
382 {
383     int currIdx;                /*Loop variable */
384
385     printf("\t%10d epoch\n", a_detP->epoch);
386
387     for (currIdx = 0; currIdx < FS_STATS_NUM_RPC_OPS; currIdx++)
388         PrintOpTiming(currIdx, &(a_detP->rpcOpTimes[currIdx]));
389
390     for (currIdx = 0; currIdx < FS_STATS_NUM_XFER_OPS; currIdx++)
391         PrintXferTiming(currIdx, &(a_detP->xferOpTimes[currIdx]));
392 }
393
394
395 /*------------------------------------------------------------------------
396  * PrintFullPerfInfo
397  *
398  * Description:
399  *      Print out the AFS_XSTATSCOLL_FULL_PERF_INFO collection we just
400  *      received.
401  *
402  * Arguments:
403  *      None.
404  *
405  * Returns:
406  *      Nothing.
407  *
408  * Environment:
409  *      All the info we need is nestled into xstat_fs_Results.
410  *
411  * Side Effects:
412  *      As advertised.
413  *------------------------------------------------------------------------*/
414
415 void
416 PrintFullPerfInfo()
417 {
418
419     static char rn[] = "PrintFullPerfInfo";     /*Routine name */
420     static afs_int32 fullPerfInt32s = (sizeof(struct fs_stats_FullPerfStats) >> 2);     /*Correct # int32s to rcv */
421     afs_int32 numInt32s;        /*# int32words received */
422     struct fs_stats_FullPerfStats *fullPerfP;   /*Ptr to full perf stats */
423     char *printableTime;        /*Ptr to printable time
424                                  * string */
425
426     numInt32s = xstat_fs_Results.data.AFS_CollData_len;
427     if (numInt32s != fullPerfInt32s) {
428         printf("** Data size mismatch in full performance collection!");
429         printf("** Expecting %d, got %d\n", fullPerfInt32s, numInt32s);
430         return;
431     }
432
433     printableTime = ctime((time_t *) & (xstat_fs_Results.probeTime));
434     printableTime[strlen(printableTime) - 1] = '\0';
435     fullPerfP = (struct fs_stats_FullPerfStats *)
436         (xstat_fs_Results.data.AFS_CollData_val);
437
438     printf
439         ("AFS_XSTATSCOLL_FULL_PERF_INFO (coll %d) for FS %s\n[Probe %d, %s]\n\n",
440          xstat_fs_Results.collectionNumber, xstat_fs_Results.connP->hostName,
441          xstat_fs_Results.probeNum, printableTime);
442
443     PrintOverallPerfInfo(&(fullPerfP->overall));
444     PrintDetailedPerfInfo(&(fullPerfP->det));
445 }
446
447
448 /*------------------------------------------------------------------------
449  * PrintPerfInfo
450  *
451  * Description:
452  *      Print out the AFS_XSTATSCOLL_PERF_INFO collection we just
453  *      received.
454  *
455  * Arguments:
456  *      None.
457  *
458  * Returns:
459  *      Nothing.
460  *
461  * Environment:
462  *      All the info we need is nestled into xstat_fs_Results.
463  *
464  * Side Effects:
465  *      As advertised.
466  *------------------------------------------------------------------------*/
467
468 void
469 PrintPerfInfo()
470 {
471     static char rn[] = "PrintPerfInfo"; /*Routine name */
472     static afs_int32 perfInt32s = (sizeof(struct afs_PerfStats) >> 2);  /*Correct # int32s to rcv */
473     afs_int32 numInt32s;        /*# int32words received */
474     struct afs_PerfStats *perfP;        /*Ptr to performance stats */
475     char *printableTime;        /*Ptr to printable time string */
476
477     numInt32s = xstat_fs_Results.data.AFS_CollData_len;
478     if (numInt32s != perfInt32s) {
479         printf("** Data size mismatch in performance collection!");
480         printf("** Expecting %d, got %d\n", perfInt32s, numInt32s);
481         return;
482     }
483
484     printableTime = ctime((time_t *) & (xstat_fs_Results.probeTime));
485     printableTime[strlen(printableTime) - 1] = '\0';
486     perfP = (struct afs_PerfStats *)
487         (xstat_fs_Results.data.AFS_CollData_val);
488
489     printf("AFS_XSTATSCOLL_PERF_INFO (coll %d) for FS %s\n[Probe %d, %s]\n\n",
490            xstat_fs_Results.collectionNumber,
491            xstat_fs_Results.connP->hostName, xstat_fs_Results.probeNum,
492            printableTime);
493
494     PrintOverallPerfInfo(perfP);
495 }
496
497
498 /*------------------------------------------------------------------------
499  * FS_Handler
500  *
501  * Description:
502  *      Handler routine passed to the xstat_fs module.  This handler is
503  *      called immediately after a poll of one of the File Servers has
504  *      taken place.  All it needs to know is exported by the xstat_fs
505  *      module, namely the data structure where the probe results are
506  *      stored.
507  *
508  * Arguments:
509  *      None.
510  *
511  * Returns:
512  *      0 on success,
513  *      -1 otherwise.
514  *
515  * Environment:
516  *      See above.  All we do now is print out what we got.
517  *
518  * Side Effects:
519  *      As advertised.
520  *------------------------------------------------------------------------*/
521
522 int
523 FS_Handler()
524 {
525     static char rn[] = "FS_Handler";    /*Routine name */
526
527     printf
528         ("\n------------------------------------------------------------\n");
529
530     /*
531      * If the probe failed, there isn't much we can do except gripe.
532      */
533     if (xstat_fs_Results.probeOK) {
534         printf("%s: Probe %d to File Server '%s' failed, code=%d\n", rn,
535                xstat_fs_Results.probeNum, xstat_fs_Results.connP->hostName,
536                xstat_fs_Results.probeOK);
537         return (0);
538     }
539
540     switch (xstat_fs_Results.collectionNumber) {
541     case AFS_XSTATSCOLL_CALL_INFO:
542         PrintCallInfo();
543         break;
544
545     case AFS_XSTATSCOLL_PERF_INFO:
546         PrintPerfInfo();
547         break;
548
549     case AFS_XSTATSCOLL_FULL_PERF_INFO:
550         PrintFullPerfInfo();
551         break;
552
553     default:
554         printf("** Unknown collection: %d\n",
555                xstat_fs_Results.collectionNumber);
556     }
557
558     /*
559      * Return the happy news.
560      */
561     return (0);
562 }
563
564
565 /*------------------------------------------------------------------------
566  * CountListItems
567  *
568  * Description:
569  *      Given a pointer to the list of File Servers we'll be polling
570  *      (or, in fact, any list at all), compute the length of the list.
571  *
572  * Arguments:
573  *      struct cmd_item *a_firstItem : Ptr to first item in list.
574  *
575  * Returns:
576  *      Length of the above list.
577  *
578  * Environment:
579  *      Nothing interesting.
580  *
581  * Side Effects:
582  *      As advertised.
583  *------------------------------------------------------------------------*/
584
585 static int
586 CountListItems(struct cmd_item *a_firstItem)
587 {
588
589     int list_len;               /*List length */
590     struct cmd_item *curr_item; /*Ptr to current item */
591
592     list_len = 0;
593     curr_item = a_firstItem;
594
595     /*
596      * Count 'em up.
597      */
598     while (curr_item) {
599         list_len++;
600         curr_item = curr_item->next;
601     }
602
603     /*
604      * Return our tally.
605      */
606     return (list_len);
607 }
608
609
610 /*------------------------------------------------------------------------
611  * RunTheTest
612  *
613  * Description:
614  *      Routine called by the command line interpreter to execute the
615  *      meat of the program.  We count the number of File Servers
616  *      to watch, allocate enough space to remember all the connection
617  *      info for them, then go for it.
618  *      
619  *
620  * Arguments:
621  *      a_s : Ptr to the command line syntax descriptor.
622  *
623  * Returns:
624  *      0, but may exit the whole program on an error!
625  *
626  * Environment:
627  *      Nothing interesting.
628  *
629  * Side Effects:
630  *      As advertised.
631  *------------------------------------------------------------------------*/
632
633 int
634 RunTheTest(struct cmd_syndesc *a_s)
635 {
636     static char rn[] = "RunTheTest";    /*Routine name */
637     int code;                   /*Return code */
638     int numFSs;                 /*# File Servers to monitor */
639     int numCollIDs;             /*# collections to fetch */
640     int currFS;                 /*Loop index */
641     int currCollIDIdx;          /*Index of current collection ID */
642     afs_int32 *collIDP;         /*Ptr to array of collection IDs */
643     afs_int32 *currCollIDP;     /*Ptr to current collection ID */
644     struct cmd_item *curr_item; /*Current FS cmd line record */
645     struct sockaddr_in FSSktArray[20];  /*File Server socket array - FIX! */
646     struct hostent *he;         /*Host entry */
647     struct timeval tv;          /*Time structure */
648     int sleep_secs;             /*Number of seconds to sleep */
649     int initFlags;              /*Flags passed to the init fcn */
650     int waitCode;               /*Result of LWP_WaitProcess() */
651     int freq;                   /*Frequency of polls */
652     int period;                 /*Time in minutes of data collection */
653
654     /*
655      * Are we doing one-shot measurements?
656      */
657     if (a_s->parms[P_ONESHOT].items != 0)
658         one_shot = 1;
659
660     /*
661      * Are we doing debugging output?
662      */
663     if (a_s->parms[P_DEBUG].items != 0)
664         debugging_on = 1;
665
666     /*
667      * Pull out the number of File Servers to watch and the number of
668      * collections to get.
669      */
670     numFSs = CountListItems(a_s->parms[P_FS_NAMES].items);
671     numCollIDs = CountListItems(a_s->parms[P_COLL_IDS].items);
672
673     /* Get the polling frequency */
674     if (a_s->parms[P_FREQUENCY].items != 0)
675         freq = atoi(a_s->parms[P_FREQUENCY].items->data);
676     else
677         freq = 30;              /* default to 30 seconds */
678
679     /* Get the time duration to run the tests */
680     if (a_s->parms[P_PERIOD].items != 0)
681         period = atoi(a_s->parms[P_PERIOD].items->data);
682     else
683         period = 10;            /* default to 10 minutes */
684
685
686     /*
687      * Fill in the socket array for each of the File Servers listed.
688      */
689     curr_item = a_s->parms[P_FS_NAMES].items;
690     for (currFS = 0; currFS < numFSs; currFS++) {
691 #if defined(AFS_DARWIN_ENV) || defined(AFS_FBSD_ENV)
692         FSSktArray[currFS].sin_family = AF_INET;        /*Internet family */
693 #else
694         FSSktArray[currFS].sin_family = htons(AF_INET); /*Internet family */
695 #endif
696         FSSktArray[currFS].sin_port = htons(7000);      /*FileServer port */
697         he = hostutil_GetHostByName(curr_item->data);
698         if (he == NULL) {
699             fprintf(stderr, "[%s] Can't get host info for '%s'\n", rn,
700                     curr_item->data);
701             exit(-1);
702         }
703         memcpy(&(FSSktArray[currFS].sin_addr.s_addr), he->h_addr, 4);
704
705         /*
706          * Move to the next File Server name.
707          */
708         curr_item = curr_item->next;
709
710     }                           /*Get socket info for each File Server */
711
712     /*
713      * Create and fill up the array of desired collection IDs.
714      */
715     if (debugging_on)
716         printf("Allocating %d long(s) for coll ID\n", numCollIDs);
717     collIDP = (afs_int32 *) (malloc(numCollIDs * sizeof(afs_int32)));
718     currCollIDP = collIDP;
719     curr_item = a_s->parms[P_COLL_IDS].items;
720     for (currCollIDIdx = 0; currCollIDIdx < numCollIDs; currCollIDIdx++) {
721         *currCollIDP = (afs_int32) (atoi(curr_item->data));
722         if (debugging_on)
723             printf("CollID at index %d is %d\n", currCollIDIdx, *currCollIDP);
724         curr_item = curr_item->next;
725         currCollIDP++;
726     };
727
728     /*
729      * Crank up the File Server prober, then sit back and have fun.
730      */
731     printf("\nStarting up the xstat_fs service, ");
732     initFlags = 0;
733     if (debugging_on) {
734         initFlags |= XSTAT_FS_INITFLAG_DEBUGGING;
735         printf("debugging enabled, ");
736     } else
737         printf("no debugging, ");
738     if (one_shot) {
739         initFlags |= XSTAT_FS_INITFLAG_ONE_SHOT;
740         printf("one-shot operation\n");
741     } else
742         printf("continuous operation\n");
743
744     code = xstat_fs_Init(numFSs,        /*Num servers */
745                          FSSktArray,    /*File Server socket array */
746                          freq,  /*Probe frequency */
747                          FS_Handler,    /*Handler routine */
748                          initFlags,     /*Initialization flags */
749                          numCollIDs,    /*Number of collection IDs */
750                          collIDP);      /*Ptr to collection ID array */
751     if (code) {
752         fprintf(stderr, "[%s] Error returned by xstat_fs_Init: %d\n", rn,
753                 code);
754         xstat_fs_Cleanup(1);    /*Get rid of malloc'ed structures */
755         exit(-1);
756     }
757
758     if (one_shot) {
759         /*
760          * One-shot operation; just wait for the collection to be done.
761          */
762         if (debugging_on)
763             printf("[%s] Calling LWP_WaitProcess() on event 0x%x\n", rn,
764                    &terminationEvent);
765         waitCode = LWP_WaitProcess(&terminationEvent);
766         if (debugging_on)
767             printf("[%s] Returned from LWP_WaitProcess()\n", rn);
768         if (waitCode) {
769             if (debugging_on)
770                 fprintf(stderr,
771                         "[%s] Error %d encountered by LWP_WaitProcess()\n",
772                         rn, waitCode);
773         }
774     } else {
775         /*
776          * Continuous operation.
777          */
778         sleep_secs = 60 * period;       /*length of data collection */
779         printf
780             ("xstat_fs service started, main thread sleeping for %d secs.\n",
781              sleep_secs);
782
783         /*
784          * Let's just fall asleep for a while, then we'll clean up.
785          */
786         tv.tv_sec = sleep_secs;
787         tv.tv_usec = 0;
788         code = IOMGR_Select(0,  /*Num fds */
789                             0,  /*Descriptors ready for reading */
790                             0,  /*Descriptors ready for writing */
791                             0,  /*Descriptors with exceptional conditions */
792                             &tv);       /*Timeout structure */
793         if (code) {
794             fprintf(stderr,
795                     "[%s] IOMGR_Select() returned non-zero value: %d\n", rn,
796                     code);
797         }
798     }
799
800     /*
801      * We're all done.  Clean up, put the last nail in Rx, then
802      * exit happily.
803      */
804     if (debugging_on)
805         printf("\nYawn, main thread just woke up.  Cleaning things out...\n");
806
807     code = xstat_fs_Cleanup(1); /*Get rid of malloc'ed data */
808     rx_Finalize();
809     return (0);
810 }
811
812
813 #include "AFS_component_version_number.c"
814
815 int
816 main(int argc, char **argv)
817 {
818     static char rn[] = "xstat_fs_test"; /*Routine name */
819     register afs_int32 code;    /*Return code */
820     struct cmd_syndesc *ts;     /*Ptr to cmd line syntax desc */
821
822     /*
823      * Set up the commands we understand.
824      */
825     ts = cmd_CreateSyntax("initcmd", RunTheTest, 0, "initialize the program");
826     cmd_AddParm(ts, "-fsname", CMD_LIST, CMD_REQUIRED,
827                 "File Server name(s) to monitor");
828     cmd_AddParm(ts, "-collID", CMD_LIST, CMD_REQUIRED,
829                 "Collection(s) to fetch");
830     cmd_AddParm(ts, "-onceonly", CMD_FLAG, CMD_OPTIONAL,
831                 "Collect results exactly once, then quit");
832     cmd_AddParm(ts, "-frequency", CMD_SINGLE, CMD_OPTIONAL,
833                 "poll frequency, in seconds");
834     cmd_AddParm(ts, "-period", CMD_SINGLE, CMD_OPTIONAL,
835                 "data collection time, in minutes");
836     cmd_AddParm(ts, "-debug", CMD_FLAG, CMD_OPTIONAL,
837                 "turn on debugging output");
838
839     /*
840      * Parse command-line switches & execute the test, then get the
841      * heck out of here.
842      */
843     code = cmd_Dispatch(argc, argv);
844     if (code) {
845         fprintf(stderr, "[%s] Call to cmd_Dispatch() failed; code is %d\n",
846                 rn, code);
847     }
848
849     exit(code);
850 }