Fix NULL pointer dereference for stdin commands.
[paraslash.git] / client.c
1 /*
2  * Copyright (C) 1997-2008 Andre Noll <maan@systemlinux.org>
3  *
4  * Licensed under the GPL v2. For licencing details see COPYING.
5  */
6
7 /** \file client.c the client program used to connect to para_server */
8
9 #include "para.h"
10 #include "list.h"
11 #include "sched.h"
12 #include "client.cmdline.h"
13 #include "string.h"
14 #include "stdin.h"
15 #include "stdout.h"
16 #include "client.h"
17 #include "error.h"
18
19 INIT_CLIENT_ERRLISTS;
20
21 static struct client_task *ct;
22 static struct stdin_task sit;
23 static struct stdout_task sot;
24
25 static void supervisor_pre_select(struct sched *s, struct task *t)
26 {
27         if (ct->task.error < 0) {
28                 t->error = ct->task.error;
29                 return;
30         }
31         if (ct->status == CL_SENDING) {
32                 stdin_set_defaults(&sit);
33                 sit.buf = para_malloc(sit.bufsize),
34                 register_task(&sit.task);
35                 ct->inbuf = sit.buf;
36                 ct->in_loaded = &sit.loaded;
37                 ct->in_error = &sit.task.error;
38                 t->error = -E_TASK_STARTED;
39                 goto min_delay;
40         }
41         if (ct->status == CL_RECEIVING) {
42                 stdout_set_defaults(&sot);
43                 sot.buf = ct->buf;
44                 sot.loaded = &ct->loaded;
45                 sot.input_error = &ct->task.error;
46                 register_task(&sot.task);
47                 t->error = -E_TASK_STARTED;
48                 goto min_delay;
49         }
50         return;
51 min_delay:
52         s->timeout.tv_sec = 0;
53         s->timeout.tv_usec = 1;
54 }
55
56 static struct task svt = {
57         .pre_select = supervisor_pre_select,
58         .status = "supervisor task"
59 };
60
61 INIT_STDERR_LOGGING(ct->conf.loglevel_arg);
62
63
64 /**
65  * The client program to connect to para_server.
66  *
67  * \param argc Usual argument count.
68  * \param argv Usual argument vector.
69  *
70  * It registers two tasks: The client task that communicates with para_server
71  * and the standard out task that writes any output produced by the client task
72  * to standard out.
73  *
74  * \return EXIT_SUCCESS or EXIT_FAILURE
75  *
76  * \sa client_open(), stdout.c, stdout.h, para_client(1), para_server(1)
77  */
78 int main(int argc, char *argv[])
79 {
80
81         int ret;
82         struct sched s;
83
84         s.default_timeout.tv_sec = 1;
85         s.default_timeout.tv_usec = 0;
86         ret = client_open(argc, argv, &ct);
87         if (ret < 0) /* can not use PARA_LOG here */
88                 exit(EXIT_FAILURE);
89         register_task(&svt);
90         ret = schedule(&s);
91         if (ret < 0)
92                 PARA_ERROR_LOG("%s\n", para_strerror(-ret));
93         client_close(ct);
94         return ret >= 0? EXIT_SUCCESS: EXIT_FAILURE;
95 }