extern int errno;
char *Progname;
int Verbose;
+dm_sessid_t sid = DM_NO_SESSION;
extern int setup_dmapi(dm_sessid_t *);
extern void err_msg(char *, ...);
int set_events(dm_sessid_t, void *, size_t);
int mk_daemon(char *);
void spawn_kid(dm_sessid_t, dm_token_t, char *);
-void migin_exit(void);
+void migin_exit(int);
void usage(char *);
+void setup_signals();
void
int c;
int error;
char *fsname, *logfile;
- dm_sessid_t sid;
void *fs_hanp;
size_t fs_hlen;
logfile = LOG_DEFAULT;
+ /*
+ * Turn ourselves into a daemon
+ */
+ if (!Verbose){
+ error = mk_daemon(logfile);
+ if (error)
+ exit(1);
+ }
+ setup_signals();
+
/*
* Now we have our filesystem name and possibly a size threshold
* to look for. Init the dmapi, and get a filesystem handle so
exit(1);
}
- /*
- * Turn ourselves into a daemon
- */
- error = mk_daemon(logfile);
- if (error)
- exit(1);
-
/*
* Set the event disposition so that our session will receive
bufsize *= 16;
msgbuf = (void *)malloc(bufsize);
if (msgbuf == NULL) {
- err_msg("Can't allocate memory for buffer");
+ err_msg("Can't allocate memory for buffer\n");
goto out;
}
free(msgbuf);
msgbuf = (void *)malloc(rlen);
if (msgbuf == NULL) {
- err_msg("Can't resize msg buffer");
+ err_msg("Can't resize msg buffer\n");
goto out;
}
continue;
if (msgbuf != NULL)
free(msgbuf);
- migin_exit();
+ migin_exit(0);
}
/*
* Fork and exec our worker bee to work on the file. If
* there is any error in fork/exec'ing the file, we have to
* supply the error return to the event. Once the child gets
- * started, he/she/it will respond to the event for us.
+ * started, they will respond to the event for us.
*/
void
spawn_kid(
char *action)
{
pid_t pid;
- char sidbuf[sizeof(dm_sessid_t)];
- char tokenbuf[sizeof(dm_token_t)];
+ char sidbuf[10];
+ char tokenbuf[10];
pid = fork();
if (pid == 0) {
sprintf(sidbuf, "%d", sid);
sprintf(tokenbuf, "%d", token);
if (Verbose) {
- fprintf(stderr, "execl(%s, %s, %s, -s, %s, -t, xs, 0)\n",
- WORKER_BEE, WORKER_BEE, action, sidbuf, tokenbuf);
- }
- if (execl(WORKER_BEE, WORKER_BEE, action, "-s", sidbuf,
- "-t", tokenbuf, NULL))
- {
- (void)dm_respond_event(sid, token, DM_RESP_ABORT,
- errno, 0, 0);
- exit(1);
+ fprintf(stderr, "execl(%s, %s, %s, -s, %s, -t, %s, 0)\n",
+ WORKER_BEE, WORKER_BEE, action, sidbuf,
+ tokenbuf);
}
+ execlp("./"WORKER_BEE, WORKER_BEE, action, "-s", sidbuf,
+ "-t", tokenbuf, NULL);
+ errno_msg("execlp of worker bee failed");
+ (void)dm_respond_event(sid, token, DM_RESP_ABORT,
+ errno, 0, 0);
+ exit(1);
}
if (pid < 0) {
- err_msg("Can't fork worker bee");
+ err_msg("Can't fork worker bee\n");
(void)dm_respond_event(sid, token, DM_RESP_ABORT, errno,
0, 0);
return;
mk_daemon(
char *logfile)
{
- int fd, pid;
+ int fd;
int i;
struct rlimit lim;
- struct sigaction act;
+ pid_t pid;
-#ifdef NOTYET
if ((pid = fork()) == -1)
return (-1);
if (pid)
(void) chdir("/");
-#endif /* NOTYET */
/*
* Determine how many open files we've got and close
* then all
(void)dup2(fd, STDERR_FILENO);
close(fd);
+ return(0);
+}
+
+
+void
+setup_signals()
+{
+ struct sigaction act;
+
/*
* Set up signals so that we can wait for spawned children
*/
(void)sigaction(SIGUSR1, &act, NULL);
(void)sigaction(SIGUSR1, &act, NULL);
(void)sigaction(SIGUSR2, &act, NULL);
-
- return(0);
}
void
-migin_exit(void)
+migin_exit(int x)
{
- dm_sessid_t *sidbuf, *sid;
- void *infobuf;
- char *cp;
- u_int nsessions, nret;
- int i, found, error;
- size_t buflen, retlen;
-
- sidbuf = NULL;
- infobuf = NULL;
-
+ int error;
+
/*
* We could try and kill off all our kids, but we're not
* 'Mr. Mean', so we just wait for them to die.
*/
- err_msg("%s: waiting for all children to die...", Progname);
+ err_msg("%s: waiting for all children to die...\n", Progname);
while (wait3((int *)0, WNOHANG, (struct rusage *)0) > 0)
;
fprintf(stdout, "\n");
- /*
- * Now search for our session and try and shut it down. We
- * could just as easily make the session ID a global, but
- * this demonstrates how sessions can be queried
- */
- nsessions = 4;
- sidbuf = (dm_sessid_t *)malloc(nsessions * sizeof(dm_sessid_t));
- if (sidbuf == NULL) {
- err_msg("Can't alloc mem to shut down session");
- goto out;
- }
- error = dm_getall_sessions(nsessions, sidbuf, &nret);
- if (error == -1) {
- if (errno != E2BIG) {
- errno_msg("Can't get list of active sessions");
- goto out;
- }
- free(sidbuf);
- nsessions = nret;
- sidbuf = (dm_sessid_t *)malloc(nsessions * sizeof(dm_sessid_t));
- if (sidbuf == NULL) {
- err_msg("Can't alloc mem to shut down session");
- goto out;
- }
- error = dm_getall_sessions(nsessions, sidbuf, &nret);
- if (error == -1) {
- errno_msg("Can't get list of active sessions");
- goto out;
- }
- }
-
- /*
- * Now we have all the active sessions in our system.
- * Query each one until we find ourselves.
- */
- sid = sidbuf;
- buflen = DM_SESSION_INFO_LEN;
- infobuf = malloc(buflen);
- if (infobuf == NULL) {
- err_msg("Can't alloc memory for session info buffer");
- goto out;
- }
-
- /*
- * When we registered our session, we just hammered the last component
- * of the path name, so that's all we look for here. This prevents
- * mismatches when running at ./migin or some other such foo
- */
- cp = strrchr(Progname, '/');
- if (cp)
- cp++;
- else
- cp = Progname;
-
-
- found = 0;
- for (i=0; i<nret; i++) {
- error = dm_query_session(*sid, buflen, infobuf, &retlen);
- if (error == -1)
- continue; /* We just punt on errors */
-
- if (strstr(infobuf, cp)) {
- found = 1;
- break;
- }
- sid++;
- }
/*
* XXXX FIXME XXX
* filesystem handle global so that we can get at it
*/
- if (!found) {
- err_msg("Can't find session to shut down");
- goto out;
- }
- error = dm_destroy_session(*sid);
+ error = dm_destroy_session(sid);
if (error == -1) {
- errno_msg("Can't shut down session");
+ errno_msg("Can't shut down session\n");
}
-
-out:
- if (infobuf)
- free(infobuf);
- if (sidbuf)
- free(sidbuf);
-
exit(0);
}
projects / xfstests-dev.git / blobdiff