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