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 9134 - (show annotations) (download) (as text)
Fri Jan 22 15:00:52 2021 UTC (3 years, 2 months ago) by nmaya
File MIME type: text/x-c++src
File size: 38655 byte(s)
CreateFile の共有モードを修正

- GENERIC_WRITE で開くときに FILE_SHARE_WRITE で共有している箇所を 0 (共有しない)に修正
- Win16 API の _lcreat, _lopen はロックしないので、互換性のため FILE_SHARE_READ | FILE_SHARE_WRITE に変更
- TTL の filecreate, fileopen のファイルハンドルは filelock でロックするので、 FILE_SHARE_READ | FILE_SHARE_WRITE に変更

dwShareMode で指定する共有モードは、後続の"ファイルを開く"操作に対して「有効にする」アクセスを指定する
> https://docs.microsoft.com/en-us/windows/win32/api/fileapi/nf-fileapi-createfilea
> Enables subsequent open operations on a file or device to request xxx access.

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

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