| /*
* sai-builder
*
* Copyright (C) 2019 - 2025 Andy Green <andy@warmcat.com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation:
* version 2.1 of the License.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301 USA
*/
#include <libwebsockets.h>
#include <string.h>
#include <signal.h>
#include <limits.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/types.h>
#if !defined(WIN32)
#include <pwd.h>
#include <grp.h>
#endif
#if defined(__linux__) || defined(__APPLE__)
#include <unistd.h>
#endif
#if defined(__APPLE__)
#include <sys/stat.h> /* for mkdir() */
#endif
#if defined(WIN32)
#include <initguid.h>
#include <KnownFolders.h>
#include <Shlobj.h>
#include <processthreadsapi.h>
#include <handleapi.h>
#if !defined(PATH_MAX)
#define PATH_MAX MAX_PATH
#endif
#endif
#include "b-private.h"
struct lws_spawn_piped *lsp_suspender;
struct lws_protocols protocol_suspender_stdxxx;
char suspender_exists;
int
saib_suspender_get_pipe(void)
{
#if defined(__linux__)
int fd = lws_spawn_get_fd_stdxxx(lsp_suspender, 0);
#else
#if defined(__APPLE__) || defined(__NetBSD__)
int fd = builder.pipe_suspender_wr;
#else
int fd = 2;
#endif
#endif
return fd;
}
void *
saib_thread_suspend(void *d)
{
return NULL;
}
static int
callback_sai_suspender_stdwsi(struct lws *wsi, enum lws_callback_reasons reason,
void *user, void *in, size_t len)
{
uint8_t buf[200];
int ilen;
switch (reason) {
case LWS_CALLBACK_RAW_CLOSE_FILE:
lws_spawn_stdwsi_closed(lsp_suspender, wsi);
break;
case LWS_CALLBACK_RAW_RX_FILE:
#if defined(WIN32)
{
DWORD rb;
if (!ReadFile((HANDLE)lws_get_socket_fd(wsi), buf, sizeof(buf), &rb, NULL)) {
lwsl_debug("%s: read on stdwsi failed\n", __func__);
return -1;
}
ilen = (int)rb;
}
#else
ilen = (int)read((int)(intptr_t)lws_get_socket_fd(wsi), buf, sizeof(buf));
if (ilen < 1) {
lwsl_debug("%s: read on stdwsi failed\n", __func__);
return -1;
}
#endif
len = (unsigned int)ilen;
lwsl_notice("%s: suspender: %.*s\n", __func__, (int)len, buf);
break;
default:
break;
}
return 0;
}
struct lws_protocols protocol_suspender_stdxxx =
{ "sai-suspender-stdxxx", callback_sai_suspender_stdwsi, 0, 0 };
#if !defined(__APPLE__) && !defined(__NetBSD__) && !defined(__OpenBSD__) && !defined(__FreeBSD__)
static void reap(void *opaque, const lws_spawn_resource_us_t *res,
siginfo_t *si, int we_killed_him)
{
// lwsl_err("%s: reaped suspender fork... %d\n", __func__, si->si_status);
}
#endif
int
saib_suspender_fork(const char *path)
{
char rpath[PATH_MAX];
#if defined(__linux__)
struct lws_spawn_piped_info info;
const char * const ea[] = { rpath, "-s", NULL };
#endif
#if !defined(WIN32)
if (!realpath(path, rpath)) {
lwsl_err("%s: failed to get realpath for %s: %s\n", __func__,
path, strerror(errno));
return 1;
}
#else
lws_strncpy(rpath, path, sizeof(rpath) - 1);
#endif
lwsl_err("%s: starting %s\n", __func__, rpath);
#if defined(__linux__)
memset(&info, 0, sizeof(info));
memset(&builder.suspend_nspawn, 0, sizeof(builder.suspend_nspawn));
info.vh = builder.vhost;
info.exec_array = ea;
info.max_log_lines = 100;
info.opaque = (void *)&builder.suspend_nspawn;
info.protocol_name = "sai-suspender-stdxxx";
info.plsp = &lsp_suspender;
info.reap_cb = reap;
lsp_suspender = lws_spawn_piped(&info);
if (!lsp_suspender) {
lwsl_err("%s: suspend spawn failed\n", __func__);
return 1;
}
#endif
#if defined(__APPLE__) || defined(__NetBSD__)
{
int pfd[2];
pid_t pid;
if (pipe(pfd) == -1) {
lwsl_err("pipe() failed\n");
return 1;
}
fcntl(pfd[0], F_SETFD, FD_CLOEXEC);
fcntl(pfd[1], F_SETFD, FD_CLOEXEC);
pid = fork();
if (pid == -1) {
lwsl_err("fork() failed\n");
return 1;
}
if (!pid) {
close(pfd[1]); /* wr */
if (dup2(pfd[0], 0) < 0)
return 1;
close(pfd[0]);
execlp(rpath, rpath, "-s", (char *)NULL);
lwsl_err("execlp failed\n");
return 1;
}
/* parent */
close(pfd[0]); /* rd */
builder.pipe_suspender_wr = pfd[1];
}
#endif
suspender_exists = 1;
/*
* We start off idle, with no tasks on any platform and doing
* the grace time before suspend. If tasks appear, the grace
* time will get cancelled.
*/
lws_sul_schedule(builder.context, 0, &builder.sul_idle,
sul_idle_cb, SAI_IDLE_GRACE_US);
lwsl_err("%s: done\n", __func__);
return 0;
}
int
saib_suspender_start(void)
{
ssize_t n = 0;
printf("%s: Spawn process creation entry...\n", __func__);
/*
* A new process gets started with this option before we drop
* privs. This allows us to suspend with root privs later.
*
* We just wait until we get a byte on stdin from the main
* process indicating we should suspend.
*/
while (n >= 0) {
#if !defined(WIN32)
int status;
pid_t p;
#endif
uint8_t d;
n = read(0, &d, 1);
lwsl_notice("%s: suspend process read returned %d\n", __func__, (int)n);
#if defined(__APPLE__)
sleep(1);
#endif
if (n <= 0)
continue;
if (d == 2) {
lwsl_warn("%s: suspend process ending\n", __func__);
break;
}
#if defined(WIN32)
/*
* On Windows, we can use the shutdown command.
* For suspend, we can use rundll32.
*/
switch(d) {
case 0:
system("shutdown /s /t 0");
break;
case 1:
system("rundll32.exe powrprof.dll,SetSuspendState 0,1,0");
break;
case 3:
if (builder.rebuild_script_user &&
builder.rebuild_script_root) {
if (system(builder.rebuild_script_user) == 0)
system(builder.rebuild_script_root);
}
break;
}
#else
p = fork();
if (!p)
switch(d) {
case 0:
#if defined(__NetBSD__)
execl("/sbin/shutdown", "/sbin/shutdown", "-h", "now", NULL);
#else
execl("/usr/sbin/shutdown", "/usr/sbin/shutdown", "--halt", "now", NULL);
#endif
break;
case 1:
execl("/usr/bin/systemctl", "/usr/bin/systemctl", "suspend", NULL);
break;
case 3:
if (builder.rebuild_script_user &&
builder.rebuild_script_root) {
char cmd[8192];
char user_buf[64];
const char *user = "sai";
int ret;
if (builder.perms) {
lws_strncpy(user_buf, builder.perms, sizeof(user_buf));
if (strchr(user_buf, ':'))
*strchr(user_buf, ':') = '\0';
user = user_buf;
}
lws_snprintf(cmd, sizeof(cmd),
"su - %s -c '%s'", user,
builder.rebuild_script_user);
ret = system(cmd);
if (ret == 0) {
lws_snprintf(cmd, sizeof(cmd),
"PATH=/usr/local/bin:/usr/bin:/bin:/sbin:/usr/sbin %s",
builder.rebuild_script_root);
system(cmd);
}
}
break;
}
else
waitpid(p, &status, 0);
#endif
}
lwsl_notice("%s: exiting suspend process\n", __func__);
return 0;
}
void
suspender_destroy()
{
if (suspender_exists) {
int fd = saib_suspender_get_pipe();
uint8_t te = 2;
/*
* Clean up after the suspend process
*/
write(fd, &te, 1);
}
}
|