3 * Bacula Director -- User Agent Server
5 * Kern Sibbald, September MM
11 Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
13 This program is free software; you can redistribute it and/or
14 modify it under the terms of the GNU General Public License as
15 published by the Free Software Foundation; either version 2 of
16 the License, or (at your option) any later version.
18 This program is distributed in the hope that it will be useful,
19 but WITHOUT ANY WARRANTY; without even the implied warranty of
20 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21 General Public License for more details.
23 You should have received a copy of the GNU General Public
24 License along with this program; if not, write to the Free
25 Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
34 /* Imported subroutines */
35 extern void run_job(JCR *jcr);
37 /* Imported variables */
40 extern struct s_res resources[];
41 extern int console_msg_pending;
42 extern char my_name[];
44 /* Static variables */
46 /* Exported variables */
47 int quit_cmd_thread = 0;
49 /* Imported functions */
51 /* Forward referenced functions */
53 static void *connect_thread(void *arg);
54 static void handle_UA_client_request(void *arg);
57 /* Global variables */
58 static int started = FALSE;
59 static workq_t ua_workq;
61 /* Called here by Director daemon to start UA (user agent)
62 * command thread. This routine creates the thread and then
65 void start_UA_server(int UA_port)
70 if ((status=pthread_create(&thid, NULL, connect_thread, (void *)UA_port)) != 0) {
71 Emsg1(M_ABORT, 0, _("Cannot create UA thread: %s\n"), strerror(status));
77 static void *connect_thread(void *arg)
79 int UA_port = (int)arg;
81 pthread_detach(pthread_self());
83 bnet_thread_server(UA_port, 5, &ua_workq, handle_UA_client_request);
88 * Handle Director User Agent commands
91 static void handle_UA_client_request(void *arg)
94 static char cmd[1000];
96 BSOCK *UA_sock = (BSOCK *) arg;
98 pthread_detach(pthread_self());
100 memset(&ua, 0, sizeof(ua));
103 ua.jcr = new_jcr(sizeof(JCR), dird_free_jcr);
104 ua.jcr->sd_auth_key = bstrdup("dummy"); /* dummy Storage daemon key */
105 ua.UA_sock = UA_sock;
106 ua.cmd = get_pool_memory(PM_FNAME);
107 ua.args = get_pool_memory(PM_FNAME);
109 create_unique_job_name(ua.jcr, "*Console*");
110 ua.jcr->sched_time = ua.jcr->start_time;
111 ua.jcr->JobType = JT_CONSOLE;
113 bnet_recv(ua.UA_sock); /* Get first message */
114 if (!authenticate_user_agent(ua.UA_sock)) {
119 stat = bnet_recv(ua.UA_sock);
120 Dmsg1(500, "stat=%d\n", stat);
122 strncpy(cmd, ua.UA_sock->msg, sizeof(cmd));
123 cmd[sizeof(cmd)-1] = 0; /* ensure it is terminated/trucated */
124 parse_command_args(&ua);
125 if (ua.argc > 0 && ua.argk[0][0] == '.') {
126 do_a_dot_command(&ua, cmd);
128 do_a_command(&ua, cmd);
131 if (ua.auto_display_messages) {
132 strcpy(cmd, "messages");
133 qmessagescmd(&ua, cmd);
134 ua.user_notified_msg_pending = FALSE;
135 } else if (!ua.user_notified_msg_pending && console_msg_pending) {
136 bsendmsg(&ua, _("You have messages.\n"));
137 ua.user_notified_msg_pending = TRUE;
139 bnet_sig(ua.UA_sock, BNET_EOD); /* send end of command */
141 } else if (stat == 0) {
142 if (ua.UA_sock->msglen == BNET_TERMINATE ||
143 ua.UA_sock->msglen == BNET_EOF) {
147 Dmsg1(000, "stat=0 msglen=%d\n", ua.UA_sock->msglen);
148 bnet_sig(ua.UA_sock, BNET_POLL);
150 break; /* error, exit */
156 bnet_close(ua.UA_sock);
160 close_db(&ua); /* do this before freeing JCR */
170 free_pool_memory(ua.cmd);
173 free_pool_memory(ua.args);
179 * Called from main Bacula thread
181 void term_ua_server()
186 quit_cmd_thread = TRUE;