1 /***
2 This file is part of avahi.
3
4 avahi is free software; you can redistribute it and/or modify it
5 under the terms of the GNU Lesser General Public License as
6 published by the Free Software Foundation; either version 2.1 of the
7 License, or (at your option) any later version.
8
9 avahi is distributed in the hope that it will be useful, but WITHOUT
10 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
12 Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with avahi; if not, write to the Free Software
16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
17 USA.
18 ***/
19
20 #ifdef HAVE_CONFIG_H
21 #include <config.h>
22 #endif
23
24 #include <stdlib.h>
25 #include <stdio.h>
26 #include <getopt.h>
27 #include <assert.h>
28 #include <string.h>
29 #include <sys/types.h>
30 #include <sys/socket.h>
31 #include <locale.h>
32
33 #include <avahi-common/simple-watch.h>
34 #include <avahi-common/error.h>
35 #include "avahi-common/avahi-malloc.h"
36 #include <avahi-common/domain.h>
37 #include <avahi-common/i18n.h>
38 #include <avahi-client/client.h>
39
40 #include "sigint.h"
41
42 typedef enum {
43 COMMAND_UNSPEC,
44 COMMAND_HELP,
45 COMMAND_VERSION
46 } Command;
47
48 typedef struct Config {
49 int verbose;
50 Command command;
51 } Config;
52
53 static AvahiSimplePoll *simple_poll = NULL;
54 static AvahiClient *client = NULL;
55
client_callback(AvahiClient * c,AvahiClientState state,AVAHI_GCC_UNUSED void * userdata)56 static void client_callback(AvahiClient *c, AvahiClientState state, AVAHI_GCC_UNUSED void * userdata) {
57 switch (state) {
58 case AVAHI_CLIENT_FAILURE:
59 fprintf(stderr, _("Client failure, exiting: %s\n"), avahi_strerror(avahi_client_errno(c)));
60 avahi_simple_poll_quit(simple_poll);
61 break;
62
63 case AVAHI_CLIENT_S_REGISTERING:
64 case AVAHI_CLIENT_S_RUNNING:
65 case AVAHI_CLIENT_S_COLLISION:
66 case AVAHI_CLIENT_CONNECTING:
67 ;
68 }
69 }
70
help(FILE * f,const char * argv0)71 static void help(FILE *f, const char *argv0) {
72 fprintf(f,
73 _("%s [options] <new host name>\n\n"
74 " -h --help Show this help\n"
75 " -V --version Show version\n"
76 " -v --verbose Enable verbose mode\n"),
77 argv0);
78 }
79
parse_command_line(Config * c,int argc,char * argv[])80 static int parse_command_line(Config *c, int argc, char *argv[]) {
81 int o;
82
83 static const struct option long_options[] = {
84 { "help", no_argument, NULL, 'h' },
85 { "version", no_argument, NULL, 'V' },
86 { "verbose", no_argument, NULL, 'v' },
87 { NULL, 0, NULL, 0 }
88 };
89
90 assert(c);
91
92 c->command = COMMAND_UNSPEC;
93 c->verbose = 0;
94
95 while ((o = getopt_long(argc, argv, "hVv", long_options, NULL)) >= 0) {
96
97 switch(o) {
98 case 'h':
99 c->command = COMMAND_HELP;
100 break;
101 case 'V':
102 c->command = COMMAND_VERSION;
103 break;
104 case 'v':
105 c->verbose = 1;
106 break;
107 default:
108 return -1;
109 }
110 }
111
112 if (c->command == COMMAND_UNSPEC) {
113 if (optind != argc-1) {
114 fprintf(stderr, _("Invalid number of arguments, expecting exactly one.\n"));
115 return -1;
116 }
117 }
118
119 return 0;
120 }
121
main(int argc,char * argv[])122 int main(int argc, char *argv[]) {
123 int ret = 1, error;
124 Config config;
125 const char *argv0;
126
127 avahi_init_i18n();
128 setlocale(LC_ALL, "");
129
130 if ((argv0 = strrchr(argv[0], '/')))
131 argv0++;
132 else
133 argv0 = argv[0];
134
135 if (parse_command_line(&config, argc, argv) < 0)
136 goto fail;
137
138 switch (config.command) {
139 case COMMAND_HELP:
140 help(stdout, argv0);
141 ret = 0;
142 break;
143
144 case COMMAND_VERSION:
145 printf("%s "PACKAGE_VERSION"\n", argv0);
146 ret = 0;
147 break;
148
149 case COMMAND_UNSPEC:
150
151 if (!(simple_poll = avahi_simple_poll_new())) {
152 fprintf(stderr, _("Failed to create simple poll object.\n"));
153 goto fail;
154 }
155
156 if (sigint_install(simple_poll) < 0)
157 goto fail;
158
159 if (!(client = avahi_client_new(avahi_simple_poll_get(simple_poll), 0, client_callback, NULL, &error))) {
160 fprintf(stderr, _("Failed to create client object: %s\n"), avahi_strerror(error));
161 goto fail;
162 }
163
164 if (config.verbose) {
165 const char *version, *hn;
166
167 if (!(version = avahi_client_get_version_string(client))) {
168 fprintf(stderr, _("Failed to query version string: %s\n"), avahi_strerror(avahi_client_errno(client)));
169 goto fail;
170 }
171
172 if (!(hn = avahi_client_get_host_name_fqdn(client))) {
173 fprintf(stderr, _("Failed to query host name: %s\n"), avahi_strerror(avahi_client_errno(client)));
174 goto fail;
175 }
176
177 fprintf(stderr, _("Server version: %s; Host name: %s\n"), version, hn);
178 }
179
180 if (avahi_client_set_host_name(client, argv[optind]) < 0) {
181 fprintf(stderr, _("Failed to create host name resolver: %s\n"), avahi_strerror(avahi_client_errno(client)));
182 goto fail;
183 }
184
185 if (config.verbose) {
186 const char *hn;
187
188 if (!(hn = avahi_client_get_host_name_fqdn(client))) {
189 fprintf(stderr, _("Failed to query host name: %s\n"), avahi_strerror(avahi_client_errno(client)));
190 goto fail;
191 }
192
193 fprintf(stderr, _("Host name successfully changed to %s\n"), hn);
194 }
195
196 ret = 0;
197 break;
198 }
199
200 fail:
201
202 if (client)
203 avahi_client_free(client);
204
205 sigint_uninstall();
206
207 if (simple_poll)
208 avahi_simple_poll_free(simple_poll);
209
210 return ret;
211 }
212