summaryrefslogtreecommitdiff
path: root/cmd/service/status.c
blob: 3be2f3efdb8497afad836fcae78e31060b8d3d13 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/* SPDX-License-Identifier: ISC */
#include "servicecmd.h"
#include "initsock.h"
#include "service.h"
#include "config.h"

#include <unistd.h>

static void free_resp(init_status_response_t *resp)
{
	free(resp->filename);
	free(resp->service_name);
}

static int cmd_status(int argc, char **argv)
{
	init_status_response_t resp;
	int fd, ret = EXIT_FAILURE;
	char tmppath[256];
	const char *state;
	bool is_tty;

	if (check_arguments(argv[0], argc, 1, 1))
		return EXIT_FAILURE;

	sprintf(tmppath, "/tmp/svcstatus.%d.sock", (int)getpid());
	fd = init_socket_open(tmppath);

	if (fd < 0) {
		unlink(tmppath);
		return EXIT_FAILURE;
	}

	if (init_socket_send_request(fd, EIR_STATUS))
		goto out;

	is_tty = (isatty(STDOUT_FILENO) == 1);

	for (;;) {
		memset(&resp, 0, sizeof(resp));

		if (init_socket_recv_status(fd, &resp)) {
			perror("reading from initd socket");
			free_resp(&resp);
			goto out;
		}

		if (resp.state == ESS_NONE) {
			free_resp(&resp);
			break;
		}

		switch (resp.state) {
		case ESS_RUNNING:
			if (!is_tty) {
				state = "UP";
				break;
			}
			state = "\033[22;32m UP \033[0m";
			break;
		case ESS_ENQUEUED:
			state = "SCHED";
			break;
		case ESS_FAILED:
			if (!is_tty) {
				state = "FAIL";
				break;
			}
			state = "\033[22;31mFAIL\033[0m";
			break;
		case ESS_DONE:
			if (!is_tty) {
				state = "DONE";
				break;
			}

			state = "\033[22;33mDONE\033[0m";
			break;
		default:
			if (!is_tty) {
				state = "UNKNOWN";
				break;
			}
			state = "\033[22;31mUNKNOWN\033[0m";
			break;
		}

		printf("[%s] %s\n", state, resp.filename);

		free_resp(&resp);
	}

	ret = EXIT_SUCCESS;
out:
	close(fd);
	unlink(tmppath);
	return ret;
}

static command_t status = {
	.cmd = "status",
	.usage = "",
	.s_desc = "report the status of the currently enabled services",
	.l_desc = "Gathers a list of all currently running services and the "
		  "state that they are in (currently running, done, failed, "
		  "wating to get scheduled).",
	.run_cmd = cmd_status,
};

REGISTER_COMMAND(status)