2 * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
3 * Copyright (C) 2013 John Crispin <blogic@openwrt.org>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU Lesser General Public License version 2.1
7 * as published by the Free Software Foundation
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
15 #include <sys/utsname.h>
17 #include <sys/sysinfo.h>
19 #include <sys/ioctl.h>
20 #include <sys/types.h>
21 #include <sys/reboot.h>
28 #include <libubox/uloop.h>
31 #include "sysupgrade.h"
34 static struct blob_buf b
;
36 static struct ubus_context
*_ctx
;
38 static int system_board(struct ubus_context
*ctx
, struct ubus_object
*obj
,
39 struct ubus_request_data
*req
, const char *method
,
40 struct blob_attr
*msg
)
44 char *key
, *val
, *next
;
45 struct utsname utsname
;
50 if (uname(&utsname
) >= 0)
52 blobmsg_add_string(&b
, "kernel", utsname
.release
);
53 blobmsg_add_string(&b
, "hostname", utsname
.nodename
);
56 if ((f
= fopen("/proc/cpuinfo", "r")) != NULL
)
58 while(fgets(line
, sizeof(line
), f
))
60 key
= strtok(line
, "\t:");
61 val
= strtok(NULL
, "\t\n");
66 if (!strcasecmp(key
, "system type") ||
67 !strcasecmp(key
, "processor") ||
68 !strcasecmp(key
, "model name"))
70 strtoul(val
+ 2, &key
, 0);
72 if (key
== (val
+ 2) || *key
!= 0)
74 blobmsg_add_string(&b
, "system", val
+ 2);
83 if ((f
= fopen("/tmp/sysinfo/model", "r")) != NULL
||
84 (f
= fopen("/proc/device-tree/model", "r")) != NULL
)
86 if (fgets(line
, sizeof(line
), f
))
88 val
= strtok(line
, "\t\n");
91 blobmsg_add_string(&b
, "model", val
);
96 else if ((f
= fopen("/proc/cpuinfo", "r")) != NULL
)
98 while(fgets(line
, sizeof(line
), f
))
100 key
= strtok(line
, "\t:");
101 val
= strtok(NULL
, "\t\n");
106 if (!strcasecmp(key
, "machine") ||
107 !strcasecmp(key
, "hardware"))
109 blobmsg_add_string(&b
, "model", val
+ 2);
117 if ((f
= fopen("/tmp/sysinfo/board_name", "r")) != NULL
)
119 if (fgets(line
, sizeof(line
), f
))
121 val
= strtok(line
, "\t\n");
124 blobmsg_add_string(&b
, "board_name", val
);
129 else if ((f
= fopen("/proc/device-tree/compatible", "r")) != NULL
)
131 if (fgets(line
, sizeof(line
), f
))
133 val
= strtok(line
, "\t\n");
138 while ((next
= strchr(next
, ',')) != NULL
)
144 blobmsg_add_string(&b
, "board_name", val
);
151 if ((f
= fopen("/etc/openwrt_release", "r")) != NULL
)
153 c
= blobmsg_open_table(&b
, "release");
155 while (fgets(line
, sizeof(line
), f
))
161 val
= strchr(line
, '=');
167 if (!strcasecmp(key
, "DISTRIB_ID"))
168 key
= "distribution";
169 else if (!strcasecmp(key
, "DISTRIB_RELEASE"))
171 else if (!strcasecmp(key
, "DISTRIB_REVISION"))
173 else if (!strcasecmp(key
, "DISTRIB_CODENAME"))
175 else if (!strcasecmp(key
, "DISTRIB_TARGET"))
177 else if (!strcasecmp(key
, "DISTRIB_DESCRIPTION"))
182 dest
= blobmsg_alloc_string_buffer(&b
, key
, strlen(val
));
184 ERROR("Failed to allocate blob.\n");
188 while (val
&& (ch
= *(val
++)) != 0) {
192 next
= strchr(val
, ch
);
201 dest
+= strlen(dest
);
204 *(dest
++) = *(val
++);
208 blobmsg_add_string_buffer(&b
);
211 blobmsg_close_array(&b
, c
);
216 ubus_send_reply(ctx
, req
, b
.head
);
218 return UBUS_STATUS_OK
;
221 static int system_info(struct ubus_context
*ctx
, struct ubus_object
*obj
,
222 struct ubus_request_data
*req
, const char *method
,
223 struct blob_attr
*msg
)
232 return UBUS_STATUS_UNKNOWN_ERROR
;
237 if (!(tm
= localtime(&now
)))
238 return UBUS_STATUS_UNKNOWN_ERROR
;
240 blob_buf_init(&b
, 0);
242 blobmsg_add_u32(&b
, "localtime", now
+ tm
->tm_gmtoff
);
245 blobmsg_add_u32(&b
, "uptime", info
.uptime
);
247 c
= blobmsg_open_array(&b
, "load");
248 blobmsg_add_u32(&b
, NULL
, info
.loads
[0]);
249 blobmsg_add_u32(&b
, NULL
, info
.loads
[1]);
250 blobmsg_add_u32(&b
, NULL
, info
.loads
[2]);
251 blobmsg_close_array(&b
, c
);
253 c
= blobmsg_open_table(&b
, "memory");
254 blobmsg_add_u64(&b
, "total", info
.mem_unit
* info
.totalram
);
255 blobmsg_add_u64(&b
, "free", info
.mem_unit
* info
.freeram
);
256 blobmsg_add_u64(&b
, "shared", info
.mem_unit
* info
.sharedram
);
257 blobmsg_add_u64(&b
, "buffered", info
.mem_unit
* info
.bufferram
);
258 blobmsg_close_table(&b
, c
);
260 c
= blobmsg_open_table(&b
, "swap");
261 blobmsg_add_u64(&b
, "total", info
.mem_unit
* info
.totalswap
);
262 blobmsg_add_u64(&b
, "free", info
.mem_unit
* info
.freeswap
);
263 blobmsg_close_table(&b
, c
);
266 ubus_send_reply(ctx
, req
, b
.head
);
268 return UBUS_STATUS_OK
;
271 static int system_reboot(struct ubus_context
*ctx
, struct ubus_object
*obj
,
272 struct ubus_request_data
*req
, const char *method
,
273 struct blob_attr
*msg
)
275 procd_shutdown(RB_AUTOBOOT
);
287 static const struct blobmsg_policy watchdog_policy
[__WDT_MAX
] = {
288 [WDT_FREQUENCY
] = { .name
= "frequency", .type
= BLOBMSG_TYPE_INT32
},
289 [WDT_TIMEOUT
] = { .name
= "timeout", .type
= BLOBMSG_TYPE_INT32
},
290 [WDT_MAGICCLOSE
] = { .name
= "magicclose", .type
= BLOBMSG_TYPE_BOOL
},
291 [WDT_STOP
] = { .name
= "stop", .type
= BLOBMSG_TYPE_BOOL
},
294 static int watchdog_set(struct ubus_context
*ctx
, struct ubus_object
*obj
,
295 struct ubus_request_data
*req
, const char *method
,
296 struct blob_attr
*msg
)
298 struct blob_attr
*tb
[__WDT_MAX
];
302 return UBUS_STATUS_INVALID_ARGUMENT
;
304 blobmsg_parse(watchdog_policy
, __WDT_MAX
, tb
, blob_data(msg
), blob_len(msg
));
305 if (tb
[WDT_FREQUENCY
]) {
306 unsigned int timeout
= watchdog_timeout(0);
307 unsigned int freq
= blobmsg_get_u32(tb
[WDT_FREQUENCY
]);
310 if (freq
> timeout
/ 2)
312 watchdog_frequency(freq
);
316 if (tb
[WDT_TIMEOUT
]) {
317 unsigned int timeout
= blobmsg_get_u32(tb
[WDT_TIMEOUT
]);
318 unsigned int frequency
= watchdog_frequency(0);
320 if (timeout
<= frequency
)
321 timeout
= frequency
* 2;
322 watchdog_timeout(timeout
);
325 if (tb
[WDT_MAGICCLOSE
])
326 watchdog_set_magicclose(blobmsg_get_bool(tb
[WDT_MAGICCLOSE
]));
329 watchdog_set_stopped(blobmsg_get_bool(tb
[WDT_STOP
]));
331 if (watchdog_fd() == NULL
)
333 else if (watchdog_get_stopped())
338 blob_buf_init(&b
, 0);
339 blobmsg_add_string(&b
, "status", status
);
340 blobmsg_add_u32(&b
, "timeout", watchdog_timeout(0));
341 blobmsg_add_u32(&b
, "frequency", watchdog_frequency(0));
342 blobmsg_add_u8(&b
, "magicclose", watchdog_get_magicclose());
343 ubus_send_reply(ctx
, req
, b
.head
);
354 static const struct blobmsg_policy signal_policy
[__SIGNAL_MAX
] = {
355 [SIGNAL_PID
] = { .name
= "pid", .type
= BLOBMSG_TYPE_INT32
},
356 [SIGNAL_NUM
] = { .name
= "signum", .type
= BLOBMSG_TYPE_INT32
},
359 static int proc_signal(struct ubus_context
*ctx
, struct ubus_object
*obj
,
360 struct ubus_request_data
*req
, const char *method
,
361 struct blob_attr
*msg
)
363 struct blob_attr
*tb
[__SIGNAL_MAX
];
366 return UBUS_STATUS_INVALID_ARGUMENT
;
368 blobmsg_parse(signal_policy
, __SIGNAL_MAX
, tb
, blob_data(msg
), blob_len(msg
));
369 if (!tb
[SIGNAL_PID
|| !tb
[SIGNAL_NUM
]])
370 return UBUS_STATUS_INVALID_ARGUMENT
;
372 kill(blobmsg_get_u32(tb
[SIGNAL_PID
]), blobmsg_get_u32(tb
[SIGNAL_NUM
]));
384 static const struct blobmsg_policy sysupgrade_policy
[__SYSUPGRADE_MAX
] = {
385 [SYSUPGRADE_PATH
] = { .name
= "path", .type
= BLOBMSG_TYPE_STRING
},
386 [SYSUPGRADE_PREFIX
] = { .name
= "prefix", .type
= BLOBMSG_TYPE_STRING
},
387 [SYSUPGRADE_COMMAND
] = { .name
= "command", .type
= BLOBMSG_TYPE_STRING
},
390 static int sysupgrade(struct ubus_context
*ctx
, struct ubus_object
*obj
,
391 struct ubus_request_data
*req
, const char *method
,
392 struct blob_attr
*msg
)
394 struct blob_attr
*tb
[__SYSUPGRADE_MAX
];
397 return UBUS_STATUS_INVALID_ARGUMENT
;
399 blobmsg_parse(sysupgrade_policy
, __SYSUPGRADE_MAX
, tb
, blob_data(msg
), blob_len(msg
));
400 if (!tb
[SYSUPGRADE_PATH
] || !tb
[SYSUPGRADE_PREFIX
])
401 return UBUS_STATUS_INVALID_ARGUMENT
;
403 sysupgrade_exec_upgraded(blobmsg_get_string(tb
[SYSUPGRADE_PREFIX
]),
404 blobmsg_get_string(tb
[SYSUPGRADE_PATH
]),
405 tb
[SYSUPGRADE_COMMAND
] ? blobmsg_get_string(tb
[SYSUPGRADE_COMMAND
]) : NULL
);
407 /* sysupgrade_exec_upgraded() will never return unless something has gone wrong */
408 return UBUS_STATUS_UNKNOWN_ERROR
;
412 procd_subscribe_cb(struct ubus_context
*ctx
, struct ubus_object
*obj
)
414 notify
= obj
->has_subscribers
;
418 static const struct ubus_method system_methods
[] = {
419 UBUS_METHOD_NOARG("board", system_board
),
420 UBUS_METHOD_NOARG("info", system_info
),
421 UBUS_METHOD_NOARG("reboot", system_reboot
),
422 UBUS_METHOD("watchdog", watchdog_set
, watchdog_policy
),
423 UBUS_METHOD("signal", proc_signal
, signal_policy
),
424 UBUS_METHOD("sysupgrade", sysupgrade
, sysupgrade_policy
),
427 static struct ubus_object_type system_object_type
=
428 UBUS_OBJECT_TYPE("system", system_methods
);
430 static struct ubus_object system_object
= {
432 .type
= &system_object_type
,
433 .methods
= system_methods
,
434 .n_methods
= ARRAY_SIZE(system_methods
),
435 .subscribe_cb
= procd_subscribe_cb
,
439 procd_bcast_event(char *event
, struct blob_attr
*msg
)
446 ret
= ubus_notify(_ctx
, &system_object
, event
, msg
, -1);
448 fprintf(stderr
, "Failed to notify log: %s\n", ubus_strerror(ret
));
451 void ubus_init_system(struct ubus_context
*ctx
)
456 ret
= ubus_add_object(ctx
, &system_object
);
458 ERROR("Failed to add object: %s\n", ubus_strerror(ret
));