| 1 |
/* |
| 2 |
* (C) 2020- TeraTerm Project |
| 3 |
* All rights reserved. |
| 4 |
* |
| 5 |
* Redistribution and use in source and binary forms, with or without |
| 6 |
* modification, are permitted provided that the following conditions |
| 7 |
* are met: |
| 8 |
* |
| 9 |
* 1. Redistributions of source code must retain the above copyright |
| 10 |
* notice, this list of conditions and the following disclaimer. |
| 11 |
* 2. Redistributions in binary form must reproduce the above copyright |
| 12 |
* notice, this list of conditions and the following disclaimer in the |
| 13 |
* documentation and/or other materials provided with the distribution. |
| 14 |
* 3. The name of the author may not be used to endorse or promote products |
| 15 |
* derived from this software without specific prior written permission. |
| 16 |
* |
| 17 |
* THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR |
| 18 |
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
| 19 |
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
| 20 |
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT, |
| 21 |
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
| 22 |
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 23 |
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 24 |
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 25 |
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| 26 |
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 27 |
*/ |
| 28 |
|
| 29 |
/* TERATERM.EXE, log routines */ |
| 30 |
#include <stdio.h> |
| 31 |
#if !defined(_CRTDBG_MAP_ALLOC) |
| 32 |
#define _CRTDBG_MAP_ALLOC |
| 33 |
#endif |
| 34 |
#include <stdlib.h> |
| 35 |
#include <crtdbg.h> |
| 36 |
#include <process.h> |
| 37 |
#include <windows.h> |
| 38 |
#include <htmlhelp.h> |
| 39 |
#include <assert.h> |
| 40 |
|
| 41 |
#include "teraterm.h" |
| 42 |
#include "tttypes.h" |
| 43 |
#include "ftdlg.h" |
| 44 |
#include "ttwinman.h" |
| 45 |
#include "commlib.h" |
| 46 |
#include "ttcommon.h" |
| 47 |
#include "ttlib.h" |
| 48 |
#include "dlglib.h" |
| 49 |
#include "vtterm.h" |
| 50 |
#include "ftlib.h" |
| 51 |
#include "buffer.h" |
| 52 |
#include "helpid.h" |
| 53 |
#include "codeconv.h" |
| 54 |
#include "asprintf.h" |
| 55 |
#include "win32helper.h" |
| 56 |
|
| 57 |
#include "filesys_log_res.h" |
| 58 |
#include "filesys_log.h" |
| 59 |
#include "filesys.h" // for ProtoGetProtoFlag() |
| 60 |
|
| 61 |
#define TitLog L"Log" |
| 62 |
|
| 63 |
/* |
| 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 |
typedef struct { |
| 74 |
wchar_t *FullName; |
| 75 |
|
| 76 |
HANDLE FileHandle; |
| 77 |
LONG FileSize, ByteCount; |
| 78 |
|
| 79 |
DWORD StartTime; |
| 80 |
|
| 81 |
enum enumLineEnd eLineEnd; |
| 82 |
|
| 83 |
// log rotate |
| 84 |
int RotateMode; // enum rotate_mode RotateMode; |
| 85 |
LONG RotateSize; |
| 86 |
int RotateStep; |
| 87 |
|
| 88 |
HANDLE LogThread; |
| 89 |
DWORD LogThreadId; |
| 90 |
HANDLE LogThreadEvent; |
| 91 |
|
| 92 |
BOOL IsPause; |
| 93 |
|
| 94 |
PFileTransDlg FLogDlg; |
| 95 |
|
| 96 |
LogCode_t log_code; |
| 97 |
BOOL bom; |
| 98 |
|
| 99 |
BOOL FileLog; |
| 100 |
BOOL BinLog; |
| 101 |
} TFileVar; |
| 102 |
typedef TFileVar *PFileVar; |
| 103 |
|
| 104 |
static PFileVar LogVar = NULL; |
| 105 |
|
| 106 |
static PCHAR cv_LogBuf; |
| 107 |
static int cv_LogPtr, cv_LStart, cv_LCount; |
| 108 |
static PCHAR cv_BinBuf; |
| 109 |
static int cv_BinPtr, cv_BStart, cv_BCount; |
| 110 |
static int cv_BinSkip; |
| 111 |
|
| 112 |
// �x�����������p�X���b�h�����b�Z�[�W |
| 113 |
#define WM_DPC_LOGTHREAD_SEND (WM_APP + 1) |
| 114 |
|
| 115 |
static void Log1Bin(BYTE b); |
| 116 |
static void LogBinSkip(int add); |
| 117 |
static BOOL CreateLogBuf(void); |
| 118 |
static BOOL CreateBinBuf(void); |
| 119 |
void LogPut1(BYTE b); |
| 120 |
static void OutputStr(const wchar_t *str); |
| 121 |
static void LogToFile(PFileVar fv); |
| 122 |
static void FLogOutputBOM(PFileVar fv); |
| 123 |
|
| 124 |
static BOOL OpenFTDlg_(PFileVar fv) |
| 125 |
{ |
| 126 |
PFileTransDlg FTDlg = new CFileTransDlg(); |
| 127 |
if (FTDlg == NULL) { |
| 128 |
return FALSE; |
| 129 |
} |
| 130 |
|
| 131 |
wchar_t *DlgCaption; |
| 132 |
wchar_t uimsg[MAX_UIMSG]; |
| 133 |
get_lang_msgW("FILEDLG_TRANS_TITLE_LOG", uimsg, _countof(uimsg), TitLog, ts.UILanguageFile); |
| 134 |
aswprintf(&DlgCaption, L"Tera Term: %s", uimsg); |
| 135 |
|
| 136 |
CFileTransDlg::Info info; |
| 137 |
info.UILanguageFile = ts.UILanguageFile; |
| 138 |
info.OpId = CFileTransDlg::OpLog; |
| 139 |
info.DlgCaption = DlgCaption; |
| 140 |
info.FileName = NULL; |
| 141 |
info.FullName = fv->FullName; |
| 142 |
info.HideDialog = ts.LogHideDialog ? TRUE : FALSE; |
| 143 |
info.HMainWin = HVTWin; |
| 144 |
FTDlg->Create(hInst, &info); |
| 145 |
FTDlg->RefreshNum(0, fv->FileSize, fv->ByteCount); |
| 146 |
|
| 147 |
fv->FLogDlg = FTDlg; |
| 148 |
|
| 149 |
free(DlgCaption); |
| 150 |
return TRUE; |
| 151 |
} |
| 152 |
|
| 153 |
/** |
| 154 |
* �t�@�C�������������u������,���O�p |
| 155 |
* �����������u�������� |
| 156 |
* &h �z�X�g�����u�� |
| 157 |
* &p TCP�|�[�g�������u�� |
| 158 |
* &u ���O�I���������[�U�� |
| 159 |
* |
| 160 |
* @param pcv |
| 161 |
* @param src �u���������O��������(�t�@�C����) |
| 162 |
* @return �u������������������ |
| 163 |
* �s�v����������free()�������� |
| 164 |
*/ |
| 165 |
static wchar_t *ConvertLognameW(const TComVar *pcv, const wchar_t *src) |
| 166 |
{ |
| 167 |
const TTTSet *pts = pcv->ts; |
| 168 |
size_t dest_len = wcslen(src) + 1; |
| 169 |
wchar_t *dest = (wchar_t *)malloc(sizeof(wchar_t) * dest_len); |
| 170 |
|
| 171 |
const wchar_t *s = src; |
| 172 |
size_t i = 0; |
| 173 |
|
| 174 |
while(*s != '\0') { |
| 175 |
if (*s == '&' && *(s+1) != '\0') { |
| 176 |
wchar_t c = *(s+1); |
| 177 |
wchar_t *add_text = NULL; |
| 178 |
switch (c) { |
| 179 |
case 'h': |
| 180 |
s += 2; |
| 181 |
if (pcv->Open) { |
| 182 |
switch(pcv->PortType) { |
| 183 |
case IdTCPIP: { |
| 184 |
// �z�X�g����IPv6�A�h���X�����A�t�@�C�������g�p������������(:)�����������u�� |
| 185 |
wchar_t *host = ToWcharA(pts->HostName); |
| 186 |
wchar_t *host_fix = replaceInvalidFileNameCharW(host, '_'); |
| 187 |
free(host); |
| 188 |
add_text = host_fix; |
| 189 |
break; |
| 190 |
} |
| 191 |
case IdSerial: { |
| 192 |
wchar_t *port; |
| 193 |
aswprintf(&port, L"COM%d", ts.ComPort); |
| 194 |
add_text = port; |
| 195 |
break; |
| 196 |
} |
| 197 |
default: |
| 198 |
; |
| 199 |
} |
| 200 |
} |
| 201 |
break; |
| 202 |
case 'p': |
| 203 |
s += 2; |
| 204 |
if (pcv->Open) { |
| 205 |
if (pcv->PortType == IdTCPIP) { |
| 206 |
wchar_t *port; |
| 207 |
aswprintf(&port, L"%d", ts.TCPPort); |
| 208 |
add_text = port; |
| 209 |
} |
| 210 |
} |
| 211 |
break; |
| 212 |
case 'u': { |
| 213 |
s += 2; |
| 214 |
wchar_t user[256 + 1]; // 256=UNLEN |
| 215 |
DWORD l = _countof(user); |
| 216 |
if (GetUserNameW(user, &l) != 0) { |
| 217 |
add_text = _wcsdup(user); |
| 218 |
} |
| 219 |
break; |
| 220 |
} |
| 221 |
default: |
| 222 |
// pass '&' |
| 223 |
s++; |
| 224 |
add_text = NULL; |
| 225 |
break; |
| 226 |
} |
| 227 |
|
| 228 |
if (add_text != NULL) { |
| 229 |
size_t l = wcslen(add_text); |
| 230 |
dest_len += l; |
| 231 |
dest = (wchar_t *)realloc(dest, sizeof(wchar_t) * dest_len); |
| 232 |
wcscpy(&dest[i], add_text); |
| 233 |
free(add_text); |
| 234 |
i += l; |
| 235 |
} |
| 236 |
} |
| 237 |
else { |
| 238 |
dest[i] = *s++; |
| 239 |
i++; |
| 240 |
} |
| 241 |
} |
| 242 |
dest[i] = 0; |
| 243 |
return dest; |
| 244 |
} |
| 245 |
|
| 246 |
static void FixLogOption(void) |
| 247 |
{ |
| 248 |
if (ts.LogBinary) { |
| 249 |
ts.LogTypePlainText = false; |
| 250 |
ts.LogTimestamp = false; |
| 251 |
} |
| 252 |
} |
| 253 |
|
| 254 |
|
| 255 |
// �X���b�h���I�����t�@�C�����N���[�Y |
| 256 |
static void CloseFileSync(PFileVar fv) |
| 257 |
{ |
| 258 |
BOOL ret; |
| 259 |
|
| 260 |
if (fv->FileHandle == INVALID_HANDLE_VALUE) { |
| 261 |
return; |
| 262 |
} |
| 263 |
|
| 264 |
if (fv->LogThread != INVALID_HANDLE_VALUE) { |
| 265 |
// �X���b�h���I������ |
| 266 |
ret = PostThreadMessage(fv->LogThreadId, WM_QUIT, 0, 0); |
| 267 |
if (ret != 0) { |
| 268 |
// �X���b�h�L���[���G���L���[���������������������������s���B |
| 269 |
WaitForSingleObject(fv->LogThread, INFINITE); |
| 270 |
} |
| 271 |
else { |
| 272 |
//DWORD code = GetLastError(); |
| 273 |
} |
| 274 |
CloseHandle(fv->LogThread); |
| 275 |
fv->LogThread = INVALID_HANDLE_VALUE; |
| 276 |
} |
| 277 |
CloseHandle(fv->FileHandle); |
| 278 |
fv->FileHandle = INVALID_HANDLE_VALUE; |
| 279 |
} |
| 280 |
|
| 281 |
// �x�����������p�X���b�h |
| 282 |
static unsigned _stdcall DeferredLogWriteThread(void *arg) |
| 283 |
{ |
| 284 |
MSG msg; |
| 285 |
PFileVar fv = (PFileVar)arg; |
| 286 |
PCHAR buf; |
| 287 |
DWORD buflen; |
| 288 |
DWORD wrote; |
| 289 |
|
| 290 |
PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE); |
| 291 |
|
| 292 |
// �X���b�h�L���[���������I�������������X���b�h�����������m�����B |
| 293 |
if (fv->LogThreadEvent != NULL) { |
| 294 |
SetEvent(fv->LogThreadEvent); |
| 295 |
} |
| 296 |
|
| 297 |
while (GetMessage(&msg, NULL, 0, 0) > 0) { |
| 298 |
switch (msg.message) { |
| 299 |
case WM_DPC_LOGTHREAD_SEND: |
| 300 |
buf = (PCHAR)msg.wParam; |
| 301 |
buflen = (DWORD)msg.lParam; |
| 302 |
WriteFile(fv->FileHandle, buf, buflen, &wrote, NULL); |
| 303 |
free(buf); // ���������������� |
| 304 |
break; |
| 305 |
|
| 306 |
case WM_QUIT: |
| 307 |
goto end; |
| 308 |
break; |
| 309 |
} |
| 310 |
} |
| 311 |
|
| 312 |
end: |
| 313 |
_endthreadex(0); |
| 314 |
return (0); |
| 315 |
} |
| 316 |
|
| 317 |
// �x�����������p�X���b�h���N�����B |
| 318 |
// (2013.4.19 yutaka) |
| 319 |
// DeferredLogWriteThread �X���b�h���N�������A�X���b�h�L���[�����������������O���A |
| 320 |
// ���O�t�@�C�����N���[�Y(CloseFileSync)���s���������A�G���L���[�����s���A�f�b�h���b�N |
| 321 |
// �����������������C�������B |
| 322 |
// �X���b�h�����������s�������A���O�����C�x���g�I�u�W�F�N�g���g�����A�X���b�h�L���[�� |
| 323 |
// ���������������������������������B���O�t���C�x���g�I�u�W�F�N�g���g���������A |
| 324 |
// �V�X�e��(Windows OS)�������j�[�N�����O�������K�v�������B |
| 325 |
// (2016.9.23 yutaka) |
| 326 |
static void StartThread(PFileVar fv) |
| 327 |
{ |
| 328 |
unsigned tid; |
| 329 |
fv->LogThreadEvent = CreateEvent(NULL, TRUE, FALSE, NULL); |
| 330 |
fv->LogThread = (HANDLE)_beginthreadex(NULL, 0, DeferredLogWriteThread, fv, 0, &tid); |
| 331 |
fv->LogThreadId = tid; |
| 332 |
if (fv->LogThreadEvent != NULL) { |
| 333 |
WaitForSingleObject(fv->LogThreadEvent, INFINITE); |
| 334 |
CloseHandle(fv->LogThreadEvent); |
| 335 |
} |
| 336 |
} |
| 337 |
|
| 338 |
/** |
| 339 |
* �_�C�A���O�����e�� ts ���������� |
| 340 |
* |
| 341 |
* TODO |
| 342 |
* �_�C�A���O�����������l�������I�������� |
| 343 |
* ����������������������������������������������? |
| 344 |
*/ |
| 345 |
static void SetLogFlags(HWND Dialog) |
| 346 |
{ |
| 347 |
WORD BinFlag, val; |
| 348 |
|
| 349 |
GetRB(Dialog, &BinFlag, IDC_FOPTBIN, IDC_FOPTBIN); |
| 350 |
ts.LogBinary = BinFlag; |
| 351 |
|
| 352 |
GetRB(Dialog, &val, IDC_APPEND, IDC_APPEND); |
| 353 |
ts.Append = val; |
| 354 |
|
| 355 |
if (!BinFlag) { |
| 356 |
GetRB(Dialog, &val, IDC_PLAINTEXT, IDC_PLAINTEXT); |
| 357 |
ts.LogTypePlainText = val; |
| 358 |
|
| 359 |
GetRB(Dialog, &val, IDC_TIMESTAMP, IDC_TIMESTAMP); |
| 360 |
ts.LogTimestamp = val; |
| 361 |
} |
| 362 |
|
| 363 |
GetRB(Dialog, &val, IDC_HIDEDIALOG, IDC_HIDEDIALOG); |
| 364 |
ts.LogHideDialog = val; |
| 365 |
|
| 366 |
GetRB(Dialog, &val, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST); |
| 367 |
ts.LogAllBuffIncludedInFirst = val; |
| 368 |
|
| 369 |
ts.LogTimestampType = (WORD)(GetCurSel(Dialog, IDC_TIMESTAMPTYPE) - 1); |
| 370 |
} |
| 371 |
|
| 372 |
/** |
| 373 |
* ���O�t�@�C���`�F�b�N |
| 374 |
* |
| 375 |
* @param[in] filename |
| 376 |
* @param[out] exist TURE/FALSE |
| 377 |
* @param[out] bom 0 no BOM (or file not exist) |
| 378 |
* 1 UTF-8 |
| 379 |
* 2 UTF-16LE |
| 380 |
* 3 UTF-16BE |
| 381 |
*/ |
| 382 |
static void CheckLogFile(const wchar_t *filename, BOOL *exist, int *bom) |
| 383 |
{ |
| 384 |
*exist = FALSE; |
| 385 |
*bom = 0; |
| 386 |
|
| 387 |
// �t�@�C������������? |
| 388 |
DWORD logdir = GetFileAttributesW(filename); |
| 389 |
if ((logdir != INVALID_FILE_ATTRIBUTES) && ((logdir & FILE_ATTRIBUTE_DIRECTORY) == 0)) { |
| 390 |
// �t�@�C���������� |
| 391 |
*exist = TRUE; |
| 392 |
|
| 393 |
// BOM�L��/�����`�F�b�N |
| 394 |
FILE *fp = _wfopen(filename, L"rb"); |
| 395 |
if (fp != NULL) { |
| 396 |
unsigned char tmp[4]; |
| 397 |
size_t l = fread(tmp, 1, sizeof(tmp), fp); |
| 398 |
fclose(fp); |
| 399 |
if (l < 2) { |
| 400 |
*bom = 0; |
| 401 |
} else if (l >= 2 && tmp[0] == 0xff && tmp[1] == 0xfe) { |
| 402 |
// UTF-16LE |
| 403 |
*bom = 2; |
| 404 |
} else if (l >= 2 && tmp[0] == 0xfe && tmp[1] == 0xff) { |
| 405 |
// UTF-16BE |
| 406 |
*bom = 3; |
| 407 |
} else if (l >= 3 && tmp[0] == 0xef && tmp[1] == 0xbb && tmp[2] == 0xbf) { |
| 408 |
// UTF-8 |
| 409 |
*bom = 1; |
| 410 |
} else { |
| 411 |
*bom = 0; |
| 412 |
} |
| 413 |
} |
| 414 |
} |
| 415 |
} |
| 416 |
|
| 417 |
typedef struct { |
| 418 |
FLogDlgInfo_t *info; |
| 419 |
// work |
| 420 |
BOOL file_exist; |
| 421 |
int current_bom; // ���������t�@�C�����G���R�[�f�B���O�i�t�@�C����BOM���������j |
| 422 |
TTTSet *pts; |
| 423 |
TComVar *pcv; |
| 424 |
} LogDlgWork_t; |
| 425 |
|
| 426 |
|
| 427 |
/* |
| 428 |
* Log �_�C�A���O�������AEnable/Disable �����������R���g���[�� |
| 429 |
* |
| 430 |
* - Append |
| 431 |
* �w���������t�@�C������������������ Enable |
| 432 |
* �w���������t�@�C�������������������� Disable |
| 433 |
* |
| 434 |
* - BOM, Encoding |
| 435 |
* Text ���� New/Overwrite �������� Enable |
| 436 |
* ���������������� Disable |
| 437 |
* BOM ���t�@�C�������������������������������������� |
| 438 |
* Encoding �����L�����������������A�����t�@�C�����G���R�[�f�B���O�� |
| 439 |
* �����I���_�C�A���O�����f���������A���[�U�������w������������ |
| 440 |
* |
| 441 |
* - Plain Text, Timestamp, Timestamp ���� |
| 442 |
* Text �������� Enable |
| 443 |
* Binary �������� Disable |
| 444 |
* |
| 445 |
* - Timestamp ���� |
| 446 |
* Timestamp=on �������� Enable |
| 447 |
* Timestamp=off �������� Disable |
| 448 |
*/ |
| 449 |
|
| 450 |
static void ArrangeControls(HWND Dialog, LogDlgWork_t *work, |
| 451 |
WORD Append, WORD LogBinary, |
| 452 |
WORD LogTypePlainText, WORD LogTimestamp) |
| 453 |
{ |
| 454 |
if (work->file_exist) { |
| 455 |
// Append ���W�I�{�^�����A�t�@�C�����������������L�������� |
| 456 |
EnableWindow(GetDlgItem(Dialog, IDC_APPEND), TRUE); |
| 457 |
|
| 458 |
if (Append > 0) { |
| 459 |
CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND); |
| 460 |
} |
| 461 |
else { |
| 462 |
CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE); |
| 463 |
} |
| 464 |
} |
| 465 |
else { // �t�@�C�������� -> �V�K |
| 466 |
EnableWindow(GetDlgItem(Dialog, IDC_APPEND), FALSE); |
| 467 |
CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_NEW_OVERWRITE); |
| 468 |
} |
| 469 |
|
| 470 |
if (!LogBinary && !Append) { |
| 471 |
EnableWindow(GetDlgItem(Dialog, IDC_BOM), TRUE); |
| 472 |
EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), TRUE); |
| 473 |
} |
| 474 |
else { |
| 475 |
EnableWindow(GetDlgItem(Dialog, IDC_BOM), FALSE); |
| 476 |
EnableWindow(GetDlgItem(Dialog, IDC_TEXTCODING_DROPDOWN), FALSE); |
| 477 |
} |
| 478 |
|
| 479 |
if (LogBinary) { |
| 480 |
CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTBIN); |
| 481 |
|
| 482 |
DisableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT); |
| 483 |
DisableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP); |
| 484 |
DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE); |
| 485 |
} |
| 486 |
else { |
| 487 |
CheckRadioButton(Dialog, IDC_FOPTBIN, IDC_FOPTTEXT, IDC_FOPTTEXT); |
| 488 |
|
| 489 |
EnableDlgItem(Dialog, IDC_PLAINTEXT, IDC_PLAINTEXT); |
| 490 |
EnableDlgItem(Dialog, IDC_TIMESTAMP, IDC_TIMESTAMP); |
| 491 |
EnableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE); |
| 492 |
|
| 493 |
if (LogTypePlainText) { |
| 494 |
SetRB(Dialog, 1, IDC_PLAINTEXT, IDC_PLAINTEXT); |
| 495 |
} |
| 496 |
if (LogTimestamp) { |
| 497 |
SetRB(Dialog, 1, IDC_TIMESTAMP, IDC_TIMESTAMP); |
| 498 |
} |
| 499 |
else { |
| 500 |
DisableDlgItem(Dialog, IDC_TIMESTAMPTYPE, IDC_TIMESTAMPTYPE); |
| 501 |
} |
| 502 |
} |
| 503 |
|
| 504 |
if (work->file_exist && Append) { |
| 505 |
// �����t�@�C�����G���R�[�f�B���O�����f���� |
| 506 |
int bom = work->current_bom; |
| 507 |
int cur = |
| 508 |
bom == 1 ? 0 : |
| 509 |
bom == 2 ? 1 : |
| 510 |
bom == 3 ? 2 : 0; |
| 511 |
SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, cur, 0); |
| 512 |
} |
| 513 |
} |
| 514 |
|
| 515 |
static void CheckLogFile(HWND Dialog, const wchar_t *filename, LogDlgWork_t *work) |
| 516 |
{ |
| 517 |
BOOL exist; |
| 518 |
int bom; |
| 519 |
CheckLogFile(filename, &exist, &bom); |
| 520 |
work->file_exist = exist; |
| 521 |
work->current_bom = bom; |
| 522 |
} |
| 523 |
|
| 524 |
static INT_PTR CALLBACK LogFnHook(HWND Dialog, UINT Message, WPARAM wParam, LPARAM lParam) |
| 525 |
{ |
| 526 |
static const DlgTextInfo TextInfos[] = { |
| 527 |
{ 0, "DLG_TABSHEET_TITLE_LOG" }, |
| 528 |
{ IDC_SENDFILE_FILENAME_TITLE, "DLG_FOPT_FILENAME_TITLE" }, |
| 529 |
{ IDC_FOPTTEXT, "DLG_FOPT_TEXT" }, |
| 530 |
{ IDC_FOPTBIN, "DLG_FOPT_BINARY" }, |
| 531 |
{ IDC_BOM, "DLG_FOPT_BOM" }, |
| 532 |
{ IDC_APPEND, "DLG_FOPT_APPEND" }, |
| 533 |
{ IDC_PLAINTEXT, "DLG_FOPT_PLAIN" }, |
| 534 |
{ IDC_HIDEDIALOG, "DLG_FOPT_HIDEDIALOG" }, |
| 535 |
{ IDC_ALLBUFF_INFIRST, "DLG_FOPT_ALLBUFFINFIRST" }, |
| 536 |
{ IDC_TIMESTAMP, "DLG_FOPT_TIMESTAMP" }, |
| 537 |
{ IDC_NEW_OVERWRITE, "DLG_FOPT_NEW_OVERWRITE" }, |
| 538 |
{ IDC_APPEND_GROUP, "DLG_FOPT_APPEND_LABEL" }, |
| 539 |
{ IDC_BINARY_GROUP, "DLG_FOPT_BINARY_LABEL" }, |
| 540 |
}; |
| 541 |
static const I18nTextInfo timestamp_list[] = { |
| 542 |
{ "DLG_FOPT_TIMESTAMP_LOCAL", L"Local Time" }, |
| 543 |
{ "DLG_FOPT_TIMESTAMP_UTC", L"UTC" }, |
| 544 |
{ "DLG_FOPT_TIMESTAMP_ELAPSED_LOGGING", L"Elapsed Time (Logging)" }, |
| 545 |
{ "DLG_FOPT_TIMESTAMP_ELAPSED_CONNECTION", L"Elapsed Time (Connection)" }, |
| 546 |
}; |
| 547 |
LogDlgWork_t *work = (LogDlgWork_t *)GetWindowLongPtr(Dialog, DWLP_USER); |
| 548 |
|
| 549 |
if (Message == RegisterWindowMessage(HELPMSGSTRING)) { |
| 550 |
// �R�����_�C�A���O�������w���v���b�Z�[�W���t�������� |
| 551 |
Message = WM_COMMAND; |
| 552 |
wParam = IDHELP; |
| 553 |
} |
| 554 |
switch (Message) { |
| 555 |
case WM_INITDIALOG: { |
| 556 |
work = (LogDlgWork_t *)lParam; |
| 557 |
TTTSet *pts = work->pts; |
| 558 |
SetWindowLongPtr(Dialog, DWLP_USER, (LONG_PTR)work); |
| 559 |
::DragAcceptFiles(Dialog, TRUE); |
| 560 |
|
| 561 |
SetDlgTextsW(Dialog, TextInfos, _countof(TextInfos), pts->UILanguageFileW); |
| 562 |
SetI18nListW("Tera Term", Dialog, IDC_TIMESTAMPTYPE, timestamp_list, _countof(timestamp_list), |
| 563 |
pts->UILanguageFileW, 0); |
| 564 |
|
| 565 |
SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-8"); |
| 566 |
SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16LE"); |
| 567 |
SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_ADDSTRING, 0, (LPARAM)"UTF-16BE"); |
| 568 |
SendDlgItemMessage(Dialog, IDC_TEXTCODING_DROPDOWN, CB_SETCURSEL, 0, 0); |
| 569 |
|
| 570 |
SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, work->info->filename); |
| 571 |
|
| 572 |
// timestamp ���� |
| 573 |
int tstype = pts->LogTimestampType == TIMESTAMP_LOCAL ? 0 : |
| 574 |
pts->LogTimestampType == TIMESTAMP_UTC ? 1 : |
| 575 |
pts->LogTimestampType == TIMESTAMP_ELAPSED_LOGSTART ? 2 : |
| 576 |
pts->LogTimestampType == TIMESTAMP_ELAPSED_CONNECTED ? 3 : 0; |
| 577 |
SendDlgItemMessage(Dialog, IDC_TIMESTAMPTYPE, CB_SETCURSEL, tstype, 0); |
| 578 |
|
| 579 |
|
| 580 |
CheckLogFile(Dialog, work->info->filename, work); |
| 581 |
ArrangeControls(Dialog, work, pts->Append, pts->LogBinary, pts->LogTypePlainText, pts->LogTimestamp); |
| 582 |
work->info->filename = NULL; |
| 583 |
|
| 584 |
// Hide dialog �`�F�b�N�{�b�N�X |
| 585 |
if (pts->LogHideDialog) { |
| 586 |
SetRB(Dialog, 1, IDC_HIDEDIALOG, IDC_HIDEDIALOG); |
| 587 |
} |
| 588 |
|
| 589 |
// Include screen buffer �`�F�b�N�{�b�N�X |
| 590 |
if (pts->LogAllBuffIncludedInFirst) { |
| 591 |
SetRB(Dialog, 1, IDC_ALLBUFF_INFIRST, IDC_ALLBUFF_INFIRST); |
| 592 |
} |
| 593 |
|
| 594 |
CenterWindow(Dialog, GetParent(Dialog)); |
| 595 |
|
| 596 |
return TRUE; |
| 597 |
} |
| 598 |
|
| 599 |
case WM_COMMAND: |
| 600 |
switch (LOWORD(wParam)) { |
| 601 |
case IDOK: { |
| 602 |
wchar_t *filename; |
| 603 |
hGetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, &filename); |
| 604 |
work->info->filename = filename; |
| 605 |
work->info->append = IsDlgButtonChecked(Dialog, IDC_APPEND) == BST_CHECKED; |
| 606 |
work->info->bom = IsDlgButtonChecked(Dialog, IDC_BOM) == BST_CHECKED; |
| 607 |
work->info->code = (LogCode_t)SendDlgItemMessageA(Dialog, IDC_TEXTCODING_DROPDOWN, CB_GETCURSEL, 0, 0); |
| 608 |
SetLogFlags(Dialog); |
| 609 |
EndDialog(Dialog, IDOK); |
| 610 |
break; |
| 611 |
} |
| 612 |
case IDCANCEL: |
| 613 |
EndDialog(Dialog, IDCANCEL); |
| 614 |
break; |
| 615 |
case IDHELP: |
| 616 |
OpenHelpCV(work->pcv, HH_HELP_CONTEXT, HlpFileLog); |
| 617 |
break; |
| 618 |
case IDC_FOPT_FILENAME_BUTTON: { |
| 619 |
/* save current dir */ |
| 620 |
const wchar_t *UILanguageFile = work->pts->UILanguageFileW; |
| 621 |
wchar_t *curdir; |
| 622 |
hGetCurrentDirectoryW(&curdir); |
| 623 |
|
| 624 |
wchar_t fname[MAX_PATH]; |
| 625 |
GetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname, _countof(fname)); |
| 626 |
|
| 627 |
const wchar_t* simple_log_filter = L"*.txt;*.log"; |
| 628 |
wchar_t *FNFilter = GetCommonDialogFilterWW(simple_log_filter, UILanguageFile); |
| 629 |
|
| 630 |
wchar_t *caption; |
| 631 |
wchar_t *uimsg; |
| 632 |
GetI18nStrWW("Tera Term", "FILEDLG_TRANS_TITLE_LOG", |
| 633 |
TitLog, UILanguageFile, &uimsg); |
| 634 |
aswprintf(&caption, L"Tera Term: %s", uimsg); |
| 635 |
free(uimsg); |
| 636 |
|
| 637 |
OPENFILENAMEW ofn = {}; |
| 638 |
ofn.lStructSize = get_OPENFILENAME_SIZEW(); |
| 639 |
//ofn.Flags = OFN_PATHMUSTEXIST | OFN_OVERWRITEPROMPT; |
| 640 |
ofn.Flags |= OFN_EXPLORER | OFN_ENABLESIZING; |
| 641 |
ofn.Flags |= OFN_SHOWHELP; |
| 642 |
ofn.Flags |= OFN_NOCHANGEDIR; // �������������������������������� |
| 643 |
ofn.hwndOwner = Dialog; |
| 644 |
ofn.lpstrFilter = FNFilter; |
| 645 |
ofn.nFilterIndex = 1; |
| 646 |
ofn.lpstrFile = fname; |
| 647 |
ofn.nMaxFile = _countof(fname); |
| 648 |
ofn.lpstrTitle = caption; |
| 649 |
ofn.lpstrInitialDir = work->pts->LogDefaultPathW; |
| 650 |
BOOL Ok = GetSaveFileNameW(&ofn); |
| 651 |
free(caption); |
| 652 |
free(FNFilter); |
| 653 |
if (Ok) { |
| 654 |
SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, fname); |
| 655 |
} |
| 656 |
|
| 657 |
/* restore dir */ |
| 658 |
SetCurrentDirectoryW(curdir); |
| 659 |
free(curdir); |
| 660 |
|
| 661 |
break; |
| 662 |
} |
| 663 |
case IDC_NEW_OVERWRITE: |
| 664 |
case IDC_APPEND: |
| 665 |
case IDC_FOPTTEXT: |
| 666 |
case IDC_FOPTBIN: |
| 667 |
case IDC_TIMESTAMP: |
| 668 |
{ |
| 669 |
WORD Appnd, LogBinary, LogTypePlainText, LogTimestamp; |
| 670 |
GetRB(Dialog, &Appnd, IDC_APPEND, IDC_APPEND); |
| 671 |
GetRB(Dialog, &LogBinary, IDC_FOPTBIN, IDC_FOPTBIN); |
| 672 |
GetRB(Dialog, &LogTypePlainText, IDC_PLAINTEXT, IDC_PLAINTEXT); |
| 673 |
GetRB(Dialog, &LogTimestamp, IDC_TIMESTAMP, IDC_TIMESTAMP); |
| 674 |
ArrangeControls(Dialog, work, Appnd, LogBinary, LogTypePlainText, LogTimestamp); |
| 675 |
} |
| 676 |
break; |
| 677 |
case IDC_FOPT_FILENAME_EDIT: |
| 678 |
if (HIWORD(wParam) == EN_CHANGE){ |
| 679 |
wchar_t *filename; |
| 680 |
hGetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, &filename); |
| 681 |
CheckLogFile(Dialog, filename, work); |
| 682 |
free(filename); |
| 683 |
{ |
| 684 |
WORD Appnd, LogBinary, LogTypePlainText, LogTimestamp; |
| 685 |
GetRB(Dialog, &Appnd, IDC_APPEND, IDC_APPEND); |
| 686 |
GetRB(Dialog, &LogBinary, IDC_FOPTBIN, IDC_FOPTBIN); |
| 687 |
GetRB(Dialog, &LogTypePlainText, IDC_PLAINTEXT, IDC_PLAINTEXT); |
| 688 |
GetRB(Dialog, &LogTimestamp, IDC_TIMESTAMP, IDC_TIMESTAMP); |
| 689 |
ArrangeControls(Dialog, work, Appnd, LogBinary, LogTypePlainText, LogTimestamp); |
| 690 |
} |
| 691 |
} |
| 692 |
break; |
| 693 |
} |
| 694 |
break; |
| 695 |
case WM_DROPFILES: { |
| 696 |
// �����h���b�v��������������1������������ |
| 697 |
HDROP hDrop = (HDROP)wParam; |
| 698 |
const UINT len = DragQueryFileW(hDrop, 0, NULL, 0); |
| 699 |
if (len == 0) { |
| 700 |
DragFinish(hDrop); |
| 701 |
return TRUE; |
| 702 |
} |
| 703 |
wchar_t *filename = (wchar_t *)malloc(sizeof(wchar_t) * (len + 1)); |
| 704 |
DragQueryFileW(hDrop, 0, filename, len + 1); |
| 705 |
filename[len] = '\0'; |
| 706 |
CheckRadioButton(Dialog, IDC_NEW_OVERWRITE, IDC_APPEND, IDC_APPEND); |
| 707 |
SetDlgItemTextW(Dialog, IDC_FOPT_FILENAME_EDIT, filename); |
| 708 |
SendDlgItemMessage(Dialog, IDC_FOPT_FILENAME_EDIT, EM_SETSEL, len, len); |
| 709 |
free(filename); |
| 710 |
DragFinish(hDrop); |
| 711 |
return TRUE; |
| 712 |
} |
| 713 |
} |
| 714 |
return FALSE; |
| 715 |
} |
| 716 |
|
| 717 |
static void OpenLogFile(PFileVar fv) |
| 718 |
{ |
| 719 |
// LogLockExclusive ���L���������������������L���������A |
| 720 |
// ���������������O�t�@�C���������G�f�B�^���J���������� |
| 721 |
int dwShareMode = FILE_SHARE_READ; |
| 722 |
if (!ts.LogLockExclusive) { |
| 723 |
dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE; |
| 724 |
} |
| 725 |
fv->FileHandle = CreateFileW(fv->FullName, GENERIC_WRITE, dwShareMode, NULL, |
| 726 |
OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); |
| 727 |
} |
| 728 |
|
| 729 |
static BOOL LogStart(PFileVar fv, const wchar_t *fname) |
| 730 |
{ |
| 731 |
fv->FullName = _wcsdup(fname); |
| 732 |
FixLogOption(); |
| 733 |
|
| 734 |
if (ts.LogBinary > 0) |
| 735 |
{ |
| 736 |
fv->BinLog = TRUE; |
| 737 |
fv->FileLog = FALSE; |
| 738 |
if (! CreateBinBuf()) |
| 739 |
{ |
| 740 |
return FALSE; |
| 741 |
} |
| 742 |
} |
| 743 |
else { |
| 744 |
fv->BinLog = FALSE; |
| 745 |
fv->FileLog = TRUE; |
| 746 |
if (! CreateLogBuf()) |
| 747 |
{ |
| 748 |
return FALSE; |
| 749 |
} |
| 750 |
} |
| 751 |
cv_LStart = cv_LogPtr; |
| 752 |
cv_LCount = 0; |
| 753 |
|
| 754 |
OpenLogFile(fv); |
| 755 |
if (fv->FileHandle == INVALID_HANDLE_VALUE) { |
| 756 |
return FALSE; |
| 757 |
} |
| 758 |
|
| 759 |
/* 2007.05.24 Gentaro */ |
| 760 |
fv->eLineEnd = Line_LineHead; |
| 761 |
if (ts.Append > 0) |
| 762 |
{ |
| 763 |
SetFilePointer(fv->FileHandle, 0, NULL, FILE_END); |
| 764 |
/* 2007.05.24 Gentaro |
| 765 |
If log file already exists, |
| 766 |
a newline is inserted before the first timestamp. |
| 767 |
*/ |
| 768 |
fv->eLineEnd = Line_FileHead; |
| 769 |
} |
| 770 |
|
| 771 |
// BOM�o�� |
| 772 |
if (ts.Append == 0 && ts.LogBinary == 0 && fv->bom) { |
| 773 |
// ���L��������(�V�K) && �o�C�i���������� && BOM ���o���� |
| 774 |
FLogOutputBOM(fv); |
| 775 |
} |
| 776 |
|
| 777 |
// Log rotate configuration |
| 778 |
fv->RotateMode = ts.LogRotate; |
| 779 |
fv->RotateSize = ts.LogRotateSize; |
| 780 |
fv->RotateStep = ts.LogRotateStep; |
| 781 |
|
| 782 |
// Log rotate���L���������A�����t�@�C���T�C�Y�����������B |
| 783 |
// �������t�@�C�������������T�C�Y�����[�e�[�g�������������C���B |
| 784 |
// (2016.4.9 yutaka) |
| 785 |
if (fv->RotateMode != ROTATE_NONE) { |
| 786 |
DWORD size = GetFileSize(fv->FileHandle, NULL); |
| 787 |
if (size == -1) { |
| 788 |
return FALSE; |
| 789 |
} |
| 790 |
fv->ByteCount = size; |
| 791 |
} |
| 792 |
else { |
| 793 |
fv->ByteCount = 0; |
| 794 |
} |
| 795 |
|
| 796 |
if (! OpenFTDlg_(fv)) { |
| 797 |
return FALSE; |
| 798 |
} |
| 799 |
|
| 800 |
fv->IsPause = FALSE; |
| 801 |
fv->StartTime = GetTickCount(); |
| 802 |
|
| 803 |
if (ts.DeferredLogWriteMode) { |
| 804 |
StartThread(fv); |
| 805 |
} |
| 806 |
|
| 807 |
if (fv->FileLog) { |
| 808 |
cv.Log1Byte = LogPut1; |
| 809 |
} |
| 810 |
if (fv->BinLog) { |
| 811 |
cv.Log1Bin = Log1Bin; |
| 812 |
cv.LogBinSkip = LogBinSkip; |
| 813 |
} |
| 814 |
|
| 815 |
return TRUE; |
| 816 |
} |
| 817 |
|
| 818 |
/** |
| 819 |
* �����o�b�t�@�������f�[�^�����������O�������o�� |
| 820 |
* (2013.9.29 yutaka) |
| 821 |
* |
| 822 |
* TODO |
| 823 |
* 1�s������ |
| 824 |
*/ |
| 825 |
void FLogOutputAllBuffer(void) |
| 826 |
{ |
| 827 |
PFileVar fv = LogVar; |
| 828 |
DWORD ofs; |
| 829 |
int size; |
| 830 |
wchar_t buf[512]; |
| 831 |
for (ofs = 0 ; ; ofs++ ) { |
| 832 |
// 1�����s�����������B���������������A�G�X�P�[�v�V�[�P���X�������������B |
| 833 |
size = BuffGetAnyLineDataW(ofs, buf, _countof(buf)); |
| 834 |
if (size == -1) |
| 835 |
break; |
| 836 |
|
| 837 |
OutputStr(buf); |
| 838 |
OutputStr(L"\r\n"); |
| 839 |
LogToFile(fv); |
| 840 |
} |
| 841 |
} |
| 842 |
|
| 843 |
/** |
| 844 |
* ���O��1byte�������� |
| 845 |
* �o�b�t�@�������������� |
| 846 |
* ���������������� LogToFile() ���s������ |
| 847 |
*/ |
| 848 |
void LogPut1(BYTE b) |
| 849 |
{ |
| 850 |
PFileVar fv = LogVar; |
| 851 |
|
| 852 |
cv_LogBuf[cv_LogPtr] = b; |
| 853 |
cv_LogPtr++; |
| 854 |
if (cv_LogPtr>=InBuffSize) |
| 855 |
cv_LogPtr = cv_LogPtr-InBuffSize; |
| 856 |
|
| 857 |
if (fv->FileLog) |
| 858 |
{ |
| 859 |
if (cv_LCount>=InBuffSize) |
| 860 |
{ |
| 861 |
cv_LCount = InBuffSize; |
| 862 |
cv_LStart = cv_LogPtr; |
| 863 |
} |
| 864 |
else |
| 865 |
cv_LCount++; |
| 866 |
} |
| 867 |
else |
| 868 |
cv_LCount = 0; |
| 869 |
} |
| 870 |
|
| 871 |
static BOOL Get1(PCHAR Buf, int *Start, int *Count, PBYTE b) |
| 872 |
{ |
| 873 |
if (*Count<=0) return FALSE; |
| 874 |
*b = Buf[*Start]; |
| 875 |
(*Start)++; |
| 876 |
if (*Start>=InBuffSize) |
| 877 |
*Start = *Start-InBuffSize; |
| 878 |
(*Count)--; |
| 879 |
return TRUE; |
| 880 |
} |
| 881 |
|
| 882 |
|
| 883 |
|
| 884 |
static CRITICAL_SECTION g_filelog_lock; /* ���b�N�p���� */ |
| 885 |
|
| 886 |
void logfile_lock_initialize(void) |
| 887 |
{ |
| 888 |
InitializeCriticalSection(&g_filelog_lock); |
| 889 |
} |
| 890 |
|
| 891 |
static inline void logfile_lock(void) |
| 892 |
{ |
| 893 |
EnterCriticalSection(&g_filelog_lock); |
| 894 |
} |
| 895 |
|
| 896 |
static inline void logfile_unlock(void) |
| 897 |
{ |
| 898 |
LeaveCriticalSection(&g_filelog_lock); |
| 899 |
} |
| 900 |
|
| 901 |
// ���O�����[�e�[�g�����B |
| 902 |
// (2013.3.21 yutaka) |
| 903 |
static void LogRotate(PFileVar fv) |
| 904 |
{ |
| 905 |
int loopmax = 10000; // XXX |
| 906 |
int i, k; |
| 907 |
|
| 908 |
if (fv->RotateMode == ROTATE_NONE) |
| 909 |
return; |
| 910 |
|
| 911 |
if (fv->RotateMode == ROTATE_SIZE) { |
| 912 |
if (fv->ByteCount <= fv->RotateSize) |
| 913 |
return; |
| 914 |
//OutputDebugPrintf("%s: mode %d size %ld\n", __FUNCTION__, fv->RotateMode, fv->ByteCount); |
| 915 |
} else { |
| 916 |
return; |
| 917 |
} |
| 918 |
|
| 919 |
logfile_lock(); |
| 920 |
// ���O�T�C�Y���������������B |
| 921 |
fv->ByteCount = 0; |
| 922 |
|
| 923 |
// �������������t�@�C�����N���[�Y�����A�������t�@�C�����I�[�v�������B |
| 924 |
CloseFileSync(fv); |
| 925 |
|
| 926 |
// �������[�e�[�V�������X�e�b�v�����w���������� |
| 927 |
if (fv->RotateStep > 0) |
| 928 |
loopmax = fv->RotateStep; |
| 929 |
|
| 930 |
for (i = 1 ; i <= loopmax ; i++) { |
| 931 |
wchar_t *filename; |
| 932 |
aswprintf(&filename, L"%s.%d", fv->FullName, i); |
| 933 |
DWORD attr = GetFileAttributesW(filename); |
| 934 |
free(filename); |
| 935 |
if (attr == INVALID_FILE_ATTRIBUTES) |
| 936 |
break; |
| 937 |
} |
| 938 |
if (i > loopmax) { |
| 939 |
// �������������������������A�������t�@�C�������p�������B |
| 940 |
i = loopmax; |
| 941 |
} |
| 942 |
|
| 943 |
// ���t�@�C�������l�[���B |
| 944 |
for (k = i-1 ; k >= 0 ; k--) { |
| 945 |
wchar_t *oldfile; |
| 946 |
if (k == 0) |
| 947 |
oldfile = _wcsdup(fv->FullName); |
| 948 |
else |
| 949 |
aswprintf(&oldfile, L"%s.%d", fv->FullName, k); |
| 950 |
wchar_t *newfile; |
| 951 |
aswprintf(&newfile, L"%s.%d", fv->FullName, k+1); |
| 952 |
DeleteFileW(newfile); |
| 953 |
if (MoveFileW(oldfile, newfile) == 0) { |
| 954 |
OutputDebugPrintf("%s: rename %d\n", __FUNCTION__, errno); |
| 955 |
} |
| 956 |
free(oldfile); |
| 957 |
free(newfile); |
| 958 |
} |
| 959 |
|
| 960 |
// ���I�[�v�� |
| 961 |
OpenLogFile(fv); |
| 962 |
if (fv->bom) { |
| 963 |
FLogOutputBOM(fv); |
| 964 |
} |
| 965 |
if (ts.DeferredLogWriteMode) { |
| 966 |
StartThread(fv); |
| 967 |
} |
| 968 |
|
| 969 |
logfile_unlock(); |
| 970 |
} |
| 971 |
|
| 972 |
static wchar_t *TimeStampStr(PFileVar fv) |
| 973 |
{ |
| 974 |
char *strtime = NULL; |
| 975 |
switch (ts.LogTimestampType) { |
| 976 |
case TIMESTAMP_LOCAL: |
| 977 |
default: |
| 978 |
strtime = mctimelocal(ts.LogTimestampFormat, FALSE); |
| 979 |
break; |
| 980 |
case TIMESTAMP_UTC: |
| 981 |
strtime = mctimelocal(ts.LogTimestampFormat, TRUE); |
| 982 |
break; |
| 983 |
case TIMESTAMP_ELAPSED_LOGSTART: |
| 984 |
strtime = strelapsed(fv->StartTime); |
| 985 |
break; |
| 986 |
case TIMESTAMP_ELAPSED_CONNECTED: |
| 987 |
strtime = strelapsed(cv.ConnectedTime); |
| 988 |
break; |
| 989 |
} |
| 990 |
|
| 991 |
char tmp[128]; |
| 992 |
tmp[0] = 0; |
| 993 |
strncat_s(tmp, sizeof(tmp), "[", _TRUNCATE); |
| 994 |
strncat_s(tmp, sizeof(tmp), strtime, _TRUNCATE); |
| 995 |
strncat_s(tmp, sizeof(tmp), "] ", _TRUNCATE); |
| 996 |
|
| 997 |
return ToWcharA(tmp); |
| 998 |
} |
| 999 |
|
| 1000 |
/** |
| 1001 |
* �o�b�t�@�������O���t�@�C������������ |
| 1002 |
*/ |
| 1003 |
static void LogToFile(PFileVar fv) |
| 1004 |
{ |
| 1005 |
PCHAR Buf; |
| 1006 |
int Start, Count; |
| 1007 |
BYTE b; |
| 1008 |
|
| 1009 |
if (fv->FileLog) |
| 1010 |
{ |
| 1011 |
Buf = cv_LogBuf; |
| 1012 |
Start = cv_LStart; |
| 1013 |
Count = cv_LCount; |
| 1014 |
} |
| 1015 |
else if (fv->BinLog) |
| 1016 |
{ |
| 1017 |
Buf = cv_BinBuf; |
| 1018 |
Start = cv_BStart; |
| 1019 |
Count = cv_BCount; |
| 1020 |
} |
| 1021 |
else |
| 1022 |
return; |
| 1023 |
|
| 1024 |
if (Buf==NULL) return; |
| 1025 |
if (Count==0) return; |
| 1026 |
|
| 1027 |
// ���b�N������(2004.8.6 yutaka) |
| 1028 |
logfile_lock(); |
| 1029 |
|
| 1030 |
// ���������f�[�^���������� |
| 1031 |
DWORD WriteBufMax = 8192; |
| 1032 |
DWORD WriteBufLen = 0; |
| 1033 |
PCHAR WriteBuf = (PCHAR)malloc(WriteBufMax); |
| 1034 |
while (Get1(Buf,&Start,&Count,&b)) { |
| 1035 |
if (FLogIsPause() || ProtoGetProtoFlag()) { |
| 1036 |
continue; |
| 1037 |
} |
| 1038 |
|
| 1039 |
if (WriteBufLen >= (WriteBufMax*4/5)) { |
| 1040 |
WriteBufMax *= 2; |
| 1041 |
WriteBuf = (PCHAR)realloc(WriteBuf, WriteBufMax); |
| 1042 |
} |
| 1043 |
|
| 1044 |
WriteBuf[WriteBufLen++] = b; |
| 1045 |
|
| 1046 |
(fv->ByteCount)++; |
| 1047 |
} |
| 1048 |
|
| 1049 |
// �������� |
| 1050 |
if (WriteBufLen > 0) { |
| 1051 |
if (ts.DeferredLogWriteMode) { |
| 1052 |
PostThreadMessage(fv->LogThreadId, WM_DPC_LOGTHREAD_SEND, (WPARAM)WriteBuf, WriteBufLen); |
| 1053 |
} |
| 1054 |
else { |
| 1055 |
DWORD wrote; |
| 1056 |
WriteFile(fv->FileHandle, WriteBuf, WriteBufLen, &wrote, NULL); |
| 1057 |
free(WriteBuf); |
| 1058 |
} |
| 1059 |
} |
| 1060 |
|
| 1061 |
logfile_unlock(); |
| 1062 |
|
| 1063 |
if (fv->FileLog) |
| 1064 |
{ |
| 1065 |
cv_LStart = Start; |
| 1066 |
cv_LCount = Count; |
| 1067 |
} |
| 1068 |
else { |
| 1069 |
cv_BStart = Start; |
| 1070 |
cv_BCount = Count; |
| 1071 |
} |
| 1072 |
if (FLogIsPause() || ProtoGetProtoFlag()) return; |
| 1073 |
fv->FLogDlg->RefreshNum(fv->StartTime, fv->FileSize, fv->ByteCount); |
| 1074 |
|
| 1075 |
|
| 1076 |
// ���O�E���[�e�[�g |
| 1077 |
LogRotate(fv); |
| 1078 |
} |
| 1079 |
|
| 1080 |
static BOOL CreateLogBuf(void) |
| 1081 |
{ |
| 1082 |
if (cv_LogBuf==NULL) |
| 1083 |
{ |
| 1084 |
cv_LogBuf = (char *)malloc(InBuffSize); |
| 1085 |
cv_LogPtr = 0; |
| 1086 |
cv_LStart = 0; |
| 1087 |
cv_LCount = 0; |
| 1088 |
} |
| 1089 |
return (cv_LogBuf!=NULL); |
| 1090 |
} |
| 1091 |
|
| 1092 |
static void FreeLogBuf(void) |
| 1093 |
{ |
| 1094 |
free(cv_LogBuf); |
| 1095 |
cv_LogBuf = NULL; |
| 1096 |
cv_LogPtr = 0; |
| 1097 |
cv_LStart = 0; |
| 1098 |
cv_LCount = 0; |
| 1099 |
} |
| 1100 |
|
| 1101 |
static BOOL CreateBinBuf(void) |
| 1102 |
{ |
| 1103 |
if (cv_BinBuf==NULL) |
| 1104 |
{ |
| 1105 |
cv_BinBuf = (PCHAR)malloc(InBuffSize); |
| 1106 |
cv_BinPtr = 0; |
| 1107 |
cv_BStart = 0; |
| 1108 |
cv_BCount = 0; |
| 1109 |
} |
| 1110 |
return (cv_BinBuf!=NULL); |
| 1111 |
} |
| 1112 |
|
| 1113 |
static void FreeBinBuf(void) |
| 1114 |
{ |
| 1115 |
free(cv_BinBuf); |
| 1116 |
cv_BinBuf = NULL; |
| 1117 |
cv_BinPtr = 0; |
| 1118 |
cv_BStart = 0; |
| 1119 |
cv_BCount = 0; |
| 1120 |
} |
| 1121 |
|
| 1122 |
static void FileTransEnd_(PFileVar fv) |
| 1123 |
{ |
| 1124 |
fv->FileLog = FALSE; |
| 1125 |
fv->BinLog = FALSE; |
| 1126 |
cv.Log1Byte = NULL; |
| 1127 |
cv.Log1Bin = NULL; |
| 1128 |
cv.LogBinSkip = NULL; |
| 1129 |
PFileTransDlg FLogDlg = fv->FLogDlg; |
| 1130 |
if (FLogDlg != NULL) { |
| 1131 |
FLogDlg->DestroyWindow(); |
| 1132 |
FLogDlg = NULL; |
| 1133 |
fv->FLogDlg = NULL; |
| 1134 |
} |
| 1135 |
CloseFileSync(fv); |
| 1136 |
FreeLogBuf(); |
| 1137 |
FreeBinBuf(); |
| 1138 |
free(fv->FullName); |
| 1139 |
fv->FullName = NULL; |
| 1140 |
free(fv); |
| 1141 |
|
| 1142 |
LogVar = NULL; |
| 1143 |
} |
| 1144 |
|
| 1145 |
/** |
| 1146 |
* ���O���|�[�Y���� |
| 1147 |
*/ |
| 1148 |
void FLogPause(BOOL Pause) |
| 1149 |
{ |
| 1150 |
PFileVar fv = LogVar; |
| 1151 |
if (fv == NULL) { |
| 1152 |
return; |
| 1153 |
} |
| 1154 |
fv->IsPause = Pause; |
| 1155 |
fv->FLogDlg->ChangeButton(Pause); |
| 1156 |
} |
| 1157 |
|
| 1158 |
/** |
| 1159 |
* ���O���[�e�[�g������ |
| 1160 |
* ���O���T�C�Y��<size>�o�C�g���������������A���[�e�[�V�������������������� |
| 1161 |
*/ |
| 1162 |
void FLogRotateSize(size_t size) |
| 1163 |
{ |
| 1164 |
PFileVar fv = LogVar; |
| 1165 |
if (fv == NULL) { |
| 1166 |
return; |
| 1167 |
} |
| 1168 |
fv->RotateMode = ROTATE_SIZE; |
| 1169 |
fv->RotateSize = (LONG)size; |
| 1170 |
} |
| 1171 |
|
| 1172 |
/** |
| 1173 |
* ���O���[�e�[�g������ |
| 1174 |
* ���O�t�@�C������������������ |
| 1175 |
*/ |
| 1176 |
void FLogRotateRotate(int step) |
| 1177 |
{ |
| 1178 |
PFileVar fv = LogVar; |
| 1179 |
if (fv == NULL) { |
| 1180 |
return; |
| 1181 |
} |
| 1182 |
fv->RotateStep = step; |
| 1183 |
} |
| 1184 |
|
| 1185 |
/** |
| 1186 |
* ���O���[�e�[�g������ |
| 1187 |
* ���[�e�[�V���������~ |
| 1188 |
*/ |
| 1189 |
void FLogRotateHalt(void) |
| 1190 |
{ |
| 1191 |
PFileVar fv = LogVar; |
| 1192 |
if (fv == NULL) { |
| 1193 |
return; |
| 1194 |
} |
| 1195 |
fv->RotateMode = ROTATE_NONE; |
| 1196 |
fv->RotateSize = 0; |
| 1197 |
fv->RotateStep = 0; |
| 1198 |
} |
| 1199 |
|
| 1200 |
static INT_PTR CALLBACK OnCommentDlgProc(HWND hDlgWnd, UINT msg, WPARAM wp, LPARAM) |
| 1201 |
{ |
| 1202 |
static const DlgTextInfo TextInfos[] = { |
| 1203 |
{ 0, "DLG_COMMENT_TITLE" }, |
| 1204 |
{ IDOK, "BTN_OK" } |
| 1205 |
}; |
| 1206 |
|
| 1207 |
switch (msg) { |
| 1208 |
case WM_INITDIALOG: |
| 1209 |
// �G�f�B�b�g�R���g���[�����t�H�[�J�X�������� |
| 1210 |
SetFocus(GetDlgItem(hDlgWnd, IDC_EDIT_COMMENT)); |
| 1211 |
SetDlgTextsW(hDlgWnd, TextInfos, _countof(TextInfos), ts.UILanguageFileW); |
| 1212 |
return FALSE; |
| 1213 |
|
| 1214 |
case WM_COMMAND: |
| 1215 |
switch (LOWORD(wp)) { |
| 1216 |
case IDOK: { |
| 1217 |
size_t len = SendDlgItemMessageW(hDlgWnd, IDC_EDIT_COMMENT, WM_GETTEXTLENGTH, 0, 0); |
| 1218 |
len += 1; |
| 1219 |
wchar_t *buf = (wchar_t *)malloc(len * sizeof(wchar_t)); |
| 1220 |
GetDlgItemTextW(hDlgWnd, IDC_EDIT_COMMENT, buf, (int)len); |
| 1221 |
FLogWriteStr(buf); |
| 1222 |
FLogWriteStr(L"\n"); // TODO ���s�R�[�h |
| 1223 |
free(buf); |
| 1224 |
TTEndDialog(hDlgWnd, IDOK); |
| 1225 |
break; |
| 1226 |
} |
| 1227 |
default: |
| 1228 |
return FALSE; |
| 1229 |
} |
| 1230 |
break; |
| 1231 |
|
| 1232 |
case WM_CLOSE: |
| 1233 |
TTEndDialog(hDlgWnd, 0); |
| 1234 |
return TRUE; |
| 1235 |
|
| 1236 |
default: |
| 1237 |
return FALSE; |
| 1238 |
} |
| 1239 |
return TRUE; |
| 1240 |
} |
| 1241 |
|
| 1242 |
/** |
| 1243 |
* ���O�t�@�C�����R�����g���������� (2004.8.6 yutaka) |
| 1244 |
*/ |
| 1245 |
void FLogAddCommentDlg(HINSTANCE hInst, HWND hWnd) |
| 1246 |
{ |
| 1247 |
PFileVar fv = LogVar; |
| 1248 |
if (fv == NULL) { |
| 1249 |
return; |
| 1250 |
} |
| 1251 |
TTDialogBox(hInst, MAKEINTRESOURCE(IDD_COMMENT_DIALOG), |
| 1252 |
hWnd, OnCommentDlgProc); |
| 1253 |
} |
| 1254 |
|
| 1255 |
void FLogClose(void) |
| 1256 |
{ |
| 1257 |
PFileVar fv = LogVar; |
| 1258 |
if (fv == NULL) { |
| 1259 |
return; |
| 1260 |
} |
| 1261 |
|
| 1262 |
FileTransEnd_(fv); |
| 1263 |
} |
| 1264 |
|
| 1265 |
/** |
| 1266 |
* ���O���I�[�v������ |
| 1267 |
* @param[in] fname ���O�t�@�C����, CreateFile()���n������ |
| 1268 |
* |
| 1269 |
* ���O�t�@�C������strftime���W�J�������s���������B |
| 1270 |
* FLogGetLogFilename() �� FLogOpenDialog() �� |
| 1271 |
* �t�@�C�����������������B |
| 1272 |
*/ |
| 1273 |
BOOL FLogOpen(const wchar_t *fname, LogCode_t code, BOOL bom) |
| 1274 |
{ |
| 1275 |
if (LogVar != NULL) { |
| 1276 |
return FALSE; |
| 1277 |
} |
| 1278 |
|
| 1279 |
// |
| 1280 |
PFileVar fv = (PFileVar)malloc(sizeof(*fv)); |
| 1281 |
if (fv == NULL) { |
| 1282 |
return FALSE; |
| 1283 |
} |
| 1284 |
LogVar = fv; |
| 1285 |
memset(fv, 0, sizeof(TFileVar)); |
| 1286 |
fv->FileHandle = INVALID_HANDLE_VALUE; |
| 1287 |
fv->LogThread = INVALID_HANDLE_VALUE; |
| 1288 |
fv->eLineEnd = Line_LineHead; |
| 1289 |
|
| 1290 |
fv->log_code = code; |
| 1291 |
fv->bom = bom; |
| 1292 |
BOOL ret = LogStart(fv, fname); |
| 1293 |
if (ret == FALSE) { |
| 1294 |
FileTransEnd_(fv); |
| 1295 |
} |
| 1296 |
|
| 1297 |
return ret; |
| 1298 |
} |
| 1299 |
|
| 1300 |
BOOL FLogIsOpend(void) |
| 1301 |
{ |
| 1302 |
return LogVar != NULL; |
| 1303 |
} |
| 1304 |
|
| 1305 |
BOOL FLogIsOpendText(void) |
| 1306 |
{ |
| 1307 |
return LogVar != NULL && LogVar->FileLog; |
| 1308 |
} |
| 1309 |
|
| 1310 |
BOOL FLogIsOpendBin(void) |
| 1311 |
{ |
| 1312 |
return LogVar != NULL && LogVar->BinLog; |
| 1313 |
} |
| 1314 |
|
| 1315 |
void FLogWriteStr(const wchar_t *str) |
| 1316 |
{ |
| 1317 |
if (LogVar != NULL) { |
| 1318 |
OutputStr(str); |
| 1319 |
} |
| 1320 |
} |
| 1321 |
|
| 1322 |
void FLogInfo(char *param_ptr, size_t param_len) |
| 1323 |
{ |
| 1324 |
PFileVar fv = LogVar; |
| 1325 |
if (fv) { |
| 1326 |
param_ptr[0] = '0' |
| 1327 |
+ (ts.LogBinary != 0) |
| 1328 |
+ ((ts.Append != 0) << 1) |
| 1329 |
+ ((ts.LogTypePlainText != 0) << 2) |
| 1330 |
+ ((ts.LogTimestamp != 0) << 3) |
| 1331 |
+ ((ts.LogHideDialog != 0) << 4); |
| 1332 |
char *filenameU8 = ToU8W(fv->FullName); |
| 1333 |
strncpy_s(param_ptr + 1, param_len - 1, filenameU8, _TRUNCATE); |
| 1334 |
free(filenameU8); |
| 1335 |
} |
| 1336 |
else { |
| 1337 |
param_ptr[0] = '0' - 1; |
| 1338 |
param_ptr[1] = 0; |
| 1339 |
} |
| 1340 |
} |
| 1341 |
|
| 1342 |
/** |
| 1343 |
* ���������O�t�@�C���������� |
| 1344 |
*/ |
| 1345 |
const wchar_t *FLogGetFilename(void) |
| 1346 |
{ |
| 1347 |
if (LogVar == NULL) { |
| 1348 |
return NULL; |
| 1349 |
} |
| 1350 |
return LogVar->FullName; |
| 1351 |
} |
| 1352 |
|
| 1353 |
/** |
| 1354 |
* ���O�_�C�A���O���J�� |
| 1355 |
* @param[in,out] info.filename �t�@�C���������l |
| 1356 |
* OK���A�t�@�C�����A�s�v����������free()�������� |
| 1357 |
* @retval TRUE [ok] ���������� |
| 1358 |
* @retval FALSE �L�����Z�������� |
| 1359 |
*/ |
| 1360 |
BOOL FLogOpenDialog(HINSTANCE hInst, HWND hWnd, FLogDlgInfo_t *info) |
| 1361 |
{ |
| 1362 |
LogDlgWork_t *work = (LogDlgWork_t *)calloc(sizeof(LogDlgWork_t), 1); |
| 1363 |
wchar_t *srcfnameW = FLogGetLogFilename(info->filename); |
| 1364 |
work->info = info; |
| 1365 |
work->info->filename = srcfnameW; |
| 1366 |
work->pts = &ts; |
| 1367 |
work->pcv = &cv; |
| 1368 |
INT_PTR ret = TTDialogBoxParam( |
| 1369 |
hInst, MAKEINTRESOURCE(IDD_LOGDLG), |
| 1370 |
hWnd, LogFnHook, (LPARAM)work); |
| 1371 |
free(srcfnameW); |
| 1372 |
free(work); |
| 1373 |
return ret == IDOK ? TRUE : FALSE; |
| 1374 |
} |
| 1375 |
|
| 1376 |
/** |
| 1377 |
* ���O�t�@�C�����p���C�����s��,�t�@�C������������ |
| 1378 |
* - strftime() ���������t�W�J |
| 1379 |
* - �������������O�t�@�C���t�H���_������ |
| 1380 |
* - �z�X�g��,�|�[�g�����W�J |
| 1381 |
* |
| 1382 |
* @param filename �t�@�C����(�p�X����������) |
| 1383 |
* @return �C�������t�@�C���� |
| 1384 |
*/ |
| 1385 |
wchar_t *FLogGetLogFilenameBase(const wchar_t *filename) |
| 1386 |
{ |
| 1387 |
// �t�@�C�������������o |
| 1388 |
const wchar_t *last_path_sep = wcsrchr(filename, L'\\'); |
| 1389 |
wchar_t *format; |
| 1390 |
if (last_path_sep == NULL) { |
| 1391 |
format = _wcsdup(filename); |
| 1392 |
} |
| 1393 |
else { |
| 1394 |
format = _wcsdup(last_path_sep + 1); |
| 1395 |
} |
| 1396 |
|
| 1397 |
// strftime ���g�p������������������ |
| 1398 |
deleteInvalidStrftimeCharW(format); |
| 1399 |
|
| 1400 |
// ����������0��������? |
| 1401 |
if (format[0] == 0) { |
| 1402 |
free(format); |
| 1403 |
return _wcsdup(L""); |
| 1404 |
} |
| 1405 |
|
| 1406 |
// �������������� |
| 1407 |
time_t time_local; |
| 1408 |
time(&time_local); |
| 1409 |
struct tm tm_local; |
| 1410 |
localtime_s(&tm_local, &time_local); |
| 1411 |
|
| 1412 |
// strftime()������ |
| 1413 |
// ���������������g�� |
| 1414 |
size_t len = 32; |
| 1415 |
wchar_t *formated = NULL; |
| 1416 |
while (1) { |
| 1417 |
wchar_t *formated_realloc = (wchar_t *)realloc(formated, sizeof(wchar_t) * len); |
| 1418 |
if (formated_realloc == NULL) { |
| 1419 |
free(format); |
| 1420 |
free(formated); |
| 1421 |
return _wcsdup(L""); |
| 1422 |
} |
| 1423 |
formated = formated_realloc; |
| 1424 |
size_t r = wcsftime(formated, len, format, &tm_local); |
| 1425 |
if (r != 0) { |
| 1426 |
// �t�H�[�}�b�g������ |
| 1427 |
break; |
| 1428 |
} |
| 1429 |
len *= 2; |
| 1430 |
} |
| 1431 |
free(format); |
| 1432 |
|
| 1433 |
// �z�X�g������ |
| 1434 |
wchar_t *host = ConvertLognameW(&cv, formated); |
| 1435 |
free(formated); |
| 1436 |
|
| 1437 |
// �t�@�C�������g�p���������������u�� |
| 1438 |
//wchar_t *replaced = replaceInvalidFileNameCharW(host, 0); // ���� |
| 1439 |
wchar_t *replaced = replaceInvalidFileNameCharW(host, L'_'); |
| 1440 |
free(host); |
| 1441 |
|
| 1442 |
return replaced; |
| 1443 |
} |
| 1444 |
|
| 1445 |
/** |
| 1446 |
* ���O�t�@�C���������� |
| 1447 |
* ���O�t�@�C�����p���C�����s�� |
| 1448 |
* - strftime() ���������t�W�J |
| 1449 |
* - �������������O�t�@�C���t�H���_������ |
| 1450 |
* - �z�X�g��,�|�[�g�����W�J |
| 1451 |
* |
| 1452 |
* @param[in] log_filename �t�@�C����(����/��������������ok) |
| 1453 |
* NULL�������f�t�H���g�t�@�C���������� |
| 1454 |
* strftime�`��ok |
| 1455 |
* @return �t���p�X�t�@�C���� |
| 1456 |
* �s�v���������� free() �������� |
| 1457 |
*/ |
| 1458 |
wchar_t *FLogGetLogFilename(const wchar_t *log_filename) |
| 1459 |
{ |
| 1460 |
wchar_t *dir; |
| 1461 |
wchar_t *fname; |
| 1462 |
if (log_filename == NULL) { |
| 1463 |
dir = _wcsdup(ts.LogDefaultPathW); |
| 1464 |
fname = _wcsdup(ts.LogDefaultNameW); |
| 1465 |
} else if (!IsRelativePathW(log_filename)) { |
| 1466 |
// �����p�X������������ |
| 1467 |
dir = ExtractDirNameW(log_filename); |
| 1468 |
fname = ExtractFileNameW(log_filename); |
| 1469 |
} |
| 1470 |
else { |
| 1471 |
dir = _wcsdup(ts.LogDefaultPathW); |
| 1472 |
fname = _wcsdup(log_filename); |
| 1473 |
} |
| 1474 |
|
| 1475 |
wchar_t *formated = FLogGetLogFilenameBase(fname); |
| 1476 |
free(fname); |
| 1477 |
|
| 1478 |
// �A������ |
| 1479 |
wchar_t *logfull = NULL; |
| 1480 |
awcscats(&logfull, dir, L"\\", formated, NULL); |
| 1481 |
free(dir); |
| 1482 |
|
| 1483 |
// ���K�� |
| 1484 |
wchar_t *normal; |
| 1485 |
hGetFullPathNameW(logfull, &normal, NULL); |
| 1486 |
free(logfull); |
| 1487 |
|
| 1488 |
return normal; |
| 1489 |
} |
| 1490 |
|
| 1491 |
BOOL FLogIsPause() |
| 1492 |
{ |
| 1493 |
if (LogVar == NULL) { |
| 1494 |
return FALSE; |
| 1495 |
} |
| 1496 |
return LogVar->IsPause; |
| 1497 |
} |
| 1498 |
|
| 1499 |
void FLogWindow(int nCmdShow) |
| 1500 |
{ |
| 1501 |
if (LogVar == NULL) { |
| 1502 |
return; |
| 1503 |
} |
| 1504 |
|
| 1505 |
HWND HWndLog = LogVar->FLogDlg->m_hWnd; |
| 1506 |
ShowWindow(HWndLog, nCmdShow); |
| 1507 |
if (nCmdShow == SW_RESTORE) { |
| 1508 |
// �g���X�^�C�� WS_EX_NOACTIVATE �������������� |
| 1509 |
SetForegroundWindow(HWndLog); |
| 1510 |
} |
| 1511 |
} |
| 1512 |
|
| 1513 |
void FLogShowDlg(void) |
| 1514 |
{ |
| 1515 |
if (LogVar == NULL) { |
| 1516 |
return; |
| 1517 |
} |
| 1518 |
HWND HWndLog = LogVar->FLogDlg->m_hWnd; |
| 1519 |
ShowWindow(HWndLog, SW_SHOWNORMAL); |
| 1520 |
SetForegroundWindow(HWndLog); |
| 1521 |
} |
| 1522 |
|
| 1523 |
/** |
| 1524 |
* ���O��1byte�������� |
| 1525 |
* LogPut1() ������? |
| 1526 |
*/ |
| 1527 |
//void Log1Bin(PComVar cv, BYTE b) |
| 1528 |
static void Log1Bin(BYTE b) |
| 1529 |
{ |
| 1530 |
if (LogVar->IsPause || ProtoGetProtoFlag()) { |
| 1531 |
return; |
| 1532 |
} |
| 1533 |
if (cv_BinSkip > 0) { |
| 1534 |
cv_BinSkip--; |
| 1535 |
return; |
| 1536 |
} |
| 1537 |
cv_BinBuf[cv_BinPtr] = b; |
| 1538 |
cv_BinPtr++; |
| 1539 |
if (cv_BinPtr>=InBuffSize) { |
| 1540 |
cv_BinPtr = cv_BinPtr-InBuffSize; |
| 1541 |
} |
| 1542 |
if (cv_BCount>=InBuffSize) { |
| 1543 |
cv_BCount = InBuffSize; |
| 1544 |
cv_BStart = cv_BinPtr; |
| 1545 |
} |
| 1546 |
else { |
| 1547 |
cv_BCount++; |
| 1548 |
} |
| 1549 |
} |
| 1550 |
|
| 1551 |
static void LogBinSkip(int add) |
| 1552 |
{ |
| 1553 |
if (cv_BinBuf != NULL) { |
| 1554 |
cv_BinSkip += add; |
| 1555 |
} |
| 1556 |
} |
| 1557 |
|
| 1558 |
/** |
| 1559 |
* ���O�o�b�t�@���������������f�[�^���o�C�g�������� |
| 1560 |
*/ |
| 1561 |
int FLogGetCount(void) |
| 1562 |
{ |
| 1563 |
PFileVar fv = LogVar; |
| 1564 |
if (fv == NULL) { |
| 1565 |
return 0; |
| 1566 |
} |
| 1567 |
if (fv->FileLog) { |
| 1568 |
return cv_LCount; |
| 1569 |
} |
| 1570 |
if (fv->BinLog) { |
| 1571 |
return cv_BCount; |
| 1572 |
} |
| 1573 |
return 0; |
| 1574 |
} |
| 1575 |
|
| 1576 |
/** |
| 1577 |
* ���O�o�b�t�@�������o�C�g�������� |
| 1578 |
*/ |
| 1579 |
int FLogGetFreeCount(void) |
| 1580 |
{ |
| 1581 |
PFileVar fv = LogVar; |
| 1582 |
if (fv == NULL) { |
| 1583 |
return 0; |
| 1584 |
} |
| 1585 |
if (fv->FileLog) { |
| 1586 |
return InBuffSize - cv_LCount; |
| 1587 |
} |
| 1588 |
if (fv->BinLog) { |
| 1589 |
return InBuffSize - cv_BCount; |
| 1590 |
} |
| 1591 |
return 0; |
| 1592 |
} |
| 1593 |
|
| 1594 |
/** |
| 1595 |
* �o�b�t�@�������O���t�@�C������������ |
| 1596 |
*/ |
| 1597 |
void FLogWriteFile(void) |
| 1598 |
{ |
| 1599 |
PFileVar fv = LogVar; |
| 1600 |
if (fv == NULL) { |
| 1601 |
return; |
| 1602 |
} |
| 1603 |
if (cv_LogBuf!=NULL) |
| 1604 |
{ |
| 1605 |
if (fv->FileLog) { |
| 1606 |
LogToFile(fv); |
| 1607 |
} |
| 1608 |
} |
| 1609 |
|
| 1610 |
if (cv_BinBuf!=NULL) |
| 1611 |
{ |
| 1612 |
if (fv->BinLog) { |
| 1613 |
LogToFile(fv); |
| 1614 |
} |
| 1615 |
} |
| 1616 |
} |
| 1617 |
|
| 1618 |
void FLogPutUTF32(unsigned int u32) |
| 1619 |
{ |
| 1620 |
PFileVar fv = LogVar; |
| 1621 |
BOOL log_available = (cv_LogBuf != 0); |
| 1622 |
|
| 1623 |
if (!log_available) { |
| 1624 |
// ���O�����o�������� |
| 1625 |
return; |
| 1626 |
} |
| 1627 |
|
| 1628 |
// �s����?(���s���o����������) |
| 1629 |
if (ts.LogTimestamp && fv->eLineEnd) { |
| 1630 |
// �^�C���X�^���v���o�� |
| 1631 |
fv->eLineEnd = Line_Other; /* clear endmark*/ |
| 1632 |
wchar_t* strtime = TimeStampStr(fv); |
| 1633 |
FLogWriteStr(strtime); |
| 1634 |
free(strtime); |
| 1635 |
} |
| 1636 |
|
| 1637 |
switch(fv->log_code) { |
| 1638 |
case LOG_UTF8: { |
| 1639 |
// UTF-8 |
| 1640 |
char u8_buf[4]; |
| 1641 |
size_t u8_len = UTF32ToUTF8(u32, u8_buf, _countof(u8_buf)); |
| 1642 |
for (size_t i = 0; i < u8_len; i++) { |
| 1643 |
BYTE b = u8_buf[i]; |
| 1644 |
LogPut1(b); |
| 1645 |
} |
| 1646 |
break; |
| 1647 |
} |
| 1648 |
case LOG_UTF16LE: |
| 1649 |
case LOG_UTF16BE: { |
| 1650 |
// UTF-16 |
| 1651 |
wchar_t u16[2]; |
| 1652 |
size_t u16_len = UTF32ToUTF16(u32, u16, _countof(u16)); |
| 1653 |
for (size_t i = 0; i < u16_len; i++) { |
| 1654 |
if (fv->log_code == LOG_UTF16LE) { |
| 1655 |
// UTF-16LE |
| 1656 |
LogPut1(u16[i] & 0xff); |
| 1657 |
LogPut1((u16[i] >> 8) & 0xff); |
| 1658 |
} |
| 1659 |
else { |
| 1660 |
// UTF-16BE |
| 1661 |
LogPut1((u16[i] >> 8) & 0xff); |
| 1662 |
LogPut1(u16[i] & 0xff); |
| 1663 |
} |
| 1664 |
} |
| 1665 |
} |
| 1666 |
} |
| 1667 |
|
| 1668 |
if (u32 == 0x0a) { |
| 1669 |
fv->eLineEnd = Line_LineHead; /* set endmark*/ |
| 1670 |
} |
| 1671 |
} |
| 1672 |
|
| 1673 |
static void FLogOutputBOM(PFileVar fv) |
| 1674 |
{ |
| 1675 |
DWORD wrote; |
| 1676 |
|
| 1677 |
switch(fv->log_code) { |
| 1678 |
case 0: { |
| 1679 |
// UTF-8 |
| 1680 |
const char *bom = "\xef\xbb\xbf"; |
| 1681 |
WriteFile(fv->FileHandle, bom, 3, &wrote, NULL); |
| 1682 |
fv->ByteCount += 3; |
| 1683 |
break; |
| 1684 |
} |
| 1685 |
case 1: { |
| 1686 |
// UTF-16LE |
| 1687 |
const char *bom = "\xff\xfe"; |
| 1688 |
WriteFile(fv->FileHandle, bom, 2, &wrote, NULL); |
| 1689 |
fv->ByteCount += 2; |
| 1690 |
break; |
| 1691 |
} |
| 1692 |
case 2: { |
| 1693 |
// UTF-16BE |
| 1694 |
const char *bom = "\xfe\xff"; |
| 1695 |
WriteFile(fv->FileHandle, bom, 2, &wrote, NULL); |
| 1696 |
fv->ByteCount += 2; |
| 1697 |
break; |
| 1698 |
} |
| 1699 |
default: |
| 1700 |
break; |
| 1701 |
} |
| 1702 |
} |
| 1703 |
|
| 1704 |
static void OutputStr(const wchar_t *str) |
| 1705 |
{ |
| 1706 |
size_t len; |
| 1707 |
|
| 1708 |
assert(str != NULL); |
| 1709 |
|
| 1710 |
len = wcslen(str); |
| 1711 |
while(*str != 0) { |
| 1712 |
unsigned int u32; |
| 1713 |
size_t u16_len = UTF16ToUTF32(str, len, &u32); |
| 1714 |
switch (u16_len) { |
| 1715 |
case 0: |
| 1716 |
default: |
| 1717 |
// ������������ |
| 1718 |
str++; |
| 1719 |
len--; |
| 1720 |
break; |
| 1721 |
case 1: |
| 1722 |
case 2: { |
| 1723 |
FLogPutUTF32(u32); |
| 1724 |
str += u16_len; |
| 1725 |
len -= u16_len; |
| 1726 |
break; |
| 1727 |
} |
| 1728 |
} |
| 1729 |
} |
| 1730 |
} |