0%

为 putty 添加保存log文件时写入时间的选项

putty 几乎是我用过的远程工具里面的最好的了。目前感觉还有的缺憾有

  1. 缺少标签页切换
  2. 没有命令窗口
  3. log文件保存时没有保存时间线的选项

今天来动手为putty增加在保存的log的前面打上时间标签。

下载putty的源代码,假设你把它解压到putty这个文件夹内。我们主要更改以下几个文件

  1. putty\windows\winhelp.h
  2. putty\putty.h
  3. putty\config.c

添加UI选项

winhelp.h

在 WINHELP_CTX_xxxxx 字段新增一条。这里我新增了一条

1
#define WINHELP_CTX_logging_timeheader "config-logtimeheader"

CTX 后面的 logging_timeheader 等下会用到。

putty.h

#define CONFIG_OPTIONS(X) \字段新增一条,这里我新增的是

1
X(BOOL, NONE, logtimeheader) \

位置大概是1389行。

config.c

位置大概 1667 行,仿造原有的代码,添加一个checkbox。这里我写成

1
ctrl_checkbox(s, "Include time header of line", 'j',HELPCTX(logging_timeheader),conf_checkbox_handler, I(CONF_logtimeheader));

上面的代码中,HELPCTX(logging_timeheader) 里面的 logging_timeheader 是在 putty\windows\winhelp.h 新增的条目。
I(CONF_logtimeheader)中的 logtimeheader是在 putty\putty.h 中新增的条目。

功能实现

这一步的主逻辑是在 putty\logging.c 这个文件内添加的。修改位置为大概 54 行。static void logwrite(LogContext *ctx, ptrlen data) 这个函数体内。修改后如下所示

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
static void logwrite(LogContext *ctx, ptrlen data)
{
/*
* In state L_CLOSED, we call logfopen, which will set the state
* to one of L_OPENING, L_OPEN or L_ERROR. Hence we process all of
* those three _after_ processing L_CLOSED.
*/
if (ctx->state == L_CLOSED)
logfopen(ctx);

if (ctx->state == L_OPENING) {
bufchain_add(&ctx->queue, data.ptr, data.len);
} else if (ctx->state == L_OPEN) {
assert(ctx->lgfp);
if (fwrite(data.ptr, 1, data.len, ctx->lgfp) < data.len) {
logfclose(ctx);
ctx->state = L_ERROR;
lp_eventlog(ctx->lp, "Disabled writing session log "
"due to error while writing");
}
/************** 新增开始 **************/
if ((strcmp(data.ptr, "\n") == 0) && (conf_get_bool(ctx->conf, CONF_logtimeheader))) {
char buf[256];
struct tm tm;
tm = ltime();
strftime(buf, 24, "%Y.%m.%d %H:%M:%S ", &tm);
fwrite(buf, 1, strlen(buf), ctx->lgfp);
}
/*************** 新增结束 *************/
} /* else L_ERROR, so ignore the write */
}

logwrite 这个函数是用来将putty窗口内显示的字符输出到log文件内的。
如果调用这个函数的地方是一行一行传进来的的话,修改的地方应该是在调用这个函数的地方。但是根据调用的情况来看,多数情况下是一个字符一个字符写的。
所以我的做法是检查到输出的字符是 \n 时,就输出一个时间,这样下一行再输出的内容就是跟在这个时间后面的。
最后实现的效果大概是下面这个样子