2 * Copyright 2006, Sine Nomine Associates and others.
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
10 /* Main program file. Define globals. */
18 #include <afsconfig.h>
19 #include <afs/param.h>
31 #include <WINNT/afsevent.h>
33 #include <sys/param.h>
37 #endif /* ITIMER_REAL */
40 #include <afs/afsint.h>
41 #include <afs/assert.h>
47 #include <afs/osi_inode.h>
51 #include <afs/afsutil.h>
52 #include <afs/fileutil.h>
57 #include <afs/ihandle.h>
58 #include <afs/vnode.h>
59 #include <afs/volume.h>
60 #include <afs/partition.h>
61 #include <afs/daemon_com.h>
62 #include <afs/salvsync.h>
67 int VolumeChanged; /* hack to make dir package happy */
70 #ifndef AFS_DEMAND_ATTACH_FS
72 main(int argc, char ** argv)
74 fprintf(stderr, "*** salvsync-debug is only supported for OpenAFS builds with the demand-attach fileserver extension\n");
77 #else /* AFS_DEMAND_ATTACH_FS */
87 struct salv_state * sop;
90 static int common_prolog(struct cmd_syndesc *, struct state *);
91 static int common_salv_prolog(struct cmd_syndesc *, struct state *);
93 static int do_salvop(struct state *, afs_int32 command, SYNC_response * res);
95 static char * response_code_to_string(afs_int32);
96 static char * command_code_to_string(afs_int32);
97 static char * reason_code_to_string(afs_int32);
98 static char * state_code_to_string(afs_int32);
101 static int OpStats(struct cmd_syndesc * as, void * rock);
102 static int OpSalvage(struct cmd_syndesc * as, void * rock);
103 static int OpCancel(struct cmd_syndesc * as, void * rock);
104 static int OpCancelAll(struct cmd_syndesc * as, void * rock);
105 static int OpRaisePrio(struct cmd_syndesc * as, void * rock);
106 static int OpQuery(struct cmd_syndesc * as, void * rock);
110 #include "AFS_component_version_number.c"
114 #define COMMON_PARMS_OFFSET 13
115 #define COMMON_PARMS(ts) \
116 cmd_Seek(ts, COMMON_PARMS_OFFSET); \
117 cmd_AddParm(ts, "-reason", CMD_SINGLE, CMD_OPTIONAL, "sync protocol reason code"); \
118 cmd_AddParm(ts, "-programtype", CMD_SINGLE, CMD_OPTIONAL, "program type code")
120 #define COMMON_SALV_PARMS_OFFSET 10
121 #define COMMON_SALV_PARMS(ts) \
122 cmd_Seek(ts, COMMON_SALV_PARMS_OFFSET); \
123 cmd_AddParm(ts, "-volumeid", CMD_SINGLE, 0, "volume id"); \
124 cmd_AddParm(ts, "-partition", CMD_SINGLE, CMD_OPTIONAL, "partition name"); \
125 cmd_AddParm(ts, "-priority", CMD_SINGLE, CMD_OPTIONAL, "priority")
127 #define SALV_PARMS_DECL(ts) \
128 COMMON_SALV_PARMS(ts); \
131 #define COMMON_PARMS_DECL(ts) \
135 main(int argc, char **argv)
137 struct cmd_syndesc *ts;
140 /* Initialize directory paths */
141 if (!(initAFSDirPath() & AFSDIR_SERVER_PATHS_OK)) {
143 ReportErrorEventAlt(AFSEVT_SVR_NO_INSTALL_DIR, 0, argv[0], 0);
145 fprintf(stderr, "%s: Unable to obtain AFS server directory.\n",
151 ts = cmd_CreateSyntax("stats", OpStats, NULL, "get salvageserver statistics (SALVSYNC_NOP opcode)");
152 COMMON_PARMS_DECL(ts);
153 cmd_CreateAlias(ts, "nop");
155 ts = cmd_CreateSyntax("salvage", OpSalvage, NULL, "schedule a salvage (SALVSYNC_SALVAGE opcode)");
158 ts = cmd_CreateSyntax("cancel", OpCancel, NULL, "cancel a salvage (SALVSYNC_CANCEL opcode)");
161 ts = cmd_CreateSyntax("raiseprio", OpRaisePrio, NULL, "raise a salvage priority (SALVSYNC_RAISEPRIO opcode)");
163 cmd_CreateAlias(ts, "rp");
165 ts = cmd_CreateSyntax("query", OpQuery, NULL, "query salvage status (SALVSYNC_QUERY opcode)");
167 cmd_CreateAlias(ts, "qry");
169 ts = cmd_CreateSyntax("kill", OpCancelAll, NULL, "cancel all scheduled salvages (SALVSYNC_CANCELALL opcode)");
170 COMMON_PARMS_DECL(ts);
172 err = cmd_Dispatch(argc, argv);
177 common_prolog(struct cmd_syndesc * as, struct state * state)
179 register struct cmd_item *ti;
180 VolumePackageOptions opts;
183 if (afs_winsockInit() < 0) {
188 VOptDefaults(debugUtility, &opts);
189 VInitVolumePackage2(debugUtility, &opts);
192 if ((ti = as->parms[COMMON_PARMS_OFFSET].items)) { /* -reason */
193 state->reason = atoi(ti->data);
195 if ((ti = as->parms[COMMON_PARMS_OFFSET+1].items)) { /* -programtype */
196 if (!strcmp(ti->data, "fileServer")) {
197 programType = fileServer;
198 } else if (!strcmp(ti->data, "volumeUtility")) {
199 programType = volumeUtility;
200 } else if (!strcmp(ti->data, "salvager")) {
201 programType = salvager;
202 } else if (!strcmp(ti->data, "salvageServer")) {
203 programType = salvageServer;
204 } else if (!strcmp(ti->data, "volumeServer")) {
205 programType = volumeServer;
206 } else if (!strcmp(ti->data, "volumeSalvager")) {
207 programType = volumeSalvager;
209 programType = (ProgramType) atoi(ti->data);
219 common_salv_prolog(struct cmd_syndesc * as, struct state * state)
221 register struct cmd_item *ti;
223 state->sop = (struct salv_state *) calloc(1, sizeof(struct salv_state));
224 assert(state->sop != NULL);
226 if ((ti = as->parms[COMMON_SALV_PARMS_OFFSET].items)) { /* -volumeid */
227 state->sop->volume = atoi(ti->data);
229 fprintf(stderr, "required argument -volumeid not given\n");
232 if ((ti = as->parms[COMMON_SALV_PARMS_OFFSET+1].items)) { /* -partition */
233 strlcpy(state->sop->partName, ti->data, sizeof(state->sop->partName));
235 memset(state->sop->partName, 0, sizeof(state->sop->partName));
238 if ((ti = as->parms[COMMON_SALV_PARMS_OFFSET+2].items)) { /* -prio */
239 state->sop->prio = atoi(ti->data);
241 state->sop->prio = 0;
248 do_salvop(struct state * state, afs_int32 command, SYNC_response * res)
251 SALVSYNC_response_hdr hdr_l, *hdr;
256 res->payload.len = sizeof(hdr_l);
257 res->payload.buf = hdr = &hdr_l;
259 hdr = (SALVSYNC_response_hdr *) res->payload.buf;
262 fprintf(stderr, "calling SALVSYNC_SalvageVolume with command code %d (%s)\n",
263 command, command_code_to_string(command));
265 code = SALVSYNC_SalvageVolume(state->sop->volume,
266 state->sop->partName,
277 fprintf(stderr, "possible sync protocol error. return code was %d\n", code);
280 fprintf(stderr, "SALVSYNC_SalvageVolume returned %d (%s)\n", code, response_code_to_string(code));
281 fprintf(stderr, "protocol response code was %d (%s)\n",
282 res->hdr.response, response_code_to_string(res->hdr.response));
283 fprintf(stderr, "protocol reason code was %d (%s)\n",
284 res->hdr.reason, reason_code_to_string(res->hdr.reason));
286 printf("state = {\n");
287 if (res->hdr.flags & SALVSYNC_FLAG_VOL_STATS_VALID) {
288 printf("\tstate = %d (%s)\n",
289 hdr->state, state_code_to_string(hdr->state));
290 printf("\tprio = %d\n", hdr->prio);
292 printf("\tsq_len = %d\n", hdr->sq_len);
293 printf("\tpq_len = %d\n", hdr->pq_len);
302 response_code_to_string(afs_int32 response)
308 return "SYNC_DENIED";
310 return "SYNC_COM_ERROR";
311 case SYNC_BAD_COMMAND:
312 return "SYNC_BAD_COMMAND";
314 return "SYNC_FAILED";
316 return "**UNKNOWN**";
321 command_code_to_string(afs_int32 command)
324 case SYNC_COM_CHANNEL_CLOSE:
325 return "SYNC_COM_CHANNEL_CLOSE";
327 return "SALVSYNC_NOP";
328 case SALVSYNC_SALVAGE:
329 return "SALVSYNC_SALVAGE";
330 case SALVSYNC_CANCEL:
331 return "SALVSYNC_CANCEL";
332 case SALVSYNC_RAISEPRIO:
333 return "SALVSYNC_RAISEPRIO";
335 return "SALVSYNC_QUERY";
336 case SALVSYNC_CANCELALL:
337 return "SALVSYNC_CANCELLALL";
339 return "**UNKNOWN**";
344 reason_code_to_string(afs_int32 reason)
347 case SALVSYNC_WHATEVER:
348 return "SALVSYNC_WHATEVER";
350 return "SALVSYNC_ERROR";
351 case SALVSYNC_OPERATOR:
352 return "SALVSYNC_OPERATOR";
353 case SALVSYNC_SHUTDOWN:
354 return "SALVSYNC_SHUTDOWN";
355 case SALVSYNC_NEEDED:
356 return "SALVSYNC_NEEDED";
358 return "**UNKNOWN**";
364 program_type_to_string(afs_int32 type)
366 switch ((ProgramType)type) {
370 return "volumeUtility";
374 return "salvageServer";
376 return "**UNKNOWN**";
382 state_code_to_string(afs_int32 state)
385 case SALVSYNC_STATE_UNKNOWN:
386 return "SALVSYNC_STATE_UNKNOWN";
387 case SALVSYNC_STATE_QUEUED:
388 return "SALVSYNC_STATE_QUEUED";
389 case SALVSYNC_STATE_SALVAGING:
390 return "SALVSYNC_STATE_SALVAGING";
391 case SALVSYNC_STATE_ERROR:
392 return "SALVSYNC_STATE_ERROR";
393 case SALVSYNC_STATE_DONE:
394 return "SALVSYNC_STATE_DONE";
396 return "**UNKNOWN**";
401 OpStats(struct cmd_syndesc * as, void * rock)
405 common_prolog(as, &state);
406 common_salv_prolog(as, &state);
408 do_salvop(&state, SALVSYNC_NOP, NULL);
414 OpSalvage(struct cmd_syndesc * as, void * rock)
418 common_prolog(as, &state);
419 common_salv_prolog(as, &state);
421 do_salvop(&state, SALVSYNC_SALVAGE, NULL);
427 OpCancel(struct cmd_syndesc * as, void * rock)
431 common_prolog(as, &state);
432 common_salv_prolog(as, &state);
434 do_salvop(&state, SALVSYNC_CANCEL, NULL);
440 OpCancelAll(struct cmd_syndesc * as, void * rock)
444 common_prolog(as, &state);
445 common_salv_prolog(as, &state);
447 do_salvop(&state, SALVSYNC_CANCELALL, NULL);
453 OpRaisePrio(struct cmd_syndesc * as, void * rock)
457 common_prolog(as, &state);
458 common_salv_prolog(as, &state);
460 do_salvop(&state, SALVSYNC_RAISEPRIO, NULL);
466 OpQuery(struct cmd_syndesc * as, void * rock)
470 common_prolog(as, &state);
471 common_salv_prolog(as, &state);
473 do_salvop(&state, SALVSYNC_QUERY, NULL);
478 #endif /* AFS_DEMAND_ATTACH_FS */