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

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