diff --git a/tools/storage_mon.c b/tools/storage_mon.c index 1aae29e58..cc415e97f 100644 --- a/tools/storage_mon.c +++ b/tools/storage_mon.c @@ -1,891 +1,898 @@ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #ifdef __FreeBSD__ #include #endif #include #include #include #include #include #include #include #include #include #define MAX_DEVICES 25 #define DEFAULT_TIMEOUT 10 #define DEFAULT_INTERVAL 30 #define DEFAULT_PIDFILE HA_VARRUNDIR "storage_mon.pid" #define DEFAULT_ATTRNAME "#health-storage_mon" #define SMON_GET_RESULT_COMMAND "get_check_value" #define SMON_RESULT_OK "green" #define SMON_RESULT_NG "red" #define SMON_RESULT_COMMAND_ERROR "unknown command" #define SMON_BUFF_1MEG 1048576 #define SMON_MAX_IPCSNAME 256 #define SMON_MAX_MSGSIZE 128 #define SMON_MAX_RESP_SIZE 100 #define PRINT_STORAGE_MON_ERR(fmt, ...) if (!daemonize) { \ fprintf(stderr, fmt"\n", __VA_ARGS__); \ } else { \ syslog(LOG_ERR, fmt, __VA_ARGS__); \ } #define PRINT_STORAGE_MON_ERR_NOARGS(str) if (!daemonize) { \ fprintf(stderr, str"\n"); \ } else { \ syslog(LOG_ERR, str); \ } #define PRINT_STORAGE_MON_INFO(fmt, ...) if (!daemonize) { \ printf(fmt"\n", __VA_ARGS__); \ } else { \ syslog(LOG_INFO, fmt, __VA_ARGS__); \ } struct storage_mon_timer_data { int interval; }; struct storage_mon_check_value_req { struct qb_ipc_request_header hdr; char message[SMON_MAX_MSGSIZE]; }; struct storage_mon_check_value_res { struct qb_ipc_response_header hdr; char message[SMON_MAX_MSGSIZE]; }; char *devices[MAX_DEVICES]; int scores[MAX_DEVICES]; size_t device_count = 0; int timeout = DEFAULT_TIMEOUT; int verbose = 0; int inject_error_percent = 0; const char *attrname = DEFAULT_ATTRNAME; gboolean daemonize = FALSE; int shutting_down = FALSE; static qb_ipcs_service_t *ipcs; int final_score = 0; int response_final_score = 0; pid_t test_forks[MAX_DEVICES]; size_t finished_count = 0; gboolean daemon_check_first_all_devices = FALSE; static qb_loop_t *storage_mon_poll_handle; static qb_loop_timer_handle timer_handle; static qb_loop_timer_handle expire_handle; static struct storage_mon_timer_data timer_d; static int test_device_main(gpointer data); static void wrap_test_device_main(void *data); static void usage(char *name, FILE *f) { fprintf(f, "usage: %s [-hv] [-d ]... [-s ]... [-t ]\n", name); fprintf(f, " --device device to test, up to %d instances\n", MAX_DEVICES); fprintf(f, " --score score if device fails the test. Must match --device count\n"); fprintf(f, " --timeout max time to wait for a device test to come back. in seconds (default %d)\n", DEFAULT_TIMEOUT); fprintf(f, " --inject-errors-percent Generate EIO errors %% of the time (for testing only)\n"); fprintf(f, " --daemonize test run in daemons.\n"); fprintf(f, " --client client connection to daemon. requires the attrname option.\n"); fprintf(f, " --interval interval to test. in seconds (default %d)(for daemonize only)\n", DEFAULT_INTERVAL); fprintf(f, " --pidfile file path to record pid (default %s)(for daemonize only)\n", DEFAULT_PIDFILE); fprintf(f, " --attrname attribute name to update test result (default %s)(for daemonize/client only)\n", DEFAULT_ATTRNAME); fprintf(f, " --verbose emit extra output to stdout\n"); fprintf(f, " --help print this message\n"); } /* Check one device */ static void *test_device(const char *device, int verbose, int inject_error_percent) { uint64_t devsize; int flags = O_RDONLY | O_DIRECT; int device_fd; int res; off_t seek_spot; if (verbose) { printf("Testing device %s\n", device); } device_fd = open(device, flags); if (device_fd < 0) { if (errno != EINVAL) { PRINT_STORAGE_MON_ERR("Failed to open %s: %s", device, strerror(errno)); exit(-1); } flags &= ~O_DIRECT; device_fd = open(device, flags); if (device_fd < 0) { PRINT_STORAGE_MON_ERR("Failed to open %s: %s", device, strerror(errno)); exit(-1); } } #ifdef __FreeBSD__ res = ioctl(device_fd, DIOCGMEDIASIZE, &devsize); #else res = ioctl(device_fd, BLKGETSIZE64, &devsize); #endif if (res < 0) { PRINT_STORAGE_MON_ERR("Failed to get device size for %s: %s", device, strerror(errno)); goto error; } if (verbose) { PRINT_STORAGE_MON_INFO("%s: opened %s O_DIRECT, size=%zu", device, (flags & O_DIRECT)?"with":"without", devsize); } /* Don't fret about real randomness */ srand(time(NULL) + getpid()); /* Pick a random place on the device - sector aligned */ seek_spot = (rand() % (devsize-1024)) & 0xFFFFFFFFFFFFFE00; res = lseek(device_fd, seek_spot, SEEK_SET); if (res < 0) { PRINT_STORAGE_MON_ERR("Failed to seek %s: %s", device, strerror(errno)); goto error; } if (verbose) { PRINT_STORAGE_MON_INFO("%s: reading from pos %ld", device, seek_spot); } if (flags & O_DIRECT) { int sec_size = 0; void *buffer; #ifdef __FreeBSD__ res = ioctl(device_fd, DIOCGSECTORSIZE, &sec_size); #else res = ioctl(device_fd, BLKSSZGET, &sec_size); #endif if (res < 0) { PRINT_STORAGE_MON_ERR("Failed to get block device sector size for %s: %s", device, strerror(errno)); goto error; } if (posix_memalign(&buffer, sysconf(_SC_PAGESIZE), sec_size) != 0) { PRINT_STORAGE_MON_ERR("Failed to allocate aligned memory: %s", strerror(errno)); goto error; } res = read(device_fd, buffer, sec_size); free(buffer); if (res < 0) { PRINT_STORAGE_MON_ERR("Failed to read %s: %s", device, strerror(errno)); goto error; } if (res < sec_size) { PRINT_STORAGE_MON_ERR("Failed to read %d bytes from %s, got %d", sec_size, device, res); goto error; } } else { char buffer[512]; res = read(device_fd, buffer, sizeof(buffer)); if (res < 0) { PRINT_STORAGE_MON_ERR("Failed to read %s: %s", device, strerror(errno)); goto error; } if (res < (int)sizeof(buffer)) { PRINT_STORAGE_MON_ERR("Failed to read %ld bytes from %s, got %d", sizeof(buffer), device, res); goto error; } } /* Fake an error */ if (inject_error_percent && ((rand() % 100) < inject_error_percent)) { PRINT_STORAGE_MON_ERR_NOARGS("People, please fasten your seatbelts, injecting errors!"); goto error; } res = close(device_fd); if (res != 0) { PRINT_STORAGE_MON_ERR("Failed to close %s: %s", device, strerror(errno)); exit(-1); } if (verbose) { PRINT_STORAGE_MON_INFO("%s: done", device); } exit(0); error: close(device_fd); exit(-1); } static gboolean is_child_runnning(void) { size_t i; for (i=0; i 0 ) { stop_child(test_forks[i], SIGTERM); } } } /* Set a timer for termination. */ qb_loop_timer_add(storage_mon_poll_handle, QB_LOOP_HIGH, 0, NULL, wrap_test_device_main, &timer_handle); return 0; } static size_t find_child_pid(int pid) { size_t i; for (i=0; i 0 ) { if (test_forks[i] == pid) { return i; } } } return -1; } static int32_t sigchld_handler(int32_t sig, void *data) { pid_t pid; size_t index; int status; if (is_child_runnning()) { while(1) { pid = waitpid(-1, &status, WNOHANG); if (pid > 0) { if (WIFEXITED(status)) { index = find_child_pid(pid); if (index >= 0) { /* If the expire timer is running, no timeout has occurred, */ /* so add the final_score from the exit code of the terminated child process. */ if (qb_loop_timer_is_running(storage_mon_poll_handle, expire_handle)) { if (WEXITSTATUS(status) !=0) { final_score += scores[index]; /* Update response values immediately in preparation for inquiries from clients. */ response_final_score = final_score; /* Even in the first demon mode check, if there is an error device, clear */ /* the flag to return the response to the client without waiting for all devices to finish. */ daemon_check_first_all_devices = TRUE; } } finished_count++; test_forks[index] = 0; } } } else { break; } } } return 0; } static void child_shutdown(int nsig) { exit(1); } static int write_pid_file(const char *pidfile) { char *pid; char *dir, *str = NULL; int fd = -1; int rc = -1; int i, len; if (asprintf(&pid, "%jd", (intmax_t)getpid()) < 0) { syslog(LOG_ERR, "Failed to allocate memory to store PID"); pid = NULL; goto done; } str = strdup(pidfile); if (str == NULL) { syslog(LOG_ERR, "Failed to duplicate string ['%s']", pidfile); goto done; } dir = dirname(str); for (i = 1, len = strlen(dir); i < len; i++) { if (dir[i] == '/') { dir[i] = 0; if ((mkdir(dir, 0640) < 0) && (errno != EEXIST)) { syslog(LOG_ERR, "Failed to create directory %s: %s", dir, strerror(errno)); goto done; } dir[i] = '/'; } } if ((mkdir(dir, 0640) < 0) && (errno != EEXIST)) { syslog(LOG_ERR, "Failed to create directory %s: %s", dir, strerror(errno)); goto done; } fd = open(pidfile, O_CREAT | O_WRONLY, 0640); if (fd < 0) { syslog(LOG_ERR, "Failed to open %s: %s", pidfile, strerror(errno)); goto done; } if (write(fd, pid, strlen(pid)) != strlen(pid)) { syslog(LOG_ERR, "Failed to write '%s' to %s: %s", pid, pidfile, strerror(errno)); goto done; } - close(fd); rc = 0; done: + if (fd != -1) { + close(fd); + } if (pid != NULL) { free(pid); } if (str != NULL) { free(str); } return rc; } static void child_timeout_handler(void *data) { size_t i; if (is_child_runnning()) { for (i=0; i 0) { /* If timeout occurs before SIGCHLD, add child process failure score to final_score. */ final_score += scores[i]; /* Update response values immediately in preparation for inquiries from clients. */ response_final_score = final_score; /* Even in the first demon mode check, if there is an error device, clear */ /* the flag to return the response to the client without waiting for all devices to finish. */ daemon_check_first_all_devices = TRUE; } } } } static void wrap_test_device_main(void *data) { struct storage_mon_timer_data *timer_data = (struct storage_mon_timer_data*)data; test_device_main((timer_data != NULL) ? &timer_data->interval : NULL); } static int test_device_main(gpointer data) { size_t i; struct timespec ts; time_t start_time; gboolean device_check = TRUE; if (daemonize) { if (shutting_down == TRUE) { goto done; } /* In the case of daemon mode, it is avoided that the timer is triggered and the number of */ /* child processes increases while the device monitoring child process is not completed. */ if (is_child_runnning()) { device_check = FALSE; } if (device_count == finished_count && device_check) { /* Update the result value for the client response once all checks have completed. */ response_final_score = final_score; if (!daemon_check_first_all_devices) { daemon_check_first_all_devices = TRUE; } } } if (device_check) { /* Reset final_score, finished_count, test_forks[] */ final_score = 0; finished_count = 0; memset(test_forks, 0, sizeof(test_forks)); for (i=0; i ts.tv_sec)) { for (i=0; i 0) { w = waitpid(test_forks[i], &wstatus, WUNTRACED | WNOHANG | WCONTINUED); if (w < 0) { PRINT_STORAGE_MON_ERR("waitpid on %s failed: %s", devices[i], strerror(errno)); return -1; } if (w == test_forks[i]) { if (WIFEXITED(wstatus)) { if (WEXITSTATUS(wstatus) != 0) { syslog(LOG_ERR, "Error reading from device %s", devices[i]); final_score += scores[i]; } finished_count++; test_forks[i] = 0; } } } } usleep(100000); clock_gettime(CLOCK_REALTIME, &ts); } /* See which threads have not finished */ for (i=0; ihdr.id, request->hdr.size, request->message); if (strcmp(request->message, SMON_GET_RESULT_COMMAND) != 0) { syslog(LOG_DEBUG, "request command is unknown."); send_score = -1; } else if (!daemon_check_first_all_devices) { send_score = -2; } resps.size = sizeof(struct qb_ipc_response_header); resps.id = 13; resps.error = 0; rc = snprintf(resp, SMON_MAX_RESP_SIZE, "%d", send_score) + 1; iov[0].iov_len = sizeof(resps); iov[0].iov_base = &resps; iov[1].iov_len = rc; iov[1].iov_base = resp; resps.size += rc; res = qb_ipcs_response_sendv(c, iov, 2); if (res < 0) { errno = -res; syslog(LOG_ERR, "qb_ipcs_response_send : errno = %d", errno); } return 0; } static int32_t storage_mon_client(void) { struct storage_mon_check_value_req request; struct storage_mon_check_value_res response; qb_ipcc_connection_t *conn; char ipcs_name[SMON_MAX_IPCSNAME]; int32_t rc; snprintf(ipcs_name, SMON_MAX_IPCSNAME, "storage_mon_%s", attrname); conn = qb_ipcc_connect(ipcs_name, 0); if (conn == NULL) { syslog(LOG_ERR, "qb_ipcc_connect error\n"); return(-1); } snprintf(request.message, SMON_MAX_MSGSIZE, "%s", SMON_GET_RESULT_COMMAND); request.hdr.id = 0; request.hdr.size = sizeof(struct storage_mon_check_value_req); + response.hdr.id = 0; rc = qb_ipcc_send(conn, &request, request.hdr.size); if (rc < 0) { syslog(LOG_ERR, "qb_ipcc_send error : %d\n", rc); return(-1); } if (rc > 0) { rc = qb_ipcc_recv(conn, &response, sizeof(response), -1); if (rc < 0) { syslog(LOG_ERR, "qb_ipcc_recv error : %d\n", rc); return(-1); } } qb_ipcc_disconnect(conn); /* Set score to result */ /* 0 : Normal. */ /* greater than 0 : monitoring error. */ /* -1 : communication system error. */ /* -2 : Not all checks completed for first device in daemon mode. */ - rc = atoi(response.message); + if (strnlen(response.message, 1)) { + rc = atoi(response.message); + } else { + rc = -1; + } syslog(LOG_DEBUG, "daemon response[%d]: %s \n", response.hdr.id, response.message); return(rc); } static int32_t storage_mon_daemon(int interval, const char *pidfile) { int32_t rc; char ipcs_name[SMON_MAX_IPCSNAME]; struct qb_ipcs_service_handlers service_handle = { .connection_accept = storage_mon_ipcs_connection_accept_fn, .connection_created = storage_mon_ipcs_connection_created_fn, .msg_process = storage_mon_ipcs_msg_process_fn, .connection_destroyed = storage_mon_ipcs_connection_destroyed_fn, .connection_closed = storage_mon_ipcs_connection_closed_fn, }; struct qb_ipcs_poll_handlers poll_handle = { .job_add = storage_mon_job_add, .dispatch_add = storage_mon_dispatch_add, .dispatch_mod = storage_mon_dispatch_mod, .dispatch_del = storage_mon_dispatch_del, }; if (daemon(0, 0) < 0) { syslog(LOG_ERR, "Failed to daemonize: %s", strerror(errno)); return -1; } umask(S_IWGRP | S_IWOTH | S_IROTH); if (write_pid_file(pidfile) < 0) { return -1; } snprintf(ipcs_name, SMON_MAX_IPCSNAME, "storage_mon_%s", attrname); ipcs = qb_ipcs_create(ipcs_name, 0, QB_IPC_NATIVE, &service_handle); if (ipcs == 0) { syslog(LOG_ERR, "qb_ipcs_create"); return -1; } qb_ipcs_enforce_buffer_size(ipcs, SMON_BUFF_1MEG); storage_mon_poll_handle = qb_loop_create(); qb_ipcs_poll_handlers_set(ipcs, &poll_handle); rc = qb_ipcs_run(ipcs); if (rc != 0) { errno = -rc; syslog(LOG_ERR, "qb_ipcs_run"); return -1; } qb_loop_signal_add(storage_mon_poll_handle, QB_LOOP_HIGH, SIGTERM, NULL, sigterm_handler, NULL); qb_loop_signal_add(storage_mon_poll_handle, QB_LOOP_MED, SIGCHLD, NULL, sigchld_handler, NULL); timer_d.interval = interval; qb_loop_timer_add(storage_mon_poll_handle, QB_LOOP_MED, 0, &timer_d, wrap_test_device_main, &timer_handle); qb_loop_run(storage_mon_poll_handle); qb_loop_destroy(storage_mon_poll_handle); unlink(pidfile); return 0; } int main(int argc, char *argv[]) { size_t score_count = 0; int opt, option_index; int interval = DEFAULT_INTERVAL; const char *pidfile = DEFAULT_PIDFILE; gboolean client = FALSE; struct option long_options[] = { {"timeout", required_argument, 0, 't' }, {"device", required_argument, 0, 'd' }, {"score", required_argument, 0, 's' }, {"inject-errors-percent", required_argument, 0, 0 }, {"daemonize", no_argument, 0, 0 }, {"client", no_argument, 0, 0 }, {"interval", required_argument, 0, 'i' }, {"pidfile", required_argument, 0, 'p' }, {"attrname", required_argument, 0, 'a' }, {"verbose", no_argument, 0, 'v' }, {"help", no_argument, 0, 'h' }, {0, 0, 0, 0 } }; while ( (opt = getopt_long(argc, argv, "hvt:d:s:i:p:a:", long_options, &option_index)) != -1 ) { switch (opt) { case 0: /* Long-only options */ if (strcmp(long_options[option_index].name, "inject-errors-percent") == 0) { inject_error_percent = atoi(optarg); if (inject_error_percent < 1 || inject_error_percent > 100) { fprintf(stderr, "inject_error_percent should be between 1 and 100\n"); return -1; } } if (strcmp(long_options[option_index].name, "daemonize") == 0) { daemonize = TRUE; } if (strcmp(long_options[option_index].name, "client") == 0) { client = TRUE; } if (daemonize && client) { fprintf(stderr,"The daemonize option and client option cannot be specified at the same time."); return -1; } break; case 'd': if (device_count < MAX_DEVICES) { devices[device_count++] = strdup(optarg); } else { fprintf(stderr, "too many devices, max is %d\n", MAX_DEVICES); return -1; } break; case 's': if (score_count < MAX_DEVICES) { int score = atoi(optarg); if (score < 1 || score > 10) { fprintf(stderr, "Score must be between 1 and 10 inclusive\n"); return -1; } scores[score_count++] = score; } else { fprintf(stderr, "too many scores, max is %d\n", MAX_DEVICES); return -1; } break; case 'v': verbose++; break; case 't': timeout = atoi(optarg); if (timeout < 1) { fprintf(stderr, "invalid timeout %d. Min 1, recommended %d (default)\n", timeout, DEFAULT_TIMEOUT); return -1; } break; case 'h': usage(argv[0], stdout); return 0; break; case 'i': interval = atoi(optarg); if (interval < 1) { fprintf(stderr, "invalid interval %d. Min 1, default is %d\n", interval, DEFAULT_INTERVAL); return -1; } break; case 'p': pidfile = strdup(optarg); if (pidfile == NULL) { fprintf(stderr, "Failed to duplicate string ['%s']\n", optarg); return -1; } break; case 'a': attrname = strdup(optarg); if (attrname == NULL) { fprintf(stderr, "Failed to duplicate string ['%s']\n", optarg); return -1; } break; default: usage(argv[0], stderr); return -1; break; } } if (client) { return(storage_mon_client()); } if (device_count == 0) { fprintf(stderr, "No devices to test, use the -d or --device argument\n"); return -1; } if (device_count != score_count) { fprintf(stderr, "There must be the same number of devices and scores\n"); return -1; } openlog("storage_mon", 0, LOG_DAEMON); if (!daemonize) { final_score = test_device_main(NULL); } else { return(storage_mon_daemon(interval, pidfile)); } return final_score; }