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 9103 - (hide annotations) (download) (as text)
Sun Dec 20 12:16:14 2020 UTC (3 years, 3 months ago) by zmatsuo
File MIME type: text/x-c++src
File size: 38591 byte(s)
ProtoFlagを TComVar から filesys_proto.cpp へ移動
1 doda 6806 /*
2 nmaya 9048 * (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 9103 #include "filesys.h" // for ProtoGetProtoFlag()
62 zmatsuo 8866
63 zmatsuo 8901 /*
64     Line Head flag for timestamping
65     2007.05.24 Gentaro
66     */
67     enum enumLineEnd {
68     Line_Other = 0,
69     Line_LineHead = 1,
70     Line_FileHead = 2,
71     };
72    
73 zmatsuo 8894 typedef struct {
74 zmatsuo 8901 wchar_t *FullName;
75 maya 3227
76 zmatsuo 8901 HANDLE FileHandle;
77     LONG FileSize, ByteCount;
78 maya 6071
79 zmatsuo 8901 DWORD StartTime;
80 maya 3227
81 zmatsuo 8901 enum enumLineEnd eLineEnd;
82 maya 7182
83 zmatsuo 8901 // log rotate
84     int RotateMode; // enum rotate_mode RotateMode;
85     LONG RotateSize;
86     int RotateStep;
87 zmatsuo 8900
88 zmatsuo 8901 HANDLE LogThread;
89     DWORD LogThreadId;
90     HANDLE LogThreadEvent;
91    
92     BOOL IsPause;
93    
94     PFileTransDlg FLogDlg;
95 zmatsuo 8904
96 zmatsuo 8912 LogCode_t log_code;
97     BOOL bom;
98 zmatsuo 8904
99 zmatsuo 8894 } TFileVar_;
100     typedef TFileVar_ *PFileVar_;
101    
102     #define PFileVar PFileVar_
103     #define TFileVar TFileVar_
104    
105     static PFileVar LogVar = NULL;
106    
107 zmatsuo 8900 static BOOL FileLog = FALSE;
108     static BOOL BinLog = FALSE;
109 zmatsuo 8894
110 zmatsuo 8906 static PCHAR cv_LogBuf;
111     static int cv_LogPtr, cv_LStart, cv_LCount;
112     static PCHAR cv_BinBuf;
113     static int cv_BinPtr, cv_BStart, cv_BCount;
114     static int cv_BinSkip;
115    
116 yutakapon 5206 // �x�����������p�X���b�h�����b�Z�[�W
117     #define WM_DPC_LOGTHREAD_SEND (WM_APP + 1)
118    
119 zmatsuo 8897 static void FileTransEnd_(void);
120     static void Log1Bin(BYTE b);
121     static void LogBinSkip(int add);
122     static BOOL CreateLogBuf(void);
123     static BOOL CreateBinBuf(void);
124 zmatsuo 8905 void LogPut1(BYTE b);
125 zmatsuo 8907 static void OutputStr(const wchar_t *str);
126 zmatsuo 8908 static void LogToFile(void);
127 zmatsuo 8912 static void FLogOutputBOM(void);
128 yutakapon 5206
129 zmatsuo 8894 static BOOL OpenFTDlg_(PFileVar fv)
130 maya 3227 {
131 zmatsuo 8899 PFileTransDlg FTDlg = new CFileTransDlg();
132     if (FTDlg == NULL) {
133     return FALSE;
134     }
135 maya 3227
136 zmatsuo 8899 wchar_t *DlgCaption;
137     wchar_t uimsg[MAX_UIMSG];
138     #define TitLogW L"Log"
139     get_lang_msgW("FILEDLG_TRANS_TITLE_LOG", uimsg, _countof(uimsg), TitLogW, ts.UILanguageFile);
140     aswprintf(&DlgCaption, L"Tera Term: %s", uimsg);
141 maya 3227
142 zmatsuo 9070 CFileTransDlg::Info info;
143 zmatsuo 8899 info.UILanguageFile = ts.UILanguageFile;
144 zmatsuo 9070 info.OpId = CFileTransDlg::OpLog;
145 zmatsuo 8899 info.DlgCaption = DlgCaption;
146 zmatsuo 8911 info.FileName = NULL;
147 zmatsuo 8899 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 doda 4454
153 zmatsuo 8901 fv->FLogDlg = FTDlg;
154 doda 5383
155 zmatsuo 8899 free(DlgCaption);
156     return TRUE;
157 maya 3227 }
158    
159 zmatsuo 8852 /**
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 maya 3227 {
167     char buf[MAXPATHLEN], buf2[MAXPATHLEN], *p = c;
168 yutakapon 5139 char tmphost[1024];
169 maya 6590 char tmpuser[256+1];
170     DWORD len_user = sizeof(tmpuser);
171 maya 3227
172     memset(buf, 0, sizeof(buf));
173    
174     while(*p != '\0') {
175     if (*p == '&' && *(p+1) != '\0') {
176     switch (*(p+1)) {
177 maya 3473 case 'h':
178 maya 3227 if (cv.Open) {
179     if (cv.PortType == IdTCPIP) {
180 yutakapon 5139 // �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 maya 5141 replaceInvalidFileNameChar(tmphost, '_');
186 yutakapon 5139 strncat_s(buf,sizeof(buf), tmphost, _TRUNCATE);
187 maya 3227 }
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 maya 3473 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 maya 6590 case 'u':
204     if (GetUserName(tmpuser, &len_user) != 0) {
205     strncat_s(buf,sizeof(buf),tmpuser,_TRUNCATE);
206     }
207     break;
208 maya 3473 default:
209 maya 3227 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 zmatsuo 8858 static void FixLogOption(void)
224 maya 3227 {
225 doda 3887 if (ts.LogBinary) {
226 maya 3227 ts.LogTypePlainText = false;
227     ts.LogTimestamp = false;
228     }
229     }
230    
231 yutakapon 5206
232     // �X���b�h���I�����t�@�C�����N���[�Y
233     static void CloseFileSync(PFileVar ptr)
234     {
235 yutakapon 6489 BOOL ret;
236    
237 zmatsuo 8901 if (ptr->FileHandle == INVALID_HANDLE_VALUE) {
238 yutakapon 5206 return;
239 zmatsuo 8901 }
240 yutakapon 5206
241 zmatsuo 7930 if (ptr->LogThread != INVALID_HANDLE_VALUE) {
242 yutakapon 5206 // �X���b�h���I������
243 yutakapon 6489 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 zmatsuo 8852 //DWORD code = GetLastError();
250 yutakapon 6489 }
251 yutakapon 5206 CloseHandle(ptr->LogThread);
252 zmatsuo 7930 ptr->LogThread = INVALID_HANDLE_VALUE;
253 yutakapon 5206 }
254 zmatsuo 7930 CloseHandle(ptr->FileHandle);
255 zmatsuo 8901 ptr->FileHandle = INVALID_HANDLE_VALUE;
256 yutakapon 5206 }
257    
258     // �x�����������p�X���b�h
259 doda 6435 static unsigned _stdcall DeferredLogWriteThread(void *arg)
260 yutakapon 5206 {
261     MSG msg;
262     PFileVar fv = (PFileVar)arg;
263     PCHAR buf;
264     DWORD buflen;
265 maya 5273 DWORD wrote;
266 yutakapon 5206
267     PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE);
268    
269 yutakapon 6489 // �X���b�h�L���[���������I�������������X���b�h�����������m�����B
270     if (fv->LogThreadEvent != NULL) {
271     SetEvent(fv->LogThreadEvent);
272     }
273    
274 yutakapon 5206 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 zmatsuo 7930 WriteFile(LogVar->FileHandle, buf, buflen, &wrote, NULL);
280 yutakapon 5206 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 zmatsuo 8901 // �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 zmatsuo 8852 /**
316     * �_�C�A���O�����e�� ts ����������
317     *
318     * TODO
319     * �_�C�A���O�����������l�������I��������
320     * ����������������������������������������������?
321     */
322     static void SetLogFlags(HWND Dialog)
323 maya 3227 {
324 zmatsuo 8852 WORD BinFlag, val;
325 maya 3227
326 zmatsuo 8852 GetRB(Dialog, &BinFlag, IDC_FOPTBIN, IDC_FOPTBIN);
327     ts.LogBinary = BinFlag;
328 maya 3227
329 zmatsuo 8866 GetRB(Dialog, &val, IDC_APPEND, IDC_APPEND);
330 zmatsuo 8852 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 maya 3227 }
339    
340 zmatsuo 8852 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 zmatsuo 8912 ts.LogTimestampType = (WORD)(GetCurSel(Dialog, IDC_TIMESTAMPTYPE) - 1);
347 zmatsuo 8852 }
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 zmatsuo 8863 static void CheckLogFile(const wchar_t *filename, BOOL *exist, int *bom)
360 zmatsuo 8852 {
361 zmatsuo 8863 *exist = FALSE;
362     *bom = 0;
363    
364 zmatsuo 8852 // �t�@�C������������?
365 zmatsuo 8863 DWORD logdir = _GetFileAttributesW(filename);
366 zmatsuo 8852 if ((logdir != INVALID_FILE_ATTRIBUTES) && ((logdir & FILE_ATTRIBUTE_DIRECTORY) == 0)) {
367 zmatsuo 8863 // �t�@�C����������
368 zmatsuo 8852 *exist = TRUE;
369    
370     // BOM�L��/�����`�F�b�N
371 zmatsuo 8863 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 zmatsuo 8852 }
391 maya 3227 }
392 zmatsuo 8863 }
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 zmatsuo 8866 if (work->pts->Append > 0) {
407     CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
408     }
409 zmatsuo 8863 }
410 zmatsuo 8852 else {
411 zmatsuo 8863 // �t�@�C�������� -> �V�K
412     EnableWindow(GetDlgItem(Dialog, IDC_APPEND), FALSE);
413     CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
414 maya 3227 }
415 zmatsuo 8852
416 zmatsuo 8863 if (work->file_exist && IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED) {
417     // �t�@�C������������ && append
418     int bom = work->current_bom;
419 zmatsuo 8852 if (bom != 0) {
420     // BOM�L��
421     CheckDlgButton(Dialog, IDC_BOM, BST_CHECKED);
422 zmatsuo 8863 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 zmatsuo 8852 }
428     else {
429     // BOM����
430     CheckDlgButton(Dialog, IDC_BOM, BST_UNCHECKED);
431 zmatsuo 8863 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
432 zmatsuo 8852 }
433 zmatsuo 8863 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 zmatsuo 8852 }
445 maya 3227 else {
446 zmatsuo 8866 // �t�@�C�������� ���� append��������(������)
447 zmatsuo 8863 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE);
448 zmatsuo 8852 CheckDlgButton(Dialog, IDC_BOM, BST_CHECKED);
449 zmatsuo 8863 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 maya 3227 }
454 zmatsuo 8852 }
455 maya 3227
456 zmatsuo 8863 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 zmatsuo 8852
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 zmatsuo 8866 { IDC_APPEND, "DLG_FOPT_APPEND" },
472 zmatsuo 8852 { 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 zmatsuo 8863 LogDlgWork_t *work = (LogDlgWork_t *)GetWindowLongPtr(Dialog, DWLP_USER);
484 zmatsuo 8852
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 zmatsuo 8863 work = (LogDlgWork_t *)lParam;
493     TTTSet *pts = work->pts;
494     const char *UILanguageFile = pts->UILanguageFile;
495     SetWindowLongPtr(Dialog, DWLP_USER, (LONG_PTR)work);
496 zmatsuo 8852 ::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 zmatsuo 8863 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16LE");
504     SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16BE");
505 zmatsuo 8852 SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
506    
507 zmatsuo 8863 _SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, work->info->filename);
508     work->info->filename = NULL;
509 zmatsuo 8852
510     // Binary/Text �`�F�b�N�{�b�N�X
511 zmatsuo 8863 if (pts->LogBinary) {
512     CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTBIN);
513 doda 6946 }
514 zmatsuo 8852 else {
515 zmatsuo 8863 CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTTEXT);
516 zmatsuo 8852 }
517    
518     // Plain Text �`�F�b�N�{�b�N�X
519 zmatsuo 8863 if (pts->LogBinary) {
520 zmatsuo 8852 // Binary�t���O���L�����������`�F�b�N��������
521     DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT);
522 doda 6946 }
523 zmatsuo 8863 else if (pts->LogTypePlainText) {
524 zmatsuo 8852 SetRB(Dialog, 1, IDC_PLAINTEXT, IDC_PLAINTEXT);
525 doda 6946 }
526 zmatsuo 8852
527     // Hide dialog�`�F�b�N�{�b�N�X (2008.1.30 maya)
528 zmatsuo 8863 if (pts->LogHideDialog) {
529 zmatsuo 8852 SetRB(Dialog, 1, IDC_HIDEDIALOG, IDC_HIDEDIALOG);
530 doda 6946 }
531 zmatsuo 8852
532     // Include screen buffer�`�F�b�N�{�b�N�X (2013.9.29 yutaka)
533 zmatsuo 8863 if (pts->LogAllBuffIncludedInFirst) {
534 zmatsuo 8852 SetRB(Dialog, 1, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST);
535 doda 6946 }
536 maya 3227
537 zmatsuo 8852 // timestamp�`�F�b�N�{�b�N�X (2006.7.23 maya)
538 zmatsuo 8863 if (pts->LogBinary) {
539 zmatsuo 8852 // Binary�t���O���L�����������`�F�b�N��������
540     DisableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP);
541     }
542 zmatsuo 8863 else if (pts->LogTimestamp) {
543 zmatsuo 8852 SetRB(Dialog, 1, IDC_TIMESTAMP, IDC_TIMESTAMP);
544     }
545    
546     // timestamp ����
547 zmatsuo 8863 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 zmatsuo 8852 SendDlgItemMessage(Dialog, IDC_TIMESTAMPTYPE, CB_SETCURSEL, tstype, 0);
552 zmatsuo 8863 if (pts->LogBinary || !pts->LogTimestamp) {
553 zmatsuo 8852 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 zmatsuo 8863 wchar_t filename[MAX_PATH];
565     _GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename, _countof(filename));
566     work->info->filename = _wcsdup(filename);
567 zmatsuo 8866 work->info->append = IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED;
568 zmatsuo 8863 work->info->bom = IsDlgButtonChecked(Dialog, IDC_BOM) == BST_CHECKED;
569 zmatsuo 8912 work->info->code = (LogCode_t)SendDlgItemMessageA(Dialog, IDC_TEXTCODING_DROPDOWN, CB_GETCURSEL, 0, 0);
570 zmatsuo 8852 SetLogFlags(Dialog);
571     EndDialog(Dialog, IDOK);
572 doda 6947 break;
573 zmatsuo 8852 }
574     case IDCANCEL:
575     EndDialog(Dialog, IDCANCEL);
576 doda 6947 break;
577 zmatsuo 8852 case IDHELP:
578 zmatsuo 8863 OpenHelp(HH_HELP_CONTEXT, HlpFileLog, work->pts->UILanguageFile);
579 doda 6947 break;
580 zmatsuo 8852 case IDC_FOPT_FILENAME_BUTTON: {
581     /* save current dir */
582 zmatsuo 8863 const char *UILanguageFile = work->pts->UILanguageFile;
583 zmatsuo 8852 wchar_t curdir[MAXPATHLEN];
584     _GetCurrentDirectoryW(_countof(curdir), curdir);
585 doda 6947
586 zmatsuo 8863 wchar_t fname[MAX_PATH];
587     GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname, _countof(fname));
588 maya 3227
589 zmatsuo 8943 wchar_t *FNFilter = GetCommonDialogFilterW(NULL, 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 8964 BOOL Ok = _GetSaveFileNameW(&ofn);
611 zmatsuo 8943 free(FNFilter);
612 zmatsuo 8852 if (Ok) {
613 zmatsuo 8863 SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname);
614 doda 6947 }
615 zmatsuo 8852
616     /* restore dir */
617     _SetCurrentDirectoryW(curdir);
618    
619     break;
620 doda 6947 }
621 zmatsuo 8852 case IDC_FOPTBIN:
622 zmatsuo 8863 EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), FALSE);
623     EnableWindow(GetDlgItem(Dialog, IDC_BOM), FALSE);
624 zmatsuo 8852 DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_TIMESTAMP);
625     DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
626     break;
627     case IDC_FOPTTEXT:
628 zmatsuo 8863 ArrangeControls(Dialog, work);
629 zmatsuo 8852 EnableDlgItem(Dialog, IDC_PLAINTEXT, IDC_TIMESTAMP);
630     // FALLTHROUGH -- BinFlag �� off ������ Timestamp �������L��/��������������
631     case IDC_TIMESTAMP:
632     if (IsDlgButtonChecked(Dialog, IDC_TIMESTAMP) == BST_CHECKED) {
633     EnableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
634 doda 6947 }
635     else {
636 zmatsuo 8852 DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE);
637 doda 6947 }
638 zmatsuo 8852 break;
639     case IDC_FOPT_FILENAME_EDIT:
640     if (HIWORD(wParam) == EN_CHANGE){
641 zmatsuo 8863 wchar_t filename[MAX_PATH];
642     GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename, _countof(filename));
643     CheckLogFile(Dialog, filename, work);
644 zmatsuo 8852 }
645     break;
646 zmatsuo 8863 case IDC_NEW_OVERWRITE:
647     if (IsDlgButtonChecked(Dialog, IDC_FOPTTEXT) == BST_CHECKED) {
648     EnableWindow(GetDlgItem(Dialog, IDC_BOM), TRUE);
649     EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), TRUE);
650     CheckDlgButton(Dialog, IDC_BOM, BST_CHECKED);
651     SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0);
652     }
653     break;
654     case IDC_APPEND:
655     ArrangeControls(Dialog, work);
656     break;
657 doda 6947 }
658 zmatsuo 8852 break;
659     case WM_DROPFILES: {
660     // �����h���b�v��������������1������������
661     HDROP hDrop = (HDROP)wParam;
662     const UINT len = _DragQueryFileW(hDrop, 0, NULL, 0);
663     if (len == 0) {
664     DragFinish(hDrop);
665     return TRUE;
666     }
667     wchar_t *filename = (wchar_t *)malloc(sizeof(wchar_t) * (len + 1));
668     _DragQueryFileW(hDrop, 0, filename, len + 1);
669     filename[len] = '\0';
670 zmatsuo 8863 CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND);
671 zmatsuo 8852 _SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename);
672     SendDlgItemMessage(Dialog, IDC_FOPT_FILENAME_EDIT, EM_SETSEL, len, len);
673     free(filename);
674     DragFinish(hDrop);
675     return TRUE;
676 maya 3227 }
677 zmatsuo 8852 }
678     return FALSE;
679     }
680 maya 3227
681 zmatsuo 8901 static void OpenLogFile(PFileVar fv)
682     {
683     int dwShareMode = FILE_SHARE_READ;
684     if (!ts.LogLockExclusive) {
685     dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
686     }
687     LogVar->FileHandle = _CreateFileW(LogVar->FullName, GENERIC_WRITE, dwShareMode, NULL,
688     OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
689     }
690    
691 zmatsuo 8899 static BOOL LogStart(const wchar_t *fname)
692 zmatsuo 8852 {
693 zmatsuo 8901 PFileVar fv = LogVar;
694 maya 3227
695 zmatsuo 8901 fv->FullName = _wcsdup(fname);
696 zmatsuo 8852 FixLogOption();
697    
698 doda 3887 if (ts.LogBinary > 0)
699 maya 3227 {
700     BinLog = TRUE;
701     FileLog = FALSE;
702     if (! CreateBinBuf())
703     {
704     return FALSE;
705     }
706     }
707     else {
708     BinLog = FALSE;
709     FileLog = TRUE;
710     if (! CreateLogBuf())
711     {
712     return FALSE;
713     }
714     }
715 zmatsuo 8906 cv_LStart = cv_LogPtr;
716     cv_LCount = 0;
717 maya 3227
718 zmatsuo 8901 OpenLogFile(fv);
719     if (LogVar->FileHandle == INVALID_HANDLE_VALUE) {
720     return FALSE;
721     }
722    
723 maya 3227 /* 2007.05.24 Gentaro */
724 zmatsuo 8901 fv->eLineEnd = Line_LineHead;
725 maya 3227 if (ts.Append > 0)
726     {
727 zmatsuo 8901 SetFilePointer(LogVar->FileHandle, 0, NULL, FILE_END);
728     /* 2007.05.24 Gentaro
729     If log file already exists,
730     a newline is inserted before the first timestamp.
731     */
732     fv->eLineEnd = Line_FileHead;
733 maya 3227 }
734 zmatsuo 8912 else {
735     // ���L��������(�V�K)������ BOM ���o������
736     if (fv->bom) {
737     FLogOutputBOM();
738     }
739     }
740 maya 3227
741 yutakapon 5171 // Log rotate configuration
742     LogVar->RotateMode = ts.LogRotate;
743     LogVar->RotateSize = ts.LogRotateSize;
744     LogVar->RotateStep = ts.LogRotateStep;
745 yutakapon 5162
746 yutakapon 6365 // Log rotate���L���������A�����t�@�C���T�C�Y�����������B
747     // �������t�@�C�������������T�C�Y�����[�e�[�g�������������C���B
748     // (2016.4.9 yutaka)
749     if (LogVar->RotateMode != ROTATE_NONE) {
750 zmatsuo 8852 DWORD size = GetFileSize(LogVar->FileHandle, NULL);
751 zmatsuo 8901 if (size == -1) {
752     return FALSE;
753     }
754     LogVar->ByteCount = size;
755 yutakapon 6365 }
756 zmatsuo 8901 else {
757     LogVar->ByteCount = 0;
758     }
759 yutakapon 6365
760 zmatsuo 8894 if (! OpenFTDlg_(LogVar)) {
761 maya 3227 return FALSE;
762     }
763    
764 zmatsuo 8900 LogVar->IsPause = FALSE;
765 zmatsuo 8899 LogVar->StartTime = GetTickCount();
766    
767 zmatsuo 8901 if (ts.DeferredLogWriteMode) {
768     StartThread(LogVar);
769 yutakapon 6489 }
770 yutakapon 5206
771 zmatsuo 8897 if (FileLog) {
772     cv.Log1Byte = LogPut1;
773     }
774     if (BinLog) {
775     cv.Log1Bin = Log1Bin;
776     cv.LogBinSkip = LogBinSkip;
777     }
778    
779 maya 3227 return TRUE;
780     }
781    
782 zmatsuo 8897 /**
783 zmatsuo 8908 * �����o�b�t�@�������f�[�^�����������O�������o��
784     * (2013.9.29 yutaka)
785     *
786     * TODO
787     * 1�s������
788     */
789     void FLogOutputAllBuffer(void)
790     {
791 zmatsuo 8909 DWORD ofs;
792 zmatsuo 8908 int size;
793     wchar_t buf[512];
794     for (ofs = 0 ; ; ofs++ ) {
795     // 1�����s�����������B���������������A�G�X�P�[�v�V�[�P���X�������������B
796     size = BuffGetAnyLineDataW(ofs, buf, _countof(buf));
797     if (size == -1)
798     break;
799    
800     OutputStr(buf);
801     OutputStr(L"\r\n");
802     LogToFile();
803     }
804     }
805    
806     /**
807 zmatsuo 8897 * ���O��1byte��������
808     * �o�b�t�@��������������
809     * ���������������� LogToFile() ���s������
810     */
811 maya 3227 void LogPut1(BYTE b)
812     {
813 zmatsuo 8906 cv_LogBuf[cv_LogPtr] = b;
814     cv_LogPtr++;
815     if (cv_LogPtr>=InBuffSize)
816     cv_LogPtr = cv_LogPtr-InBuffSize;
817 maya 3227
818     if (FileLog)
819     {
820 zmatsuo 8906 if (cv_LCount>=InBuffSize)
821 maya 3227 {
822 zmatsuo 8906 cv_LCount = InBuffSize;
823     cv_LStart = cv_LogPtr;
824 maya 3227 }
825     else
826 zmatsuo 8906 cv_LCount++;
827 maya 3227 }
828     else
829 zmatsuo 8906 cv_LCount = 0;
830 maya 3227 }
831    
832     static BOOL Get1(PCHAR Buf, int *Start, int *Count, PBYTE b)
833     {
834     if (*Count<=0) return FALSE;
835     *b = Buf[*Start];
836     (*Start)++;
837     if (*Start>=InBuffSize)
838     *Start = *Start-InBuffSize;
839     (*Count)--;
840     return TRUE;
841     }
842    
843    
844    
845     static CRITICAL_SECTION g_filelog_lock; /* ���b�N�p���� */
846    
847     void logfile_lock_initialize(void)
848     {
849     InitializeCriticalSection(&g_filelog_lock);
850     }
851    
852     static inline void logfile_lock(void)
853     {
854     EnterCriticalSection(&g_filelog_lock);
855     }
856    
857     static inline void logfile_unlock(void)
858     {
859     LeaveCriticalSection(&g_filelog_lock);
860     }
861    
862 yutakapon 5162 // ���O�����[�e�[�g�����B
863     // (2013.3.21 yutaka)
864     static void LogRotate(void)
865     {
866 yutakapon 5165 int loopmax = 10000; // XXX
867     int i, k;
868 yutakapon 5162
869     if (LogVar->RotateMode == ROTATE_NONE)
870     return;
871    
872     if (LogVar->RotateMode == ROTATE_SIZE) {
873 doda 6435 if (LogVar->ByteCount <= LogVar->RotateSize)
874 yutakapon 5162 return;
875     //OutputDebugPrintf("%s: mode %d size %ld\n", __FUNCTION__, LogVar->RotateMode, LogVar->ByteCount);
876     } else {
877     return;
878     }
879    
880     logfile_lock();
881     // ���O�T�C�Y���������������B
882     LogVar->ByteCount = 0;
883    
884     // �������������t�@�C�����N���[�Y�����A�������t�@�C�����I�[�v�������B
885 yutakapon 5206 CloseFileSync(LogVar);
886 yutakapon 5162
887 yutakapon 5165 // �������[�e�[�V�������X�e�b�v�����w����������
888     if (LogVar->RotateStep > 0)
889     loopmax = LogVar->RotateStep;
890    
891     for (i = 1 ; i <= loopmax ; i++) {
892 zmatsuo 8899 wchar_t *filename;
893     aswprintf(&filename, L"%s.%d", LogVar->FullName, i);
894     DWORD attr = _GetFileAttributesW(filename);
895     free(filename);
896 zmatsuo 8901 if (attr == INVALID_FILE_ATTRIBUTES)
897 yutakapon 5162 break;
898     }
899 yutakapon 5165 if (i > loopmax) {
900     // �������������������������A�������t�@�C�������p�������B
901     i = loopmax;
902 yutakapon 5162 }
903    
904     // ���t�@�C�������l�[���B
905 yutakapon 5165 for (k = i-1 ; k >= 0 ; k--) {
906 zmatsuo 8899 wchar_t *oldfile;
907 yutakapon 5165 if (k == 0)
908 zmatsuo 8899 oldfile = _wcsdup(LogVar->FullName);
909 yutakapon 5165 else
910 zmatsuo 8899 aswprintf(&oldfile, L"%s.%d", LogVar->FullName, k);
911     wchar_t *newfile;
912     aswprintf(&newfile, L"%s.%d", LogVar->FullName, k+1);
913     _DeleteFileW(newfile);
914     if (_MoveFileW(oldfile, newfile) == 0) {
915 yutakapon 5165 OutputDebugPrintf("%s: rename %d\n", __FUNCTION__, errno);
916     }
917 zmatsuo 8899 free(oldfile);
918     free(newfile);
919 yutakapon 5165 }
920 yutakapon 5162
921     // ���I�[�v��
922 zmatsuo 8901 OpenLogFile(LogVar);
923 zmatsuo 8912 if (LogVar->bom) {
924     FLogOutputBOM();
925     }
926 zmatsuo 8901 if (ts.DeferredLogWriteMode) {
927     StartThread(LogVar);
928 yutakapon 5162 }
929    
930     logfile_unlock();
931     }
932    
933 zmatsuo 8910 static wchar_t *TimeStampStr()
934 zmatsuo 8909 {
935     char *strtime = NULL;
936     switch (ts.LogTimestampType) {
937     case TIMESTAMP_LOCAL:
938     default:
939     strtime = mctimelocal(ts.LogTimestampFormat, FALSE);
940     break;
941     case TIMESTAMP_UTC:
942     strtime = mctimelocal(ts.LogTimestampFormat, TRUE);
943     break;
944     case TIMESTAMP_ELAPSED_LOGSTART:
945     strtime = strelapsed(LogVar->StartTime);
946     break;
947     case TIMESTAMP_ELAPSED_CONNECTED:
948     strtime = strelapsed(cv.ConnectedTime);
949     break;
950     }
951    
952     char tmp[128];
953     tmp[0] = 0;
954     strncat_s(tmp, sizeof(tmp), "[", _TRUNCATE);
955     strncat_s(tmp, sizeof(tmp), strtime, _TRUNCATE);
956     strncat_s(tmp, sizeof(tmp), "] ", _TRUNCATE);
957    
958 zmatsuo 8910 return ToWcharA(tmp);
959 zmatsuo 8909 }
960    
961 zmatsuo 8897 /**
962     * �o�b�t�@�������O���t�@�C������������
963     */
964     static void LogToFile(void)
965 maya 3227 {
966     PCHAR Buf;
967     int Start, Count;
968     BYTE b;
969    
970     if (FileLog)
971     {
972 zmatsuo 8906 Buf = cv_LogBuf;
973     Start = cv_LStart;
974     Count = cv_LCount;
975 maya 3227 }
976     else if (BinLog)
977     {
978 zmatsuo 8906 Buf = cv_BinBuf;
979     Start = cv_BStart;
980     Count = cv_BCount;
981 maya 3227 }
982     else
983     return;
984    
985     if (Buf==NULL) return;
986     if (Count==0) return;
987    
988     // ���b�N������(2004.8.6 yutaka)
989     logfile_lock();
990    
991 zmatsuo 8909 // ���������f�[�^����������
992     DWORD WriteBufMax = 8192;
993     DWORD WriteBufLen = 0;
994     PCHAR WriteBuf = (PCHAR)malloc(WriteBufMax);
995     while (Get1(Buf,&Start,&Count,&b)) {
996 zmatsuo 9103 if (FLogIsPause() || ProtoGetProtoFlag()) {
997 zmatsuo 8909 continue;
998     }
999 doda 6947
1000 zmatsuo 8909 if (WriteBufLen >= (WriteBufMax*4/5)) {
1001     WriteBufMax *= 2;
1002     WriteBuf = (PCHAR)realloc(WriteBuf, WriteBufMax);
1003     }
1004 doda 6947
1005 zmatsuo 8909 WriteBuf[WriteBufLen++] = b;
1006 yutakapon 5206
1007 zmatsuo 8909 (LogVar->ByteCount)++;
1008     }
1009 yutakapon 5206
1010 zmatsuo 8909 // ��������
1011     if (WriteBufLen > 0) {
1012     if (ts.DeferredLogWriteMode) {
1013     PostThreadMessage(LogVar->LogThreadId, WM_DPC_LOGTHREAD_SEND, (WPARAM)WriteBuf, WriteBufLen);
1014 yutakapon 5206 }
1015 zmatsuo 8909 else {
1016     DWORD wrote;
1017     WriteFile(LogVar->FileHandle, WriteBuf, WriteBufLen, &wrote, NULL);
1018     free(WriteBuf);
1019     }
1020 maya 3227 }
1021    
1022     logfile_unlock();
1023    
1024     if (FileLog)
1025     {
1026 zmatsuo 8906 cv_LStart = Start;
1027     cv_LCount = Count;
1028 maya 3227 }
1029     else {
1030 zmatsuo 8906 cv_BStart = Start;
1031     cv_BCount = Count;
1032 maya 3227 }
1033 zmatsuo 9103 if (FLogIsPause() || ProtoGetProtoFlag()) return;
1034 zmatsuo 8901 LogVar->FLogDlg->RefreshNum(LogVar->StartTime, LogVar->FileSize, LogVar->ByteCount);
1035 yutakapon 5162
1036 zmatsuo 8896
1037 yutakapon 5162 // ���O�E���[�e�[�g
1038     LogRotate();
1039 maya 3227 }
1040    
1041 zmatsuo 8897 static BOOL CreateLogBuf(void)
1042 maya 3227 {
1043 zmatsuo 8906 if (cv_LogBuf==NULL)
1044 maya 3227 {
1045 zmatsuo 8906 cv_LogBuf = (char *)malloc(InBuffSize);
1046     cv_LogPtr = 0;
1047     cv_LStart = 0;
1048     cv_LCount = 0;
1049 maya 3227 }
1050 zmatsuo 8906 return (cv_LogBuf!=NULL);
1051 maya 3227 }
1052    
1053 zmatsuo 8897 static void FreeLogBuf(void)
1054 maya 3227 {
1055 zmatsuo 8906 free(cv_LogBuf);
1056     cv_LogBuf = NULL;
1057     cv_LogPtr = 0;
1058     cv_LStart = 0;
1059     cv_LCount = 0;
1060 maya 3227 }
1061    
1062 zmatsuo 8897 static BOOL CreateBinBuf(void)
1063 maya 3227 {
1064 zmatsuo 8906 if (cv_BinBuf==NULL)
1065 maya 3227 {
1066 zmatsuo 8906 cv_BinBuf = (PCHAR)malloc(InBuffSize);
1067     cv_BinPtr = 0;
1068     cv_BStart = 0;
1069     cv_BCount = 0;
1070 maya 3227 }
1071 zmatsuo 8906 return (cv_BinBuf!=NULL);
1072 maya 3227 }
1073    
1074 zmatsuo 8897 static void FreeBinBuf(void)
1075 maya 3227 {
1076 zmatsuo 8906 free(cv_BinBuf);
1077     cv_BinBuf = NULL;
1078     cv_BinPtr = 0;
1079     cv_BStart = 0;
1080     cv_BCount = 0;
1081 maya 3227 }
1082    
1083 zmatsuo 8897 static void FileTransEnd_(void)
1084 maya 3227 {
1085 zmatsuo 8900 if (LogVar == NULL) {
1086     return;
1087     }
1088 zmatsuo 8897 FileLog = FALSE;
1089     BinLog = FALSE;
1090     cv.Log1Byte = NULL;
1091     cv.Log1Bin = NULL;
1092     cv.LogBinSkip = NULL;
1093 zmatsuo 8901 PFileTransDlg FLogDlg = LogVar->FLogDlg;
1094     if (FLogDlg != NULL) {
1095 zmatsuo 8897 FLogDlg->DestroyWindow();
1096     FLogDlg = NULL;
1097 zmatsuo 8916 LogVar->FLogDlg = NULL;
1098 maya 3227 }
1099 zmatsuo 8900 CloseFileSync(LogVar);
1100 zmatsuo 8897 FreeLogBuf();
1101     FreeBinBuf();
1102 zmatsuo 8916 free(LogVar->FullName);
1103     LogVar->FullName = NULL;
1104 zmatsuo 8900 free(LogVar);
1105     LogVar = NULL;
1106 maya 3227 }
1107    
1108 zmatsuo 8857 /**
1109     * ���O���|�[�Y����
1110     */
1111     void FLogPause(BOOL Pause)
1112 maya 3227 {
1113 zmatsuo 8899 if (LogVar == NULL) {
1114     return;
1115     }
1116 zmatsuo 8900 LogVar->IsPause = Pause;
1117 zmatsuo 8901 LogVar->FLogDlg->ChangeButton(Pause);
1118 maya 3227 }
1119    
1120 zmatsuo 8857 /**
1121 zmatsuo 8852 * ���O���[�e�[�g������
1122     * ���O���T�C�Y��<size>�o�C�g���������������A���[�e�[�V��������������������
1123     */
1124 zmatsuo 8858 void FLogRotateSize(size_t size)
1125 zmatsuo 8852 {
1126     if (LogVar == NULL) {
1127     return;
1128     }
1129     LogVar->RotateMode = ROTATE_SIZE;
1130 zmatsuo 8912 LogVar->RotateSize = (LONG)size;
1131 zmatsuo 8852 }
1132    
1133     /**
1134     * ���O���[�e�[�g������
1135     * ���O�t�@�C������������������
1136     */
1137 zmatsuo 8858 void FLogRotateRotate(int step)
1138 zmatsuo 8852 {
1139     if (LogVar == NULL) {
1140     return;
1141     }
1142     LogVar->RotateStep = step;
1143     }
1144    
1145     /**
1146     * ���O���[�e�[�g������
1147     * ���[�e�[�V���������~
1148     */
1149 zmatsuo 8858 void FLogRotateHalt(void)
1150 zmatsuo 8852 {
1151     if (LogVar == NULL) {
1152     return;
1153     }
1154     LogVar->RotateMode = ROTATE_NONE;
1155     LogVar->RotateSize = 0;
1156     LogVar->RotateStep = 0;
1157     }
1158    
1159     static INT_PTR CALLBACK OnCommentDlgProc(HWND hDlgWnd, UINT msg, WPARAM wp, LPARAM lp)
1160     {
1161     static const DlgTextInfo TextInfos[] = {
1162     { 0, "DLG_COMMENT_TITLE" },
1163     { IDOK, "BTN_OK" }
1164     };
1165    
1166     switch (msg) {
1167     case WM_INITDIALOG:
1168     // �G�f�B�b�g�R���g���[�����t�H�[�J�X��������
1169     SetFocus(GetDlgItem(hDlgWnd, IDC_EDIT_COMMENT));
1170     SetDlgTexts(hDlgWnd, TextInfos, _countof(TextInfos), ts.UILanguageFile);
1171     return FALSE;
1172    
1173     case WM_COMMAND:
1174     switch (LOWORD(wp)) {
1175 zmatsuo 8901 case IDOK: {
1176 zmatsuo 8905 size_t len = _SendDlgItemMessageW(hDlgWnd, IDC_EDIT_COMMENT, WM_GETTEXTLENGTH, 0, 0);
1177 zmatsuo 8907 len += 1;
1178     wchar_t *buf = (wchar_t *)malloc(len * sizeof(wchar_t));
1179 zmatsuo 8911 _GetDlgItemTextW(hDlgWnd, IDC_EDIT_COMMENT, buf, (int)len);
1180 zmatsuo 8905 FLogWriteStr(buf);
1181     FLogWriteStr(L"\n"); // TODO ���s�R�[�h
1182     free(buf);
1183 zmatsuo 8852 TTEndDialog(hDlgWnd, IDOK);
1184     break;
1185 zmatsuo 8901 }
1186 zmatsuo 8852 default:
1187     return FALSE;
1188     }
1189     break;
1190 zmatsuo 8907
1191 zmatsuo 8852 case WM_CLOSE:
1192     TTEndDialog(hDlgWnd, 0);
1193     return TRUE;
1194    
1195     default:
1196     return FALSE;
1197     }
1198     return TRUE;
1199     }
1200    
1201 zmatsuo 8905 /**
1202     * ���O�t�@�C�����R�����g���������� (2004.8.6 yutaka)
1203     */
1204 zmatsuo 8858 void FLogAddCommentDlg(HINSTANCE hInst, HWND hWnd)
1205 zmatsuo 8852 {
1206 zmatsuo 8905 if (LogVar == NULL) {
1207     return;
1208     }
1209 zmatsuo 8852 TTDialogBox(hInst, MAKEINTRESOURCE(IDD_COMMENT_DIALOG),
1210     HVTWin, OnCommentDlgProc);
1211     }
1212    
1213 zmatsuo 8858 void FLogClose(void)
1214 zmatsuo 8852 {
1215 zmatsuo 8897 if (LogVar == NULL) {
1216     return;
1217     }
1218    
1219     FileTransEnd_();
1220 zmatsuo 8852 }
1221    
1222 zmatsuo 8897 /**
1223     * ���O���I�[�v������
1224     * @param[in] fname ���O�t�@�C����, CreateFile()���n������
1225     *
1226     * ���O�t�@�C������strftime���W�J�������s���������B
1227     * FLogGetLogFilename() �� FLogOpenDialog() ��
1228     * �t�@�C�����������������B
1229     */
1230 zmatsuo 8912 BOOL FLogOpen(const wchar_t *fname, LogCode_t code, BOOL bom)
1231 zmatsuo 8852 {
1232     BOOL ret;
1233    
1234 zmatsuo 8894 if (LogVar != NULL) {
1235 zmatsuo 8852 return FALSE;
1236     }
1237 zmatsuo 8897 if ((FileLog) || (BinLog)) return FALSE;
1238 zmatsuo 8852
1239 zmatsuo 8897 //
1240     PFileVar fv = (PFileVar)malloc(sizeof(TFileVar));
1241     if (fv == NULL) {
1242     return FALSE;
1243     }
1244     LogVar = fv;
1245     memset(fv, 0, sizeof(TFileVar));
1246 zmatsuo 8901 fv->FileHandle = INVALID_HANDLE_VALUE;
1247     fv->LogThread = INVALID_HANDLE_VALUE;
1248     fv->eLineEnd = Line_LineHead;
1249 zmatsuo 8897
1250 zmatsuo 8912 fv->log_code = code;
1251     fv->bom = bom;
1252 zmatsuo 8901 ret = LogStart(fname);
1253     if (ret == FALSE) {
1254     FileTransEnd_();
1255     }
1256 zmatsuo 8897
1257 zmatsuo 8852 return ret;
1258     }
1259    
1260 zmatsuo 8858 BOOL FLogIsOpend(void)
1261 zmatsuo 8852 {
1262     return LogVar != NULL;
1263     }
1264    
1265 zmatsuo 8904 BOOL FLogIsOpendText(void)
1266     {
1267     return LogVar != NULL && FileLog;
1268     }
1269    
1270     BOOL FLogIsOpendBin(void)
1271     {
1272     return LogVar != NULL && BinLog;
1273     }
1274    
1275 zmatsuo 8905 void FLogWriteStr(const wchar_t *str)
1276 zmatsuo 8852 {
1277 zmatsuo 8905 if (LogVar != NULL) {
1278 zmatsuo 8907 OutputStr(str);
1279 zmatsuo 8852 }
1280     }
1281    
1282 zmatsuo 8858 void FLogInfo(char *param_ptr, size_t param_len)
1283 zmatsuo 8852 {
1284     if (LogVar) {
1285     param_ptr[0] = '0'
1286     + (ts.LogBinary != 0)
1287     + ((ts.Append != 0) << 1)
1288     + ((ts.LogTypePlainText != 0) << 2)
1289     + ((ts.LogTimestamp != 0) << 3)
1290     + ((ts.LogHideDialog != 0) << 4);
1291 zmatsuo 8899 char *filenameU8 = ToU8W(LogVar->FullName);
1292     strncpy_s(param_ptr + 1, param_len - 1, filenameU8, _TRUNCATE);
1293     free(filenameU8);
1294 zmatsuo 8852 }
1295     else {
1296     param_ptr[0] = '0' - 1;
1297     param_ptr[1] = 0;
1298     }
1299     }
1300    
1301     /**
1302     * ���������O�t�@�C����������
1303     */
1304 zmatsuo 8899 const wchar_t *FLogGetFilename(void)
1305 zmatsuo 8852 {
1306     if (LogVar == NULL) {
1307     return NULL;
1308     }
1309     return LogVar->FullName;
1310     }
1311    
1312     /**
1313     * ���O�_�C�A���O���J��
1314 zmatsuo 8863 * @param[in,out] info.filename �t�@�C���������l
1315     * OK���A�t�@�C�����A�s�v����������free()��������
1316 zmatsuo 8852 * @retval TRUE [ok] ����������
1317     * @retval FALSE �L�����Z��������
1318     */
1319 zmatsuo 8863 BOOL FLogOpenDialog(HINSTANCE hInst, HWND hWnd, FLogDlgInfo_t *info)
1320 zmatsuo 8852 {
1321 zmatsuo 8863 LogDlgWork_t *work = (LogDlgWork_t *)calloc(sizeof(LogDlgWork_t), 1);
1322 zmatsuo 8899 wchar_t *srcfnameW = FLogGetLogFilename(info->filename);
1323 zmatsuo 8863 work->info = info;
1324     work->info->filename = srcfnameW;
1325     work->pts = &ts;
1326 zmatsuo 8852 INT_PTR ret = TTDialogBoxParam(
1327     hInst, MAKEINTRESOURCE(IDD_LOGDLG),
1328 zmatsuo 8863 hWnd, LogFnHook, (LPARAM)work);
1329     free(srcfnameW);
1330     free(work);
1331 zmatsuo 8852 return ret == IDOK ? TRUE : FALSE;
1332     }
1333    
1334     /**
1335     * ���O�t�@�C����������
1336     * ���O�t�@�C�����p���C�����s��
1337     * - strftime() ���������t�W�J
1338     * - �������������O�t�@�C���t�H���_������
1339     * - �z�X�g��,�|�[�g�����W�J
1340     *
1341     * @param[in] log_filename �t�@�C����(����/��������������ok)
1342     * NULL�������f�t�H���g�t�@�C����������
1343     * strftime�`��ok
1344     * @return �t���p�X�t�@�C����
1345     * �s�v���������� free() ��������
1346     */
1347 zmatsuo 8899 wchar_t *FLogGetLogFilename(const wchar_t *log_filename)
1348 zmatsuo 8852 {
1349     // �t�H���_
1350     char FileDirExpanded[MAX_PATH];
1351 zmatsuo 8899 const char *logdir;
1352 zmatsuo 8852 if (strlen(ts.LogDefaultPath) > 0) {
1353     logdir = ts.LogDefaultPath;
1354     }
1355     else if (strlen(ts.FileDir) > 0) {
1356     ExpandEnvironmentStrings(ts.FileDir, FileDirExpanded, sizeof(FileDirExpanded));
1357     logdir = FileDirExpanded;
1358     }
1359     else {
1360     logdir = ts.HomeDir;
1361     }
1362    
1363     // ���������t�@�C����
1364     char base_name[MAX_PATH];
1365     if (log_filename == NULL) {
1366     strncpy_s(base_name, _countof(base_name), ts.LogDefaultName, _TRUNCATE);
1367     }
1368     else {
1369 zmatsuo 8899 char *filenameA = ToCharW(log_filename);
1370     strncpy_s(base_name, _countof(base_name), filenameA, _TRUNCATE);
1371     free(filenameA);
1372 zmatsuo 8852 }
1373    
1374     // �t���p�X��
1375     char full_path[MAX_PATH];
1376     ConvFName(logdir, base_name, sizeof(base_name), "", full_path, sizeof(full_path));
1377     ParseStrftimeFileName(full_path, sizeof(full_path));
1378     ConvertLogname(full_path, sizeof(full_path));
1379    
1380 zmatsuo 8899 return ToWcharA(full_path);
1381 zmatsuo 8852 }
1382 zmatsuo 8857
1383     BOOL FLogIsPause()
1384     {
1385 zmatsuo 8900 if (LogVar == NULL) {
1386     return FALSE;
1387     }
1388     return LogVar->IsPause;
1389 zmatsuo 8857 }
1390    
1391     void FLogWindow(int nCmdShow)
1392     {
1393 zmatsuo 8900 if (LogVar == NULL) {
1394     return;
1395     }
1396 zmatsuo 8857
1397 zmatsuo 8901 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1398 zmatsuo 8857 ShowWindow(HWndLog, nCmdShow);
1399     if (nCmdShow == SW_RESTORE) {
1400     // �g���X�^�C�� WS_EX_NOACTIVATE ��������������
1401     SetForegroundWindow(HWndLog);
1402     }
1403     }
1404    
1405     void FLogShowDlg(void)
1406     {
1407 zmatsuo 8900 if (LogVar == NULL) {
1408     return;
1409     }
1410 zmatsuo 8901 HWND HWndLog = LogVar->FLogDlg->m_hWnd;
1411     ShowWindow(HWndLog, SW_SHOWNORMAL);
1412     SetForegroundWindow(HWndLog);
1413 zmatsuo 8857 }
1414 zmatsuo 8897
1415     /**
1416     * ���O��1byte��������
1417     * LogPut1() ������?
1418     */
1419     //void Log1Bin(PComVar cv, BYTE b)
1420     static void Log1Bin(BYTE b)
1421     {
1422 zmatsuo 9103 if (LogVar->IsPause || ProtoGetProtoFlag()) {
1423 zmatsuo 8897 return;
1424     }
1425 zmatsuo 8906 if (cv_BinSkip > 0) {
1426     cv_BinSkip--;
1427 zmatsuo 8897 return;
1428     }
1429 zmatsuo 8906 cv_BinBuf[cv_BinPtr] = b;
1430     cv_BinPtr++;
1431     if (cv_BinPtr>=InBuffSize) {
1432     cv_BinPtr = cv_BinPtr-InBuffSize;
1433 zmatsuo 8897 }
1434 zmatsuo 8906 if (cv_BCount>=InBuffSize) {
1435     cv_BCount = InBuffSize;
1436     cv_BStart = cv_BinPtr;
1437 zmatsuo 8897 }
1438     else {
1439 zmatsuo 8906 cv_BCount++;
1440 zmatsuo 8897 }
1441     }
1442    
1443     static void LogBinSkip(int add)
1444     {
1445 zmatsuo 8906 if (cv_BinBuf != NULL) {
1446     cv_BinSkip += add;
1447 zmatsuo 8897 }
1448     }
1449    
1450     /**
1451     * ���O�o�b�t�@���������������f�[�^���o�C�g��������
1452     */
1453     int FLogGetCount(void)
1454     {
1455     if (FileLog) {
1456 zmatsuo 8906 return cv_LCount;
1457 zmatsuo 8897 }
1458     if (BinLog) {
1459 zmatsuo 8906 return cv_BCount;
1460 zmatsuo 8897 }
1461     return 0;
1462     }
1463    
1464     /**
1465 zmatsuo 8906 * ���O�o�b�t�@�������o�C�g��������
1466     */
1467     int FLogGetFreeCount(void)
1468     {
1469     if (FileLog) {
1470     return InBuffSize - cv_LCount;
1471     }
1472     if (BinLog) {
1473     return InBuffSize - cv_BCount;
1474     }
1475     return 0;
1476     }
1477    
1478     /**
1479 zmatsuo 8897 * �o�b�t�@�������O���t�@�C������������
1480     */
1481     void FLogWriteFile(void)
1482     {
1483 zmatsuo 8906 if (cv_LogBuf!=NULL)
1484 zmatsuo 8897 {
1485     if (FileLog) {
1486     LogToFile();
1487     }
1488     }
1489    
1490 zmatsuo 8906 if (cv_BinBuf!=NULL)
1491 zmatsuo 8897 {
1492     if (BinLog) {
1493     LogToFile();
1494     }
1495     }
1496     }
1497 zmatsuo 8904
1498     void FLogPutUTF32(unsigned int u32)
1499     {
1500     PFileVar fv = LogVar;
1501     size_t i;
1502 zmatsuo 8906 BOOL log_available = (cv_LogBuf != 0);
1503 zmatsuo 8904
1504     if (!log_available) {
1505 zmatsuo 8905 // ���O�����o��������
1506 zmatsuo 8904 return;
1507     }
1508    
1509 zmatsuo 8912 // �s����?(���s���o����������)
1510 zmatsuo 8910 if (ts.LogTimestamp && fv->eLineEnd) {
1511 zmatsuo 8912 // �^�C���X�^���v���o��
1512 zmatsuo 8910 fv->eLineEnd = Line_Other; /* clear endmark*/
1513     wchar_t* strtime = TimeStampStr();
1514     FLogWriteStr(strtime);
1515     free(strtime);
1516     }
1517    
1518 zmatsuo 8904 switch(fv->log_code) {
1519 zmatsuo 8912 case LOG_UTF8: {
1520 zmatsuo 8904 // UTF-8
1521     char u8_buf[4];
1522     size_t u8_len = UTF32ToUTF8(u32, u8_buf, _countof(u8_buf));
1523     for (i = 0; i < u8_len; i++) {
1524     BYTE b = u8_buf[i];
1525     LogPut1(b);
1526     }
1527     break;
1528     }
1529 zmatsuo 8912 case LOG_UTF16LE:
1530     case LOG_UTF16BE: {
1531 zmatsuo 8904 // UTF-16
1532     wchar_t u16[2];
1533     size_t u16_len = UTF32ToUTF16(u32, u16, _countof(u16));
1534     size_t i;
1535     for (i = 0; i < u16_len; i++) {
1536 zmatsuo 8912 if (fv->log_code == LOG_UTF16LE) {
1537 zmatsuo 8904 // UTF-16LE
1538     LogPut1(u16[i] & 0xff);
1539     LogPut1((u16[i] >> 8) & 0xff);
1540     }
1541     else {
1542     // UTF-16BE
1543     LogPut1((u16[i] >> 8) & 0xff);
1544     LogPut1(u16[i] & 0xff);
1545     }
1546     }
1547     }
1548     }
1549 zmatsuo 8910
1550     if (u32 == 0x0a) {
1551     fv->eLineEnd = Line_LineHead; /* set endmark*/
1552     }
1553 zmatsuo 8904 }
1554    
1555 zmatsuo 8912 static void FLogOutputBOM(void)
1556 zmatsuo 8904 {
1557     PFileVar fv = LogVar;
1558 zmatsuo 8910 DWORD wrote;
1559 zmatsuo 8904
1560     switch(fv->log_code) {
1561 zmatsuo 8910 case 0: {
1562 zmatsuo 8904 // UTF-8
1563 zmatsuo 8910 const char *bom = "\xef\xbb\xbf";
1564     WriteFile(LogVar->FileHandle, bom, 3, &wrote, NULL);
1565     LogVar->ByteCount += 3;
1566 zmatsuo 8904 break;
1567 zmatsuo 8910 }
1568     case 1: {
1569 zmatsuo 8904 // UTF-16LE
1570 zmatsuo 8910 const char *bom = "\xff\xfe";
1571     WriteFile(LogVar->FileHandle, bom, 2, &wrote, NULL);
1572     LogVar->ByteCount += 2;
1573 zmatsuo 8904 break;
1574 zmatsuo 8910 }
1575     case 2: {
1576 zmatsuo 8904 // UTF-16BE
1577 zmatsuo 8910 const char *bom = "\xfe\xff";
1578     WriteFile(LogVar->FileHandle, bom, 2, &wrote, NULL);
1579     LogVar->ByteCount += 2;
1580 zmatsuo 8904 break;
1581 zmatsuo 8910 }
1582 zmatsuo 8904 default:
1583     break;
1584     }
1585     }
1586    
1587 zmatsuo 8907 static void OutputStr(const wchar_t *str)
1588     {
1589     size_t len;
1590    
1591     assert(str != NULL);
1592    
1593     len = wcslen(str);
1594     while(*str != 0) {
1595     unsigned int u32;
1596     size_t u16_len = UTF16ToUTF32(str, len, &u32);
1597     switch (u16_len) {
1598     case 0:
1599     default:
1600     // ������������
1601     str++;
1602     len--;
1603     break;
1604     case 1:
1605     case 2: {
1606     FLogPutUTF32(u32);
1607     str += u16_len;
1608     len -= u16_len;
1609     break;
1610     }
1611     }
1612     }
1613     }

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