Develop and Download Open Source Software

Browse Subversion Repository

Contents of /trunk/teraterm/teraterm/filesys_log.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 10443 - (show annotations) (download) (as text)
Thu Dec 29 16:43:20 2022 UTC (15 months, 1 week ago) by zmatsuo
File MIME type: text/x-c++src
File size: 39240 byte(s)
ログタブの標準ファイル名に設定できない文字があったので修正

- 修正前の動作
  - strftime() で %c %x %X %Z などにファイル名に使えない文字が含まれる時、
  - ダイアログで設定できない
    - OKボタンを押した後エラーが出て設定されない
    - 時間や日付の区切りに':','/'が使われるため(ロケール=日本)
  - 設定ファイル(TERATERM.INI)に直接設定すると使用できる
- ダイアログでのファイル名の判定方法を変更
  - strftime()でフォーマット後に使用できない文字を置換するようにした
  - 日本以外のロケールでも使用できる
- フォーマット後のプレビューを出すようにした
- ダイアログでのファイル名の扱いUnicode化した
- ログファイル名でよく使うと思われるパターンをドロップダウンで選択できるようにした

ticket #45553
1 /*
2 * (C) 2020- TeraTerm Project
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * 3. The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 /* TERATERM.EXE, log routines */
30 #include <stdio.h>
31 #if !defined(_CRTDBG_MAP_ALLOC)
32 #define _CRTDBG_MAP_ALLOC
33 #endif
34 #include <stdlib.h>
35 #include <crtdbg.h>
36 #include <process.h>
37 #include <windows.h>
38 #include <htmlhelp.h>
39 #include <assert.h>
40
41 #include "teraterm.h"
42 #include "tttypes.h"
43 #include "ftdlg.h"
44 #include "ttwinman.h"
45 #include "commlib.h"
46 #include "ttcommon.h"
47 #include "ttlib.h"
48 #include "dlglib.h"
49 #include "vtterm.h"
50 #include "ftlib.h"
51 #include "buffer.h"
52 #include "helpid.h"
53 #include "codeconv.h"
54 #include "asprintf.h"
55 #include "win32helper.h"
56
57 #include "filesys_log_res.h"
58 #include "filesys_log.h"
59 #include "filesys.h" // for ProtoGetProtoFlag()
60
61 #define TitLog L"Log"
62
63 /*
64 Line Head flag for timestamping
65 2007.05.24 Gentaro
66 */
67 enum enumLineEnd {
68 Line_Other = 0,
69 Line_LineHead = 1,
70 Line_FileHead = 2,
71 };
72
73 typedef struct {
74 wchar_t *FullName;
75
76 HANDLE FileHandle;
77 LONG FileSize, ByteCount;
78
79 DWORD StartTime;
80
81 enum enumLineEnd eLineEnd;
82
83 // log rotate
84 int RotateMode; // enum rotate_mode RotateMode;
85 LONG RotateSize;
86 int RotateStep;
87
88 HANDLE LogThread;
89 DWORD LogThreadId;
90 HANDLE LogThreadEvent;
91
92 BOOL IsPause;
93
94 PFileTransDlg FLogDlg;
95
96 LogCode_t log_code;
97 BOOL bom;
98
99 BOOL FileLog;
100 BOOL BinLog;
101 } TFileVar;
102 typedef TFileVar *PFileVar;
103
104 static PFileVar LogVar = NULL;
105
106 static PCHAR cv_LogBuf;
107 static int cv_LogPtr, cv_LStart, cv_LCount;
108 static PCHAR cv_BinBuf;
109 static int cv_BinPtr, cv_BStart, cv_BCount;
110 static int cv_BinSkip;
111
112 // �x�����������p�X���b�h�����b�Z�[�W
113 #define WM_DPC_LOGTHREAD_SEND (WM_APP + 1)
114
115 static void Log1Bin(BYTE b);
116 static void LogBinSkip(int add);
117 static BOOL CreateLogBuf(void);
118 static BOOL CreateBinBuf(void);
119 void LogPut1(BYTE b);
120 static void OutputStr(const wchar_t *str);
121 static void LogToFile(PFileVar fv);
122 static void FLogOutputBOM(PFileVar fv);
123
124 static BOOL OpenFTDlg_(PFileVar fv)
125 {
126 PFileTransDlg FTDlg = new CFileTransDlg();
127 if (FTDlg == NULL) {
128 return FALSE;
129 }
130
131 wchar_t *DlgCaption;
132 wchar_t uimsg[MAX_UIMSG];
133 get_lang_msgW("FILEDLG_TRANS_TITLE_LOG", uimsg, _countof(uimsg), TitLog, ts.UILanguageFile);
134 aswprintf(&DlgCaption, L"Tera Term: %s", uimsg);
135
136 CFileTransDlg::Info info;
137 info.UILanguageFile = ts.UILanguageFile;
138 info.OpId = CFileTransDlg::OpLog;
139 info.DlgCaption = DlgCaption;
140 info.FileName = NULL;
141 info.FullName = fv->FullName;
142 info.HideDialog = ts.LogHideDialog ? TRUE : FALSE;
143 info.HMainWin = HVTWin;
144 FTDlg->Create(hInst, &info);
145 FTDlg->RefreshNum(0, fv->FileSize, fv->ByteCount);
146
147 fv->FLogDlg = FTDlg;
148
149 free(DlgCaption);
150 return TRUE;
151 }
152
153 /**
154 * �t�@�C�������������u������,���O�p
155 * �����������u��������
156 * &h �z�X�g�����u��
157 * &p TCP�|�[�g�������u��
158 * &u ���O�I���������[�U��
159 *
160 * @param pcv
161 * @param src �u���������O��������(�t�@�C����)
162 * @return �u������������������
163 * �s�v����������free()��������
164 */
165 wchar_t *ConvertLognameW(const TComVar *pcv, const wchar_t *src)
166 {
167 const TTTSet *pts = pcv->ts;
168 size_t dest_len = wcslen(src) + 1;
169 wchar_t *dest = (wchar_t *)malloc(sizeof(wchar_t) * dest_len);
170
171 const wchar_t *s = src;
172 size_t i = 0;
173
174 while(*s != '\0') {
175 if (*s == '&' && *(s+1) != '\0') {
176 wchar_t c = *(s+1);
177 wchar_t *add_text = NULL;
178 switch (c) {
179 case 'h':
180 s += 2;
181 if (pcv->Open) {
182 switch(pcv->PortType) {
183 case IdTCPIP: {
184 // �z�X�g����IPv6�A�h���X�����A�t�@�C�������g�p������������(:)�����������u��
185 wchar_t *host = ToWcharA(pts->HostName);
186 wchar_t *host_fix = replaceInvalidFileNameCharW(host, '_');
187 free(host);
188 add_text = host_fix;
189 break;
190 }
191 case IdSerial: {
192 wchar_t *port;
193 aswprintf(&port, L"COM%d", ts.ComPort);
194 add_text = port;
195 break;
196 }
197 default:
198 ;
199 }
200 }
201 break;
202 case 'p':
203 s += 2;
204 if (pcv->Open) {
205 if (pcv->PortType == IdTCPIP) {
206 wchar_t *port;
207 aswprintf(&port, L"%d", ts.TCPPort);
208 add_text = port;
209 }
210 }
211 break;
212 case 'u': {
213 s += 2;
214 wchar_t user[256 + 1]; // 256=UNLEN
215 DWORD l = _countof(user);
216 if (GetUserNameW(user, &l) != 0) {
217 add_text = wcsdup(user);
218 }
219 break;
220 }
221 default:
222 // pass '&'
223 s++;
224 add_text = NULL;
225 break;
226 }
227
228 if (add_text != 0) {
229 size_t l = wcslen(add_text);
230 dest_len += l;
231 dest = (wchar_t *)realloc(dest, sizeof(wchar_t) * dest_len);
232 wcscpy(&dest[i], add_text);
233 i += l;
234 }
235 }
236 else {
237 dest[i] = *s++;
238 i++;
239 }
240 }
241 dest[i] = 0;
242 return dest;
243 }
244
245 /**
246 * �t�@�C�������������u������ ANSI��
247 */
248 static void ConvertLogname(char *c, int destlen)
249 {
250 wchar_t *cW = ToWcharA(c);
251 wchar_t *filenameW = ConvertLognameW(&cv, cW);
252 WideCharToACP_t(cW, c, destlen);
253 free(filenameW);
254 free(cW);
255 }
256
257 static void FixLogOption(void)
258 {
259 if (ts.LogBinary) {
260 ts.LogTypePlainText = false;
261 ts.LogTimestamp = false;
262 }
263 }
264
265
266 // �X���b�h���I�����t�@�C�����N���[�Y
267 static void CloseFileSync(PFileVar fv)
268 {
269 BOOL ret;
270
271 if (fv->FileHandle == INVALID_HANDLE_VALUE) {
272 return;
273 }
274
275 if (fv->LogThread != INVALID_HANDLE_VALUE) {
276 // �X���b�h���I������
277 ret = PostThreadMessage(fv->LogThreadId, WM_QUIT, 0, 0);
278 if (ret != 0) {
279 // �X���b�h�L���[���G���L���[���������������������������s���B
280 WaitForSingleObject(fv->LogThread, INFINITE);
281 }
282 else {
283 //DWORD code = GetLastError();
284 }
285 CloseHandle(fv->LogThread);
286 fv->LogThread = INVALID_HANDLE_VALUE;
287 }
288 CloseHandle(fv->FileHandle);
289 fv->FileHandle = INVALID_HANDLE_VALUE;
290 }
291
292 // �x�����������p�X���b�h
293 static unsigned _stdcall DeferredLogWriteThread(void *arg)
294 {
295 MSG msg;
296 PFileVar fv = (PFileVar)arg;
297 PCHAR buf;
298 DWORD buflen;
299 DWORD wrote;
300
301 PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE);
302
303 // �X���b�h�L���[���������I�������������X���b�h�����������m�����B
304 if (fv->LogThreadEvent != NULL) {
305 SetEvent(fv->LogThreadEvent);
306 }
307
308 while (GetMessage(&msg, NULL, 0, 0) > 0) {
309 switch (msg.message) {
310 case WM_DPC_LOGTHREAD_SEND:
311 buf = (PCHAR)msg.wParam;
312 buflen = (DWORD)msg.lParam;
313 WriteFile(fv->FileHandle, buf, buflen, &wrote, NULL);
314 free(buf); // ����������������
315 break;
316
317 case WM_QUIT:
318 goto end;
319 break;
320 }
321 }
322
323 end:
324 _endthreadex(0);
325 return (0);
326 }
327
328 // �x�����������p�X���b�h���N�����B
329 // (2013.4.19 yutaka)
330 // DeferredLogWriteThread �X���b�h���N�������A�X���b�h�L���[�����������������O���A
331 // ���O�t�@�C�����N���[�Y(CloseFileSync)���s���������A�G���L���[�����s���A�f�b�h���b�N
332 // �����������������C�������B
333 // �X���b�h�����������s�������A���O�����C�x���g�I�u�W�F�N�g���g�����A�X���b�h�L���[��
334 // ���������������������������������B���O�t���C�x���g�I�u�W�F�N�g���g���������A
335 // �V�X�e��(Windows OS)�������j�[�N�����O�������K�v�������B
336 // (2016.9.23 yutaka)
337 static void StartThread(PFileVar fv)
338 {
339 unsigned tid;
340 fv->LogThreadEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
341 fv->LogThread = (HANDLE)_beginthreadex(NULL, 0, DeferredLogWriteThread, fv, 0, &tid);
342 fv->LogThreadId = tid;
343 if (fv->LogThreadEvent != NULL) {
344 WaitForSingleObject(fv->LogThreadEvent, INFINITE);
345 CloseHandle(fv->LogThreadEvent);
346 }
347 }
348
349 /**
350 * �_�C�A���O�����e�� ts ����������
351 *
352 * TODO
353 * �_�C�A���O�����������l�������I��������
354 * ����������������������������������������������?
355 */
356 static void SetLogFlags(HWND Dialog)
357 {
358 WORD BinFlag, val;
359
360 GetRB(Dialog, &BinFlag, IDC_FOPTBIN, IDC_FOPTBIN);
361 ts.LogBinary = BinFlag;
362
363 GetRB(Dialog, &val, IDC_APPEND, IDC_APPEND);
364 ts.Append = val;
365
366 if (!BinFlag) {
367 GetRB(Dialog, &val, IDC_PLAINTEXT, IDC_PLAINTEXT);
368 ts.LogTypePlainText = val;
369
370 GetRB(Dialog, &val, IDC_TIMESTAMP, IDC_TIMESTAMP);
371 ts.LogTimestamp = val;
372 }
373
374 GetRB(Dialog, &val, IDC_HIDEDIALOG, IDC_HIDEDIALOG);
375 ts.LogHideDialog = val;
376
377 GetRB(Dialog, &val, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST);
378 ts.LogAllBuffIncludedInFirst = val;
379
380 ts.LogTimestampType = (WORD)(GetCurSel(Dialog, IDC_TIMESTAMPTYPE) - 1);
381 }
382
383 /**
384 * ���O�t�@�C���`�F�b�N
385 *
386 * @param[in] filename
387 * @param[out] exist TURE/FALSE
388 * @param[out] bom 0 no BOM (or file not exist)
389 * 1 UTF-8
390 * 2 UTF-16LE
391 * 3 UTF-16BE
392 */
393 static void CheckLogFile(const wchar_t *filename, BOOL *exist, int *bom)
394 {
395 *exist = FALSE;
396 *bom = 0;
397
398 // �t�@�C������������?
399 DWORD logdir = GetFileAttributesW(filename);
400 if ((logdir != INVALID_FILE_ATTRIBUTES) && ((logdir & FILE_ATTRIBUTE_DIRECTORY) == 0)) {
401 // �t�@�C����������
402 *exist = TRUE;
403
404 // BOM�L��/�����`�F�b�N
405 FILE *fp = _wfopen(filename, L"rb");
406 if (fp != NULL) {
407 unsigned char tmp[4];
408 size_t l = fread(tmp, 1, sizeof(tmp), fp);
409 fclose(fp);
410 if (l < 2) {
411 *bom = 0;
412 } else if (l >= 2 && tmp[0] == 0xff && tmp[1] == 0xfe) {
413 // UTF-16LE
414 *bom = 2;
415 } else if (l >= 2 && tmp[0] == 0xfe && tmp[1] == 0xff) {
416 // UTF-16BE
417 *bom = 3;
418 } else if (l >= 3 && tmp[0] == 0xef && tmp[1] == 0xbb && tmp[2] == 0xbf) {
419 // UTF-8
420 *bom = 1;
421 } else {
422 *bom = 0;
423 }
424 }
425 }
426 }
427
428 typedef struct {
429 FLogDlgInfo_t *info;
430 // work
431 BOOL file_exist;
432 int current_bom; // ���������t�@�C�����G���R�[�f�B���O�i�t�@�C����BOM���������j
433 TTTSet *pts;
434 TComVar *pcv;
435 } LogDlgWork_t;
436
437
438 /*
439 * Log �_�C�A���O�������AEnable/Disable �����������R���g���[��
440 *
441 * - Append
442 * �w���������t�@�C������������������ Enable
443 * �w���������t�@�C�������������������� Disable
444 *
445 * - BOM, Encoding
446 * Text ���� New/Overwrite �������� Enable
447 * ���������������� Disable
448 * BOM ���t�@�C��������������������������������������
449 * Encoding �����L�����������������A�����t�@�C�����G���R�[�f�B���O��
450 * �����I���_�C�A���O�����f���������A���[�U�������w������������
451 *
452 * - Plain Text, Timestamp, Timestamp ����
453 * Text �������� Enable
454 * Binary �������� Disable
455 *
456 * - Timestamp ����
457 * Timestamp=on �������� Enable
458 * Timestamp=off �������� Disable
459 */
460
461 static void ArrangeControls(HWND Dialog, LogDlgWork_t *work,
462 WORD Append, WORD LogBinary,
463 WORD LogTypePlainText, WORD LogTimestamp)
464 {
465 if (work->file_exist) {
466 // Append ���W�I�{�^�����A�t�@�C�����������������L��������
467 EnableWindow(GetDlgItem(Dialog, IDC_APPEND), TRUE);
468
469 if (Append > 0) {
470 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
471 }
472 else {
473 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
474 }
475 }
476 else { // �t�@�C�������� -> �V�K
477 EnableWindow(GetDlgItem(Dialog, IDC_APPEND), FALSE);
478 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
479 }
480
481 if (!LogBinary && !Append) {
482 EnableWindow(GetDlgItem(Dialog, IDC_BOM), TRUE);
483 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), TRUE);
484 }
485 else {
486 EnableWindow(GetDlgItem(Dialog, IDC_BOM), FALSE);
487 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), FALSE);
488 }
489
490 if (LogBinary) {
491 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTBIN);
492
493 DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT);
494 DisableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP);
495 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
496 }
497 else {
498 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTTEXT);
499
500 EnableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT);
501 EnableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP);
502 EnableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
503
504 if (LogTypePlainText) {
505 SetRB(Dialog, 1, IDC_PLAINTEXT, IDC_PLAINTEXT);
506 }
507 if (LogTimestamp) {
508 SetRB(Dialog, 1, IDC_TIMESTAMP, IDC_TIMESTAMP);
509 }
510 else {
511 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
512 }
513 }
514
515 if (work->file_exist && Append) {
516 // �����t�@�C�����G���R�[�f�B���O�����f����
517 int bom = work->current_bom;
518 int cur =
519 bom == 1 ? 0 :
520 bom == 2 ? 1 :
521 bom == 3 ? 2 : 0;
522 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, cur, 0);
523 }
524 }
525
526 static void CheckLogFile(HWND Dialog, const wchar_t *filename, LogDlgWork_t *work)
527 {
528 BOOL exist;
529 int bom;
530 CheckLogFile(filename, &exist, &bom);
531 work->file_exist = exist;
532 work->current_bom = bom;
533 }
534
535 static INT_PTR CALLBACK LogFnHook(HWND Dialog, UINT Message, WPARAM wParam, LPARAM lParam)
536 {
537 static const DlgTextInfo TextInfos[] = {
538 { 0, "DLG_TABSHEET_TITLE_LOG" },
539 { IDC_SENDFILE_FILENAME_TITLE, "DLG_FOPT_FILENAME_TITLE" },
540 { IDC_FOPTTEXT, "DLG_FOPT_TEXT" },
541 { IDC_FOPTBIN, "DLG_FOPT_BINARY" },
542 { IDC_BOM, "DLG_FOPT_BOM" },
543 { IDC_APPEND, "DLG_FOPT_APPEND" },
544 { IDC_PLAINTEXT, "DLG_FOPT_PLAIN" },
545 { IDC_HIDEDIALOG, "DLG_FOPT_HIDEDIALOG" },
546 { IDC_ALLBUFF_INFIRST, "DLG_FOPT_ALLBUFFINFIRST" },
547 { IDC_TIMESTAMP, "DLG_FOPT_TIMESTAMP" },
548 { IDC_NEW_OVERWRITE, "DLG_FOPT_NEW_OVERWRITE" },
549 { IDC_APPEND_GROUP, "DLG_FOPT_APPEND_LABEL" },
550 { IDC_BINARY_GROUP, "DLG_FOPT_BINARY_LABEL" },
551 };
552 static const I18nTextInfo timestamp_list[] = {
553 { "DLG_FOPT_TIMESTAMP_LOCAL", L"Local Time" },
554 { "DLG_FOPT_TIMESTAMP_UTC", L"UTC" },
555 { "DLG_FOPT_TIMESTAMP_ELAPSED_LOGGING", L"Elapsed Time (Logging)" },
556 { "DLG_FOPT_TIMESTAMP_ELAPSED_CONNECTION", L"Elapsed Time (Connection)" },
557 };
558 LogDlgWork_t *work = (LogDlgWork_t *)GetWindowLongPtr(Dialog, DWLP_USER);
559
560 if (Message == RegisterWindowMessage(HELPMSGSTRING)) {
561 // �R�����_�C�A���O�������w���v���b�Z�[�W���t��������
562 Message = WM_COMMAND;
563 wParam = IDHELP;
564 }
565 switch (Message) {
566 case WM_INITDIALOG: {
567 work = (LogDlgWork_t *)lParam;
568 TTTSet *pts = work->pts;
569 SetWindowLongPtr(Dialog, DWLP_USER, (LONG_PTR)work);
570 ::DragAcceptFiles(Dialog, TRUE);
571
572 SetDlgTextsW(Dialog, TextInfos, _countof(TextInfos), pts->UILanguageFileW);
573 SetI18nListW("Tera Term", Dialog, IDC_TIMESTAMPTYPE, timestamp_list, _countof(timestamp_list),
574 pts->UILanguageFileW, 0);
575
576 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-8");
577 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16LE");
578 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16BE");
579 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
580
581 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, work->info->filename);
582
583 // timestamp ����
584 int tstype = pts->LogTimestampType == TIMESTAMP_LOCAL ? 0 :
585 pts->LogTimestampType == TIMESTAMP_UTC ? 1 :
586 pts->LogTimestampType == TIMESTAMP_ELAPSED_LOGSTART ? 2 :
587 pts->LogTimestampType == TIMESTAMP_ELAPSED_CONNECTED ? 3 : 0;
588 SendDlgItemMessage(Dialog, IDC_TIMESTAMPTYPE, CB_SETCURSEL, tstype, 0);
589
590
591 CheckLogFile(Dialog, work->info->filename, work);
592 ArrangeControls(Dialog, work, pts->Append, pts->LogBinary, pts->LogTypePlainText, pts->LogTimestamp);
593 work->info->filename = NULL;
594
595 // Hide dialog �`�F�b�N�{�b�N�X
596 if (pts->LogHideDialog) {
597 SetRB(Dialog, 1, IDC_HIDEDIALOG, IDC_HIDEDIALOG);
598 }
599
600 // Include screen buffer �`�F�b�N�{�b�N�X
601 if (pts->LogAllBuffIncludedInFirst) {
602 SetRB(Dialog, 1, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST);
603 }
604
605 CenterWindow(Dialog, GetParent(Dialog));
606
607 return TRUE;
608 }
609
610 case WM_COMMAND:
611 switch (LOWORD(wParam)) {
612 case IDOK: {
613 wchar_t *filename;
614 hGetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, &filename);
615 work->info->filename = filename;
616 work->info->append = IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED;
617 work->info->bom = IsDlgButtonChecked(Dialog, IDC_BOM) == BST_CHECKED;
618 work->info->code = (LogCode_t)SendDlgItemMessageA(Dialog, IDC_TEXTCODING_DROPDOWN, CB_GETCURSEL, 0, 0);
619 SetLogFlags(Dialog);
620 EndDialog(Dialog, IDOK);
621 break;
622 }
623 case IDCANCEL:
624 EndDialog(Dialog, IDCANCEL);
625 break;
626 case IDHELP:
627 OpenHelpCV(work->pcv, HH_HELP_CONTEXT, HlpFileLog);
628 break;
629 case IDC_FOPT_FILENAME_BUTTON: {
630 /* save current dir */
631 const wchar_t *UILanguageFile = work->pts->UILanguageFileW;
632 wchar_t *curdir;
633 hGetCurrentDirectoryW(&curdir);
634
635 wchar_t fname[MAX_PATH];
636 GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname, _countof(fname));
637
638 const wchar_t* simple_log_filter = L"*.txt;*.log";
639 wchar_t *FNFilter = GetCommonDialogFilterWW(simple_log_filter, UILanguageFile);
640
641 wchar_t *caption;
642 wchar_t *uimsg;
643 GetI18nStrWW("Tera Term", "FILEDLG_TRANS_TITLE_LOG",
644 TitLog, UILanguageFile, &uimsg);
645 aswprintf(&caption, L"Tera Term: %s", uimsg);
646 free(uimsg);
647
648 OPENFILENAMEW ofn = {};
649 ofn.lStructSize = get_OPENFILENAME_SIZEW();
650 //ofn.Flags = OFN_PATHMUSTEXIST | OFN_OVERWRITEPROMPT;
651 ofn.Flags |= OFN_EXPLORER | OFN_ENABLESIZING;
652 ofn.Flags |= OFN_SHOWHELP;
653 ofn.Flags |= OFN_NOCHANGEDIR; // ��������������������������������
654 ofn.hwndOwner = Dialog;
655 ofn.lpstrFilter = FNFilter;
656 ofn.nFilterIndex = 1;
657 ofn.lpstrFile = fname;
658 ofn.nMaxFile = _countof(fname);
659 ofn.lpstrTitle = caption;
660 ofn.lpstrInitialDir = work->pts->LogDefaultPathW;
661 BOOL Ok = GetSaveFileNameW(&ofn);
662 free(caption);
663 free(FNFilter);
664 if (Ok) {
665 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname);
666 }
667
668 /* restore dir */
669 SetCurrentDirectoryW(curdir);
670 free(curdir);
671
672 break;
673 }
674 case IDC_NEW_OVERWRITE:
675 case IDC_APPEND:
676 case IDC_FOPTTEXT:
677 case IDC_FOPTBIN:
678 case IDC_TIMESTAMP:
679 {
680 WORD Appnd, LogBinary, LogTypePlainText, LogTimestamp;
681 GetRB(Dialog, &Appnd, IDC_APPEND, IDC_APPEND);
682 GetRB(Dialog, &LogBinary, IDC_FOPTBIN, IDC_FOPTBIN);
683 GetRB(Dialog, &LogTypePlainText, IDC_PLAINTEXT, IDC_PLAINTEXT);
684 GetRB(Dialog, &LogTimestamp, IDC_TIMESTAMP, IDC_TIMESTAMP);
685 ArrangeControls(Dialog, work, Appnd, LogBinary, LogTypePlainText, LogTimestamp);
686 }
687 break;
688 case IDC_FOPT_FILENAME_EDIT:
689 if (HIWORD(wParam) == EN_CHANGE){
690 wchar_t *filename;
691 hGetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, &filename);
692 CheckLogFile(Dialog, filename, work);
693 free(filename);
694 {
695 WORD Appnd, LogBinary, LogTypePlainText, LogTimestamp;
696 GetRB(Dialog, &Appnd, IDC_APPEND, IDC_APPEND);
697 GetRB(Dialog, &LogBinary, IDC_FOPTBIN, IDC_FOPTBIN);
698 GetRB(Dialog, &LogTypePlainText, IDC_PLAINTEXT, IDC_PLAINTEXT);
699 GetRB(Dialog, &LogTimestamp, IDC_TIMESTAMP, IDC_TIMESTAMP);
700 ArrangeControls(Dialog, work, Appnd, LogBinary, LogTypePlainText, LogTimestamp);
701 }
702 }
703 break;
704 }
705 break;
706 case WM_DROPFILES: {
707 // �����h���b�v��������������1������������
708 HDROP hDrop = (HDROP)wParam;
709 const UINT len = DragQueryFileW(hDrop, 0, NULL, 0);
710 if (len == 0) {
711 DragFinish(hDrop);
712 return TRUE;
713 }
714 wchar_t *filename = (wchar_t *)malloc(sizeof(wchar_t) * (len + 1));
715 DragQueryFileW(hDrop, 0, filename, len + 1);
716 filename[len] = '\0';
717 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
718 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename);
719 SendDlgItemMessage(Dialog, IDC_FOPT_FILENAME_EDIT, EM_SETSEL, len, len);
720 free(filename);
721 DragFinish(hDrop);
722 return TRUE;
723 }
724 }
725 return FALSE;
726 }
727
728 static void OpenLogFile(PFileVar fv)
729 {
730 // LogLockExclusive ���L���������������������L���������A
731 // ���������������O�t�@�C���������G�f�B�^���J����������
732 int dwShareMode = FILE_SHARE_READ;
733 if (!ts.LogLockExclusive) {
734 dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
735 }
736 fv->FileHandle = CreateFileW(fv->FullName, GENERIC_WRITE, dwShareMode, NULL,
737 OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
738 }
739
740 static BOOL LogStart(PFileVar fv, const wchar_t *fname)
741 {
742 fv->FullName = _wcsdup(fname);
743 FixLogOption();
744
745 if (ts.LogBinary > 0)
746 {
747 fv->BinLog = TRUE;
748 fv->FileLog = FALSE;
749 if (! CreateBinBuf())
750 {
751 return FALSE;
752 }
753 }
754 else {
755 fv->BinLog = FALSE;
756 fv->FileLog = TRUE;
757 if (! CreateLogBuf())
758 {
759 return FALSE;
760 }
761 }
762 cv_LStart = cv_LogPtr;
763 cv_LCount = 0;
764
765 OpenLogFile(fv);
766 if (fv->FileHandle == INVALID_HANDLE_VALUE) {
767 return FALSE;
768 }
769
770 /* 2007.05.24 Gentaro */
771 fv->eLineEnd = Line_LineHead;
772 if (ts.Append > 0)
773 {
774 SetFilePointer(fv->FileHandle, 0, NULL, FILE_END);
775 /* 2007.05.24 Gentaro
776 If log file already exists,
777 a newline is inserted before the first timestamp.
778 */
779 fv->eLineEnd = Line_FileHead;
780 }
781
782 // BOM�o��
783 if (ts.Append == 0 && ts.LogBinary == 0 && fv->bom) {
784 // ���L��������(�V�K) && �o�C�i���������� && BOM ���o����
785 FLogOutputBOM(fv);
786 }
787
788 // Log rotate configuration
789 fv->RotateMode = ts.LogRotate;
790 fv->RotateSize = ts.LogRotateSize;
791 fv->RotateStep = ts.LogRotateStep;
792
793 // Log rotate���L���������A�����t�@�C���T�C�Y�����������B
794 // �������t�@�C�������������T�C�Y�����[�e�[�g�������������C���B
795 // (2016.4.9 yutaka)
796 if (fv->RotateMode != ROTATE_NONE) {
797 DWORD size = GetFileSize(fv->FileHandle, NULL);
798 if (size == -1) {
799 return FALSE;
800 }
801 fv->ByteCount = size;
802 }
803 else {
804 fv->ByteCount = 0;
805 }
806
807 if (! OpenFTDlg_(fv)) {
808 return FALSE;
809 }
810
811 fv->IsPause = FALSE;
812 fv->StartTime = GetTickCount();
813
814 if (ts.DeferredLogWriteMode) {
815 StartThread(fv);
816 }
817
818 if (fv->FileLog) {
819 cv.Log1Byte = LogPut1;
820 }
821 if (fv->BinLog) {
822 cv.Log1Bin = Log1Bin;
823 cv.LogBinSkip = LogBinSkip;
824 }
825
826 return TRUE;
827 }
828
829 /**
830 * �����o�b�t�@�������f�[�^�����������O�������o��
831 * (2013.9.29 yutaka)
832 *
833 * TODO
834 * 1�s������
835 */
836 void FLogOutputAllBuffer(void)
837 {
838 PFileVar fv = LogVar;
839 DWORD ofs;
840 int size;
841 wchar_t buf[512];
842 for (ofs = 0 ; ; ofs++ ) {
843 // 1�����s�����������B���������������A�G�X�P�[�v�V�[�P���X�������������B
844 size = BuffGetAnyLineDataW(ofs, buf, _countof(buf));
845 if (size == -1)
846 break;
847
848 OutputStr(buf);
849 OutputStr(L"\r\n");
850 LogToFile(fv);
851 }
852 }
853
854 /**
855 * ���O��1byte��������
856 * �o�b�t�@��������������
857 * ���������������� LogToFile() ���s������
858 */
859 void LogPut1(BYTE b)
860 {
861 PFileVar fv = LogVar;
862
863 cv_LogBuf[cv_LogPtr] = b;
864 cv_LogPtr++;
865 if (cv_LogPtr>=InBuffSize)
866 cv_LogPtr = cv_LogPtr-InBuffSize;
867
868 if (fv->FileLog)
869 {
870 if (cv_LCount>=InBuffSize)
871 {
872 cv_LCount = InBuffSize;
873 cv_LStart = cv_LogPtr;
874 }
875 else
876 cv_LCount++;
877 }
878 else
879 cv_LCount = 0;
880 }
881
882 static BOOL Get1(PCHAR Buf, int *Start, int *Count, PBYTE b)
883 {
884 if (*Count<=0) return FALSE;
885 *b = Buf[*Start];
886 (*Start)++;
887 if (*Start>=InBuffSize)
888 *Start = *Start-InBuffSize;
889 (*Count)--;
890 return TRUE;
891 }
892
893
894
895 static CRITICAL_SECTION g_filelog_lock; /* ���b�N�p���� */
896
897 void logfile_lock_initialize(void)
898 {
899 InitializeCriticalSection(&g_filelog_lock);
900 }
901
902 static inline void logfile_lock(void)
903 {
904 EnterCriticalSection(&g_filelog_lock);
905 }
906
907 static inline void logfile_unlock(void)
908 {
909 LeaveCriticalSection(&g_filelog_lock);
910 }
911
912 // ���O�����[�e�[�g�����B
913 // (2013.3.21 yutaka)
914 static void LogRotate(PFileVar fv)
915 {
916 int loopmax = 10000; // XXX
917 int i, k;
918
919 if (fv->RotateMode == ROTATE_NONE)
920 return;
921
922 if (fv->RotateMode == ROTATE_SIZE) {
923 if (fv->ByteCount <= fv->RotateSize)
924 return;
925 //OutputDebugPrintf("%s: mode %d size %ld\n", __FUNCTION__, fv->RotateMode, fv->ByteCount);
926 } else {
927 return;
928 }
929
930 logfile_lock();
931 // ���O�T�C�Y���������������B
932 fv->ByteCount = 0;
933
934 // �������������t�@�C�����N���[�Y�����A�������t�@�C�����I�[�v�������B
935 CloseFileSync(fv);
936
937 // �������[�e�[�V�������X�e�b�v�����w����������
938 if (fv->RotateStep > 0)
939 loopmax = fv->RotateStep;
940
941 for (i = 1 ; i <= loopmax ; i++) {
942 wchar_t *filename;
943 aswprintf(&filename, L"%s.%d", fv->FullName, i);
944 DWORD attr = GetFileAttributesW(filename);
945 free(filename);
946 if (attr == INVALID_FILE_ATTRIBUTES)
947 break;
948 }
949 if (i > loopmax) {
950 // �������������������������A�������t�@�C�������p�������B
951 i = loopmax;
952 }
953
954 // ���t�@�C�������l�[���B
955 for (k = i-1 ; k >= 0 ; k--) {
956 wchar_t *oldfile;
957 if (k == 0)
958 oldfile = _wcsdup(fv->FullName);
959 else
960 aswprintf(&oldfile, L"%s.%d", fv->FullName, k);
961 wchar_t *newfile;
962 aswprintf(&newfile, L"%s.%d", fv->FullName, k+1);
963 DeleteFileW(newfile);
964 if (MoveFileW(oldfile, newfile) == 0) {
965 OutputDebugPrintf("%s: rename %d\n", __FUNCTION__, errno);
966 }
967 free(oldfile);
968 free(newfile);
969 }
970
971 // ���I�[�v��
972 OpenLogFile(fv);
973 if (fv->bom) {
974 FLogOutputBOM(fv);
975 }
976 if (ts.DeferredLogWriteMode) {
977 StartThread(fv);
978 }
979
980 logfile_unlock();
981 }
982
983 static wchar_t *TimeStampStr(PFileVar fv)
984 {
985 char *strtime = NULL;
986 switch (ts.LogTimestampType) {
987 case TIMESTAMP_LOCAL:
988 default:
989 strtime = mctimelocal(ts.LogTimestampFormat, FALSE);
990 break;
991 case TIMESTAMP_UTC:
992 strtime = mctimelocal(ts.LogTimestampFormat, TRUE);
993 break;
994 case TIMESTAMP_ELAPSED_LOGSTART:
995 strtime = strelapsed(fv->StartTime);
996 break;
997 case TIMESTAMP_ELAPSED_CONNECTED:
998 strtime = strelapsed(cv.ConnectedTime);
999 break;
1000 }
1001
1002 char tmp[128];
1003 tmp[0] = 0;
1004 strncat_s(tmp, sizeof(tmp), "[", _TRUNCATE);
1005 strncat_s(tmp, sizeof(tmp), strtime, _TRUNCATE);
1006 strncat_s(tmp, sizeof(tmp), "] ", _TRUNCATE);
1007
1008 return ToWcharA(tmp);
1009 }
1010
1011 /**
1012 * �o�b�t�@�������O���t�@�C������������
1013 */
1014 static void LogToFile(PFileVar fv)
1015 {
1016 PCHAR Buf;
1017 int Start, Count;
1018 BYTE b;
1019
1020 if (fv->FileLog)
1021 {
1022 Buf = cv_LogBuf;
1023 Start = cv_LStart;
1024 Count = cv_LCount;
1025 }
1026 else if (fv->BinLog)
1027 {
1028 Buf = cv_BinBuf;
1029 Start = cv_BStart;
1030 Count = cv_BCount;
1031 }
1032 else
1033 return;
1034
1035 if (Buf==NULL) return;
1036 if (Count==0) return;
1037
1038 // ���b�N������(2004.8.6 yutaka)
1039 logfile_lock();
1040
1041 // ���������f�[�^����������
1042 DWORD WriteBufMax = 8192;
1043 DWORD WriteBufLen = 0;
1044 PCHAR WriteBuf = (PCHAR)malloc(WriteBufMax);
1045 while (Get1(Buf,&Start,&Count,&b)) {
1046 if (FLogIsPause() || ProtoGetProtoFlag()) {
1047 continue;
1048 }
1049
1050 if (WriteBufLen >= (WriteBufMax*4/5)) {
1051 WriteBufMax *= 2;
1052 WriteBuf = (PCHAR)realloc(WriteBuf, WriteBufMax);
1053 }
1054
1055 WriteBuf[WriteBufLen++] = b;
1056
1057 (fv->ByteCount)++;
1058 }
1059
1060 // ��������
1061 if (WriteBufLen > 0) {
1062 if (ts.DeferredLogWriteMode) {
1063 PostThreadMessage(fv->LogThreadId, WM_DPC_LOGTHREAD_SEND, (WPARAM)WriteBuf, WriteBufLen);
1064 }
1065 else {
1066 DWORD wrote;
1067 WriteFile(fv->FileHandle, WriteBuf, WriteBufLen, &wrote, NULL);
1068 free(WriteBuf);
1069 }
1070 }
1071
1072 logfile_unlock();
1073
1074 if (fv->FileLog)
1075 {
1076 cv_LStart = Start;
1077 cv_LCount = Count;
1078 }
1079 else {
1080 cv_BStart = Start;
1081 cv_BCount = Count;
1082 }
1083 if (FLogIsPause() || ProtoGetProtoFlag()) return;
1084 fv->FLogDlg->RefreshNum(fv->StartTime, fv->FileSize, fv->ByteCount);
1085
1086
1087 // ���O�E���[�e�[�g
1088 LogRotate(fv);
1089 }
1090
1091 static BOOL CreateLogBuf(void)
1092 {
1093 if (cv_LogBuf==NULL)
1094 {
1095 cv_LogBuf = (char *)malloc(InBuffSize);
1096 cv_LogPtr = 0;
1097 cv_LStart = 0;
1098 cv_LCount = 0;
1099 }
1100 return (cv_LogBuf!=NULL);
1101 }
1102
1103 static void FreeLogBuf(void)
1104 {
1105 free(cv_LogBuf);
1106 cv_LogBuf = NULL;
1107 cv_LogPtr = 0;
1108 cv_LStart = 0;
1109 cv_LCount = 0;
1110 }
1111
1112 static BOOL CreateBinBuf(void)
1113 {
1114 if (cv_BinBuf==NULL)
1115 {
1116 cv_BinBuf = (PCHAR)malloc(InBuffSize);
1117 cv_BinPtr = 0;
1118 cv_BStart = 0;
1119 cv_BCount = 0;
1120 }
1121 return (cv_BinBuf!=NULL);
1122 }
1123
1124 static void FreeBinBuf(void)
1125 {
1126 free(cv_BinBuf);
1127 cv_BinBuf = NULL;
1128 cv_BinPtr = 0;
1129 cv_BStart = 0;
1130 cv_BCount = 0;
1131 }
1132
1133 static void FileTransEnd_(PFileVar fv)
1134 {
1135 fv->FileLog = FALSE;
1136 fv->BinLog = FALSE;
1137 cv.Log1Byte = NULL;
1138 cv.Log1Bin = NULL;
1139 cv.LogBinSkip = NULL;
1140 PFileTransDlg FLogDlg = fv->FLogDlg;
1141 if (FLogDlg != NULL) {
1142 FLogDlg->DestroyWindow();
1143 FLogDlg = NULL;
1144 fv->FLogDlg = NULL;
1145 }
1146 CloseFileSync(fv);
1147 FreeLogBuf();
1148 FreeBinBuf();
1149 free(fv->FullName);
1150 fv->FullName = NULL;
1151 free(fv);
1152
1153 LogVar = NULL;
1154 }
1155
1156 /**
1157 * ���O���|�[�Y����
1158 */
1159 void FLogPause(BOOL Pause)
1160 {
1161 PFileVar fv = LogVar;
1162 if (fv == NULL) {
1163 return;
1164 }
1165 fv->IsPause = Pause;
1166 fv->FLogDlg->ChangeButton(Pause);
1167 }
1168
1169 /**
1170 * ���O���[�e�[�g������
1171 * ���O���T�C�Y��<size>�o�C�g���������������A���[�e�[�V��������������������
1172 */
1173 void FLogRotateSize(size_t size)
1174 {
1175 PFileVar fv = LogVar;
1176 if (fv == NULL) {
1177 return;
1178 }
1179 fv->RotateMode = ROTATE_SIZE;
1180 fv->RotateSize = (LONG)size;
1181 }
1182
1183 /**
1184 * ���O���[�e�[�g������
1185 * ���O�t�@�C������������������
1186 */
1187 void FLogRotateRotate(int step)
1188 {
1189 PFileVar fv = LogVar;
1190 if (fv == NULL) {
1191 return;
1192 }
1193 fv->RotateStep = step;
1194 }
1195
1196 /**
1197 * ���O���[�e�[�g������
1198 * ���[�e�[�V���������~
1199 */
1200 void FLogRotateHalt(void)
1201 {
1202 PFileVar fv = LogVar;
1203 if (fv == NULL) {
1204 return;
1205 }
1206 fv->RotateMode = ROTATE_NONE;
1207 fv->RotateSize = 0;
1208 fv->RotateStep = 0;
1209 }
1210
1211 static INT_PTR CALLBACK OnCommentDlgProc(HWND hDlgWnd, UINT msg, WPARAM wp, LPARAM)
1212 {
1213 static const DlgTextInfo TextInfos[] = {
1214 { 0, "DLG_COMMENT_TITLE" },
1215 { IDOK, "BTN_OK" }
1216 };
1217
1218 switch (msg) {
1219 case WM_INITDIALOG:
1220 // �G�f�B�b�g�R���g���[�����t�H�[�J�X��������
1221 SetFocus(GetDlgItem(hDlgWnd, IDC_EDIT_COMMENT));
1222 SetDlgTextsW(hDlgWnd, TextInfos, _countof(TextInfos), ts.UILanguageFileW);
1223 return FALSE;
1224
1225 case WM_COMMAND:
1226 switch (LOWORD(wp)) {
1227 case IDOK: {
1228 size_t len = SendDlgItemMessageW(hDlgWnd, IDC_EDIT_COMMENT, WM_GETTEXTLENGTH, 0, 0);
1229 len += 1;
1230 wchar_t *buf = (wchar_t *)malloc(len * sizeof(wchar_t));
1231 GetDlgItemTextW(hDlgWnd, IDC_EDIT_COMMENT, buf, (int)len);
1232 FLogWriteStr(buf);
1233 FLogWriteStr(L"\n"); // TODO ���s�R�[�h
1234 free(buf);
1235 TTEndDialog(hDlgWnd, IDOK);
1236 break;
1237 }
1238 default:
1239 return FALSE;
1240 }
1241 break;
1242
1243 case WM_CLOSE:
1244 TTEndDialog(hDlgWnd, 0);
1245 return TRUE;
1246
1247 default:
1248 return FALSE;
1249 }
1250 return TRUE;
1251 }
1252
1253 /**
1254 * ���O�t�@�C�����R�����g���������� (2004.8.6 yutaka)
1255 */
1256 void FLogAddCommentDlg(HINSTANCE hInst, HWND hWnd)
1257 {
1258 PFileVar fv = LogVar;
1259 if (fv == NULL) {
1260 return;
1261 }
1262 TTDialogBox(hInst, MAKEINTRESOURCE(IDD_COMMENT_DIALOG),
1263 hWnd, OnCommentDlgProc);
1264 }
1265
1266 void FLogClose(void)
1267 {
1268 PFileVar fv = LogVar;
1269 if (fv == NULL) {
1270 return;
1271 }
1272
1273 FileTransEnd_(fv);
1274 }
1275
1276 /**
1277 * ���O���I�[�v������
1278 * @param[in] fname ���O�t�@�C����, CreateFile()���n������
1279 *
1280 * ���O�t�@�C������strftime���W�J�������s���������B
1281 * FLogGetLogFilename() �� FLogOpenDialog() ��
1282 * �t�@�C�����������������B
1283 */
1284 BOOL FLogOpen(const wchar_t *fname, LogCode_t code, BOOL bom)
1285 {
1286 if (LogVar != NULL) {
1287 return FALSE;
1288 }
1289
1290 //
1291 PFileVar fv = (PFileVar)malloc(sizeof(*fv));
1292 if (fv == NULL) {
1293 return FALSE;
1294 }
1295 LogVar = fv;
1296 memset(fv, 0, sizeof(TFileVar));
1297 fv->FileHandle = INVALID_HANDLE_VALUE;
1298 fv->LogThread = INVALID_HANDLE_VALUE;
1299 fv->eLineEnd = Line_LineHead;
1300
1301 fv->log_code = code;
1302 fv->bom = bom;
1303 BOOL ret = LogStart(fv, fname);
1304 if (ret == FALSE) {
1305 FileTransEnd_(fv);
1306 }
1307
1308 return ret;
1309 }
1310
1311 BOOL FLogIsOpend(void)
1312 {
1313 return LogVar != NULL;
1314 }
1315
1316 BOOL FLogIsOpendText(void)
1317 {
1318 return LogVar != NULL && LogVar->FileLog;
1319 }
1320
1321 BOOL FLogIsOpendBin(void)
1322 {
1323 return LogVar != NULL && LogVar->BinLog;
1324 }
1325
1326 void FLogWriteStr(const wchar_t *str)
1327 {
1328 if (LogVar != NULL) {
1329 OutputStr(str);
1330 }
1331 }
1332
1333 void FLogInfo(char *param_ptr, size_t param_len)
1334 {
1335 PFileVar fv = LogVar;
1336 if (fv) {
1337 param_ptr[0] = '0'
1338 + (ts.LogBinary != 0)
1339 + ((ts.Append != 0) << 1)
1340 + ((ts.LogTypePlainText != 0) << 2)
1341 + ((ts.LogTimestamp != 0) << 3)
1342 + ((ts.LogHideDialog != 0) << 4);
1343 char *filenameU8 = ToU8W(fv->FullName);
1344 strncpy_s(param_ptr + 1, param_len - 1, filenameU8, _TRUNCATE);
1345 free(filenameU8);
1346 }
1347 else {
1348 param_ptr[0] = '0' - 1;
1349 param_ptr[1] = 0;
1350 }
1351 }
1352
1353 /**
1354 * ���������O�t�@�C����������
1355 */
1356 const wchar_t *FLogGetFilename(void)
1357 {
1358 if (LogVar == NULL) {
1359 return NULL;
1360 }
1361 return LogVar->FullName;
1362 }
1363
1364 /**
1365 * ���O�_�C�A���O���J��
1366 * @param[in,out] info.filename �t�@�C���������l
1367 * OK���A�t�@�C�����A�s�v����������free()��������
1368 * @retval TRUE [ok] ����������
1369 * @retval FALSE �L�����Z��������
1370 */
1371 BOOL FLogOpenDialog(HINSTANCE hInst, HWND hWnd, FLogDlgInfo_t *info)
1372 {
1373 LogDlgWork_t *work = (LogDlgWork_t *)calloc(sizeof(LogDlgWork_t), 1);
1374 wchar_t *srcfnameW = FLogGetLogFilename(info->filename);
1375 work->info = info;
1376 work->info->filename = srcfnameW;
1377 work->pts = &ts;
1378 work->pcv = &cv;
1379 INT_PTR ret = TTDialogBoxParam(
1380 hInst, MAKEINTRESOURCE(IDD_LOGDLG),
1381 hWnd, LogFnHook, (LPARAM)work);
1382 free(srcfnameW);
1383 free(work);
1384 return ret == IDOK ? TRUE : FALSE;
1385 }
1386
1387 /**
1388 * ���O�t�@�C����������
1389 * ���O�t�@�C�����p���C�����s��
1390 * - strftime() ���������t�W�J
1391 * - �������������O�t�@�C���t�H���_������
1392 * - �z�X�g��,�|�[�g�����W�J
1393 *
1394 * @param[in] log_filename �t�@�C����(����/��������������ok)
1395 * NULL�������f�t�H���g�t�@�C����������
1396 * strftime�`��ok
1397 * @return �t���p�X�t�@�C����
1398 * �s�v���������� free() ��������
1399 */
1400 wchar_t *FLogGetLogFilename(const wchar_t *log_filename)
1401 {
1402 const char *logdir = ToCharW(ts.LogDefaultPathW);
1403
1404 // ���������t�@�C����
1405 char base_name[MAX_PATH];
1406 if (log_filename == NULL) {
1407 strncpy_s(base_name, _countof(base_name), ts.LogDefaultName, _TRUNCATE);
1408 }
1409 else {
1410 char *filenameA = ToCharW(log_filename);
1411 strncpy_s(base_name, _countof(base_name), filenameA, _TRUNCATE);
1412 free(filenameA);
1413 }
1414
1415 // �t���p�X��
1416 char full_path[MAX_PATH];
1417 ConvFName(logdir, base_name, sizeof(base_name), "", full_path, sizeof(full_path));
1418 ParseStrftimeFileName(full_path, sizeof(full_path));
1419 ConvertLogname(full_path, sizeof(full_path));
1420
1421 free((void *)logdir);
1422 return ToWcharA(full_path);
1423 }
1424
1425 BOOL FLogIsPause()
1426 {
1427 if (LogVar == NULL) {
1428 return FALSE;
1429 }
1430 return LogVar->IsPause;
1431 }
1432
1433 void FLogWindow(int nCmdShow)
1434 {
1435 if (LogVar == NULL) {
1436 return;
1437 }
1438
1439 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1440 ShowWindow(HWndLog, nCmdShow);
1441 if (nCmdShow == SW_RESTORE) {
1442 // �g���X�^�C�� WS_EX_NOACTIVATE ��������������
1443 SetForegroundWindow(HWndLog);
1444 }
1445 }
1446
1447 void FLogShowDlg(void)
1448 {
1449 if (LogVar == NULL) {
1450 return;
1451 }
1452 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1453 ShowWindow(HWndLog, SW_SHOWNORMAL);
1454 SetForegroundWindow(HWndLog);
1455 }
1456
1457 /**
1458 * ���O��1byte��������
1459 * LogPut1() ������?
1460 */
1461 //void Log1Bin(PComVar cv, BYTE b)
1462 static void Log1Bin(BYTE b)
1463 {
1464 if (LogVar->IsPause || ProtoGetProtoFlag()) {
1465 return;
1466 }
1467 if (cv_BinSkip > 0) {
1468 cv_BinSkip--;
1469 return;
1470 }
1471 cv_BinBuf[cv_BinPtr] = b;
1472 cv_BinPtr++;
1473 if (cv_BinPtr>=InBuffSize) {
1474 cv_BinPtr = cv_BinPtr-InBuffSize;
1475 }
1476 if (cv_BCount>=InBuffSize) {
1477 cv_BCount = InBuffSize;
1478 cv_BStart = cv_BinPtr;
1479 }
1480 else {
1481 cv_BCount++;
1482 }
1483 }
1484
1485 static void LogBinSkip(int add)
1486 {
1487 if (cv_BinBuf != NULL) {
1488 cv_BinSkip += add;
1489 }
1490 }
1491
1492 /**
1493 * ���O�o�b�t�@���������������f�[�^���o�C�g��������
1494 */
1495 int FLogGetCount(void)
1496 {
1497 PFileVar fv = LogVar;
1498 if (fv == NULL) {
1499 return 0;
1500 }
1501 if (fv->FileLog) {
1502 return cv_LCount;
1503 }
1504 if (fv->BinLog) {
1505 return cv_BCount;
1506 }
1507 return 0;
1508 }
1509
1510 /**
1511 * ���O�o�b�t�@�������o�C�g��������
1512 */
1513 int FLogGetFreeCount(void)
1514 {
1515 PFileVar fv = LogVar;
1516 if (fv == NULL) {
1517 return 0;
1518 }
1519 if (fv->FileLog) {
1520 return InBuffSize - cv_LCount;
1521 }
1522 if (fv->BinLog) {
1523 return InBuffSize - cv_BCount;
1524 }
1525 return 0;
1526 }
1527
1528 /**
1529 * �o�b�t�@�������O���t�@�C������������
1530 */
1531 void FLogWriteFile(void)
1532 {
1533 PFileVar fv = LogVar;
1534 if (fv == NULL) {
1535 return;
1536 }
1537 if (cv_LogBuf!=NULL)
1538 {
1539 if (fv->FileLog) {
1540 LogToFile(fv);
1541 }
1542 }
1543
1544 if (cv_BinBuf!=NULL)
1545 {
1546 if (fv->BinLog) {
1547 LogToFile(fv);
1548 }
1549 }
1550 }
1551
1552 void FLogPutUTF32(unsigned int u32)
1553 {
1554 PFileVar fv = LogVar;
1555 BOOL log_available = (cv_LogBuf != 0);
1556
1557 if (!log_available) {
1558 // ���O�����o��������
1559 return;
1560 }
1561
1562 // �s����?(���s���o����������)
1563 if (ts.LogTimestamp && fv->eLineEnd) {
1564 // �^�C���X�^���v���o��
1565 fv->eLineEnd = Line_Other; /* clear endmark*/
1566 wchar_t* strtime = TimeStampStr(fv);
1567 FLogWriteStr(strtime);
1568 free(strtime);
1569 }
1570
1571 switch(fv->log_code) {
1572 case LOG_UTF8: {
1573 // UTF-8
1574 char u8_buf[4];
1575 size_t u8_len = UTF32ToUTF8(u32, u8_buf, _countof(u8_buf));
1576 for (size_t i = 0; i < u8_len; i++) {
1577 BYTE b = u8_buf[i];
1578 LogPut1(b);
1579 }
1580 break;
1581 }
1582 case LOG_UTF16LE:
1583 case LOG_UTF16BE: {
1584 // UTF-16
1585 wchar_t u16[2];
1586 size_t u16_len = UTF32ToUTF16(u32, u16, _countof(u16));
1587 for (size_t i = 0; i < u16_len; i++) {
1588 if (fv->log_code == LOG_UTF16LE) {
1589 // UTF-16LE
1590 LogPut1(u16[i] & 0xff);
1591 LogPut1((u16[i] >> 8) & 0xff);
1592 }
1593 else {
1594 // UTF-16BE
1595 LogPut1((u16[i] >> 8) & 0xff);
1596 LogPut1(u16[i] & 0xff);
1597 }
1598 }
1599 }
1600 }
1601
1602 if (u32 == 0x0a) {
1603 fv->eLineEnd = Line_LineHead; /* set endmark*/
1604 }
1605 }
1606
1607 static void FLogOutputBOM(PFileVar fv)
1608 {
1609 DWORD wrote;
1610
1611 switch(fv->log_code) {
1612 case 0: {
1613 // UTF-8
1614 const char *bom = "\xef\xbb\xbf";
1615 WriteFile(fv->FileHandle, bom, 3, &wrote, NULL);
1616 fv->ByteCount += 3;
1617 break;
1618 }
1619 case 1: {
1620 // UTF-16LE
1621 const char *bom = "\xff\xfe";
1622 WriteFile(fv->FileHandle, bom, 2, &wrote, NULL);
1623 fv->ByteCount += 2;
1624 break;
1625 }
1626 case 2: {
1627 // UTF-16BE
1628 const char *bom = "\xfe\xff";
1629 WriteFile(fv->FileHandle, bom, 2, &wrote, NULL);
1630 fv->ByteCount += 2;
1631 break;
1632 }
1633 default:
1634 break;
1635 }
1636 }
1637
1638 static void OutputStr(const wchar_t *str)
1639 {
1640 size_t len;
1641
1642 assert(str != NULL);
1643
1644 len = wcslen(str);
1645 while(*str != 0) {
1646 unsigned int u32;
1647 size_t u16_len = UTF16ToUTF32(str, len, &u32);
1648 switch (u16_len) {
1649 case 0:
1650 default:
1651 // ������������
1652 str++;
1653 len--;
1654 break;
1655 case 1:
1656 case 2: {
1657 FLogPutUTF32(u32);
1658 str += u16_len;
1659 len -= u16_len;
1660 break;
1661 }
1662 }
1663 }
1664 }

Back to OSDN">Back to OSDN
ViewVC Help
Powered by ViewVC 1.1.26