Implement sk_addr_dup().
[u/mdw/putty] / logging.c
CommitLineData
eaf1e20a 1/*
2 * Session logging.
3 */
4
00db133f 5#include <stdio.h>
6#include <stdlib.h>
7#include <ctype.h>
8
9#include <time.h>
10#include <assert.h>
11
12#include "putty.h"
13
14/* log session to file stuff ... */
a8327734 15struct LogContext {
16 FILE *lgfp;
8b029099 17 enum { L_CLOSED, L_OPENING, L_OPEN, L_ERROR } state;
919baedb 18 bufchain queue;
9a30e26b 19 Filename currlogfilename;
a8327734 20 void *frontend;
c229ef97 21 Config cfg;
a8327734 22};
00db133f 23
9a30e26b 24static void xlatlognam(Filename *d, Filename s, char *hostname, struct tm *tm);
00db133f 25
26/*
919baedb 27 * Internal wrapper function which must be called for _all_ output
28 * to the log file. It takes care of opening the log file if it
29 * isn't open, buffering data if it's in the process of being
30 * opened asynchronously, etc.
00db133f 31 */
919baedb 32static void logwrite(struct LogContext *ctx, void *data, int len)
00db133f 33{
919baedb 34 /*
8b029099 35 * In state L_CLOSED, we call logfopen, which will set the state
36 * to one of L_OPENING, L_OPEN or L_ERROR. Hence we process all of
37 * those three _after_ processing L_CLOSED.
919baedb 38 */
8b029099 39 if (ctx->state == L_CLOSED)
919baedb 40 logfopen(ctx);
41
8b029099 42 if (ctx->state == L_OPENING) {
919baedb 43 bufchain_add(&ctx->queue, data, len);
8b029099 44 } else if (ctx->state == L_OPEN) {
919baedb 45 assert(ctx->lgfp);
46 fwrite(data, 1, len, ctx->lgfp);
8b029099 47 } /* else L_ERROR, so ignore the write */
919baedb 48}
49
50/*
51 * Convenience wrapper on logwrite() which printf-formats the
52 * string.
53 */
54static void logprintf(struct LogContext *ctx, const char *fmt, ...)
55{
56 va_list ap;
57 char *data;
58
59 va_start(ap, fmt);
60 data = dupvprintf(fmt, ap);
61 va_end(ap);
62
63 logwrite(ctx, data, strlen(data));
64 sfree(data);
00db133f 65}
66
67/*
11cc5e30 68 * Flush any open log file.
69 */
70void logflush(void *handle) {
71 struct LogContext *ctx = (struct LogContext *)handle;
72 if (ctx->cfg.logtype > 0)
8b029099 73 if (ctx->state == L_OPEN)
11cc5e30 74 fflush(ctx->lgfp);
75}
76
919baedb 77static void logfopen_callback(void *handle, int mode)
78{
79 struct LogContext *ctx = (struct LogContext *)handle;
80 char buf[256], *event;
81 struct tm tm;
82 const char *fmode;
83
84 if (mode == 0) {
8b029099 85 ctx->state = L_ERROR; /* disable logging */
919baedb 86 } else {
9cd439a3 87 fmode = (mode == 1 ? "ab" : "wb");
c6940f12 88 ctx->lgfp = f_open(ctx->currlogfilename, fmode, TRUE);
919baedb 89 if (ctx->lgfp)
8b029099 90 ctx->state = L_OPEN;
919baedb 91 else
8b029099 92 ctx->state = L_ERROR;
919baedb 93 }
94
8b029099 95 if (ctx->state == L_OPEN) {
919baedb 96 /* Write header line into log file. */
97 tm = ltime();
98 strftime(buf, 24, "%Y.%m.%d %H:%M:%S", &tm);
99 logprintf(ctx, "=~=~=~=~=~=~=~=~=~=~=~= PuTTY log %s"
100 " =~=~=~=~=~=~=~=~=~=~=~=\r\n", buf);
101 }
102
103 event = dupprintf("%s session log (%s mode) to file: %s",
104 (mode == 0 ? "Disabled writing" :
105 mode == 1 ? "Appending" : "Writing new"),
106 (ctx->cfg.logtype == LGTYP_ASCII ? "ASCII" :
107 ctx->cfg.logtype == LGTYP_DEBUG ? "raw" :
108 ctx->cfg.logtype == LGTYP_PACKETS ? "SSH packets" :
bf8a49a1 109 ctx->cfg.logtype == LGTYP_SSHRAW ? "SSH raw data" :
919baedb 110 "unknown"),
111 filename_to_str(&ctx->currlogfilename));
112 logevent(ctx->frontend, event);
113 sfree(event);
114
115 /*
116 * Having either succeeded or failed in opening the log file,
117 * we should write any queued data out.
118 */
8b029099 119 assert(ctx->state != L_OPENING); /* make _sure_ it won't be requeued */
919baedb 120 while (bufchain_size(&ctx->queue)) {
121 void *data;
122 int len;
123 bufchain_prefix(&ctx->queue, &data, &len);
124 logwrite(ctx, data, len);
125 bufchain_consume(&ctx->queue, len);
126 }
127}
128
129/*
130 * Open the log file. Takes care of detecting an already-existing
131 * file and asking the user whether they want to append, overwrite
132 * or cancel logging.
133 */
134void logfopen(void *handle)
135{
136 struct LogContext *ctx = (struct LogContext *)handle;
137 struct tm tm;
138 int mode;
139
140 /* Prevent repeat calls */
8b029099 141 if (ctx->state != L_CLOSED)
919baedb 142 return;
143
144 if (!ctx->cfg.logtype)
145 return;
146
147 tm = ltime();
148
149 /* substitute special codes in file name */
150 xlatlognam(&ctx->currlogfilename, ctx->cfg.logfilename,ctx->cfg.host, &tm);
151
c6940f12 152 ctx->lgfp = f_open(ctx->currlogfilename, "r", FALSE); /* file already present? */
919baedb 153 if (ctx->lgfp) {
154 fclose(ctx->lgfp);
155 if (ctx->cfg.logxfovr != LGXF_ASK) {
156 mode = ((ctx->cfg.logxfovr == LGXF_OVR) ? 2 : 1);
157 } else
158 mode = askappend(ctx->frontend, ctx->currlogfilename,
159 logfopen_callback, ctx);
160 } else
161 mode = 2; /* create == overwrite */
162
163 if (mode < 0)
8b029099 164 ctx->state = L_OPENING;
919baedb 165 else
166 logfopen_callback(ctx, mode); /* open the file */
167}
168
169void logfclose(void *handle)
170{
171 struct LogContext *ctx = (struct LogContext *)handle;
172 if (ctx->lgfp) {
173 fclose(ctx->lgfp);
174 ctx->lgfp = NULL;
175 }
8b029099 176 ctx->state = L_CLOSED;
919baedb 177}
178
179/*
180 * Log session traffic.
181 */
182void logtraffic(void *handle, unsigned char c, int logmode)
183{
184 struct LogContext *ctx = (struct LogContext *)handle;
185 if (ctx->cfg.logtype > 0) {
186 if (ctx->cfg.logtype == logmode)
187 logwrite(ctx, &c, 1);
188 }
189}
190
11cc5e30 191/*
382908ad 192 * Log an Event Log entry. Used in SSH packet logging mode; this is
193 * also as convenient a place as any to put the output of Event Log
194 * entries to stderr when a command-line tool is in verbose mode.
195 * (In particular, this is a better place to put it than in the
196 * front ends, because it only has to be done once for all
197 * platforms. Platforms which don't have a meaningful stderr can
198 * just avoid defining FLAG_STDERR.
fb89f7ff 199 */
cbe2d68f 200void log_eventlog(void *handle, const char *event)
fb89f7ff 201{
a8327734 202 struct LogContext *ctx = (struct LogContext *)handle;
382908ad 203 if ((flags & FLAG_STDERR) && (flags & FLAG_VERBOSE)) {
204 fprintf(stderr, "%s\n", event);
205 fflush(stderr);
206 }
c9ef99df 207 /* If we don't have a context yet (eg winnet.c init) then skip entirely */
208 if (!ctx)
209 return;
bf8a49a1 210 if (ctx->cfg.logtype != LGTYP_PACKETS &&
211 ctx->cfg.logtype != LGTYP_SSHRAW)
fb89f7ff 212 return;
919baedb 213 logprintf(ctx, "Event Log: %s\r\n", event);
bf8a49a1 214 logflush(ctx);
fb89f7ff 215}
216
217/*
00db133f 218 * Log an SSH packet.
9a10ecf4 219 * If n_blanks != 0, blank or omit some parts.
220 * Set of blanking areas must be in increasing order.
00db133f 221 */
a8327734 222void log_packet(void *handle, int direction, int type,
9a10ecf4 223 char *texttype, void *data, int len,
224 int n_blanks, const struct logblank_t *blanks)
00db133f 225{
a8327734 226 struct LogContext *ctx = (struct LogContext *)handle;
00db133f 227 char dumpdata[80], smalldata[5];
919baedb 228 int p = 0, b = 0, omitted = 0;
229 int output_pos = 0; /* NZ if pending output in dumpdata */
00db133f 230
bf8a49a1 231 if (!(ctx->cfg.logtype == LGTYP_SSHRAW ||
232 (ctx->cfg.logtype == LGTYP_PACKETS && texttype)))
00db133f 233 return;
9a10ecf4 234
919baedb 235 /* Packet header. */
bf8a49a1 236 if (texttype)
237 logprintf(ctx, "%s packet type %d / 0x%02x (%s)\r\n",
238 direction == PKT_INCOMING ? "Incoming" : "Outgoing",
239 type, type, texttype);
240 else
241 logprintf(ctx, "%s raw data\r\n",
242 direction == PKT_INCOMING ? "Incoming" : "Outgoing");
919baedb 243
244 /*
245 * Output a hex/ASCII dump of the packet body, blanking/omitting
246 * parts as specified.
247 */
248 while (p < len) {
249 int blktype;
250
251 /* Move to a current entry in the blanking array. */
252 while ((b < n_blanks) &&
253 (p >= blanks[b].offset + blanks[b].len))
254 b++;
255 /* Work out what type of blanking to apply to
256 * this byte. */
257 blktype = PKTLOG_EMIT; /* default */
258 if ((b < n_blanks) &&
259 (p >= blanks[b].offset) &&
260 (p < blanks[b].offset + blanks[b].len))
261 blktype = blanks[b].type;
262
263 /* If we're about to stop omitting, it's time to say how
264 * much we omitted. */
265 if ((blktype != PKTLOG_OMIT) && omitted) {
266 logprintf(ctx, " (%d byte%s omitted)\r\n",
267 omitted, (omitted==1?"":"s"));
268 omitted = 0;
269 }
9a10ecf4 270
919baedb 271 /* (Re-)initialise dumpdata as necessary
272 * (start of row, or if we've just stopped omitting) */
273 if (!output_pos && !omitted)
274 sprintf(dumpdata, " %08x%*s\r\n", p-(p%16), 1+3*16+2+16, "");
275
276 /* Deal with the current byte. */
277 if (blktype == PKTLOG_OMIT) {
278 omitted++;
279 } else {
280 int c;
281 if (blktype == PKTLOG_BLANK) {
282 c = 'X';
283 sprintf(smalldata, "XX");
284 } else { /* PKTLOG_EMIT */
285 c = ((unsigned char *)data)[p];
286 sprintf(smalldata, "%02x", c);
00db133f 287 }
919baedb 288 dumpdata[10+2+3*(p%16)] = smalldata[0];
289 dumpdata[10+2+3*(p%16)+1] = smalldata[1];
290 dumpdata[10+1+3*16+2+(p%16)] = (isprint(c) ? c : '.');
291 output_pos = (p%16) + 1;
00db133f 292 }
9a10ecf4 293
919baedb 294 p++;
00db133f 295
919baedb 296 /* Flush row if necessary */
297 if (((p % 16) == 0) || (p == len) || omitted) {
298 if (output_pos) {
299 strcpy(dumpdata + 10+1+3*16+2+output_pos, "\r\n");
300 logwrite(ctx, dumpdata, strlen(dumpdata));
301 output_pos = 0;
302 }
00db133f 303 }
00db133f 304
00db133f 305 }
00db133f 306
919baedb 307 /* Tidy up */
308 if (omitted)
309 logprintf(ctx, " (%d byte%s omitted)\r\n",
310 omitted, (omitted==1?"":"s"));
311 logflush(ctx);
00db133f 312}
313
c229ef97 314void *log_init(void *frontend, Config *cfg)
a8327734 315{
3d88e64d 316 struct LogContext *ctx = snew(struct LogContext);
a8327734 317 ctx->lgfp = NULL;
8b029099 318 ctx->state = L_CLOSED;
a8327734 319 ctx->frontend = frontend;
c229ef97 320 ctx->cfg = *cfg; /* STRUCTURE COPY */
919baedb 321 bufchain_init(&ctx->queue);
a8327734 322 return ctx;
323}
324
fabd1805 325void log_free(void *handle)
326{
327 struct LogContext *ctx = (struct LogContext *)handle;
328
329 logfclose(ctx);
919baedb 330 bufchain_clear(&ctx->queue);
fabd1805 331 sfree(ctx);
332}
333
c229ef97 334void log_reconfig(void *handle, Config *cfg)
335{
336 struct LogContext *ctx = (struct LogContext *)handle;
337 int reset_logging;
338
9a30e26b 339 if (!filename_equal(ctx->cfg.logfilename, cfg->logfilename) ||
c229ef97 340 ctx->cfg.logtype != cfg->logtype)
341 reset_logging = TRUE;
342 else
343 reset_logging = FALSE;
344
345 if (reset_logging)
4512d432 346 logfclose(ctx);
c229ef97 347
348 ctx->cfg = *cfg; /* STRUCTURE COPY */
349
350 if (reset_logging)
4512d432 351 logfopen(ctx);
c229ef97 352}
353
00db133f 354/*
355 * translate format codes into time/date strings
356 * and insert them into log file name
357 *
6e8f9cab 358 * "&Y":YYYY "&m":MM "&d":DD "&T":hhmmss "&h":<hostname> "&&":&
00db133f 359 */
9a30e26b 360static void xlatlognam(Filename *dest, Filename src,
361 char *hostname, struct tm *tm) {
00db133f 362 char buf[10], *bufp;
363 int size;
9a30e26b 364 char buffer[FILENAME_MAX];
365 int len = sizeof(buffer)-1;
9fab77dc 366 char *d;
367 const char *s;
9a30e26b 368
369 d = buffer;
9fab77dc 370 s = filename_to_str(&src);
00db133f 371
372 while (*s) {
373 /* Let (bufp, len) be the string to append. */
374 bufp = buf; /* don't usually override this */
375 if (*s == '&') {
376 char c;
377 s++;
1709795f 378 size = 0;
00db133f 379 if (*s) switch (c = *s++, tolower(c)) {
380 case 'y':
381 size = strftime(buf, sizeof(buf), "%Y", tm);
382 break;
383 case 'm':
384 size = strftime(buf, sizeof(buf), "%m", tm);
385 break;
386 case 'd':
387 size = strftime(buf, sizeof(buf), "%d", tm);
388 break;
389 case 't':
390 size = strftime(buf, sizeof(buf), "%H%M%S", tm);
391 break;
392 case 'h':
393 bufp = hostname;
394 size = strlen(bufp);
395 break;
396 default:
397 buf[0] = '&';
398 size = 1;
399 if (c != '&')
400 buf[size++] = c;
401 }
402 } else {
403 buf[0] = *s++;
404 size = 1;
405 }
406 if (size > len)
407 size = len;
408 memcpy(d, bufp, size);
409 d += size;
410 len -= size;
411 }
412 *d = '\0';
9a30e26b 413
0ac2edb3 414 *dest = filename_from_str(buffer);
00db133f 415}