1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
/*
Copyright (C) 2016 Jeremy White <jwhite@codeweavers.com>
All rights reserved.
This file is part of x11spice
x11spice is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
x11spice 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with x11spice. If not, see <http://www.gnu.org/licenses/>.
*/
/*----------------------------------------------------------------------------
** main.c
** The main program for x11spice. The hope is that this code
** remains simple, and easy to follow, and that logic is broken neatly
** down in subordinate modules.
**--------------------------------------------------------------------------*/
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include "x11spice.h"
#include "options.h"
#include "local_spice.h"
#include "display.h"
#include "agent.h"
#include "gui.h"
#include "session.h"
static void sigterm_handler(int arg G_GNUC_UNUSED)
{
gui_sigterm();
}
static void handle_sigterm(void)
{
struct sigaction act = { };
act.sa_handler = sigterm_handler;
sigaction(SIGTERM, &act, NULL);
}
int main(int argc, char *argv[])
{
int rc;
session_t session = { };
int display_opened = 0;
int spice_started = 0;
int gui_created = 0;
int session_created = 0;
int session_started = 0;
/*------------------------------------------------------------------------
** Parse arguments
**----------------------------------------------------------------------*/
options_init(&session.options);
rc = options_load(&session.options, argc, argv);
if (rc)
goto exit;
/*------------------------------------------------------------------------
** Create the session
**----------------------------------------------------------------------*/
rc = session_create(&session);
if (rc)
goto exit;
session_created = 1;
/*------------------------------------------------------------------------
** Open the display
**----------------------------------------------------------------------*/
rc = display_open(&session.display, &session);
if (rc)
goto exit;
display_opened = 1;
/*------------------------------------------------------------------------
** Initialize the GUI
**----------------------------------------------------------------------*/
rc = gui_create(&session.gui, &session, argc, argv);
if (rc)
goto exit;
gui_created = 1;
/*------------------------------------------------------------------------
** Make sure we have a way to operate.
**----------------------------------------------------------------------*/
if (options_impossible_config(&session.options)) {
gui_report_error(&session.gui,
"You must provide some authentication method.\n"
"Refer to the x11spice man page for more details.");
goto exit;
}
/*------------------------------------------------------------------------
** Start up a spice server
**----------------------------------------------------------------------*/
rc = spice_start(&session.spice, &session.options, session.display.primary);
if (rc)
goto exit;
spice_started = 1;
agent_start(&session.spice, &session.options, &session.agent);
/*------------------------------------------------------------------------
** Start our session and leave the GUI running until we have
** a reason to quit
**----------------------------------------------------------------------*/
rc = session_start(&session);
if (rc)
goto exit;
session_started = 1;
handle_sigterm();
/*------------------------------------------------------------------------
** The spice server (prior to November 2016) can call exit() prior to
** exiting, most notably when we have exit-on-disconnect turned on.
** By registering this handler, we ensure that disconnect scripts are
** called even if that happens
**----------------------------------------------------------------------*/
if (session.options.on_disconnect || session.options.audit)
atexit(session_remote_disconnected);
gui_run(&session.gui);
/*------------------------------------------------------------------------
** Clean up, go home
**----------------------------------------------------------------------*/
exit:
if (session_started)
session_end(&session);
if (spice_started) {
agent_stop(&session.agent);
spice_end(&session.spice);
}
if (gui_created)
gui_destroy(&session.gui);
if (display_opened)
display_close(&session.display);
if (session_created)
session_destroy(&session);
options_free(&session.options);
return rc;
}
|
