prevent double free
[samba.git] / source3 / utils / net_time.c
1 /* 
2    Samba Unix/Linux SMB client library 
3    Version 3.0
4    net time command
5    Copyright (C) 2001 Andrew Tridgell (tridge@samba.org)
6
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
20
21 #include "includes.h"
22 #include "../utils/net.h"
23
24
25 /*
26   return the time on a server. This does not require any authentication
27 */
28 static time_t cli_servertime(const char *host, struct in_addr *ip)
29 {
30         struct nmb_name calling, called;
31         time_t ret = 0;
32         extern pstring global_myname;
33         struct cli_state *cli = NULL;
34
35         cli = cli_initialise(NULL);
36         if (!cli) goto done;
37
38         if (!cli_connect(cli, host, ip)) {
39                 fprintf(stderr,"Can't contact server\n");
40                 goto done;
41         }
42
43         make_nmb_name(&calling, global_myname, 0x0);
44         if (host) {
45                 make_nmb_name(&called, host, 0x20);
46         } else {
47                 make_nmb_name(&called, "*SMBSERVER", 0x20);
48         }
49
50         if (!cli_session_request(cli, &calling, &called)) {
51                 fprintf(stderr,"Session request failed\n");
52                 goto done;
53         }
54         if (!cli_negprot(cli)) {
55                 fprintf(stderr,"Protocol negotiation failed\n");
56                 goto done;
57         }
58
59         ret = cli->servertime;
60
61 done:
62         if (cli) cli_shutdown(cli);
63         return ret;
64 }
65
66 /* find the servers time on the opt_host host */
67 static time_t nettime(void)
68 {
69         extern BOOL opt_have_ip;
70         extern struct in_addr opt_dest_ip;
71         extern char *opt_host; 
72         return cli_servertime(opt_host, opt_have_ip? &opt_dest_ip : NULL);
73 }
74
75 /* return a time as a string ready to be passed to date -u */
76 static char *systime(time_t t)
77 {
78         static char s[100];
79         struct tm *tm;
80
81         tm = gmtime(&t);
82         
83         snprintf(s, sizeof(s), "%02d%02d%02d%02d%04d.%02d", 
84                  tm->tm_mon+1, tm->tm_mday, tm->tm_hour, 
85                  tm->tm_min, tm->tm_year + 1900, tm->tm_sec);
86         return s;
87 }
88
89 int net_time_usage(int argc, const char **argv)
90 {
91         d_printf(
92 "net time\n\tdisplays time on a server\n\n"\
93 "net time system\n\tdisplays time on a server in a format ready for /bin/date\n\n"\
94 "net time set\n\truns /bin/date -u with the time from the server\n\n"\
95 "\n");
96         general_rap_usage(argc, argv);
97         return -1;
98 }
99
100 /* try to set the system clock using /bin/date */
101 static int net_time_set(int argc, const char **argv)
102 {
103         time_t t = nettime();
104         char *cmd;
105
106         if (t == 0) return -1;
107         
108         /* yes, I know this is cheesy. Use "net time system" if you want to 
109            roll your own. I'm putting this in as it works on a large number
110            of systems and the user has a choice in whether its used or not */
111         asprintf(&cmd, "/bin/date -u %s", systime(t));
112         system(cmd);
113         free(cmd);
114
115         return 0;
116 }
117
118 /* display the time on a remote box in a format ready for /bin/date */
119 static int net_time_system(int argc, const char **argv)
120 {
121         time_t t = nettime();
122
123         if (t == 0) return -1;
124
125         printf("%s\n", systime(t));
126
127         return 0;
128 }
129
130 /* display or set the time on a host */
131 int net_time(int argc, const char **argv)
132 {
133         time_t t;
134         extern BOOL opt_have_ip;
135         extern struct in_addr opt_dest_ip;
136         extern char *opt_host; 
137         struct functable func[] = {
138                 {"SYSTEM", net_time_system},
139                 {"SET", net_time_set},
140                 {NULL, NULL}
141         };
142
143         if (!opt_host && !opt_have_ip) {
144                 d_printf("You must specify a hostname or IP\n");
145                 return -1;
146         }
147
148         if (argc != 0) {
149                 return net_run_function(argc, argv, func, net_time_usage);
150         }
151
152         /* default - print the time */
153         t = cli_servertime(opt_host, opt_have_ip? &opt_dest_ip : NULL);
154         if (t == 0) return -1;
155
156         d_printf("%s", ctime(&t));
157         return 0;
158 }