Develop and Download Open Source Software

Browse Subversion Repository

Contents of /branches/ttcomtester/teraterm/teraterm/filesys_log.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 10521 - (show annotations) (download) (as text)
Fri Jan 20 16:03:38 2023 UTC (13 months, 2 weeks ago) by zmatsuo
File MIME type: text/x-c++src
File size: 40665 byte(s)
add communication test tool
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 static 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 != NULL) {
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 free(add_text);
234 i += l;
235 }
236 }
237 else {
238 dest[i] = *s++;
239 i++;
240 }
241 }
242 dest[i] = 0;
243 return dest;
244 }
245
246 static void FixLogOption(void)
247 {
248 if (ts.LogBinary) {
249 ts.LogTypePlainText = false;
250 ts.LogTimestamp = false;
251 }
252 }
253
254
255 // �X���b�h���I�����t�@�C�����N���[�Y
256 static void CloseFileSync(PFileVar fv)
257 {
258 BOOL ret;
259
260 if (fv->FileHandle == INVALID_HANDLE_VALUE) {
261 return;
262 }
263
264 if (fv->LogThread != INVALID_HANDLE_VALUE) {
265 // �X���b�h���I������
266 ret = PostThreadMessage(fv->LogThreadId, WM_QUIT, 0, 0);
267 if (ret != 0) {
268 // �X���b�h�L���[���G���L���[���������������������������s���B
269 WaitForSingleObject(fv->LogThread, INFINITE);
270 }
271 else {
272 //DWORD code = GetLastError();
273 }
274 CloseHandle(fv->LogThread);
275 fv->LogThread = INVALID_HANDLE_VALUE;
276 }
277 CloseHandle(fv->FileHandle);
278 fv->FileHandle = INVALID_HANDLE_VALUE;
279 }
280
281 // �x�����������p�X���b�h
282 static unsigned _stdcall DeferredLogWriteThread(void *arg)
283 {
284 MSG msg;
285 PFileVar fv = (PFileVar)arg;
286 PCHAR buf;
287 DWORD buflen;
288 DWORD wrote;
289
290 PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE);
291
292 // �X���b�h�L���[���������I�������������X���b�h�����������m�����B
293 if (fv->LogThreadEvent != NULL) {
294 SetEvent(fv->LogThreadEvent);
295 }
296
297 while (GetMessage(&msg, NULL, 0, 0) > 0) {
298 switch (msg.message) {
299 case WM_DPC_LOGTHREAD_SEND:
300 buf = (PCHAR)msg.wParam;
301 buflen = (DWORD)msg.lParam;
302 WriteFile(fv->FileHandle, buf, buflen, &wrote, NULL);
303 free(buf); // ����������������
304 break;
305
306 case WM_QUIT:
307 goto end;
308 break;
309 }
310 }
311
312 end:
313 _endthreadex(0);
314 return (0);
315 }
316
317 // �x�����������p�X���b�h���N�����B
318 // (2013.4.19 yutaka)
319 // DeferredLogWriteThread �X���b�h���N�������A�X���b�h�L���[�����������������O���A
320 // ���O�t�@�C�����N���[�Y(CloseFileSync)���s���������A�G���L���[�����s���A�f�b�h���b�N
321 // �����������������C�������B
322 // �X���b�h�����������s�������A���O�����C�x���g�I�u�W�F�N�g���g�����A�X���b�h�L���[��
323 // ���������������������������������B���O�t���C�x���g�I�u�W�F�N�g���g���������A
324 // �V�X�e��(Windows OS)�������j�[�N�����O�������K�v�������B
325 // (2016.9.23 yutaka)
326 static void StartThread(PFileVar fv)
327 {
328 unsigned tid;
329 fv->LogThreadEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
330 fv->LogThread = (HANDLE)_beginthreadex(NULL, 0, DeferredLogWriteThread, fv, 0, &tid);
331 fv->LogThreadId = tid;
332 if (fv->LogThreadEvent != NULL) {
333 WaitForSingleObject(fv->LogThreadEvent, INFINITE);
334 CloseHandle(fv->LogThreadEvent);
335 }
336 }
337
338 /**
339 * �_�C�A���O�����e�� ts ����������
340 *
341 * TODO
342 * �_�C�A���O�����������l�������I��������
343 * ����������������������������������������������?
344 */
345 static void SetLogFlags(HWND Dialog)
346 {
347 WORD BinFlag, val;
348
349 GetRB(Dialog, &BinFlag, IDC_FOPTBIN, IDC_FOPTBIN);
350 ts.LogBinary = BinFlag;
351
352 GetRB(Dialog, &val, IDC_APPEND, IDC_APPEND);
353 ts.Append = val;
354
355 if (!BinFlag) {
356 GetRB(Dialog, &val, IDC_PLAINTEXT, IDC_PLAINTEXT);
357 ts.LogTypePlainText = val;
358
359 GetRB(Dialog, &val, IDC_TIMESTAMP, IDC_TIMESTAMP);
360 ts.LogTimestamp = val;
361 }
362
363 GetRB(Dialog, &val, IDC_HIDEDIALOG, IDC_HIDEDIALOG);
364 ts.LogHideDialog = val;
365
366 GetRB(Dialog, &val, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST);
367 ts.LogAllBuffIncludedInFirst = val;
368
369 ts.LogTimestampType = (WORD)(GetCurSel(Dialog, IDC_TIMESTAMPTYPE) - 1);
370 }
371
372 /**
373 * ���O�t�@�C���`�F�b�N
374 *
375 * @param[in] filename
376 * @param[out] exist TURE/FALSE
377 * @param[out] bom 0 no BOM (or file not exist)
378 * 1 UTF-8
379 * 2 UTF-16LE
380 * 3 UTF-16BE
381 */
382 static void CheckLogFile(const wchar_t *filename, BOOL *exist, int *bom)
383 {
384 *exist = FALSE;
385 *bom = 0;
386
387 // �t�@�C������������?
388 DWORD logdir = GetFileAttributesW(filename);
389 if ((logdir != INVALID_FILE_ATTRIBUTES) && ((logdir & FILE_ATTRIBUTE_DIRECTORY) == 0)) {
390 // �t�@�C����������
391 *exist = TRUE;
392
393 // BOM�L��/�����`�F�b�N
394 FILE *fp = _wfopen(filename, L"rb");
395 if (fp != NULL) {
396 unsigned char tmp[4];
397 size_t l = fread(tmp, 1, sizeof(tmp), fp);
398 fclose(fp);
399 if (l < 2) {
400 *bom = 0;
401 } else if (l >= 2 && tmp[0] == 0xff && tmp[1] == 0xfe) {
402 // UTF-16LE
403 *bom = 2;
404 } else if (l >= 2 && tmp[0] == 0xfe && tmp[1] == 0xff) {
405 // UTF-16BE
406 *bom = 3;
407 } else if (l >= 3 && tmp[0] == 0xef && tmp[1] == 0xbb && tmp[2] == 0xbf) {
408 // UTF-8
409 *bom = 1;
410 } else {
411 *bom = 0;
412 }
413 }
414 }
415 }
416
417 typedef struct {
418 FLogDlgInfo_t *info;
419 // work
420 BOOL file_exist;
421 int current_bom; // ���������t�@�C�����G���R�[�f�B���O�i�t�@�C����BOM���������j
422 TTTSet *pts;
423 TComVar *pcv;
424 } LogDlgWork_t;
425
426
427 /*
428 * Log �_�C�A���O�������AEnable/Disable �����������R���g���[��
429 *
430 * - Append
431 * �w���������t�@�C������������������ Enable
432 * �w���������t�@�C�������������������� Disable
433 *
434 * - BOM, Encoding
435 * Text ���� New/Overwrite �������� Enable
436 * ���������������� Disable
437 * BOM ���t�@�C��������������������������������������
438 * Encoding �����L�����������������A�����t�@�C�����G���R�[�f�B���O��
439 * �����I���_�C�A���O�����f���������A���[�U�������w������������
440 *
441 * - Plain Text, Timestamp, Timestamp ����
442 * Text �������� Enable
443 * Binary �������� Disable
444 *
445 * - Timestamp ����
446 * Timestamp=on �������� Enable
447 * Timestamp=off �������� Disable
448 */
449
450 static void ArrangeControls(HWND Dialog, LogDlgWork_t *work,
451 WORD Append, WORD LogBinary,
452 WORD LogTypePlainText, WORD LogTimestamp)
453 {
454 if (work->file_exist) {
455 // Append ���W�I�{�^�����A�t�@�C�����������������L��������
456 EnableWindow(GetDlgItem(Dialog, IDC_APPEND), TRUE);
457
458 if (Append > 0) {
459 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
460 }
461 else {
462 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
463 }
464 }
465 else { // �t�@�C�������� -> �V�K
466 EnableWindow(GetDlgItem(Dialog, IDC_APPEND), FALSE);
467 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
468 }
469
470 if (!LogBinary && !Append) {
471 EnableWindow(GetDlgItem(Dialog, IDC_BOM), TRUE);
472 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), TRUE);
473 }
474 else {
475 EnableWindow(GetDlgItem(Dialog, IDC_BOM), FALSE);
476 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), FALSE);
477 }
478
479 if (LogBinary) {
480 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTBIN);
481
482 DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT);
483 DisableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP);
484 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
485 }
486 else {
487 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTTEXT);
488
489 EnableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT);
490 EnableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP);
491 EnableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
492
493 if (LogTypePlainText) {
494 SetRB(Dialog, 1, IDC_PLAINTEXT, IDC_PLAINTEXT);
495 }
496 if (LogTimestamp) {
497 SetRB(Dialog, 1, IDC_TIMESTAMP, IDC_TIMESTAMP);
498 }
499 else {
500 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
501 }
502 }
503
504 if (work->file_exist && Append) {
505 // �����t�@�C�����G���R�[�f�B���O�����f����
506 int bom = work->current_bom;
507 int cur =
508 bom == 1 ? 0 :
509 bom == 2 ? 1 :
510 bom == 3 ? 2 : 0;
511 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, cur, 0);
512 }
513 }
514
515 static void CheckLogFile(HWND Dialog, const wchar_t *filename, LogDlgWork_t *work)
516 {
517 BOOL exist;
518 int bom;
519 CheckLogFile(filename, &exist, &bom);
520 work->file_exist = exist;
521 work->current_bom = bom;
522 }
523
524 static INT_PTR CALLBACK LogFnHook(HWND Dialog, UINT Message, WPARAM wParam, LPARAM lParam)
525 {
526 static const DlgTextInfo TextInfos[] = {
527 { 0, "DLG_TABSHEET_TITLE_LOG" },
528 { IDC_SENDFILE_FILENAME_TITLE, "DLG_FOPT_FILENAME_TITLE" },
529 { IDC_FOPTTEXT, "DLG_FOPT_TEXT" },
530 { IDC_FOPTBIN, "DLG_FOPT_BINARY" },
531 { IDC_BOM, "DLG_FOPT_BOM" },
532 { IDC_APPEND, "DLG_FOPT_APPEND" },
533 { IDC_PLAINTEXT, "DLG_FOPT_PLAIN" },
534 { IDC_HIDEDIALOG, "DLG_FOPT_HIDEDIALOG" },
535 { IDC_ALLBUFF_INFIRST, "DLG_FOPT_ALLBUFFINFIRST" },
536 { IDC_TIMESTAMP, "DLG_FOPT_TIMESTAMP" },
537 { IDC_NEW_OVERWRITE, "DLG_FOPT_NEW_OVERWRITE" },
538 { IDC_APPEND_GROUP, "DLG_FOPT_APPEND_LABEL" },
539 { IDC_BINARY_GROUP, "DLG_FOPT_BINARY_LABEL" },
540 };
541 static const I18nTextInfo timestamp_list[] = {
542 { "DLG_FOPT_TIMESTAMP_LOCAL", L"Local Time" },
543 { "DLG_FOPT_TIMESTAMP_UTC", L"UTC" },
544 { "DLG_FOPT_TIMESTAMP_ELAPSED_LOGGING", L"Elapsed Time (Logging)" },
545 { "DLG_FOPT_TIMESTAMP_ELAPSED_CONNECTION", L"Elapsed Time (Connection)" },
546 };
547 LogDlgWork_t *work = (LogDlgWork_t *)GetWindowLongPtr(Dialog, DWLP_USER);
548
549 if (Message == RegisterWindowMessage(HELPMSGSTRING)) {
550 // �R�����_�C�A���O�������w���v���b�Z�[�W���t��������
551 Message = WM_COMMAND;
552 wParam = IDHELP;
553 }
554 switch (Message) {
555 case WM_INITDIALOG: {
556 work = (LogDlgWork_t *)lParam;
557 TTTSet *pts = work->pts;
558 SetWindowLongPtr(Dialog, DWLP_USER, (LONG_PTR)work);
559 ::DragAcceptFiles(Dialog, TRUE);
560
561 SetDlgTextsW(Dialog, TextInfos, _countof(TextInfos), pts->UILanguageFileW);
562 SetI18nListW("Tera Term", Dialog, IDC_TIMESTAMPTYPE, timestamp_list, _countof(timestamp_list),
563 pts->UILanguageFileW, 0);
564
565 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-8");
566 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16LE");
567 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16BE");
568 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
569
570 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, work->info->filename);
571
572 // timestamp ����
573 int tstype = pts->LogTimestampType == TIMESTAMP_LOCAL ? 0 :
574 pts->LogTimestampType == TIMESTAMP_UTC ? 1 :
575 pts->LogTimestampType == TIMESTAMP_ELAPSED_LOGSTART ? 2 :
576 pts->LogTimestampType == TIMESTAMP_ELAPSED_CONNECTED ? 3 : 0;
577 SendDlgItemMessage(Dialog, IDC_TIMESTAMPTYPE, CB_SETCURSEL, tstype, 0);
578
579
580 CheckLogFile(Dialog, work->info->filename, work);
581 ArrangeControls(Dialog, work, pts->Append, pts->LogBinary, pts->LogTypePlainText, pts->LogTimestamp);
582 work->info->filename = NULL;
583
584 // Hide dialog �`�F�b�N�{�b�N�X
585 if (pts->LogHideDialog) {
586 SetRB(Dialog, 1, IDC_HIDEDIALOG, IDC_HIDEDIALOG);
587 }
588
589 // Include screen buffer �`�F�b�N�{�b�N�X
590 if (pts->LogAllBuffIncludedInFirst) {
591 SetRB(Dialog, 1, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST);
592 }
593
594 CenterWindow(Dialog, GetParent(Dialog));
595
596 return TRUE;
597 }
598
599 case WM_COMMAND:
600 switch (LOWORD(wParam)) {
601 case IDOK: {
602 wchar_t *filename;
603 hGetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, &filename);
604 work->info->filename = filename;
605 work->info->append = IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED;
606 work->info->bom = IsDlgButtonChecked(Dialog, IDC_BOM) == BST_CHECKED;
607 work->info->code = (LogCode_t)SendDlgItemMessageA(Dialog, IDC_TEXTCODING_DROPDOWN, CB_GETCURSEL, 0, 0);
608 SetLogFlags(Dialog);
609 EndDialog(Dialog, IDOK);
610 break;
611 }
612 case IDCANCEL:
613 EndDialog(Dialog, IDCANCEL);
614 break;
615 case IDHELP:
616 OpenHelpCV(work->pcv, HH_HELP_CONTEXT, HlpFileLog);
617 break;
618 case IDC_FOPT_FILENAME_BUTTON: {
619 /* save current dir */
620 const wchar_t *UILanguageFile = work->pts->UILanguageFileW;
621 wchar_t *curdir;
622 hGetCurrentDirectoryW(&curdir);
623
624 wchar_t fname[MAX_PATH];
625 GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname, _countof(fname));
626
627 const wchar_t* simple_log_filter = L"*.txt;*.log";
628 wchar_t *FNFilter = GetCommonDialogFilterWW(simple_log_filter, UILanguageFile);
629
630 wchar_t *caption;
631 wchar_t *uimsg;
632 GetI18nStrWW("Tera Term", "FILEDLG_TRANS_TITLE_LOG",
633 TitLog, UILanguageFile, &uimsg);
634 aswprintf(&caption, L"Tera Term: %s", uimsg);
635 free(uimsg);
636
637 OPENFILENAMEW ofn = {};
638 ofn.lStructSize = get_OPENFILENAME_SIZEW();
639 //ofn.Flags = OFN_PATHMUSTEXIST | OFN_OVERWRITEPROMPT;
640 ofn.Flags |= OFN_EXPLORER | OFN_ENABLESIZING;
641 ofn.Flags |= OFN_SHOWHELP;
642 ofn.Flags |= OFN_NOCHANGEDIR; // ��������������������������������
643 ofn.hwndOwner = Dialog;
644 ofn.lpstrFilter = FNFilter;
645 ofn.nFilterIndex = 1;
646 ofn.lpstrFile = fname;
647 ofn.nMaxFile = _countof(fname);
648 ofn.lpstrTitle = caption;
649 ofn.lpstrInitialDir = work->pts->LogDefaultPathW;
650 BOOL Ok = GetSaveFileNameW(&ofn);
651 free(caption);
652 free(FNFilter);
653 if (Ok) {
654 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname);
655 }
656
657 /* restore dir */
658 SetCurrentDirectoryW(curdir);
659 free(curdir);
660
661 break;
662 }
663 case IDC_NEW_OVERWRITE:
664 case IDC_APPEND:
665 case IDC_FOPTTEXT:
666 case IDC_FOPTBIN:
667 case IDC_TIMESTAMP:
668 {
669 WORD Appnd, LogBinary, LogTypePlainText, LogTimestamp;
670 GetRB(Dialog, &Appnd, IDC_APPEND, IDC_APPEND);
671 GetRB(Dialog, &LogBinary, IDC_FOPTBIN, IDC_FOPTBIN);
672 GetRB(Dialog, &LogTypePlainText, IDC_PLAINTEXT, IDC_PLAINTEXT);
673 GetRB(Dialog, &LogTimestamp, IDC_TIMESTAMP, IDC_TIMESTAMP);
674 ArrangeControls(Dialog, work, Appnd, LogBinary, LogTypePlainText, LogTimestamp);
675 }
676 break;
677 case IDC_FOPT_FILENAME_EDIT:
678 if (HIWORD(wParam) == EN_CHANGE){
679 wchar_t *filename;
680 hGetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, &filename);
681 CheckLogFile(Dialog, filename, work);
682 free(filename);
683 {
684 WORD Appnd, LogBinary, LogTypePlainText, LogTimestamp;
685 GetRB(Dialog, &Appnd, IDC_APPEND, IDC_APPEND);
686 GetRB(Dialog, &LogBinary, IDC_FOPTBIN, IDC_FOPTBIN);
687 GetRB(Dialog, &LogTypePlainText, IDC_PLAINTEXT, IDC_PLAINTEXT);
688 GetRB(Dialog, &LogTimestamp, IDC_TIMESTAMP, IDC_TIMESTAMP);
689 ArrangeControls(Dialog, work, Appnd, LogBinary, LogTypePlainText, LogTimestamp);
690 }
691 }
692 break;
693 }
694 break;
695 case WM_DROPFILES: {
696 // �����h���b�v��������������1������������
697 HDROP hDrop = (HDROP)wParam;
698 const UINT len = DragQueryFileW(hDrop, 0, NULL, 0);
699 if (len == 0) {
700 DragFinish(hDrop);
701 return TRUE;
702 }
703 wchar_t *filename = (wchar_t *)malloc(sizeof(wchar_t) * (len + 1));
704 DragQueryFileW(hDrop, 0, filename, len + 1);
705 filename[len] = '\0';
706 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
707 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename);
708 SendDlgItemMessage(Dialog, IDC_FOPT_FILENAME_EDIT, EM_SETSEL, len, len);
709 free(filename);
710 DragFinish(hDrop);
711 return TRUE;
712 }
713 }
714 return FALSE;
715 }
716
717 static void OpenLogFile(PFileVar fv)
718 {
719 // LogLockExclusive ���L���������������������L���������A
720 // ���������������O�t�@�C���������G�f�B�^���J����������
721 int dwShareMode = FILE_SHARE_READ;
722 if (!ts.LogLockExclusive) {
723 dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
724 }
725 fv->FileHandle = CreateFileW(fv->FullName, GENERIC_WRITE, dwShareMode, NULL,
726 OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
727 }
728
729 static BOOL LogStart(PFileVar fv, const wchar_t *fname)
730 {
731 fv->FullName = _wcsdup(fname);
732 FixLogOption();
733
734 if (ts.LogBinary > 0)
735 {
736 fv->BinLog = TRUE;
737 fv->FileLog = FALSE;
738 if (! CreateBinBuf())
739 {
740 return FALSE;
741 }
742 }
743 else {
744 fv->BinLog = FALSE;
745 fv->FileLog = TRUE;
746 if (! CreateLogBuf())
747 {
748 return FALSE;
749 }
750 }
751 cv_LStart = cv_LogPtr;
752 cv_LCount = 0;
753
754 OpenLogFile(fv);
755 if (fv->FileHandle == INVALID_HANDLE_VALUE) {
756 return FALSE;
757 }
758
759 /* 2007.05.24 Gentaro */
760 fv->eLineEnd = Line_LineHead;
761 if (ts.Append > 0)
762 {
763 SetFilePointer(fv->FileHandle, 0, NULL, FILE_END);
764 /* 2007.05.24 Gentaro
765 If log file already exists,
766 a newline is inserted before the first timestamp.
767 */
768 fv->eLineEnd = Line_FileHead;
769 }
770
771 // BOM�o��
772 if (ts.Append == 0 && ts.LogBinary == 0 && fv->bom) {
773 // ���L��������(�V�K) && �o�C�i���������� && BOM ���o����
774 FLogOutputBOM(fv);
775 }
776
777 // Log rotate configuration
778 fv->RotateMode = ts.LogRotate;
779 fv->RotateSize = ts.LogRotateSize;
780 fv->RotateStep = ts.LogRotateStep;
781
782 // Log rotate���L���������A�����t�@�C���T�C�Y�����������B
783 // �������t�@�C�������������T�C�Y�����[�e�[�g�������������C���B
784 // (2016.4.9 yutaka)
785 if (fv->RotateMode != ROTATE_NONE) {
786 DWORD size = GetFileSize(fv->FileHandle, NULL);
787 if (size == -1) {
788 return FALSE;
789 }
790 fv->ByteCount = size;
791 }
792 else {
793 fv->ByteCount = 0;
794 }
795
796 if (! OpenFTDlg_(fv)) {
797 return FALSE;
798 }
799
800 fv->IsPause = FALSE;
801 fv->StartTime = GetTickCount();
802
803 if (ts.DeferredLogWriteMode) {
804 StartThread(fv);
805 }
806
807 if (fv->FileLog) {
808 cv.Log1Byte = LogPut1;
809 }
810 if (fv->BinLog) {
811 cv.Log1Bin = Log1Bin;
812 cv.LogBinSkip = LogBinSkip;
813 }
814
815 return TRUE;
816 }
817
818 /**
819 * �����o�b�t�@�������f�[�^�����������O�������o��
820 * (2013.9.29 yutaka)
821 *
822 * TODO
823 * 1�s������
824 */
825 void FLogOutputAllBuffer(void)
826 {
827 PFileVar fv = LogVar;
828 DWORD ofs;
829 int size;
830 wchar_t buf[512];
831 for (ofs = 0 ; ; ofs++ ) {
832 // 1�����s�����������B���������������A�G�X�P�[�v�V�[�P���X�������������B
833 size = BuffGetAnyLineDataW(ofs, buf, _countof(buf));
834 if (size == -1)
835 break;
836
837 OutputStr(buf);
838 OutputStr(L"\r\n");
839 LogToFile(fv);
840 }
841 }
842
843 /**
844 * ���O��1byte��������
845 * �o�b�t�@��������������
846 * ���������������� LogToFile() ���s������
847 */
848 void LogPut1(BYTE b)
849 {
850 PFileVar fv = LogVar;
851
852 cv_LogBuf[cv_LogPtr] = b;
853 cv_LogPtr++;
854 if (cv_LogPtr>=InBuffSize)
855 cv_LogPtr = cv_LogPtr-InBuffSize;
856
857 if (fv->FileLog)
858 {
859 if (cv_LCount>=InBuffSize)
860 {
861 cv_LCount = InBuffSize;
862 cv_LStart = cv_LogPtr;
863 }
864 else
865 cv_LCount++;
866 }
867 else
868 cv_LCount = 0;
869 }
870
871 static BOOL Get1(PCHAR Buf, int *Start, int *Count, PBYTE b)
872 {
873 if (*Count<=0) return FALSE;
874 *b = Buf[*Start];
875 (*Start)++;
876 if (*Start>=InBuffSize)
877 *Start = *Start-InBuffSize;
878 (*Count)--;
879 return TRUE;
880 }
881
882
883
884 static CRITICAL_SECTION g_filelog_lock; /* ���b�N�p���� */
885
886 void logfile_lock_initialize(void)
887 {
888 InitializeCriticalSection(&g_filelog_lock);
889 }
890
891 static inline void logfile_lock(void)
892 {
893 EnterCriticalSection(&g_filelog_lock);
894 }
895
896 static inline void logfile_unlock(void)
897 {
898 LeaveCriticalSection(&g_filelog_lock);
899 }
900
901 // ���O�����[�e�[�g�����B
902 // (2013.3.21 yutaka)
903 static void LogRotate(PFileVar fv)
904 {
905 int loopmax = 10000; // XXX
906 int i, k;
907
908 if (fv->RotateMode == ROTATE_NONE)
909 return;
910
911 if (fv->RotateMode == ROTATE_SIZE) {
912 if (fv->ByteCount <= fv->RotateSize)
913 return;
914 //OutputDebugPrintf("%s: mode %d size %ld\n", __FUNCTION__, fv->RotateMode, fv->ByteCount);
915 } else {
916 return;
917 }
918
919 logfile_lock();
920 // ���O�T�C�Y���������������B
921 fv->ByteCount = 0;
922
923 // �������������t�@�C�����N���[�Y�����A�������t�@�C�����I�[�v�������B
924 CloseFileSync(fv);
925
926 // �������[�e�[�V�������X�e�b�v�����w����������
927 if (fv->RotateStep > 0)
928 loopmax = fv->RotateStep;
929
930 for (i = 1 ; i <= loopmax ; i++) {
931 wchar_t *filename;
932 aswprintf(&filename, L"%s.%d", fv->FullName, i);
933 DWORD attr = GetFileAttributesW(filename);
934 free(filename);
935 if (attr == INVALID_FILE_ATTRIBUTES)
936 break;
937 }
938 if (i > loopmax) {
939 // �������������������������A�������t�@�C�������p�������B
940 i = loopmax;
941 }
942
943 // ���t�@�C�������l�[���B
944 for (k = i-1 ; k >= 0 ; k--) {
945 wchar_t *oldfile;
946 if (k == 0)
947 oldfile = _wcsdup(fv->FullName);
948 else
949 aswprintf(&oldfile, L"%s.%d", fv->FullName, k);
950 wchar_t *newfile;
951 aswprintf(&newfile, L"%s.%d", fv->FullName, k+1);
952 DeleteFileW(newfile);
953 if (MoveFileW(oldfile, newfile) == 0) {
954 OutputDebugPrintf("%s: rename %d\n", __FUNCTION__, errno);
955 }
956 free(oldfile);
957 free(newfile);
958 }
959
960 // ���I�[�v��
961 OpenLogFile(fv);
962 if (fv->bom) {
963 FLogOutputBOM(fv);
964 }
965 if (ts.DeferredLogWriteMode) {
966 StartThread(fv);
967 }
968
969 logfile_unlock();
970 }
971
972 static wchar_t *TimeStampStr(PFileVar fv)
973 {
974 char *strtime = NULL;
975 switch (ts.LogTimestampType) {
976 case TIMESTAMP_LOCAL:
977 default:
978 strtime = mctimelocal(ts.LogTimestampFormat, FALSE);
979 break;
980 case TIMESTAMP_UTC:
981 strtime = mctimelocal(ts.LogTimestampFormat, TRUE);
982 break;
983 case TIMESTAMP_ELAPSED_LOGSTART:
984 strtime = strelapsed(fv->StartTime);
985 break;
986 case TIMESTAMP_ELAPSED_CONNECTED:
987 strtime = strelapsed(cv.ConnectedTime);
988 break;
989 }
990
991 char tmp[128];
992 tmp[0] = 0;
993 strncat_s(tmp, sizeof(tmp), "[", _TRUNCATE);
994 strncat_s(tmp, sizeof(tmp), strtime, _TRUNCATE);
995 strncat_s(tmp, sizeof(tmp), "] ", _TRUNCATE);
996
997 return ToWcharA(tmp);
998 }
999
1000 /**
1001 * �o�b�t�@�������O���t�@�C������������
1002 */
1003 static void LogToFile(PFileVar fv)
1004 {
1005 PCHAR Buf;
1006 int Start, Count;
1007 BYTE b;
1008
1009 if (fv->FileLog)
1010 {
1011 Buf = cv_LogBuf;
1012 Start = cv_LStart;
1013 Count = cv_LCount;
1014 }
1015 else if (fv->BinLog)
1016 {
1017 Buf = cv_BinBuf;
1018 Start = cv_BStart;
1019 Count = cv_BCount;
1020 }
1021 else
1022 return;
1023
1024 if (Buf==NULL) return;
1025 if (Count==0) return;
1026
1027 // ���b�N������(2004.8.6 yutaka)
1028 logfile_lock();
1029
1030 // ���������f�[�^����������
1031 DWORD WriteBufMax = 8192;
1032 DWORD WriteBufLen = 0;
1033 PCHAR WriteBuf = (PCHAR)malloc(WriteBufMax);
1034 while (Get1(Buf,&Start,&Count,&b)) {
1035 if (FLogIsPause() || ProtoGetProtoFlag()) {
1036 continue;
1037 }
1038
1039 if (WriteBufLen >= (WriteBufMax*4/5)) {
1040 WriteBufMax *= 2;
1041 WriteBuf = (PCHAR)realloc(WriteBuf, WriteBufMax);
1042 }
1043
1044 WriteBuf[WriteBufLen++] = b;
1045
1046 (fv->ByteCount)++;
1047 }
1048
1049 // ��������
1050 if (WriteBufLen > 0) {
1051 if (ts.DeferredLogWriteMode) {
1052 PostThreadMessage(fv->LogThreadId, WM_DPC_LOGTHREAD_SEND, (WPARAM)WriteBuf, WriteBufLen);
1053 }
1054 else {
1055 DWORD wrote;
1056 WriteFile(fv->FileHandle, WriteBuf, WriteBufLen, &wrote, NULL);
1057 free(WriteBuf);
1058 }
1059 }
1060
1061 logfile_unlock();
1062
1063 if (fv->FileLog)
1064 {
1065 cv_LStart = Start;
1066 cv_LCount = Count;
1067 }
1068 else {
1069 cv_BStart = Start;
1070 cv_BCount = Count;
1071 }
1072 if (FLogIsPause() || ProtoGetProtoFlag()) return;
1073 fv->FLogDlg->RefreshNum(fv->StartTime, fv->FileSize, fv->ByteCount);
1074
1075
1076 // ���O�E���[�e�[�g
1077 LogRotate(fv);
1078 }
1079
1080 static BOOL CreateLogBuf(void)
1081 {
1082 if (cv_LogBuf==NULL)
1083 {
1084 cv_LogBuf = (char *)malloc(InBuffSize);
1085 cv_LogPtr = 0;
1086 cv_LStart = 0;
1087 cv_LCount = 0;
1088 }
1089 return (cv_LogBuf!=NULL);
1090 }
1091
1092 static void FreeLogBuf(void)
1093 {
1094 free(cv_LogBuf);
1095 cv_LogBuf = NULL;
1096 cv_LogPtr = 0;
1097 cv_LStart = 0;
1098 cv_LCount = 0;
1099 }
1100
1101 static BOOL CreateBinBuf(void)
1102 {
1103 if (cv_BinBuf==NULL)
1104 {
1105 cv_BinBuf = (PCHAR)malloc(InBuffSize);
1106 cv_BinPtr = 0;
1107 cv_BStart = 0;
1108 cv_BCount = 0;
1109 }
1110 return (cv_BinBuf!=NULL);
1111 }
1112
1113 static void FreeBinBuf(void)
1114 {
1115 free(cv_BinBuf);
1116 cv_BinBuf = NULL;
1117 cv_BinPtr = 0;
1118 cv_BStart = 0;
1119 cv_BCount = 0;
1120 }
1121
1122 static void FileTransEnd_(PFileVar fv)
1123 {
1124 fv->FileLog = FALSE;
1125 fv->BinLog = FALSE;
1126 cv.Log1Byte = NULL;
1127 cv.Log1Bin = NULL;
1128 cv.LogBinSkip = NULL;
1129 PFileTransDlg FLogDlg = fv->FLogDlg;
1130 if (FLogDlg != NULL) {
1131 FLogDlg->DestroyWindow();
1132 FLogDlg = NULL;
1133 fv->FLogDlg = NULL;
1134 }
1135 CloseFileSync(fv);
1136 FreeLogBuf();
1137 FreeBinBuf();
1138 free(fv->FullName);
1139 fv->FullName = NULL;
1140 free(fv);
1141
1142 LogVar = NULL;
1143 }
1144
1145 /**
1146 * ���O���|�[�Y����
1147 */
1148 void FLogPause(BOOL Pause)
1149 {
1150 PFileVar fv = LogVar;
1151 if (fv == NULL) {
1152 return;
1153 }
1154 fv->IsPause = Pause;
1155 fv->FLogDlg->ChangeButton(Pause);
1156 }
1157
1158 /**
1159 * ���O���[�e�[�g������
1160 * ���O���T�C�Y��<size>�o�C�g���������������A���[�e�[�V��������������������
1161 */
1162 void FLogRotateSize(size_t size)
1163 {
1164 PFileVar fv = LogVar;
1165 if (fv == NULL) {
1166 return;
1167 }
1168 fv->RotateMode = ROTATE_SIZE;
1169 fv->RotateSize = (LONG)size;
1170 }
1171
1172 /**
1173 * ���O���[�e�[�g������
1174 * ���O�t�@�C������������������
1175 */
1176 void FLogRotateRotate(int step)
1177 {
1178 PFileVar fv = LogVar;
1179 if (fv == NULL) {
1180 return;
1181 }
1182 fv->RotateStep = step;
1183 }
1184
1185 /**
1186 * ���O���[�e�[�g������
1187 * ���[�e�[�V���������~
1188 */
1189 void FLogRotateHalt(void)
1190 {
1191 PFileVar fv = LogVar;
1192 if (fv == NULL) {
1193 return;
1194 }
1195 fv->RotateMode = ROTATE_NONE;
1196 fv->RotateSize = 0;
1197 fv->RotateStep = 0;
1198 }
1199
1200 static INT_PTR CALLBACK OnCommentDlgProc(HWND hDlgWnd, UINT msg, WPARAM wp, LPARAM)
1201 {
1202 static const DlgTextInfo TextInfos[] = {
1203 { 0, "DLG_COMMENT_TITLE" },
1204 { IDOK, "BTN_OK" }
1205 };
1206
1207 switch (msg) {
1208 case WM_INITDIALOG:
1209 // �G�f�B�b�g�R���g���[�����t�H�[�J�X��������
1210 SetFocus(GetDlgItem(hDlgWnd, IDC_EDIT_COMMENT));
1211 SetDlgTextsW(hDlgWnd, TextInfos, _countof(TextInfos), ts.UILanguageFileW);
1212 return FALSE;
1213
1214 case WM_COMMAND:
1215 switch (LOWORD(wp)) {
1216 case IDOK: {
1217 size_t len = SendDlgItemMessageW(hDlgWnd, IDC_EDIT_COMMENT, WM_GETTEXTLENGTH, 0, 0);
1218 len += 1;
1219 wchar_t *buf = (wchar_t *)malloc(len * sizeof(wchar_t));
1220 GetDlgItemTextW(hDlgWnd, IDC_EDIT_COMMENT, buf, (int)len);
1221 FLogWriteStr(buf);
1222 FLogWriteStr(L"\n"); // TODO ���s�R�[�h
1223 free(buf);
1224 TTEndDialog(hDlgWnd, IDOK);
1225 break;
1226 }
1227 default:
1228 return FALSE;
1229 }
1230 break;
1231
1232 case WM_CLOSE:
1233 TTEndDialog(hDlgWnd, 0);
1234 return TRUE;
1235
1236 default:
1237 return FALSE;
1238 }
1239 return TRUE;
1240 }
1241
1242 /**
1243 * ���O�t�@�C�����R�����g���������� (2004.8.6 yutaka)
1244 */
1245 void FLogAddCommentDlg(HINSTANCE hInst, HWND hWnd)
1246 {
1247 PFileVar fv = LogVar;
1248 if (fv == NULL) {
1249 return;
1250 }
1251 TTDialogBox(hInst, MAKEINTRESOURCE(IDD_COMMENT_DIALOG),
1252 hWnd, OnCommentDlgProc);
1253 }
1254
1255 void FLogClose(void)
1256 {
1257 PFileVar fv = LogVar;
1258 if (fv == NULL) {
1259 return;
1260 }
1261
1262 FileTransEnd_(fv);
1263 }
1264
1265 /**
1266 * ���O���I�[�v������
1267 * @param[in] fname ���O�t�@�C����, CreateFile()���n������
1268 *
1269 * ���O�t�@�C������strftime���W�J�������s���������B
1270 * FLogGetLogFilename() �� FLogOpenDialog() ��
1271 * �t�@�C�����������������B
1272 */
1273 BOOL FLogOpen(const wchar_t *fname, LogCode_t code, BOOL bom)
1274 {
1275 if (LogVar != NULL) {
1276 return FALSE;
1277 }
1278
1279 //
1280 PFileVar fv = (PFileVar)malloc(sizeof(*fv));
1281 if (fv == NULL) {
1282 return FALSE;
1283 }
1284 LogVar = fv;
1285 memset(fv, 0, sizeof(TFileVar));
1286 fv->FileHandle = INVALID_HANDLE_VALUE;
1287 fv->LogThread = INVALID_HANDLE_VALUE;
1288 fv->eLineEnd = Line_LineHead;
1289
1290 fv->log_code = code;
1291 fv->bom = bom;
1292 BOOL ret = LogStart(fv, fname);
1293 if (ret == FALSE) {
1294 FileTransEnd_(fv);
1295 }
1296
1297 return ret;
1298 }
1299
1300 BOOL FLogIsOpend(void)
1301 {
1302 return LogVar != NULL;
1303 }
1304
1305 BOOL FLogIsOpendText(void)
1306 {
1307 return LogVar != NULL && LogVar->FileLog;
1308 }
1309
1310 BOOL FLogIsOpendBin(void)
1311 {
1312 return LogVar != NULL && LogVar->BinLog;
1313 }
1314
1315 void FLogWriteStr(const wchar_t *str)
1316 {
1317 if (LogVar != NULL) {
1318 OutputStr(str);
1319 }
1320 }
1321
1322 void FLogInfo(char *param_ptr, size_t param_len)
1323 {
1324 PFileVar fv = LogVar;
1325 if (fv) {
1326 param_ptr[0] = '0'
1327 + (ts.LogBinary != 0)
1328 + ((ts.Append != 0) << 1)
1329 + ((ts.LogTypePlainText != 0) << 2)
1330 + ((ts.LogTimestamp != 0) << 3)
1331 + ((ts.LogHideDialog != 0) << 4);
1332 char *filenameU8 = ToU8W(fv->FullName);
1333 strncpy_s(param_ptr + 1, param_len - 1, filenameU8, _TRUNCATE);
1334 free(filenameU8);
1335 }
1336 else {
1337 param_ptr[0] = '0' - 1;
1338 param_ptr[1] = 0;
1339 }
1340 }
1341
1342 /**
1343 * ���������O�t�@�C����������
1344 */
1345 const wchar_t *FLogGetFilename(void)
1346 {
1347 if (LogVar == NULL) {
1348 return NULL;
1349 }
1350 return LogVar->FullName;
1351 }
1352
1353 /**
1354 * ���O�_�C�A���O���J��
1355 * @param[in,out] info.filename �t�@�C���������l
1356 * OK���A�t�@�C�����A�s�v����������free()��������
1357 * @retval TRUE [ok] ����������
1358 * @retval FALSE �L�����Z��������
1359 */
1360 BOOL FLogOpenDialog(HINSTANCE hInst, HWND hWnd, FLogDlgInfo_t *info)
1361 {
1362 LogDlgWork_t *work = (LogDlgWork_t *)calloc(sizeof(LogDlgWork_t), 1);
1363 wchar_t *srcfnameW = FLogGetLogFilename(info->filename);
1364 work->info = info;
1365 work->info->filename = srcfnameW;
1366 work->pts = &ts;
1367 work->pcv = &cv;
1368 INT_PTR ret = TTDialogBoxParam(
1369 hInst, MAKEINTRESOURCE(IDD_LOGDLG),
1370 hWnd, LogFnHook, (LPARAM)work);
1371 free(srcfnameW);
1372 free(work);
1373 return ret == IDOK ? TRUE : FALSE;
1374 }
1375
1376 /**
1377 * ���O�t�@�C�����p���C�����s��,�t�@�C������������
1378 * - strftime() ���������t�W�J
1379 * - �������������O�t�@�C���t�H���_������
1380 * - �z�X�g��,�|�[�g�����W�J
1381 *
1382 * @param filename �t�@�C����(�p�X����������)
1383 * @return �C�������t�@�C����
1384 */
1385 wchar_t *FLogGetLogFilenameBase(const wchar_t *filename)
1386 {
1387 // �t�@�C�������������o
1388 const wchar_t *last_path_sep = wcsrchr(filename, L'\\');
1389 wchar_t *format;
1390 if (last_path_sep == NULL) {
1391 format = _wcsdup(filename);
1392 }
1393 else {
1394 format = _wcsdup(last_path_sep + 1);
1395 }
1396
1397 // strftime ���g�p������������������
1398 deleteInvalidStrftimeCharW(format);
1399
1400 // ����������0��������?
1401 if (format[0] == 0) {
1402 free(format);
1403 return _wcsdup(L"");
1404 }
1405
1406 // ��������������
1407 time_t time_local;
1408 time(&time_local);
1409 struct tm tm_local;
1410 localtime_s(&tm_local, &time_local);
1411
1412 // strftime()������
1413 // ���������������g��
1414 size_t len = 32;
1415 wchar_t *formated = NULL;
1416 while (1) {
1417 wchar_t *formated_realloc = (wchar_t *)realloc(formated, sizeof(wchar_t) * len);
1418 if (formated_realloc == NULL) {
1419 free(format);
1420 free(formated);
1421 return _wcsdup(L"");
1422 }
1423 formated = formated_realloc;
1424 size_t r = wcsftime(formated, len, format, &tm_local);
1425 if (r != 0) {
1426 // �t�H�[�}�b�g������
1427 break;
1428 }
1429 len *= 2;
1430 }
1431 free(format);
1432
1433 // �z�X�g������
1434 wchar_t *host = ConvertLognameW(&cv, formated);
1435 free(formated);
1436
1437 // �t�@�C�������g�p���������������u��
1438 //wchar_t *replaced = replaceInvalidFileNameCharW(host, 0); // ����
1439 wchar_t *replaced = replaceInvalidFileNameCharW(host, L'_');
1440 free(host);
1441
1442 return replaced;
1443 }
1444
1445 /**
1446 * ���O�t�@�C����������
1447 * ���O�t�@�C�����p���C�����s��
1448 * - strftime() ���������t�W�J
1449 * - �������������O�t�@�C���t�H���_������
1450 * - �z�X�g��,�|�[�g�����W�J
1451 *
1452 * @param[in] log_filename �t�@�C����(����/��������������ok)
1453 * NULL�������f�t�H���g�t�@�C����������
1454 * strftime�`��ok
1455 * @return �t���p�X�t�@�C����
1456 * �s�v���������� free() ��������
1457 */
1458 wchar_t *FLogGetLogFilename(const wchar_t *log_filename)
1459 {
1460 wchar_t *dir;
1461 wchar_t *fname;
1462 if (log_filename == NULL) {
1463 dir = _wcsdup(ts.LogDefaultPathW);
1464 fname = _wcsdup(ts.LogDefaultNameW);
1465 } else if (!IsRelativePathW(log_filename)) {
1466 // �����p�X������������
1467 dir = ExtractDirNameW(log_filename);
1468 fname = ExtractFileNameW(log_filename);
1469 }
1470 else {
1471 dir = _wcsdup(ts.LogDefaultPathW);
1472 fname = _wcsdup(log_filename);
1473 }
1474
1475 wchar_t *formated = FLogGetLogFilenameBase(fname);
1476 free(fname);
1477
1478 // �A������
1479 wchar_t *logfull = NULL;
1480 awcscats(&logfull, dir, L"\\", formated, NULL);
1481 free(dir);
1482
1483 // ���K��
1484 wchar_t *normal;
1485 hGetFullPathNameW(logfull, &normal, NULL);
1486 free(logfull);
1487
1488 return normal;
1489 }
1490
1491 BOOL FLogIsPause()
1492 {
1493 if (LogVar == NULL) {
1494 return FALSE;
1495 }
1496 return LogVar->IsPause;
1497 }
1498
1499 void FLogWindow(int nCmdShow)
1500 {
1501 if (LogVar == NULL) {
1502 return;
1503 }
1504
1505 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1506 ShowWindow(HWndLog, nCmdShow);
1507 if (nCmdShow == SW_RESTORE) {
1508 // �g���X�^�C�� WS_EX_NOACTIVATE ��������������
1509 SetForegroundWindow(HWndLog);
1510 }
1511 }
1512
1513 void FLogShowDlg(void)
1514 {
1515 if (LogVar == NULL) {
1516 return;
1517 }
1518 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1519 ShowWindow(HWndLog, SW_SHOWNORMAL);
1520 SetForegroundWindow(HWndLog);
1521 }
1522
1523 /**
1524 * ���O��1byte��������
1525 * LogPut1() ������?
1526 */
1527 //void Log1Bin(PComVar cv, BYTE b)
1528 static void Log1Bin(BYTE b)
1529 {
1530 if (LogVar->IsPause || ProtoGetProtoFlag()) {
1531 return;
1532 }
1533 if (cv_BinSkip > 0) {
1534 cv_BinSkip--;
1535 return;
1536 }
1537 cv_BinBuf[cv_BinPtr] = b;
1538 cv_BinPtr++;
1539 if (cv_BinPtr>=InBuffSize) {
1540 cv_BinPtr = cv_BinPtr-InBuffSize;
1541 }
1542 if (cv_BCount>=InBuffSize) {
1543 cv_BCount = InBuffSize;
1544 cv_BStart = cv_BinPtr;
1545 }
1546 else {
1547 cv_BCount++;
1548 }
1549 }
1550
1551 static void LogBinSkip(int add)
1552 {
1553 if (cv_BinBuf != NULL) {
1554 cv_BinSkip += add;
1555 }
1556 }
1557
1558 /**
1559 * ���O�o�b�t�@���������������f�[�^���o�C�g��������
1560 */
1561 int FLogGetCount(void)
1562 {
1563 PFileVar fv = LogVar;
1564 if (fv == NULL) {
1565 return 0;
1566 }
1567 if (fv->FileLog) {
1568 return cv_LCount;
1569 }
1570 if (fv->BinLog) {
1571 return cv_BCount;
1572 }
1573 return 0;
1574 }
1575
1576 /**
1577 * ���O�o�b�t�@�������o�C�g��������
1578 */
1579 int FLogGetFreeCount(void)
1580 {
1581 PFileVar fv = LogVar;
1582 if (fv == NULL) {
1583 return 0;
1584 }
1585 if (fv->FileLog) {
1586 return InBuffSize - cv_LCount;
1587 }
1588 if (fv->BinLog) {
1589 return InBuffSize - cv_BCount;
1590 }
1591 return 0;
1592 }
1593
1594 /**
1595 * �o�b�t�@�������O���t�@�C������������
1596 */
1597 void FLogWriteFile(void)
1598 {
1599 PFileVar fv = LogVar;
1600 if (fv == NULL) {
1601 return;
1602 }
1603 if (cv_LogBuf!=NULL)
1604 {
1605 if (fv->FileLog) {
1606 LogToFile(fv);
1607 }
1608 }
1609
1610 if (cv_BinBuf!=NULL)
1611 {
1612 if (fv->BinLog) {
1613 LogToFile(fv);
1614 }
1615 }
1616 }
1617
1618 void FLogPutUTF32(unsigned int u32)
1619 {
1620 PFileVar fv = LogVar;
1621 BOOL log_available = (cv_LogBuf != 0);
1622
1623 if (!log_available) {
1624 // ���O�����o��������
1625 return;
1626 }
1627
1628 // �s����?(���s���o����������)
1629 if (ts.LogTimestamp && fv->eLineEnd) {
1630 // �^�C���X�^���v���o��
1631 fv->eLineEnd = Line_Other; /* clear endmark*/
1632 wchar_t* strtime = TimeStampStr(fv);
1633 FLogWriteStr(strtime);
1634 free(strtime);
1635 }
1636
1637 switch(fv->log_code) {
1638 case LOG_UTF8: {
1639 // UTF-8
1640 char u8_buf[4];
1641 size_t u8_len = UTF32ToUTF8(u32, u8_buf, _countof(u8_buf));
1642 for (size_t i = 0; i < u8_len; i++) {
1643 BYTE b = u8_buf[i];
1644 LogPut1(b);
1645 }
1646 break;
1647 }
1648 case LOG_UTF16LE:
1649 case LOG_UTF16BE: {
1650 // UTF-16
1651 wchar_t u16[2];
1652 size_t u16_len = UTF32ToUTF16(u32, u16, _countof(u16));
1653 for (size_t i = 0; i < u16_len; i++) {
1654 if (fv->log_code == LOG_UTF16LE) {
1655 // UTF-16LE
1656 LogPut1(u16[i] & 0xff);
1657 LogPut1((u16[i] >> 8) & 0xff);
1658 }
1659 else {
1660 // UTF-16BE
1661 LogPut1((u16[i] >> 8) & 0xff);
1662 LogPut1(u16[i] & 0xff);
1663 }
1664 }
1665 }
1666 }
1667
1668 if (u32 == 0x0a) {
1669 fv->eLineEnd = Line_LineHead; /* set endmark*/
1670 }
1671 }
1672
1673 static void FLogOutputBOM(PFileVar fv)
1674 {
1675 DWORD wrote;
1676
1677 switch(fv->log_code) {
1678 case 0: {
1679 // UTF-8
1680 const char *bom = "\xef\xbb\xbf";
1681 WriteFile(fv->FileHandle, bom, 3, &wrote, NULL);
1682 fv->ByteCount += 3;
1683 break;
1684 }
1685 case 1: {
1686 // UTF-16LE
1687 const char *bom = "\xff\xfe";
1688 WriteFile(fv->FileHandle, bom, 2, &wrote, NULL);
1689 fv->ByteCount += 2;
1690 break;
1691 }
1692 case 2: {
1693 // UTF-16BE
1694 const char *bom = "\xfe\xff";
1695 WriteFile(fv->FileHandle, bom, 2, &wrote, NULL);
1696 fv->ByteCount += 2;
1697 break;
1698 }
1699 default:
1700 break;
1701 }
1702 }
1703
1704 static void OutputStr(const wchar_t *str)
1705 {
1706 size_t len;
1707
1708 assert(str != NULL);
1709
1710 len = wcslen(str);
1711 while(*str != 0) {
1712 unsigned int u32;
1713 size_t u16_len = UTF16ToUTF32(str, len, &u32);
1714 switch (u16_len) {
1715 case 0:
1716 default:
1717 // ������������
1718 str++;
1719 len--;
1720 break;
1721 case 1:
1722 case 2: {
1723 FLogPutUTF32(u32);
1724 str += u16_len;
1725 len -= u16_len;
1726 break;
1727 }
1728 }
1729 }
1730 }

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