Develop and Download Open Source Software

Browse Subversion Repository

Annotation of /trunk/teraterm/teraterm/filesys_log.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 8912 - (hide annotations) (download) (as text)
Tue Aug 18 15:32:23 2020 UTC (3 years, 7 months ago) by zmatsuo
File MIME type: text/x-c++src
File size: 38472 byte(s)
ログローテート時、新しいファイルにBOMが出力されるよう修正
1 doda 6806 /*
2 zmatsuo 8894 * (C) 2020 TeraTerm Project
3 doda 6806 * All rights reserved.
4     *
5 doda 6841 * Redistribution and use in source and binary forms, with or without
6     * modification, are permitted provided that the following conditions
7     * are met:
8 doda 6806 *
9 doda 6841 * 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 doda 6806 *
17 doda 6841 * 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 doda 6806 */
28 maya 3227
29 zmatsuo 8897 /* TERATERM.EXE, log routines */
30 zmatsuo 7526 #include <stdio.h>
31 zmatsuo 8900 #if !defined(_CRTDBG_MAP_ALLOC)
32     #define _CRTDBG_MAP_ALLOC
33     #endif
34     #include <stdlib.h>
35     #include <crtdbg.h>
36 zmatsuo 7526 #include <process.h>
37 zmatsuo 8852 #include <windows.h>
38     #include <htmlhelp.h>
39     #include <assert.h>
40 zmatsuo 7526
41 maya 3227 #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 doda 3904 #include "vtterm.h"
51 maya 3227 #include "ftlib.h"
52 yutakapon 5392 #include "buffer.h"
53 zmatsuo 8852 #include "helpid.h"
54     #include "layer_for_unicode.h"
55 zmatsuo 8863 #include "layer_for_unicode_crt.h"
56     #include "codeconv.h"
57 zmatsuo 8899 #include "asprintf.h"
58 yutakapon 5392
59 zmatsuo 8852 #include "filesys_log_res.h"
60 zmatsuo 8900 #include "filesys_log.h"
61 zmatsuo 8866
62 zmatsuo 8901 /*
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 zmatsuo 8894 typedef struct {
73 zmatsuo 8901 wchar_t *FullName;
74 maya 3227
75 zmatsuo 8901 HANDLE FileHandle;
76     LONG FileSize, ByteCount;
77 maya 6071
78 zmatsuo 8901 DWORD StartTime;
79 maya 3227
80 zmatsuo 8901 enum enumLineEnd eLineEnd;
81 maya 7182
82 zmatsuo 8901 // log rotate
83     int RotateMode; // enum rotate_mode RotateMode;
84     LONG RotateSize;
85     int RotateStep;
86 zmatsuo 8900
87 zmatsuo 8901 HANDLE LogThread;
88     DWORD LogThreadId;
89     HANDLE LogThreadEvent;
90    
91     BOOL IsPause;
92    
93     PFileTransDlg FLogDlg;
94 zmatsuo 8904
95 zmatsuo 8912 LogCode_t log_code;
96     BOOL bom;
97 zmatsuo 8904
98 zmatsuo 8894 } TFileVar_;
99     typedef TFileVar_ *PFileVar_;
100    
101     #define PFileVar PFileVar_
102     #define TFileVar TFileVar_
103    
104     static PFileVar LogVar = NULL;
105    
106 zmatsuo 8900 static BOOL FileLog = FALSE;
107     static BOOL BinLog = FALSE;
108 zmatsuo 8894
109 zmatsuo 8906 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 yutakapon 5206 // �x�����������p�X���b�h�����b�Z�[�W
116     #define WM_DPC_LOGTHREAD_SEND (WM_APP + 1)
117    
118 zmatsuo 8897 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 zmatsuo 8905 void LogPut1(BYTE b);
124 zmatsuo 8907 static void OutputStr(const wchar_t *str);
125 zmatsuo 8908 static void LogToFile(void);
126 zmatsuo 8912 static void FLogOutputBOM(void);
127 yutakapon 5206
128 zmatsuo 8894 static BOOL OpenFTDlg_(PFileVar fv)
129 maya 3227 {
130 zmatsuo 8899 PFileTransDlg FTDlg = new CFileTransDlg();
131     if (FTDlg == NULL) {
132     return FALSE;
133     }
134 maya 3227
135 zmatsuo 8899 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 maya 3227
141 zmatsuo 8899 CFileTransDlgInfo info;
142     info.UILanguageFile = ts.UILanguageFile;
143     info.OpId = OpLog;
144     info.DlgCaption = DlgCaption;
145 zmatsuo 8911 info.FileName = NULL;
146 zmatsuo 8899 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 doda 4454
152 zmatsuo 8901 fv->FLogDlg = FTDlg;
153 doda 5383
154 zmatsuo 8899 free(DlgCaption);
155     return TRUE;
156 maya 3227 }
157    
158 zmatsuo 8852 /**
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 maya 3227 {
166     char buf[MAXPATHLEN], buf2[MAXPATHLEN], *p = c;
167 yutakapon 5139 char tmphost[1024];
168 maya 6590 char tmpuser[256+1];
169     DWORD len_user = sizeof(tmpuser);
170 maya 3227
171     memset(buf, 0, sizeof(buf));
172    
173     while(*p != '\0') {
174     if (*p == '&' && *(p+1) != '\0') {
175     switch (*(p+1)) {
176 maya 3473 case 'h':
177 maya 3227 if (cv.Open) {
178     if (cv.PortType == IdTCPIP) {
179 yutakapon 5139 // �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 maya 5141 replaceInvalidFileNameChar(tmphost, '_');
185 yutakapon 5139 strncat_s(buf,sizeof(buf), tmphost, _TRUNCATE);
186 maya 3227 }
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 maya 3473 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 maya 6590 case 'u':
203     if (GetUserName(tmpuser, &len_user) != 0) {
204     strncat_s(buf,sizeof(buf),tmpuser,_TRUNCATE);
205     }
206     break;
207 maya 3473 default:
208 maya 3227 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 zmatsuo 8858 static void FixLogOption(void)
223 maya 3227 {
224 doda 3887 if (ts.LogBinary) {
225 maya 3227 ts.LogTypePlainText = false;
226     ts.LogTimestamp = false;
227     }
228     }
229    
230 yutakapon 5206
231     // �X���b�h���I�����t�@�C�����N���[�Y
232     static void CloseFileSync(PFileVar ptr)
233     {
234 yutakapon 6489 BOOL ret;
235    
236 zmatsuo 8901 if (ptr->FileHandle == INVALID_HANDLE_VALUE) {
237 yutakapon 5206 return;
238 zmatsuo 8901 }
239 yutakapon 5206
240 zmatsuo 7930 if (ptr->LogThread != INVALID_HANDLE_VALUE) {
241 yutakapon 5206 // �X���b�h���I������
242 yutakapon 6489 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 zmatsuo 8852 //DWORD code = GetLastError();
249 yutakapon 6489 }
250 yutakapon 5206 CloseHandle(ptr->LogThread);
251 zmatsuo 7930 ptr->LogThread = INVALID_HANDLE_VALUE;
252 yutakapon 5206 }
253 zmatsuo 7930 CloseHandle(ptr->FileHandle);
254 zmatsuo 8901 ptr->FileHandle = INVALID_HANDLE_VALUE;
255 yutakapon 5206 }
256    
257     // �x�����������p�X���b�h
258 doda 6435 static unsigned _stdcall DeferredLogWriteThread(void *arg)
259 yutakapon 5206 {
260     MSG msg;
261     PFileVar fv = (PFileVar)arg;
262     PCHAR buf;
263     DWORD buflen;
264 maya 5273 DWORD wrote;
265 yutakapon 5206
266     PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE);
267    
268 yutakapon 6489 // �X���b�h�L���[���������I�������������X���b�h�����������m�����B
269     if (fv->LogThreadEvent != NULL) {
270     SetEvent(fv->LogThreadEvent);
271     }
272    
273 yutakapon 5206 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 zmatsuo 7930 WriteFile(LogVar->FileHandle, buf, buflen, &wrote, NULL);
279 yutakapon 5206 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 zmatsuo 8901 // �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 zmatsuo 8852 /**
315     * �_�C�A���O�����e�� ts ����������
316     *
317     * TODO
318     * �_�C�A���O�����������l�������I��������
319     * ����������������������������������������������?
320     */
321     static void SetLogFlags(HWND Dialog)
322 maya 3227 {
323 zmatsuo 8852 WORD BinFlag, val;
324 maya 3227
325 zmatsuo 8852 GetRB(Dialog, &BinFlag, IDC_FOPTBIN, IDC_FOPTBIN);
326     ts.LogBinary = BinFlag;
327 maya 3227
328 zmatsuo 8866 GetRB(Dialog, &val, IDC_APPEND, IDC_APPEND);
329 zmatsuo 8852 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 maya 3227 }
338    
339 zmatsuo 8852 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 zmatsuo 8912 ts.LogTimestampType = (WORD)(GetCurSel(Dialog, IDC_TIMESTAMPTYPE) - 1);
346 zmatsuo 8852 }
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 zmatsuo 8863 static void CheckLogFile(const wchar_t *filename, BOOL *exist, int *bom)
359 zmatsuo 8852 {
360 zmatsuo 8863 *exist = FALSE;
361     *bom = 0;
362    
363 zmatsuo 8852 // �t�@�C������������?
364 zmatsuo 8863 DWORD logdir = _GetFileAttributesW(filename);
365 zmatsuo 8852 if ((logdir != INVALID_FILE_ATTRIBUTES) && ((logdir & FILE_ATTRIBUTE_DIRECTORY) == 0)) {
366 zmatsuo 8863 // �t�@�C����������
367 zmatsuo 8852 *exist = TRUE;
368    
369     // BOM�L��/�����`�F�b�N
370 zmatsuo 8863 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 zmatsuo 8852 }
390 maya 3227 }
391 zmatsuo 8863 }
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 zmatsuo 8866 if (work->pts->Append > 0) {
406     CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
407     }
408 zmatsuo 8863 }
409 zmatsuo 8852 else {
410 zmatsuo 8863 // �t�@�C�������� -> �V�K
411     EnableWindow(GetDlgItem(Dialog, IDC_APPEND), FALSE);
412     CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
413 maya 3227 }
414 zmatsuo 8852
415 zmatsuo 8863 if (work->file_exist && IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED) {
416     // �t�@�C������������ && append
417     int bom = work->current_bom;
418 zmatsuo 8852 if (bom != 0) {
419     // BOM�L��
420     CheckDlgButton(Dialog, IDC_BOM, BST_CHECKED);
421 zmatsuo 8863 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 zmatsuo 8852 }
427     else {
428     // BOM����
429     CheckDlgButton(Dialog, IDC_BOM, BST_UNCHECKED);
430 zmatsuo 8863 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
431 zmatsuo 8852 }
432 zmatsuo 8863 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 zmatsuo 8852 }
444 maya 3227 else {
445 zmatsuo 8866 // �t�@�C�������� ���� append��������(������)
446 zmatsuo 8863 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
447 zmatsuo 8852 CheckDlgButton(Dialog, IDC_BOM, BST_CHECKED);
448 zmatsuo 8863 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 maya 3227 }
453 zmatsuo 8852 }
454 maya 3227
455 zmatsuo 8863 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 zmatsuo 8852
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 zmatsuo 8866 { IDC_APPEND, "DLG_FOPT_APPEND" },
471 zmatsuo 8852 { 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 zmatsuo 8863 LogDlgWork_t *work = (LogDlgWork_t *)GetWindowLongPtr(Dialog, DWLP_USER);
483 zmatsuo 8852
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 zmatsuo 8863 work = (LogDlgWork_t *)lParam;
492     TTTSet *pts = work->pts;
493     const char *UILanguageFile = pts->UILanguageFile;
494     SetWindowLongPtr(Dialog, DWLP_USER, (LONG_PTR)work);
495 zmatsuo 8852 ::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 zmatsuo 8863 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16LE");
503     SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16BE");
504 zmatsuo 8852 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
505    
506 zmatsuo 8863 _SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, work->info->filename);
507     work->info->filename = NULL;
508 zmatsuo 8852
509     // Binary/Text �`�F�b�N�{�b�N�X
510 zmatsuo 8863 if (pts->LogBinary) {
511     CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTBIN);
512 doda 6946 }
513 zmatsuo 8852 else {
514 zmatsuo 8863 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTTEXT);
515 zmatsuo 8852 }
516    
517     // Plain Text �`�F�b�N�{�b�N�X
518 zmatsuo 8863 if (pts->LogBinary) {
519 zmatsuo 8852 // Binary�t���O���L�����������`�F�b�N��������
520     DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT);
521 doda 6946 }
522 zmatsuo 8863 else if (pts->LogTypePlainText) {
523 zmatsuo 8852 SetRB(Dialog, 1, IDC_PLAINTEXT, IDC_PLAINTEXT);
524 doda 6946 }
525 zmatsuo 8852
526     // Hide dialog�`�F�b�N�{�b�N�X (2008.1.30 maya)
527 zmatsuo 8863 if (pts->LogHideDialog) {
528 zmatsuo 8852 SetRB(Dialog, 1, IDC_HIDEDIALOG, IDC_HIDEDIALOG);
529 doda 6946 }
530 zmatsuo 8852
531     // Include screen buffer�`�F�b�N�{�b�N�X (2013.9.29 yutaka)
532 zmatsuo 8863 if (pts->LogAllBuffIncludedInFirst) {
533 zmatsuo 8852 SetRB(Dialog, 1, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST);
534 doda 6946 }
535 maya 3227
536 zmatsuo 8852 // timestamp�`�F�b�N�{�b�N�X (2006.7.23 maya)
537 zmatsuo 8863 if (pts->LogBinary) {
538 zmatsuo 8852 // Binary�t���O���L�����������`�F�b�N��������
539     DisableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP);
540     }
541 zmatsuo 8863 else if (pts->LogTimestamp) {
542 zmatsuo 8852 SetRB(Dialog, 1, IDC_TIMESTAMP, IDC_TIMESTAMP);
543     }
544    
545     // timestamp ����
546 zmatsuo 8863 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 zmatsuo 8852 SendDlgItemMessage(Dialog, IDC_TIMESTAMPTYPE, CB_SETCURSEL, tstype, 0);
551 zmatsuo 8863 if (pts->LogBinary || !pts->LogTimestamp) {
552 zmatsuo 8852 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 zmatsuo 8863 wchar_t filename[MAX_PATH];
564     _GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename, _countof(filename));
565     work->info->filename = _wcsdup(filename);
566 zmatsuo 8866 work->info->append = IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED;
567 zmatsuo 8863 work->info->bom = IsDlgButtonChecked(Dialog, IDC_BOM) == BST_CHECKED;
568 zmatsuo 8912 work->info->code = (LogCode_t)SendDlgItemMessageA(Dialog, IDC_TEXTCODING_DROPDOWN, CB_GETCURSEL, 0, 0);
569 zmatsuo 8852 SetLogFlags(Dialog);
570     EndDialog(Dialog, IDOK);
571 doda 6947 break;
572 zmatsuo 8852 }
573     case IDCANCEL:
574     EndDialog(Dialog, IDCANCEL);
575 doda 6947 break;
576 zmatsuo 8852 case IDHELP:
577 zmatsuo 8863 OpenHelp(HH_HELP_CONTEXT, HlpFileLog, work->pts->UILanguageFile);
578 doda 6947 break;
579 zmatsuo 8852 case IDC_FOPT_FILENAME_BUTTON: {
580     /* save current dir */
581 zmatsuo 8863 const char *UILanguageFile = work->pts->UILanguageFile;
582 zmatsuo 8852 wchar_t curdir[MAXPATHLEN];
583     _GetCurrentDirectoryW(_countof(curdir), curdir);
584 doda 6947
585 zmatsuo 8863 wchar_t fname[MAX_PATH];
586     GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname, _countof(fname));
587 maya 3227
588 zmatsuo 8863 wchar_t FNFilter[128*3];
589     get_lang_msgW("FILEDLG_ALL_FILTER", FNFilter, sizeof(FNFilter), L"All(*.*)\\0*.*\\0\\0", UILanguageFile);
590 maya 3227
591 zmatsuo 8863 wchar_t caption[MAX_PATH];
592     wchar_t uimsg[MAX_UIMSG];
593     #define TitLogW L"Log"
594     get_lang_msgW("FILEDLG_TRANS_TITLE_LOG", uimsg, _countof(uimsg), TitLogW, UILanguageFile);
595     wcsncpy_s(caption, _countof(caption), L"Tera Term: ", _TRUNCATE);
596     wcsncat_s(caption, _countof(caption), uimsg, _TRUNCATE);
597 doda 6947
598 zmatsuo 8863 OPENFILENAMEW ofn = {};
599     ofn.lStructSize = get_OPENFILENAME_SIZEW();
600 zmatsuo 8852 //ofn.Flags = OFN_PATHMUSTEXIST | OFN_OVERWRITEPROMPT;
601     ofn.Flags |= OFN_EXPLORER | OFN_ENABLESIZING;
602     ofn.Flags |= OFN_SHOWHELP;
603     ofn.Flags |= OFN_NOCHANGEDIR; // ��������������������������������
604     ofn.hwndOwner = Dialog;
605     ofn.lpstrFilter = FNFilter;
606     ofn.nFilterIndex = 1;
607     ofn.lpstrFile = fname;
608 zmatsuo 8863 ofn.nMaxFile = _countof(fname);
609 zmatsuo 8852 ofn.lpstrTitle = caption;
610 zmatsuo 8863 BOOL Ok = GetSaveFileNameW(&ofn);
611 zmatsuo 8852 if (Ok) {
612 zmatsuo 8863 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname);
613 doda 6947 }
614 zmatsuo 8852
615     /* restore dir */
616     _SetCurrentDirectoryW(curdir);
617    
618     break;
619 doda 6947 }
620 zmatsuo 8852 case IDC_FOPTBIN:
621 zmatsuo 8863 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), FALSE);
622     EnableWindow(GetDlgItem(Dialog, IDC_BOM), FALSE);
623 zmatsuo 8852 DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_TIMESTAMP);
624     DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
625     break;
626     case IDC_FOPTTEXT:
627 zmatsuo 8863 ArrangeControls(Dialog, work);
628 zmatsuo 8852 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 doda 6947 }
634     else {
635 zmatsuo 8852 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
636 doda 6947 }
637 zmatsuo 8852 break;
638     case IDC_FOPT_FILENAME_EDIT:
639     if (HIWORD(wParam) == EN_CHANGE){
640 zmatsuo 8863 wchar_t filename[MAX_PATH];
641     GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename, _countof(filename));
642     CheckLogFile(Dialog, filename, work);
643 zmatsuo 8852 }
644     break;
645 zmatsuo 8863 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 doda 6947 }
657 zmatsuo 8852 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 zmatsuo 8863 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
670 zmatsuo 8852 _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 maya 3227 }
676 zmatsuo 8852 }
677     return FALSE;
678     }
679 maya 3227
680 zmatsuo 8901 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 zmatsuo 8899 static BOOL LogStart(const wchar_t *fname)
691 zmatsuo 8852 {
692 zmatsuo 8901 PFileVar fv = LogVar;
693 maya 3227
694 zmatsuo 8901 fv->FullName = _wcsdup(fname);
695 zmatsuo 8852 FixLogOption();
696    
697 doda 3887 if (ts.LogBinary > 0)
698 maya 3227 {
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 zmatsuo 8906 cv_LStart = cv_LogPtr;
715     cv_LCount = 0;
716 maya 3227
717 zmatsuo 8901 OpenLogFile(fv);
718     if (LogVar->FileHandle == INVALID_HANDLE_VALUE) {
719     return FALSE;
720     }
721    
722 maya 3227 /* 2007.05.24 Gentaro */
723 zmatsuo 8901 fv->eLineEnd = Line_LineHead;
724 maya 3227 if (ts.Append > 0)
725     {
726 zmatsuo 8901 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 maya 3227 }
733 zmatsuo 8912 else {
734     // ���L��������(�V�K)������ BOM ���o������
735     if (fv->bom) {
736     FLogOutputBOM();
737     }
738     }
739 maya 3227
740 yutakapon 5171 // Log rotate configuration
741     LogVar->RotateMode = ts.LogRotate;
742     LogVar->RotateSize = ts.LogRotateSize;
743     LogVar->RotateStep = ts.LogRotateStep;
744 yutakapon 5162
745 yutakapon 6365 // 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 zmatsuo 8852 DWORD size = GetFileSize(LogVar->FileHandle, NULL);
750 zmatsuo 8901 if (size == -1) {
751     return FALSE;
752     }
753     LogVar->ByteCount = size;
754 yutakapon 6365 }
755 zmatsuo 8901 else {
756     LogVar->ByteCount = 0;
757     }
758 yutakapon 6365
759 zmatsuo 8894 if (! OpenFTDlg_(LogVar)) {
760 maya 3227 return FALSE;
761     }
762    
763 zmatsuo 8900 LogVar->IsPause = FALSE;
764 zmatsuo 8899 LogVar->StartTime = GetTickCount();
765    
766 zmatsuo 8901 if (ts.DeferredLogWriteMode) {
767     StartThread(LogVar);
768 yutakapon 6489 }
769 yutakapon 5206
770 zmatsuo 8897 if (FileLog) {
771     cv.Log1Byte = LogPut1;
772     }
773     if (BinLog) {
774     cv.Log1Bin = Log1Bin;
775     cv.LogBinSkip = LogBinSkip;
776     }
777    
778 maya 3227 return TRUE;
779     }
780    
781 zmatsuo 8897 /**
782 zmatsuo 8908 * �����o�b�t�@�������f�[�^�����������O�������o��
783     * (2013.9.29 yutaka)
784     *
785     * TODO
786     * 1�s������
787     */
788     void FLogOutputAllBuffer(void)
789     {
790 zmatsuo 8909 DWORD ofs;
791 zmatsuo 8908 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 zmatsuo 8897 * ���O��1byte��������
807     * �o�b�t�@��������������
808     * ���������������� LogToFile() ���s������
809     */
810 maya 3227 void LogPut1(BYTE b)
811     {
812 zmatsuo 8906 cv_LogBuf[cv_LogPtr] = b;
813     cv_LogPtr++;
814     if (cv_LogPtr>=InBuffSize)
815     cv_LogPtr = cv_LogPtr-InBuffSize;
816 maya 3227
817     if (FileLog)
818     {
819 zmatsuo 8906 if (cv_LCount>=InBuffSize)
820 maya 3227 {
821 zmatsuo 8906 cv_LCount = InBuffSize;
822     cv_LStart = cv_LogPtr;
823 maya 3227 }
824     else
825 zmatsuo 8906 cv_LCount++;
826 maya 3227 }
827     else
828 zmatsuo 8906 cv_LCount = 0;
829 maya 3227 }
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 yutakapon 5162 // ���O�����[�e�[�g�����B
862     // (2013.3.21 yutaka)
863     static void LogRotate(void)
864     {
865 yutakapon 5165 int loopmax = 10000; // XXX
866     int i, k;
867 yutakapon 5162
868     if (LogVar->RotateMode == ROTATE_NONE)
869     return;
870    
871     if (LogVar->RotateMode == ROTATE_SIZE) {
872 doda 6435 if (LogVar->ByteCount <= LogVar->RotateSize)
873 yutakapon 5162 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 yutakapon 5206 CloseFileSync(LogVar);
885 yutakapon 5162
886 yutakapon 5165 // �������[�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 zmatsuo 8899 wchar_t *filename;
892     aswprintf(&filename, L"%s.%d", LogVar->FullName, i);
893     DWORD attr = _GetFileAttributesW(filename);
894     free(filename);
895 zmatsuo 8901 if (attr == INVALID_FILE_ATTRIBUTES)
896 yutakapon 5162 break;
897     }
898 yutakapon 5165 if (i > loopmax) {
899     // �������������������������A�������t�@�C�������p�������B
900     i = loopmax;
901 yutakapon 5162 }
902    
903     // ���t�@�C�������l�[���B
904 yutakapon 5165 for (k = i-1 ; k >= 0 ; k--) {
905 zmatsuo 8899 wchar_t *oldfile;
906 yutakapon 5165 if (k == 0)
907 zmatsuo 8899 oldfile = _wcsdup(LogVar->FullName);
908 yutakapon 5165 else
909 zmatsuo 8899 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 yutakapon 5165 OutputDebugPrintf("%s: rename %d\n", __FUNCTION__, errno);
915     }
916 zmatsuo 8899 free(oldfile);
917     free(newfile);
918 yutakapon 5165 }
919 yutakapon 5162
920     // ���I�[�v��
921 zmatsuo 8901 OpenLogFile(LogVar);
922 zmatsuo 8912 if (LogVar->bom) {
923     FLogOutputBOM();
924     }
925 zmatsuo 8901 if (ts.DeferredLogWriteMode) {
926     StartThread(LogVar);
927 yutakapon 5162 }
928    
929     logfile_unlock();
930     }
931    
932 zmatsuo 8910 static wchar_t *TimeStampStr()
933 zmatsuo 8909 {
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 zmatsuo 8910 return ToWcharA(tmp);
958 zmatsuo 8909 }
959    
960 zmatsuo 8897 /**
961     * �o�b�t�@�������O���t�@�C������������
962     */
963     static void LogToFile(void)
964 maya 3227 {
965     PCHAR Buf;
966     int Start, Count;
967     BYTE b;
968    
969     if (FileLog)
970     {
971 zmatsuo 8906 Buf = cv_LogBuf;
972     Start = cv_LStart;
973     Count = cv_LCount;
974 maya 3227 }
975     else if (BinLog)
976     {
977 zmatsuo 8906 Buf = cv_BinBuf;
978     Start = cv_BStart;
979     Count = cv_BCount;
980 maya 3227 }
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 zmatsuo 8909 // ���������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 doda 6947
999 zmatsuo 8909 if (WriteBufLen >= (WriteBufMax*4/5)) {
1000     WriteBufMax *= 2;
1001     WriteBuf = (PCHAR)realloc(WriteBuf, WriteBufMax);
1002     }
1003 doda 6947
1004 zmatsuo 8909 WriteBuf[WriteBufLen++] = b;
1005 yutakapon 5206
1006 zmatsuo 8909 (LogVar->ByteCount)++;
1007     }
1008 yutakapon 5206
1009 zmatsuo 8909 // ��������
1010     if (WriteBufLen > 0) {
1011     if (ts.DeferredLogWriteMode) {
1012     PostThreadMessage(LogVar->LogThreadId, WM_DPC_LOGTHREAD_SEND, (WPARAM)WriteBuf, WriteBufLen);
1013 yutakapon 5206 }
1014 zmatsuo 8909 else {
1015     DWORD wrote;
1016     WriteFile(LogVar->FileHandle, WriteBuf, WriteBufLen, &wrote, NULL);
1017     free(WriteBuf);
1018     }
1019 maya 3227 }
1020    
1021     logfile_unlock();
1022    
1023     if (FileLog)
1024     {
1025 zmatsuo 8906 cv_LStart = Start;
1026     cv_LCount = Count;
1027 maya 3227 }
1028     else {
1029 zmatsuo 8906 cv_BStart = Start;
1030     cv_BCount = Count;
1031 maya 3227 }
1032 zmatsuo 8857 if (FLogIsPause() || cv.ProtoFlag) return;
1033 zmatsuo 8901 LogVar->FLogDlg->RefreshNum(LogVar->StartTime, LogVar->FileSize, LogVar->ByteCount);
1034 yutakapon 5162
1035 zmatsuo 8896
1036 yutakapon 5162 // ���O�E���[�e�[�g
1037     LogRotate();
1038 maya 3227 }
1039    
1040 zmatsuo 8897 static BOOL CreateLogBuf(void)
1041 maya 3227 {
1042 zmatsuo 8906 if (cv_LogBuf==NULL)
1043 maya 3227 {
1044 zmatsuo 8906 cv_LogBuf = (char *)malloc(InBuffSize);
1045     cv_LogPtr = 0;
1046     cv_LStart = 0;
1047     cv_LCount = 0;
1048 maya 3227 }
1049 zmatsuo 8906 return (cv_LogBuf!=NULL);
1050 maya 3227 }
1051    
1052 zmatsuo 8897 static void FreeLogBuf(void)
1053 maya 3227 {
1054 zmatsuo 8906 free(cv_LogBuf);
1055     cv_LogBuf = NULL;
1056     cv_LogPtr = 0;
1057     cv_LStart = 0;
1058     cv_LCount = 0;
1059 maya 3227 }
1060    
1061 zmatsuo 8897 static BOOL CreateBinBuf(void)
1062 maya 3227 {
1063 zmatsuo 8906 if (cv_BinBuf==NULL)
1064 maya 3227 {
1065 zmatsuo 8906 cv_BinBuf = (PCHAR)malloc(InBuffSize);
1066     cv_BinPtr = 0;
1067     cv_BStart = 0;
1068     cv_BCount = 0;
1069 maya 3227 }
1070 zmatsuo 8906 return (cv_BinBuf!=NULL);
1071 maya 3227 }
1072    
1073 zmatsuo 8897 static void FreeBinBuf(void)
1074 maya 3227 {
1075 zmatsuo 8906 free(cv_BinBuf);
1076     cv_BinBuf = NULL;
1077     cv_BinPtr = 0;
1078     cv_BStart = 0;
1079     cv_BCount = 0;
1080 maya 3227 }
1081    
1082 zmatsuo 8897 static void FileTransEnd_(void)
1083 maya 3227 {
1084 zmatsuo 8900 if (LogVar == NULL) {
1085     return;
1086     }
1087 zmatsuo 8897 FileLog = FALSE;
1088     BinLog = FALSE;
1089     cv.Log1Byte = NULL;
1090     cv.Log1Bin = NULL;
1091     cv.LogBinSkip = NULL;
1092 zmatsuo 8901 PFileTransDlg FLogDlg = LogVar->FLogDlg;
1093     if (FLogDlg != NULL) {
1094 zmatsuo 8897 FLogDlg->DestroyWindow();
1095     FLogDlg = NULL;
1096 maya 3227 }
1097 zmatsuo 8900 CloseFileSync(LogVar);
1098 zmatsuo 8897 FreeLogBuf();
1099     FreeBinBuf();
1100 zmatsuo 8900 free(LogVar);
1101     LogVar = NULL;
1102 maya 3227 }
1103    
1104 zmatsuo 8857 /**
1105     * ���O���|�[�Y����
1106     */
1107     void FLogPause(BOOL Pause)
1108 maya 3227 {
1109 zmatsuo 8899 if (LogVar == NULL) {
1110     return;
1111     }
1112 zmatsuo 8900 LogVar->IsPause = Pause;
1113 zmatsuo 8901 LogVar->FLogDlg->ChangeButton(Pause);
1114 maya 3227 }
1115    
1116 zmatsuo 8857 /**
1117 zmatsuo 8852 * ���O���[�e�[�g������
1118     * ���O���T�C�Y��<size>�o�C�g���������������A���[�e�[�V��������������������
1119     */
1120 zmatsuo 8858 void FLogRotateSize(size_t size)
1121 zmatsuo 8852 {
1122     if (LogVar == NULL) {
1123     return;
1124     }
1125     LogVar->RotateMode = ROTATE_SIZE;
1126 zmatsuo 8912 LogVar->RotateSize = (LONG)size;
1127 zmatsuo 8852 }
1128    
1129     /**
1130     * ���O���[�e�[�g������
1131     * ���O�t�@�C������������������
1132     */
1133 zmatsuo 8858 void FLogRotateRotate(int step)
1134 zmatsuo 8852 {
1135     if (LogVar == NULL) {
1136     return;
1137     }
1138     LogVar->RotateStep = step;
1139     }
1140    
1141     /**
1142     * ���O���[�e�[�g������
1143     * ���[�e�[�V���������~
1144     */
1145 zmatsuo 8858 void FLogRotateHalt(void)
1146 zmatsuo 8852 {
1147     if (LogVar == NULL) {
1148     return;
1149     }
1150     LogVar->RotateMode = ROTATE_NONE;
1151     LogVar->RotateSize = 0;
1152     LogVar->RotateStep = 0;
1153     }
1154    
1155     static INT_PTR CALLBACK OnCommentDlgProc(HWND hDlgWnd, UINT msg, WPARAM wp, LPARAM lp)
1156     {
1157     static const DlgTextInfo TextInfos[] = {
1158     { 0, "DLG_COMMENT_TITLE" },
1159     { IDOK, "BTN_OK" }
1160     };
1161    
1162     switch (msg) {
1163     case WM_INITDIALOG:
1164     // �G�f�B�b�g�R���g���[�����t�H�[�J�X��������
1165     SetFocus(GetDlgItem(hDlgWnd, IDC_EDIT_COMMENT));
1166     SetDlgTexts(hDlgWnd, TextInfos, _countof(TextInfos), ts.UILanguageFile);
1167     return FALSE;
1168    
1169     case WM_COMMAND:
1170     switch (LOWORD(wp)) {
1171 zmatsuo 8901 case IDOK: {
1172 zmatsuo 8905 size_t len = _SendDlgItemMessageW(hDlgWnd, IDC_EDIT_COMMENT, WM_GETTEXTLENGTH, 0, 0);
1173 zmatsuo 8907 len += 1;
1174     wchar_t *buf = (wchar_t *)malloc(len * sizeof(wchar_t));
1175 zmatsuo 8911 _GetDlgItemTextW(hDlgWnd, IDC_EDIT_COMMENT, buf, (int)len);
1176 zmatsuo 8905 FLogWriteStr(buf);
1177     FLogWriteStr(L"\n"); // TODO ���s�R�[�h
1178     free(buf);
1179 zmatsuo 8852 TTEndDialog(hDlgWnd, IDOK);
1180     break;
1181 zmatsuo 8901 }
1182 zmatsuo 8852 default:
1183     return FALSE;
1184     }
1185     break;
1186 zmatsuo 8907
1187 zmatsuo 8852 case WM_CLOSE:
1188     TTEndDialog(hDlgWnd, 0);
1189     return TRUE;
1190    
1191     default:
1192     return FALSE;
1193     }
1194     return TRUE;
1195     }
1196    
1197 zmatsuo 8905 /**
1198     * ���O�t�@�C�����R�����g���������� (2004.8.6 yutaka)
1199     */
1200 zmatsuo 8858 void FLogAddCommentDlg(HINSTANCE hInst, HWND hWnd)
1201 zmatsuo 8852 {
1202 zmatsuo 8905 if (LogVar == NULL) {
1203     return;
1204     }
1205 zmatsuo 8852 TTDialogBox(hInst, MAKEINTRESOURCE(IDD_COMMENT_DIALOG),
1206     HVTWin, OnCommentDlgProc);
1207     }
1208    
1209 zmatsuo 8858 void FLogClose(void)
1210 zmatsuo 8852 {
1211 zmatsuo 8897 if (LogVar == NULL) {
1212     return;
1213     }
1214    
1215     FileTransEnd_();
1216 zmatsuo 8852 }
1217    
1218 zmatsuo 8897 /**
1219     * ���O���I�[�v������
1220     * @param[in] fname ���O�t�@�C����, CreateFile()���n������
1221     *
1222     * ���O�t�@�C������strftime���W�J�������s���������B
1223     * FLogGetLogFilename() �� FLogOpenDialog() ��
1224     * �t�@�C�����������������B
1225     */
1226 zmatsuo 8912 BOOL FLogOpen(const wchar_t *fname, LogCode_t code, BOOL bom)
1227 zmatsuo 8852 {
1228     BOOL ret;
1229    
1230 zmatsuo 8894 if (LogVar != NULL) {
1231 zmatsuo 8852 return FALSE;
1232     }
1233 zmatsuo 8897 if ((FileLog) || (BinLog)) return FALSE;
1234 zmatsuo 8852
1235 zmatsuo 8897 //
1236     PFileVar fv = (PFileVar)malloc(sizeof(TFileVar));
1237     if (fv == NULL) {
1238     return FALSE;
1239     }
1240     LogVar = fv;
1241     memset(fv, 0, sizeof(TFileVar));
1242 zmatsuo 8901 fv->FileHandle = INVALID_HANDLE_VALUE;
1243     fv->LogThread = INVALID_HANDLE_VALUE;
1244     fv->eLineEnd = Line_LineHead;
1245 zmatsuo 8897
1246 zmatsuo 8912 fv->log_code = code;
1247     fv->bom = bom;
1248 zmatsuo 8901 ret = LogStart(fname);
1249     if (ret == FALSE) {
1250     FileTransEnd_();
1251     }
1252 zmatsuo 8897
1253 zmatsuo 8852 return ret;
1254     }
1255    
1256 zmatsuo 8858 BOOL FLogIsOpend(void)
1257 zmatsuo 8852 {
1258     return LogVar != NULL;
1259     }
1260    
1261 zmatsuo 8904 BOOL FLogIsOpendText(void)
1262     {
1263     return LogVar != NULL && FileLog;
1264     }
1265    
1266     BOOL FLogIsOpendBin(void)
1267     {
1268     return LogVar != NULL && BinLog;
1269     }
1270    
1271 zmatsuo 8905 void FLogWriteStr(const wchar_t *str)
1272 zmatsuo 8852 {
1273 zmatsuo 8905 if (LogVar != NULL) {
1274 zmatsuo 8907 OutputStr(str);
1275 zmatsuo 8852 }
1276     }
1277    
1278 zmatsuo 8858 void FLogInfo(char *param_ptr, size_t param_len)
1279 zmatsuo 8852 {
1280     if (LogVar) {
1281     param_ptr[0] = '0'
1282     + (ts.LogBinary != 0)
1283     + ((ts.Append != 0) << 1)
1284     + ((ts.LogTypePlainText != 0) << 2)
1285     + ((ts.LogTimestamp != 0) << 3)
1286     + ((ts.LogHideDialog != 0) << 4);
1287 zmatsuo 8899 char *filenameU8 = ToU8W(LogVar->FullName);
1288     strncpy_s(param_ptr + 1, param_len - 1, filenameU8, _TRUNCATE);
1289     free(filenameU8);
1290 zmatsuo 8852 }
1291     else {
1292     param_ptr[0] = '0' - 1;
1293     param_ptr[1] = 0;
1294     }
1295     }
1296    
1297     /**
1298     * ���������O�t�@�C����������
1299     */
1300 zmatsuo 8899 const wchar_t *FLogGetFilename(void)
1301 zmatsuo 8852 {
1302     if (LogVar == NULL) {
1303     return NULL;
1304     }
1305     return LogVar->FullName;
1306     }
1307    
1308     /**
1309     * ���O�_�C�A���O���J��
1310 zmatsuo 8863 * @param[in,out] info.filename �t�@�C���������l
1311     * OK���A�t�@�C�����A�s�v����������free()��������
1312 zmatsuo 8852 * @retval TRUE [ok] ����������
1313     * @retval FALSE �L�����Z��������
1314     */
1315 zmatsuo 8863 BOOL FLogOpenDialog(HINSTANCE hInst, HWND hWnd, FLogDlgInfo_t *info)
1316 zmatsuo 8852 {
1317 zmatsuo 8863 LogDlgWork_t *work = (LogDlgWork_t *)calloc(sizeof(LogDlgWork_t), 1);
1318 zmatsuo 8899 wchar_t *srcfnameW = FLogGetLogFilename(info->filename);
1319 zmatsuo 8863 work->info = info;
1320     work->info->filename = srcfnameW;
1321     work->pts = &ts;
1322 zmatsuo 8852 INT_PTR ret = TTDialogBoxParam(
1323     hInst, MAKEINTRESOURCE(IDD_LOGDLG),
1324 zmatsuo 8863 hWnd, LogFnHook, (LPARAM)work);
1325     free(srcfnameW);
1326     free(work);
1327 zmatsuo 8852 return ret == IDOK ? TRUE : FALSE;
1328     }
1329    
1330     /**
1331     * ���O�t�@�C����������
1332     * ���O�t�@�C�����p���C�����s��
1333     * - strftime() ���������t�W�J
1334     * - �������������O�t�@�C���t�H���_������
1335     * - �z�X�g��,�|�[�g�����W�J
1336     *
1337     * @param[in] log_filename �t�@�C����(����/��������������ok)
1338     * NULL�������f�t�H���g�t�@�C����������
1339     * strftime�`��ok
1340     * @return �t���p�X�t�@�C����
1341     * �s�v���������� free() ��������
1342     */
1343 zmatsuo 8899 wchar_t *FLogGetLogFilename(const wchar_t *log_filename)
1344 zmatsuo 8852 {
1345     // �t�H���_
1346     char FileDirExpanded[MAX_PATH];
1347 zmatsuo 8899 const char *logdir;
1348 zmatsuo 8852 if (strlen(ts.LogDefaultPath) > 0) {
1349     logdir = ts.LogDefaultPath;
1350     }
1351     else if (strlen(ts.FileDir) > 0) {
1352     ExpandEnvironmentStrings(ts.FileDir, FileDirExpanded, sizeof(FileDirExpanded));
1353     logdir = FileDirExpanded;
1354     }
1355     else {
1356     logdir = ts.HomeDir;
1357     }
1358    
1359     // ���������t�@�C����
1360     char base_name[MAX_PATH];
1361     if (log_filename == NULL) {
1362     strncpy_s(base_name, _countof(base_name), ts.LogDefaultName, _TRUNCATE);
1363     }
1364     else {
1365 zmatsuo 8899 char *filenameA = ToCharW(log_filename);
1366     strncpy_s(base_name, _countof(base_name), filenameA, _TRUNCATE);
1367     free(filenameA);
1368 zmatsuo 8852 }
1369    
1370     // �t���p�X��
1371     char full_path[MAX_PATH];
1372     ConvFName(logdir, base_name, sizeof(base_name), "", full_path, sizeof(full_path));
1373     ParseStrftimeFileName(full_path, sizeof(full_path));
1374     ConvertLogname(full_path, sizeof(full_path));
1375    
1376 zmatsuo 8899 return ToWcharA(full_path);
1377 zmatsuo 8852 }
1378 zmatsuo 8857
1379     BOOL FLogIsPause()
1380     {
1381 zmatsuo 8900 if (LogVar == NULL) {
1382     return FALSE;
1383     }
1384     return LogVar->IsPause;
1385 zmatsuo 8857 }
1386    
1387     void FLogWindow(int nCmdShow)
1388     {
1389 zmatsuo 8900 if (LogVar == NULL) {
1390     return;
1391     }
1392 zmatsuo 8857
1393 zmatsuo 8901 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1394 zmatsuo 8857 ShowWindow(HWndLog, nCmdShow);
1395     if (nCmdShow == SW_RESTORE) {
1396     // �g���X�^�C�� WS_EX_NOACTIVATE ��������������
1397     SetForegroundWindow(HWndLog);
1398     }
1399     }
1400    
1401     void FLogShowDlg(void)
1402     {
1403 zmatsuo 8900 if (LogVar == NULL) {
1404     return;
1405     }
1406 zmatsuo 8901 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1407     ShowWindow(HWndLog, SW_SHOWNORMAL);
1408     SetForegroundWindow(HWndLog);
1409 zmatsuo 8857 }
1410 zmatsuo 8897
1411     /**
1412     * ���O��1byte��������
1413     * LogPut1() ������?
1414     */
1415     //void Log1Bin(PComVar cv, BYTE b)
1416     static void Log1Bin(BYTE b)
1417     {
1418 zmatsuo 8900 if (LogVar->IsPause || cv.ProtoFlag) {
1419 zmatsuo 8897 return;
1420     }
1421 zmatsuo 8906 if (cv_BinSkip > 0) {
1422     cv_BinSkip--;
1423 zmatsuo 8897 return;
1424     }
1425 zmatsuo 8906 cv_BinBuf[cv_BinPtr] = b;
1426     cv_BinPtr++;
1427     if (cv_BinPtr>=InBuffSize) {
1428     cv_BinPtr = cv_BinPtr-InBuffSize;
1429 zmatsuo 8897 }
1430 zmatsuo 8906 if (cv_BCount>=InBuffSize) {
1431     cv_BCount = InBuffSize;
1432     cv_BStart = cv_BinPtr;
1433 zmatsuo 8897 }
1434     else {
1435 zmatsuo 8906 cv_BCount++;
1436 zmatsuo 8897 }
1437     }
1438    
1439     static void LogBinSkip(int add)
1440     {
1441 zmatsuo 8906 if (cv_BinBuf != NULL) {
1442     cv_BinSkip += add;
1443 zmatsuo 8897 }
1444     }
1445    
1446     /**
1447     * ���O�o�b�t�@���������������f�[�^���o�C�g��������
1448     */
1449     int FLogGetCount(void)
1450     {
1451     if (FileLog) {
1452 zmatsuo 8906 return cv_LCount;
1453 zmatsuo 8897 }
1454     if (BinLog) {
1455 zmatsuo 8906 return cv_BCount;
1456 zmatsuo 8897 }
1457     return 0;
1458     }
1459    
1460     /**
1461 zmatsuo 8906 * ���O�o�b�t�@�������o�C�g��������
1462     */
1463     int FLogGetFreeCount(void)
1464     {
1465     if (FileLog) {
1466     return InBuffSize - cv_LCount;
1467     }
1468     if (BinLog) {
1469     return InBuffSize - cv_BCount;
1470     }
1471     return 0;
1472     }
1473    
1474     /**
1475 zmatsuo 8897 * �o�b�t�@�������O���t�@�C������������
1476     */
1477     void FLogWriteFile(void)
1478     {
1479 zmatsuo 8906 if (cv_LogBuf!=NULL)
1480 zmatsuo 8897 {
1481     if (FileLog) {
1482     LogToFile();
1483     }
1484     }
1485    
1486 zmatsuo 8906 if (cv_BinBuf!=NULL)
1487 zmatsuo 8897 {
1488     if (BinLog) {
1489     LogToFile();
1490     }
1491     }
1492     }
1493 zmatsuo 8904
1494     void FLogPutUTF32(unsigned int u32)
1495     {
1496     PFileVar fv = LogVar;
1497     size_t i;
1498 zmatsuo 8906 BOOL log_available = (cv_LogBuf != 0);
1499 zmatsuo 8904
1500     if (!log_available) {
1501 zmatsuo 8905 // ���O�����o��������
1502 zmatsuo 8904 return;
1503     }
1504    
1505 zmatsuo 8912 // �s����?(���s���o����������)
1506 zmatsuo 8910 if (ts.LogTimestamp && fv->eLineEnd) {
1507 zmatsuo 8912 // �^�C���X�^���v���o��
1508 zmatsuo 8910 fv->eLineEnd = Line_Other; /* clear endmark*/
1509     wchar_t* strtime = TimeStampStr();
1510     FLogWriteStr(strtime);
1511     free(strtime);
1512     }
1513    
1514 zmatsuo 8904 switch(fv->log_code) {
1515 zmatsuo 8912 case LOG_UTF8: {
1516 zmatsuo 8904 // UTF-8
1517     char u8_buf[4];
1518     size_t u8_len = UTF32ToUTF8(u32, u8_buf, _countof(u8_buf));
1519     for (i = 0; i < u8_len; i++) {
1520     BYTE b = u8_buf[i];
1521     LogPut1(b);
1522     }
1523     break;
1524     }
1525 zmatsuo 8912 case LOG_UTF16LE:
1526     case LOG_UTF16BE: {
1527 zmatsuo 8904 // UTF-16
1528     wchar_t u16[2];
1529     size_t u16_len = UTF32ToUTF16(u32, u16, _countof(u16));
1530     size_t i;
1531     for (i = 0; i < u16_len; i++) {
1532 zmatsuo 8912 if (fv->log_code == LOG_UTF16LE) {
1533 zmatsuo 8904 // UTF-16LE
1534     LogPut1(u16[i] & 0xff);
1535     LogPut1((u16[i] >> 8) & 0xff);
1536     }
1537     else {
1538     // UTF-16BE
1539     LogPut1((u16[i] >> 8) & 0xff);
1540     LogPut1(u16[i] & 0xff);
1541     }
1542     }
1543     }
1544     }
1545 zmatsuo 8910
1546     if (u32 == 0x0a) {
1547     fv->eLineEnd = Line_LineHead; /* set endmark*/
1548     }
1549 zmatsuo 8904 }
1550    
1551 zmatsuo 8912 static void FLogOutputBOM(void)
1552 zmatsuo 8904 {
1553     PFileVar fv = LogVar;
1554 zmatsuo 8910 DWORD wrote;
1555 zmatsuo 8904
1556     switch(fv->log_code) {
1557 zmatsuo 8910 case 0: {
1558 zmatsuo 8904 // UTF-8
1559 zmatsuo 8910 const char *bom = "\xef\xbb\xbf";
1560     WriteFile(LogVar->FileHandle, bom, 3, &wrote, NULL);
1561     LogVar->ByteCount += 3;
1562 zmatsuo 8904 break;
1563 zmatsuo 8910 }
1564     case 1: {
1565 zmatsuo 8904 // UTF-16LE
1566 zmatsuo 8910 const char *bom = "\xff\xfe";
1567     WriteFile(LogVar->FileHandle, bom, 2, &wrote, NULL);
1568     LogVar->ByteCount += 2;
1569 zmatsuo 8904 break;
1570 zmatsuo 8910 }
1571     case 2: {
1572 zmatsuo 8904 // UTF-16BE
1573 zmatsuo 8910 const char *bom = "\xfe\xff";
1574     WriteFile(LogVar->FileHandle, bom, 2, &wrote, NULL);
1575     LogVar->ByteCount += 2;
1576 zmatsuo 8904 break;
1577 zmatsuo 8910 }
1578 zmatsuo 8904 default:
1579     break;
1580     }
1581     }
1582    
1583 zmatsuo 8907 static void OutputStr(const wchar_t *str)
1584     {
1585     size_t len;
1586    
1587     assert(str != NULL);
1588    
1589     len = wcslen(str);
1590     while(*str != 0) {
1591     unsigned int u32;
1592     size_t u16_len = UTF16ToUTF32(str, len, &u32);
1593     switch (u16_len) {
1594     case 0:
1595     default:
1596     // ������������
1597     str++;
1598     len--;
1599     break;
1600     case 1:
1601     case 2: {
1602     FLogPutUTF32(u32);
1603     str += u16_len;
1604     len -= u16_len;
1605     break;
1606     }
1607     }
1608     }
1609     }

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