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 8908 - (show annotations) (download) (as text)
Tue Aug 18 15:31:51 2020 UTC (3 years, 7 months ago) by zmatsuo
File MIME type: text/x-c++src
File size: 39623 byte(s)
現在バッファをすべてログへ書き出しを Unicode化した

- 別関数に分離, FLogOutputAllBuffer()
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 "ttftypes.h"
44 #include "ftdlg.h"
45 #include "ttwinman.h"
46 #include "commlib.h"
47 #include "ttcommon.h"
48 #include "ttlib.h"
49 #include "dlglib.h"
50 #include "vtterm.h"
51 #include "ftlib.h"
52 #include "buffer.h"
53 #include "helpid.h"
54 #include "layer_for_unicode.h"
55 #include "layer_for_unicode_crt.h"
56 #include "codeconv.h"
57 #include "asprintf.h"
58
59 #include "filesys_log_res.h"
60 #include "filesys_log.h"
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 wchar_t *FileName;
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 int log_code;
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
127 static BOOL OpenFTDlg_(PFileVar fv)
128 {
129 PFileTransDlg FTDlg = new CFileTransDlg();
130 if (FTDlg == NULL) {
131 return FALSE;
132 }
133
134 wchar_t *DlgCaption;
135 wchar_t uimsg[MAX_UIMSG];
136 #define TitLogW L"Log"
137 get_lang_msgW("FILEDLG_TRANS_TITLE_LOG", uimsg, _countof(uimsg), TitLogW, ts.UILanguageFile);
138 aswprintf(&DlgCaption, L"Tera Term: %s", uimsg);
139
140 CFileTransDlgInfo info;
141 info.UILanguageFile = ts.UILanguageFile;
142 info.OpId = OpLog;
143 info.DlgCaption = DlgCaption;
144 info.FileName = fv->FileName;
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 = (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 const char *UILanguageFile = pts->UILanguageFile;
493 SetWindowLongPtr(Dialog, DWLP_USER, (LONG_PTR)work);
494 ::DragAcceptFiles(Dialog, TRUE);
495
496 SetDlgTexts(Dialog, TextInfos, _countof(TextInfos), UILanguageFile);
497 SetI18nList("Tera Term", Dialog, IDC_TIMESTAMPTYPE, timestamp_list, _countof(timestamp_list),
498 UILanguageFile, 0);
499
500 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-8");
501 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16LE");
502 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16BE");
503 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
504
505 _SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, work->info->filename);
506 work->info->filename = NULL;
507
508 // Binary/Text �`�F�b�N�{�b�N�X
509 if (pts->LogBinary) {
510 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTBIN);
511 }
512 else {
513 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTTEXT);
514 }
515
516 // Plain Text �`�F�b�N�{�b�N�X
517 if (pts->LogBinary) {
518 // Binary�t���O���L�����������`�F�b�N��������
519 DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT);
520 }
521 else if (pts->LogTypePlainText) {
522 SetRB(Dialog, 1, IDC_PLAINTEXT, IDC_PLAINTEXT);
523 }
524
525 // Hide dialog�`�F�b�N�{�b�N�X (2008.1.30 maya)
526 if (pts->LogHideDialog) {
527 SetRB(Dialog, 1, IDC_HIDEDIALOG, IDC_HIDEDIALOG);
528 }
529
530 // Include screen buffer�`�F�b�N�{�b�N�X (2013.9.29 yutaka)
531 if (pts->LogAllBuffIncludedInFirst) {
532 SetRB(Dialog, 1, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST);
533 }
534
535 // timestamp�`�F�b�N�{�b�N�X (2006.7.23 maya)
536 if (pts->LogBinary) {
537 // Binary�t���O���L�����������`�F�b�N��������
538 DisableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP);
539 }
540 else if (pts->LogTimestamp) {
541 SetRB(Dialog, 1, IDC_TIMESTAMP, IDC_TIMESTAMP);
542 }
543
544 // timestamp ����
545 int tstype = pts->LogTimestampType == TIMESTAMP_LOCAL ? 0 :
546 pts->LogTimestampType == TIMESTAMP_UTC ? 1 :
547 pts->LogTimestampType == TIMESTAMP_ELAPSED_LOGSTART ? 2 :
548 pts->LogTimestampType == TIMESTAMP_ELAPSED_CONNECTED ? 3 : 0;
549 SendDlgItemMessage(Dialog, IDC_TIMESTAMPTYPE, CB_SETCURSEL, tstype, 0);
550 if (pts->LogBinary || !pts->LogTimestamp) {
551 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
552 }
553
554 CenterWindow(Dialog, GetParent(Dialog));
555
556 return TRUE;
557 }
558
559 case WM_COMMAND:
560 switch (LOWORD(wParam)) {
561 case IDOK: {
562 wchar_t filename[MAX_PATH];
563 _GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename, _countof(filename));
564 work->info->filename = _wcsdup(filename);
565 work->info->append = IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED;
566 work->info->bom = IsDlgButtonChecked(Dialog, IDC_BOM) == BST_CHECKED;
567 work->info->code = (int)SendDlgItemMessageA(Dialog, IDC_TEXTCODING_DROPDOWN, CB_GETCURSEL, 0, 0);
568 SetLogFlags(Dialog);
569 EndDialog(Dialog, IDOK);
570 break;
571 }
572 case IDCANCEL:
573 EndDialog(Dialog, IDCANCEL);
574 break;
575 case IDHELP:
576 OpenHelp(HH_HELP_CONTEXT, HlpFileLog, work->pts->UILanguageFile);
577 break;
578 case IDC_FOPT_FILENAME_BUTTON: {
579 /* save current dir */
580 const char *UILanguageFile = work->pts->UILanguageFile;
581 wchar_t curdir[MAXPATHLEN];
582 _GetCurrentDirectoryW(_countof(curdir), curdir);
583
584 wchar_t fname[MAX_PATH];
585 GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname, _countof(fname));
586
587 wchar_t FNFilter[128*3];
588 get_lang_msgW("FILEDLG_ALL_FILTER", FNFilter, sizeof(FNFilter), L"All(*.*)\\0*.*\\0\\0", UILanguageFile);
589
590 wchar_t caption[MAX_PATH];
591 wchar_t uimsg[MAX_UIMSG];
592 #define TitLogW L"Log"
593 get_lang_msgW("FILEDLG_TRANS_TITLE_LOG", uimsg, _countof(uimsg), TitLogW, UILanguageFile);
594 wcsncpy_s(caption, _countof(caption), L"Tera Term: ", _TRUNCATE);
595 wcsncat_s(caption, _countof(caption), uimsg, _TRUNCATE);
596
597 OPENFILENAMEW ofn = {};
598 ofn.lStructSize = get_OPENFILENAME_SIZEW();
599 //ofn.Flags = OFN_PATHMUSTEXIST | OFN_OVERWRITEPROMPT;
600 ofn.Flags |= OFN_EXPLORER | OFN_ENABLESIZING;
601 ofn.Flags |= OFN_SHOWHELP;
602 ofn.Flags |= OFN_NOCHANGEDIR; // ��������������������������������
603 ofn.hwndOwner = Dialog;
604 ofn.lpstrFilter = FNFilter;
605 ofn.nFilterIndex = 1;
606 ofn.lpstrFile = fname;
607 ofn.nMaxFile = _countof(fname);
608 ofn.lpstrTitle = caption;
609 BOOL Ok = GetSaveFileNameW(&ofn);
610 if (Ok) {
611 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname);
612 }
613
614 /* restore dir */
615 _SetCurrentDirectoryW(curdir);
616
617 break;
618 }
619 case IDC_FOPTBIN:
620 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), FALSE);
621 EnableWindow(GetDlgItem(Dialog, IDC_BOM), FALSE);
622 DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_TIMESTAMP);
623 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
624 break;
625 case IDC_FOPTTEXT:
626 ArrangeControls(Dialog, work);
627 EnableDlgItem(Dialog, IDC_PLAINTEXT, IDC_TIMESTAMP);
628 // FALLTHROUGH -- BinFlag �� off ������ Timestamp �������L��/��������������
629 case IDC_TIMESTAMP:
630 if (IsDlgButtonChecked(Dialog, IDC_TIMESTAMP) == BST_CHECKED) {
631 EnableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
632 }
633 else {
634 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
635 }
636 break;
637 case IDC_FOPT_FILENAME_EDIT:
638 if (HIWORD(wParam) == EN_CHANGE){
639 wchar_t filename[MAX_PATH];
640 GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename, _countof(filename));
641 CheckLogFile(Dialog, filename, work);
642 }
643 break;
644 case IDC_NEW_OVERWRITE:
645 if (IsDlgButtonChecked(Dialog, IDC_FOPTTEXT) == BST_CHECKED) {
646 EnableWindow(GetDlgItem(Dialog, IDC_BOM), TRUE);
647 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), TRUE);
648 CheckDlgButton(Dialog, IDC_BOM, BST_CHECKED);
649 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
650 }
651 break;
652 case IDC_APPEND:
653 ArrangeControls(Dialog, work);
654 break;
655 }
656 break;
657 case WM_DROPFILES: {
658 // �����h���b�v��������������1������������
659 HDROP hDrop = (HDROP)wParam;
660 const UINT len = _DragQueryFileW(hDrop, 0, NULL, 0);
661 if (len == 0) {
662 DragFinish(hDrop);
663 return TRUE;
664 }
665 wchar_t *filename = (wchar_t *)malloc(sizeof(wchar_t) * (len + 1));
666 _DragQueryFileW(hDrop, 0, filename, len + 1);
667 filename[len] = '\0';
668 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
669 _SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename);
670 SendDlgItemMessage(Dialog, IDC_FOPT_FILENAME_EDIT, EM_SETSEL, len, len);
671 free(filename);
672 DragFinish(hDrop);
673 return TRUE;
674 }
675 }
676 return FALSE;
677 }
678
679 static void OpenLogFile(PFileVar fv)
680 {
681 int dwShareMode = FILE_SHARE_READ;
682 if (!ts.LogLockExclusive) {
683 dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
684 }
685 LogVar->FileHandle = _CreateFileW(LogVar->FullName, GENERIC_WRITE, dwShareMode, NULL,
686 OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
687 }
688
689 static BOOL LogStart(const wchar_t *fname)
690 {
691 PFileVar fv = LogVar;
692
693 fv->FullName = _wcsdup(fname);
694 fv->FileName = NULL;
695 FixLogOption();
696
697 if (ts.LogBinary > 0)
698 {
699 BinLog = TRUE;
700 FileLog = FALSE;
701 if (! CreateBinBuf())
702 {
703 return FALSE;
704 }
705 }
706 else {
707 BinLog = FALSE;
708 FileLog = TRUE;
709 if (! CreateLogBuf())
710 {
711 return FALSE;
712 }
713 }
714 cv_LStart = cv_LogPtr;
715 cv_LCount = 0;
716
717 OpenLogFile(fv);
718 if (LogVar->FileHandle == INVALID_HANDLE_VALUE) {
719 return FALSE;
720 }
721
722 /* 2007.05.24 Gentaro */
723 fv->eLineEnd = Line_LineHead;
724 if (ts.Append > 0)
725 {
726 SetFilePointer(LogVar->FileHandle, 0, NULL, FILE_END);
727 /* 2007.05.24 Gentaro
728 If log file already exists,
729 a newline is inserted before the first timestamp.
730 */
731 fv->eLineEnd = Line_FileHead;
732 }
733
734 // Log rotate configuration
735 LogVar->RotateMode = ts.LogRotate;
736 LogVar->RotateSize = ts.LogRotateSize;
737 LogVar->RotateStep = ts.LogRotateStep;
738
739 // Log rotate���L���������A�����t�@�C���T�C�Y�����������B
740 // �������t�@�C�������������T�C�Y�����[�e�[�g�������������C���B
741 // (2016.4.9 yutaka)
742 if (LogVar->RotateMode != ROTATE_NONE) {
743 DWORD size = GetFileSize(LogVar->FileHandle, NULL);
744 if (size == -1) {
745 return FALSE;
746 }
747 LogVar->ByteCount = size;
748 }
749 else {
750 LogVar->ByteCount = 0;
751 }
752
753 if (! OpenFTDlg_(LogVar)) {
754 return FALSE;
755 }
756
757 LogVar->IsPause = FALSE;
758 LogVar->StartTime = GetTickCount();
759
760 if (ts.DeferredLogWriteMode) {
761 StartThread(LogVar);
762 }
763
764 if (FileLog) {
765 cv.Log1Byte = LogPut1;
766 }
767 if (BinLog) {
768 cv.Log1Bin = Log1Bin;
769 cv.LogBinSkip = LogBinSkip;
770 }
771
772 return TRUE;
773 }
774
775 /**
776 * �����o�b�t�@�������f�[�^�����������O�������o��
777 * (2013.9.29 yutaka)
778 *
779 * TODO
780 * 1�s������
781 */
782 void FLogOutputAllBuffer(void)
783 {
784 DWORD ofs, written_size;
785 int size;
786 wchar_t buf[512];
787 for (ofs = 0 ; ; ofs++ ) {
788 // 1�����s�����������B���������������A�G�X�P�[�v�V�[�P���X�������������B
789 size = BuffGetAnyLineDataW(ofs, buf, _countof(buf));
790 if (size == -1)
791 break;
792
793 OutputStr(buf);
794 OutputStr(L"\r\n");
795 LogToFile();
796 }
797 }
798
799 /**
800 * ���O��1byte��������
801 * �o�b�t�@��������������
802 * ���������������� LogToFile() ���s������
803 */
804 void LogPut1(BYTE b)
805 {
806 cv_LogBuf[cv_LogPtr] = b;
807 cv_LogPtr++;
808 if (cv_LogPtr>=InBuffSize)
809 cv_LogPtr = cv_LogPtr-InBuffSize;
810
811 if (FileLog)
812 {
813 if (cv_LCount>=InBuffSize)
814 {
815 cv_LCount = InBuffSize;
816 cv_LStart = cv_LogPtr;
817 }
818 else
819 cv_LCount++;
820 }
821 else
822 cv_LCount = 0;
823 }
824
825 static BOOL Get1(PCHAR Buf, int *Start, int *Count, PBYTE b)
826 {
827 if (*Count<=0) return FALSE;
828 *b = Buf[*Start];
829 (*Start)++;
830 if (*Start>=InBuffSize)
831 *Start = *Start-InBuffSize;
832 (*Count)--;
833 return TRUE;
834 }
835
836
837
838 static CRITICAL_SECTION g_filelog_lock; /* ���b�N�p���� */
839
840 void logfile_lock_initialize(void)
841 {
842 InitializeCriticalSection(&g_filelog_lock);
843 }
844
845 static inline void logfile_lock(void)
846 {
847 EnterCriticalSection(&g_filelog_lock);
848 }
849
850 static inline void logfile_unlock(void)
851 {
852 LeaveCriticalSection(&g_filelog_lock);
853 }
854
855 // ���O�����[�e�[�g�����B
856 // (2013.3.21 yutaka)
857 static void LogRotate(void)
858 {
859 int loopmax = 10000; // XXX
860 int i, k;
861
862 if (LogVar->RotateMode == ROTATE_NONE)
863 return;
864
865 if (LogVar->RotateMode == ROTATE_SIZE) {
866 if (LogVar->ByteCount <= LogVar->RotateSize)
867 return;
868 //OutputDebugPrintf("%s: mode %d size %ld\n", __FUNCTION__, LogVar->RotateMode, LogVar->ByteCount);
869 } else {
870 return;
871 }
872
873 logfile_lock();
874 // ���O�T�C�Y���������������B
875 LogVar->ByteCount = 0;
876
877 // �������������t�@�C�����N���[�Y�����A�������t�@�C�����I�[�v�������B
878 CloseFileSync(LogVar);
879
880 // �������[�e�[�V�������X�e�b�v�����w����������
881 if (LogVar->RotateStep > 0)
882 loopmax = LogVar->RotateStep;
883
884 for (i = 1 ; i <= loopmax ; i++) {
885 wchar_t *filename;
886 aswprintf(&filename, L"%s.%d", LogVar->FullName, i);
887 DWORD attr = _GetFileAttributesW(filename);
888 free(filename);
889 if (attr == INVALID_FILE_ATTRIBUTES)
890 break;
891 }
892 if (i > loopmax) {
893 // �������������������������A�������t�@�C�������p�������B
894 i = loopmax;
895 }
896
897 // ���t�@�C�������l�[���B
898 for (k = i-1 ; k >= 0 ; k--) {
899 wchar_t *oldfile;
900 if (k == 0)
901 oldfile = _wcsdup(LogVar->FullName);
902 else
903 aswprintf(&oldfile, L"%s.%d", LogVar->FullName, k);
904 wchar_t *newfile;
905 aswprintf(&newfile, L"%s.%d", LogVar->FullName, k+1);
906 _DeleteFileW(newfile);
907 if (_MoveFileW(oldfile, newfile) == 0) {
908 OutputDebugPrintf("%s: rename %d\n", __FUNCTION__, errno);
909 }
910 free(oldfile);
911 free(newfile);
912 }
913
914 // ���I�[�v��
915 OpenLogFile(LogVar);
916 if (ts.DeferredLogWriteMode) {
917 StartThread(LogVar);
918 }
919
920 logfile_unlock();
921 }
922
923 /**
924 * �o�b�t�@�������O���t�@�C������������
925 */
926 static void LogToFile(void)
927 {
928 PCHAR Buf;
929 int Start, Count;
930 BYTE b;
931 PCHAR WriteBuf;
932 DWORD WriteBufMax, WriteBufLen;
933 CHAR tmp[128];
934 DWORD wrote;
935 PFileVar fv = LogVar;
936
937 if (FileLog)
938 {
939 Buf = cv_LogBuf;
940 Start = cv_LStart;
941 Count = cv_LCount;
942 }
943 else if (BinLog)
944 {
945 Buf = cv_BinBuf;
946 Start = cv_BStart;
947 Count = cv_BCount;
948 }
949 else
950 return;
951
952 if (Buf==NULL) return;
953 if (Count==0) return;
954
955 // ���b�N������(2004.8.6 yutaka)
956 logfile_lock();
957
958 if (ts.DeferredLogWriteMode) {
959 WriteBufMax = 8192;
960 WriteBufLen = 0;
961 WriteBuf = (PCHAR)malloc(WriteBufMax);
962 while (Get1(Buf,&Start,&Count,&b)) {
963 if (!FLogIsPause() && (! cv.ProtoFlag))
964 {
965 tmp[0] = 0;
966 if ( ts.LogTimestamp && fv->eLineEnd ) {
967 char *strtime = NULL;
968
969 switch (ts.LogTimestampType) {
970 case TIMESTAMP_LOCAL:
971 strtime = mctimelocal(ts.LogTimestampFormat, FALSE);
972 break;
973 case TIMESTAMP_UTC:
974 strtime = mctimelocal(ts.LogTimestampFormat, TRUE);
975 break;
976 case TIMESTAMP_ELAPSED_LOGSTART:
977 strtime = strelapsed(LogVar->StartTime);
978 break;
979 case TIMESTAMP_ELAPSED_CONNECTED:
980 strtime = strelapsed(cv.ConnectedTime);
981 break;
982 }
983
984 /* 2007.05.24 Gentaro */
985 if(fv->eLineEnd == Line_FileHead ){
986 strncat_s(tmp, sizeof(tmp), "\r\n", _TRUNCATE);
987 }
988 strncat_s(tmp, sizeof(tmp), "[", _TRUNCATE);
989 strncat_s(tmp, sizeof(tmp), strtime, _TRUNCATE);
990 strncat_s(tmp, sizeof(tmp), "] ", _TRUNCATE);
991 }
992
993 /* 2007.05.24 Gentaro */
994 if( b == 0x0a ){
995 fv->eLineEnd = Line_LineHead; /* set endmark*/
996 }
997 else {
998 fv->eLineEnd = Line_Other; /* clear endmark*/
999 }
1000
1001 if (WriteBufLen >= (WriteBufMax*4/5)) {
1002 WriteBufMax *= 2;
1003 WriteBuf = (PCHAR)realloc(WriteBuf, WriteBufMax);
1004 }
1005 memcpy(&WriteBuf[WriteBufLen], tmp, strlen(tmp));
1006 WriteBufLen += strlen(tmp);
1007 WriteBuf[WriteBufLen++] = b;
1008
1009 (LogVar->ByteCount)++;
1010 }
1011 }
1012
1013 PostThreadMessage(LogVar->LogThreadId, WM_DPC_LOGTHREAD_SEND, (WPARAM)WriteBuf, WriteBufLen);
1014
1015 } else {
1016
1017 while (Get1(Buf,&Start,&Count,&b))
1018 {
1019 if (!FLogIsPause() && (! cv.ProtoFlag))
1020 {
1021 if ( ts.LogTimestamp && fv->eLineEnd ) {
1022 char *strtime = NULL;
1023
1024 switch (ts.LogTimestampType) {
1025 case TIMESTAMP_LOCAL:
1026 strtime = mctimelocal(ts.LogTimestampFormat, FALSE);
1027 break;
1028 case TIMESTAMP_UTC:
1029 strtime = mctimelocal(ts.LogTimestampFormat, TRUE);
1030 break;
1031 case TIMESTAMP_ELAPSED_LOGSTART:
1032 strtime = strelapsed(LogVar->StartTime);
1033 break;
1034 case TIMESTAMP_ELAPSED_CONNECTED:
1035 strtime = strelapsed(cv.ConnectedTime);
1036 break;
1037 }
1038 WriteFile(LogVar->FileHandle, "[", 1, &wrote, NULL);
1039 WriteFile(LogVar->FileHandle, strtime, strlen(strtime), &wrote, NULL);
1040 WriteFile(LogVar->FileHandle, "] ", 2, &wrote, NULL);
1041 }
1042
1043 /* 2007.05.24 Gentaro */
1044 if( b == 0x0a ){
1045 fv->eLineEnd = Line_LineHead; /* set endmark*/
1046 }
1047 else {
1048 fv->eLineEnd = Line_Other; /* clear endmark*/
1049 }
1050
1051 WriteFile(LogVar->FileHandle, (PCHAR)&b, 1, &wrote, NULL);
1052 (LogVar->ByteCount)++;
1053 }
1054 }
1055
1056 }
1057
1058 logfile_unlock();
1059
1060 if (FileLog)
1061 {
1062 cv_LStart = Start;
1063 cv_LCount = Count;
1064 }
1065 else {
1066 cv_BStart = Start;
1067 cv_BCount = Count;
1068 }
1069 if (FLogIsPause() || cv.ProtoFlag) return;
1070 LogVar->FLogDlg->RefreshNum(LogVar->StartTime, LogVar->FileSize, LogVar->ByteCount);
1071
1072
1073 // ���O�E���[�e�[�g
1074 LogRotate();
1075 }
1076
1077 static BOOL CreateLogBuf(void)
1078 {
1079 if (cv_LogBuf==NULL)
1080 {
1081 cv_LogBuf = (char *)malloc(InBuffSize);
1082 cv_LogPtr = 0;
1083 cv_LStart = 0;
1084 cv_LCount = 0;
1085 }
1086 return (cv_LogBuf!=NULL);
1087 }
1088
1089 static void FreeLogBuf(void)
1090 {
1091 free(cv_LogBuf);
1092 cv_LogBuf = NULL;
1093 cv_LogPtr = 0;
1094 cv_LStart = 0;
1095 cv_LCount = 0;
1096 }
1097
1098 static BOOL CreateBinBuf(void)
1099 {
1100 if (cv_BinBuf==NULL)
1101 {
1102 cv_BinBuf = (PCHAR)malloc(InBuffSize);
1103 cv_BinPtr = 0;
1104 cv_BStart = 0;
1105 cv_BCount = 0;
1106 }
1107 return (cv_BinBuf!=NULL);
1108 }
1109
1110 static void FreeBinBuf(void)
1111 {
1112 free(cv_BinBuf);
1113 cv_BinBuf = NULL;
1114 cv_BinPtr = 0;
1115 cv_BStart = 0;
1116 cv_BCount = 0;
1117 }
1118
1119 static void FileTransEnd_(void)
1120 {
1121 if (LogVar == NULL) {
1122 return;
1123 }
1124 FileLog = FALSE;
1125 BinLog = FALSE;
1126 cv.Log1Byte = NULL;
1127 cv.Log1Bin = NULL;
1128 cv.LogBinSkip = NULL;
1129 PFileTransDlg FLogDlg = LogVar->FLogDlg;
1130 if (FLogDlg != NULL) {
1131 FLogDlg->DestroyWindow();
1132 FLogDlg = NULL;
1133 }
1134 CloseFileSync(LogVar);
1135 FreeLogBuf();
1136 FreeBinBuf();
1137 free(LogVar);
1138 LogVar = NULL;
1139 }
1140
1141 /**
1142 * ���O���|�[�Y����
1143 */
1144 void FLogPause(BOOL Pause)
1145 {
1146 if (LogVar == NULL) {
1147 return;
1148 }
1149 LogVar->IsPause = Pause;
1150 LogVar->FLogDlg->ChangeButton(Pause);
1151 }
1152
1153 /**
1154 * ���O���[�e�[�g������
1155 * ���O���T�C�Y��<size>�o�C�g���������������A���[�e�[�V��������������������
1156 */
1157 void FLogRotateSize(size_t size)
1158 {
1159 if (LogVar == NULL) {
1160 return;
1161 }
1162 LogVar->RotateMode = ROTATE_SIZE;
1163 LogVar->RotateSize = size;
1164 }
1165
1166 /**
1167 * ���O���[�e�[�g������
1168 * ���O�t�@�C������������������
1169 */
1170 void FLogRotateRotate(int step)
1171 {
1172 if (LogVar == NULL) {
1173 return;
1174 }
1175 LogVar->RotateStep = step;
1176 }
1177
1178 /**
1179 * ���O���[�e�[�g������
1180 * ���[�e�[�V���������~
1181 */
1182 void FLogRotateHalt(void)
1183 {
1184 if (LogVar == NULL) {
1185 return;
1186 }
1187 LogVar->RotateMode = ROTATE_NONE;
1188 LogVar->RotateSize = 0;
1189 LogVar->RotateStep = 0;
1190 }
1191
1192 static INT_PTR CALLBACK OnCommentDlgProc(HWND hDlgWnd, UINT msg, WPARAM wp, LPARAM lp)
1193 {
1194 static const DlgTextInfo TextInfos[] = {
1195 { 0, "DLG_COMMENT_TITLE" },
1196 { IDOK, "BTN_OK" }
1197 };
1198
1199 switch (msg) {
1200 case WM_INITDIALOG:
1201 // �G�f�B�b�g�R���g���[�����t�H�[�J�X��������
1202 SetFocus(GetDlgItem(hDlgWnd, IDC_EDIT_COMMENT));
1203 SetDlgTexts(hDlgWnd, TextInfos, _countof(TextInfos), ts.UILanguageFile);
1204 return FALSE;
1205
1206 case WM_COMMAND:
1207 switch (LOWORD(wp)) {
1208 case IDOK: {
1209 size_t len = _SendDlgItemMessageW(hDlgWnd, IDC_EDIT_COMMENT, WM_GETTEXTLENGTH, 0, 0);
1210 len += 1;
1211 wchar_t *buf = (wchar_t *)malloc(len * sizeof(wchar_t));
1212 _GetDlgItemTextW(hDlgWnd, IDC_EDIT_COMMENT, buf, len);
1213 FLogWriteStr(buf);
1214 FLogWriteStr(L"\n"); // TODO ���s�R�[�h
1215 free(buf);
1216 TTEndDialog(hDlgWnd, IDOK);
1217 break;
1218 }
1219 default:
1220 return FALSE;
1221 }
1222 break;
1223
1224 case WM_CLOSE:
1225 TTEndDialog(hDlgWnd, 0);
1226 return TRUE;
1227
1228 default:
1229 return FALSE;
1230 }
1231 return TRUE;
1232 }
1233
1234 /**
1235 * ���O�t�@�C�����R�����g���������� (2004.8.6 yutaka)
1236 */
1237 void FLogAddCommentDlg(HINSTANCE hInst, HWND hWnd)
1238 {
1239 if (LogVar == NULL) {
1240 return;
1241 }
1242 TTDialogBox(hInst, MAKEINTRESOURCE(IDD_COMMENT_DIALOG),
1243 HVTWin, OnCommentDlgProc);
1244 }
1245
1246 void FLogClose(void)
1247 {
1248 if (LogVar == NULL) {
1249 return;
1250 }
1251
1252 FileTransEnd_();
1253 }
1254
1255 /**
1256 * ���O���I�[�v������
1257 * @param[in] fname ���O�t�@�C����, CreateFile()���n������
1258 *
1259 * ���O�t�@�C������strftime���W�J�������s���������B
1260 * FLogGetLogFilename() �� FLogOpenDialog() ��
1261 * �t�@�C�����������������B
1262 */
1263 BOOL FLogOpen(const wchar_t *fname)
1264 {
1265 BOOL ret;
1266
1267 if (LogVar != NULL) {
1268 return FALSE;
1269 }
1270 if ((FileLog) || (BinLog)) return FALSE;
1271
1272 //
1273 PFileVar fv = (PFileVar)malloc(sizeof(TFileVar));
1274 if (fv == NULL) {
1275 return FALSE;
1276 }
1277 LogVar = fv;
1278 memset(fv, 0, sizeof(TFileVar));
1279 fv->FileHandle = INVALID_HANDLE_VALUE;
1280 fv->LogThread = INVALID_HANDLE_VALUE;
1281 fv->eLineEnd = Line_LineHead;
1282 fv->log_code = 0; // UTF-8
1283
1284 ret = LogStart(fname);
1285 if (ret == FALSE) {
1286 FileTransEnd_();
1287 }
1288
1289 return ret;
1290 }
1291
1292 BOOL FLogIsOpend(void)
1293 {
1294 return LogVar != NULL;
1295 }
1296
1297 BOOL FLogIsOpendText(void)
1298 {
1299 return LogVar != NULL && FileLog;
1300 }
1301
1302 BOOL FLogIsOpendBin(void)
1303 {
1304 return LogVar != NULL && BinLog;
1305 }
1306
1307 void FLogWriteStr(const wchar_t *str)
1308 {
1309 if (LogVar != NULL) {
1310 #if 0
1311 DWORD wrote;
1312 size_t len = wcslen(str) * sizeof(wchar_t);
1313 logfile_lock();
1314 WriteFile(LogVar->FileHandle, str, len, &wrote, NULL);
1315 LogVar->eLineEnd = Line_LineHead;
1316 logfile_unlock();
1317 LogVar->ByteCount =
1318 LogVar->ByteCount + len;
1319 LogVar->FLogDlg->RefreshNum(LogVar->StartTime, LogVar->FileSize, LogVar->ByteCount);
1320 #endif
1321 OutputStr(str);
1322 }
1323 }
1324
1325 void FLogInfo(char *param_ptr, size_t param_len)
1326 {
1327 if (LogVar) {
1328 param_ptr[0] = '0'
1329 + (ts.LogBinary != 0)
1330 + ((ts.Append != 0) << 1)
1331 + ((ts.LogTypePlainText != 0) << 2)
1332 + ((ts.LogTimestamp != 0) << 3)
1333 + ((ts.LogHideDialog != 0) << 4);
1334 char *filenameU8 = ToU8W(LogVar->FullName);
1335 strncpy_s(param_ptr + 1, param_len - 1, filenameU8, _TRUNCATE);
1336 free(filenameU8);
1337 }
1338 else {
1339 param_ptr[0] = '0' - 1;
1340 param_ptr[1] = 0;
1341 }
1342 }
1343
1344 /**
1345 * ���������O�t�@�C����������
1346 */
1347 const wchar_t *FLogGetFilename(void)
1348 {
1349 if (LogVar == NULL) {
1350 return NULL;
1351 }
1352 return LogVar->FullName;
1353 }
1354
1355 /**
1356 * ���O�_�C�A���O���J��
1357 * @param[in,out] info.filename �t�@�C���������l
1358 * OK���A�t�@�C�����A�s�v����������free()��������
1359 * @retval TRUE [ok] ����������
1360 * @retval FALSE �L�����Z��������
1361 */
1362 BOOL FLogOpenDialog(HINSTANCE hInst, HWND hWnd, FLogDlgInfo_t *info)
1363 {
1364 LogDlgWork_t *work = (LogDlgWork_t *)calloc(sizeof(LogDlgWork_t), 1);
1365 wchar_t *srcfnameW = FLogGetLogFilename(info->filename);
1366 work->info = info;
1367 work->info->filename = srcfnameW;
1368 work->pts = &ts;
1369 INT_PTR ret = TTDialogBoxParam(
1370 hInst, MAKEINTRESOURCE(IDD_LOGDLG),
1371 hWnd, LogFnHook, (LPARAM)work);
1372 free(srcfnameW);
1373 free(work);
1374 return ret == IDOK ? TRUE : FALSE;
1375 }
1376
1377 /**
1378 * ���O�t�@�C����������
1379 * ���O�t�@�C�����p���C�����s��
1380 * - strftime() ���������t�W�J
1381 * - �������������O�t�@�C���t�H���_������
1382 * - �z�X�g��,�|�[�g�����W�J
1383 *
1384 * @param[in] log_filename �t�@�C����(����/��������������ok)
1385 * NULL�������f�t�H���g�t�@�C����������
1386 * strftime�`��ok
1387 * @return �t���p�X�t�@�C����
1388 * �s�v���������� free() ��������
1389 */
1390 wchar_t *FLogGetLogFilename(const wchar_t *log_filename)
1391 {
1392 // �t�H���_
1393 char FileDirExpanded[MAX_PATH];
1394 const char *logdir;
1395 if (strlen(ts.LogDefaultPath) > 0) {
1396 logdir = ts.LogDefaultPath;
1397 }
1398 else if (strlen(ts.FileDir) > 0) {
1399 ExpandEnvironmentStrings(ts.FileDir, FileDirExpanded, sizeof(FileDirExpanded));
1400 logdir = FileDirExpanded;
1401 }
1402 else {
1403 logdir = ts.HomeDir;
1404 }
1405
1406 // ���������t�@�C����
1407 char base_name[MAX_PATH];
1408 if (log_filename == NULL) {
1409 strncpy_s(base_name, _countof(base_name), ts.LogDefaultName, _TRUNCATE);
1410 }
1411 else {
1412 char *filenameA = ToCharW(log_filename);
1413 strncpy_s(base_name, _countof(base_name), filenameA, _TRUNCATE);
1414 free(filenameA);
1415 }
1416
1417 // �t���p�X��
1418 char full_path[MAX_PATH];
1419 ConvFName(logdir, base_name, sizeof(base_name), "", full_path, sizeof(full_path));
1420 ParseStrftimeFileName(full_path, sizeof(full_path));
1421 ConvertLogname(full_path, sizeof(full_path));
1422
1423 return ToWcharA(full_path);
1424 }
1425
1426 BOOL FLogIsPause()
1427 {
1428 if (LogVar == NULL) {
1429 return FALSE;
1430 }
1431 return LogVar->IsPause;
1432 }
1433
1434 void FLogWindow(int nCmdShow)
1435 {
1436 if (LogVar == NULL) {
1437 return;
1438 }
1439
1440 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1441 ShowWindow(HWndLog, nCmdShow);
1442 if (nCmdShow == SW_RESTORE) {
1443 // �g���X�^�C�� WS_EX_NOACTIVATE ��������������
1444 SetForegroundWindow(HWndLog);
1445 }
1446 }
1447
1448 void FLogShowDlg(void)
1449 {
1450 if (LogVar == NULL) {
1451 return;
1452 }
1453 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1454 ShowWindow(HWndLog, SW_SHOWNORMAL);
1455 SetForegroundWindow(HWndLog);
1456 }
1457
1458 /**
1459 * ���O��1byte��������
1460 * LogPut1() ������?
1461 */
1462 //void Log1Bin(PComVar cv, BYTE b)
1463 static void Log1Bin(BYTE b)
1464 {
1465 if (LogVar->IsPause || cv.ProtoFlag) {
1466 return;
1467 }
1468 if (cv_BinSkip > 0) {
1469 cv_BinSkip--;
1470 return;
1471 }
1472 cv_BinBuf[cv_BinPtr] = b;
1473 cv_BinPtr++;
1474 if (cv_BinPtr>=InBuffSize) {
1475 cv_BinPtr = cv_BinPtr-InBuffSize;
1476 }
1477 if (cv_BCount>=InBuffSize) {
1478 cv_BCount = InBuffSize;
1479 cv_BStart = cv_BinPtr;
1480 }
1481 else {
1482 cv_BCount++;
1483 }
1484 }
1485
1486 static void LogBinSkip(int add)
1487 {
1488 if (cv_BinBuf != NULL) {
1489 cv_BinSkip += add;
1490 }
1491 }
1492
1493 /**
1494 * ���O�o�b�t�@���������������f�[�^���o�C�g��������
1495 */
1496 int FLogGetCount(void)
1497 {
1498 if (FileLog) {
1499 return cv_LCount;
1500 }
1501 if (BinLog) {
1502 return cv_BCount;
1503 }
1504 return 0;
1505 }
1506
1507 /**
1508 * ���O�o�b�t�@�������o�C�g��������
1509 */
1510 int FLogGetFreeCount(void)
1511 {
1512 if (FileLog) {
1513 return InBuffSize - cv_LCount;
1514 }
1515 if (BinLog) {
1516 return InBuffSize - cv_BCount;
1517 }
1518 return 0;
1519 }
1520
1521 /**
1522 * �o�b�t�@�������O���t�@�C������������
1523 */
1524 void FLogWriteFile(void)
1525 {
1526 if (cv_LogBuf!=NULL)
1527 {
1528 if (FileLog) {
1529 LogToFile();
1530 }
1531 }
1532
1533 if (cv_BinBuf!=NULL)
1534 {
1535 if (BinLog) {
1536 LogToFile();
1537 }
1538 // GlobalUnlock(cv_HBinBuf);
1539 // cv_BinBuf = NULL;
1540 }
1541 }
1542
1543 void FLogPutUTF32(unsigned int u32)
1544 {
1545 PFileVar fv = LogVar;
1546 size_t i;
1547 BOOL log_available = (cv_LogBuf != 0);
1548
1549 if (!log_available) {
1550 // ���O�����o��������
1551 return;
1552 }
1553
1554 switch(fv->log_code) {
1555 case 0: {
1556 // UTF-8
1557 char u8_buf[4];
1558 size_t u8_len = UTF32ToUTF8(u32, u8_buf, _countof(u8_buf));
1559 for (i = 0; i < u8_len; i++) {
1560 BYTE b = u8_buf[i];
1561 LogPut1(b);
1562 }
1563 break;
1564 }
1565 case 1:
1566 case 2: {
1567 // UTF-16
1568 wchar_t u16[2];
1569 size_t u16_len = UTF32ToUTF16(u32, u16, _countof(u16));
1570 size_t i;
1571 for (i = 0; i < u16_len; i++) {
1572 if (fv->log_code == 1) {
1573 // UTF-16LE
1574 LogPut1(u16[i] & 0xff);
1575 LogPut1((u16[i] >> 8) & 0xff);
1576 }
1577 else {
1578 // UTF-16BE
1579 LogPut1((u16[i] >> 8) & 0xff);
1580 LogPut1(u16[i] & 0xff);
1581 }
1582 }
1583 }
1584 }
1585 }
1586
1587 void FLogOutputBOM(void)
1588 {
1589 PFileVar fv = LogVar;
1590
1591 switch(fv->log_code) {
1592 case 0:
1593 // UTF-8
1594 LogPut1(0xef);
1595 LogPut1(0xbb);
1596 LogPut1(0xbf);
1597 break;
1598 case 1:
1599 // UTF-16LE
1600 LogPut1(0xff);
1601 LogPut1(0xfe);
1602 break;
1603 case 2:
1604 // UTF-16BE
1605 LogPut1(0xfe);
1606 LogPut1(0xff);
1607 break;
1608 default:
1609 break;
1610 }
1611 }
1612
1613 void FLogSetCode(int code)
1614 {
1615 PFileVar fv = LogVar;
1616 fv->log_code = code;
1617 }
1618
1619 static void OutputStr(const wchar_t *str)
1620 {
1621 size_t len;
1622
1623 assert(str != NULL);
1624
1625 len = wcslen(str);
1626 while(*str != 0) {
1627 unsigned int u32;
1628 size_t u16_len = UTF16ToUTF32(str, len, &u32);
1629 switch (u16_len) {
1630 case 0:
1631 default:
1632 // ������������
1633 str++;
1634 len--;
1635 break;
1636 case 1:
1637 case 2: {
1638 FLogPutUTF32(u32);
1639 str += u16_len;
1640 len -= u16_len;
1641 break;
1642 }
1643 }
1644 }
1645 }

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