67293a075b7583e387e4aa19e4d1314487b6422f
[openafs.git] / src / libadmin / samples / rxstat_query_peer.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  * Portions Copyright (c) 2003 Apple Computer, Inc.
10  */
11
12 /*
13  * This file contains sample code for the rxstats interface 
14  */
15
16 #include <afsconfig.h>
17 #include <afs/param.h>
18
19
20 #ifdef AFS_NT40_ENV
21 #include <winsock2.h>
22 #include <pthread.h>
23 #endif
24
25 #include <rx/rx.h>
26 #include <rx/rxstat.h>
27
28 #include <afs/afs_Admin.h>
29 #include <afs/afs_clientAdmin.h>
30 #include <afs/afs_utilAdmin.h>
31
32 #ifdef AFS_DARWIN_ENV
33 pthread_mutex_t des_init_mutex = PTHREAD_MUTEX_INITIALIZER;
34 pthread_mutex_t des_random_mutex = PTHREAD_MUTEX_INITIALIZER;
35 pthread_mutex_t rxkad_random_mutex = PTHREAD_MUTEX_INITIALIZER;
36 #endif /* AFS_DARWIN_ENV */
37
38 void
39 Usage(void)
40 {
41     fprintf(stderr, "Usage: rxstat_query_peer <host> <port>\n");
42     exit(1);
43 }
44
45 void
46 ParseArgs(int argc, char *argv[], char **srvrName, long *srvrPort)
47 {
48     char **argp = argv;
49
50     if (!*(++argp))
51         Usage();
52     *srvrName = *(argp++);
53     if (!*(argp))
54         Usage();
55     *srvrPort = strtol(*(argp++), NULL, 0);
56     if (*srvrPort <= 0 || *srvrPort >= 65536)
57         Usage();
58     if (*(argp))
59         Usage();
60 }
61
62 int
63 main(int argc, char *argv[])
64 {
65     int rc;
66     afs_status_t st = 0;
67     struct rx_connection *conn;
68     char *srvrName;
69     long srvrPort;
70     void *cellHandle;
71     afs_RPCStatsState_t state;
72
73     ParseArgs(argc, argv, &srvrName, &srvrPort);
74
75     rc = afsclient_Init(&st);
76     if (!rc) {
77         fprintf(stderr, "afsclient_Init, status %d\n", st);
78         exit(1);
79     }
80
81     rc = afsclient_NullCellOpen(&cellHandle, &st);
82     if (!rc) {
83         fprintf(stderr, "afsclient_NullCellOpen, status %d\n", st);
84         exit(1);
85     }
86
87     rc = afsclient_RPCStatOpenPort(cellHandle, srvrName, srvrPort, &conn,
88                                    &st);
89     if (!rc) {
90         fprintf(stderr, "afsclient_RPCStatOpenPort, status %d\n", st);
91         exit(1);
92     }
93
94     rc = util_RPCStatsStateGet(conn, RXSTATS_QueryPeerRPCStats, &state, &st);
95     if (!rc) {
96         fprintf(stderr, "util_RPCStatsStateGet, status %d\n", st);
97         exit(1);
98     }
99
100     rc = afsclient_RPCStatClose(conn, &st);
101     if (!rc) {
102         fprintf(stderr, "afsclient_RPCStatClose, status %d\n", st);
103         exit(1);
104     }
105
106     rc = afsclient_CellClose(cellHandle, &st);
107     if (!rc) {
108         fprintf(stderr, "afsclient_CellClose, status %d\n", st);
109         exit(1);
110     }
111
112     printf("\n");
113     printf("Peer RPC stats are ");
114     switch (state) {
115     case AFS_RPC_STATS_DISABLED:
116         printf("disabled\n");
117         break;
118     case AFS_RPC_STATS_ENABLED:
119         printf("enabled\n");
120         break;
121     default:
122         printf("INVALID\n");
123         break;
124     }
125     printf("\n");
126
127     exit(0);
128 }