416a790098d43416cdf3480d1b431f1749a24725
1 /* Copyright (C) 2018 Thomas Balzer */
3 /* This file is part of tomd. */
5 /* tomd is free software: you can redistribute it and/or modify */
6 /* it under the terms of the GNU General Public License as published by */
7 /* the Free Software Foundation, either version 3 of the License, or */
8 /* (at your option) any later version. */
10 /* tomd is distributed in the hope that it will be useful, */
11 /* but WITHOUT ANY WARRANTY; without even the implied warranty of */
12 /* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the */
13 /* GNU General Public License for more details. */
15 /* You should have received a copy of the GNU General Public License */
16 /* along with tomd. If not, see <http://www.gnu.org/licenses/>. */
24 #include "../../include/job.h"
25 #include "../../include/macros.h"
28 #include "../../include/scm_interface.h"
31 #define MANIFEST_LOC "/.config/tomd/init/manifest.scm"
35 static int root_job
= -1;
38 static struct job jobs
[MAX_JOBS
] =
42 "/bin/bash", /* cmd */
53 static struct job jobs
[MAX_JOBS
] = {};
57 static struct job
load_job(SCM job_list
, int index
)
59 struct job ret
= { NULL
};
60 SCM scm_cur_job
= SCM_ARR(job_list
, index
);
62 if(scm_is_false(job_predicate(scm_cur_job
))){
63 tomd_p("job %d wasn't a real job type.", index
);
67 SCM scm_name
= get_name(scm_cur_job
);
68 SCM scm_cmd
= get_cmd(scm_cur_job
);
69 SCM scm_args
= get_args(scm_cur_job
);
71 /* TODO > Handle these. */
72 /* planned is to be able to run at different points, and at */
73 /* the request of a user. for testing we do things all at */
75 SCM scm_start_trigger
= get_start_trigger(scm_cur_job
);
76 SCM scm_end_trigger
= get_end_trigger(scm_cur_job
);
78 char *job_name
= scm_to_locale_string(scm_name
);
79 char *job_cmd
= scm_to_locale_string(scm_cmd
);
81 int jlen
= scm_to_int(scm_length(scm_args
));
86 scm_to_locale_string(SCM_ARR(scm_args
, j
));
95 ret
.args
[j
] = real_args
[j
- 1];
97 ret
.args
[jlen
+ 1] = NULL
;
98 ret
.args
[0] = ret
.cmd
;
103 static void *load_manifest(void *args
)
107 tomd_p("arg to load_manifest is NULL");
110 manifest_loc
= (char *)args
;
112 scm_c_primitive_load(args
);
115 scm_c_public_ref("tomd manifest", "job-list");
117 if(scm_is_false(scm_job_list
)){
118 tomd_p("no job-list found in manifest.scm");
122 if(scm_is_false(scm_list_p(scm_job_list
))){
123 tomd_p("job-list found, but isn't a list.");
128 int len
= SCM_LIST_LEN(scm_job_list
);
129 tomd_p("len=%d, max=%d", len
, MAX_JOBS
);
131 i
< len
&& i
< MAX_JOBS
;
133 jobs
[i
] = load_job(scm_job_list
, i
);
135 tomd_p("looked at %d jobs.", i
);
140 static char *lookup_user_folder(void)
143 uid_t user
= getuid();
144 struct passwd
*userpasswd
=
147 /* take the manifest location and sub ~ for pw_dir */
148 char *buffer
= (char *)malloc(sizeof(char) * 300);
149 strcpy(buffer
, userpasswd
->pw_dir
);
150 int len
= strlen(userpasswd
->pw_dir
);
151 strcpy(buffer
+ len
, MANIFEST_LOC
);
160 char *manifest
= lookup_user_folder();
162 tomd_p("Loading jobs from '%s'", manifest
);
165 scm_with_guile(load_manifest
,
168 tomd_p("Stubbed job loading.");
170 tomd_p("Finished loading.");
175 jobs
[i
].cmd
== NULL
) {
178 tomd_p("JOB <%d>:", i
);
179 tomd_p(" cmd:'%s'", jobs
[i
].cmd
);
183 jobs
[i
].args
[j
] == NULL
){
186 tomd_p(" arg [%d]:'%s'", j
, jobs
[i
].args
[j
]);
194 static int stdincache
;
195 static int stdoutcache
;
196 static int stderrcache
;
200 /* cache all filedes */
201 stdincache
= dup(STDIN_FILENO
);
202 stdoutcache
= dup(STDOUT_FILENO
);
203 stderrcache
= dup(STDERR_FILENO
);
205 tomd_p("going silent.");
207 /* close default filedes */
209 close(STDOUT_FILENO
);
210 close(STDERR_FILENO
);
212 tomd_p("now silent.");
215 #define LOG_DIR "/var/log/tomd/"
217 void run_job(int index
)
219 struct job
*job
= &jobs
[index
];
222 if(pid
== 0){ /* child */
224 /* redirect to a file */
226 strcpy(buf
, LOG_DIR
);
227 strcpy(buf
+ strlen(LOG_DIR
), job
->name
);
228 tomd_p("redirecting stdout to %s.", buf
);
229 if(unlink(buf
) != 0){
230 tomd_p("file couldn't be removed.");
233 int fd
= open(buf
, O_WRONLY
| O_CREAT
, 0644);
235 tomd_p("couldn't open file.");
239 dup2(fd
, STDOUT_FILENO
);
240 dup2(fd
, STDERR_FILENO
);
243 execvp(job
->cmd
, job
->args
);
244 tomd_panic("execvp for '%s' failed", job
->cmd
);
246 tomd_p("forked [%d] to run '%s'", pid
, job
->cmd
);
248 waitpid(job
->pid
, &job
->last_status
, WNOHANG
);
252 struct job
*lookup_job(char *my_job_name
)
257 if(jobs
[i
].name
== NULL
){
260 if(strcmp(jobs
[i
].name
, my_job_name
) == 0){
273 if(jobs
[i
].cmd
== NULL
){
274 tomd_p("out of jobs");
277 tomd_p("running job [%d] '%s'", i
, jobs
[i
].cmd
);
280 if(root_job
!= i
&& root_job
!= -1){
281 tomd_p("error! only one job can be root.");
290 static int delay
= 0;
292 void check_root_job(void)
295 if(delay
% 100 != 0){
301 int n_pid
= waitpid(jobs
[root_job
].pid
, &jobs
[root_job
].last_status
, WNOHANG
);
302 if(n_pid
!= jobs
[root_job
].pid
){
306 if(WIFEXITED(jobs
[root_job
].last_status
)){
307 /* we died, restart it */
308 tomd_p("restarting root job. (last pid = %d, last status = %d)",
309 jobs
[root_job
].pid
, jobs
[root_job
].last_status
);