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 9856 - (show annotations) (download) (as text)
Tue Apr 12 13:21:09 2022 UTC (23 months, 4 weeks ago) by zmatsuo
File MIME type: text/x-c++src
File size: 38513 byte(s)
ファイル選択時にデフォルトのログフォルダを開くようにした

- フィルタに txt, log を追加
- GetCommonDialogFilterWW() のプロトタイプが誤っていたので修正

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

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