347 lines
No EOL
8 KiB
C++
347 lines
No EOL
8 KiB
C++
#include "StdAfx.h"
|
||
#include "PropSignalManager.h"
|
||
#include <winsock2.h>
|
||
#include <process.h>
|
||
#include "PropSignalItem.h"
|
||
#include "FileServer.h"
|
||
|
||
int CPropSignalManager::s_nUniqueIndex = 0;
|
||
|
||
CPropSignalManager::CPropSignalManager()
|
||
{
|
||
m_szWorkingFolder = CFileServer::GetInstance().GetWorkingFolder();
|
||
m_szWorkingFolder += "\\Resource\\PropInfo";
|
||
|
||
m_szFileName = "PropInfo.dat";
|
||
|
||
CString szIP, szHost;
|
||
|
||
GetLocalIP( szIP );
|
||
GetHostName( szHost );
|
||
|
||
m_szMyName = szHost;
|
||
m_szMyName += ";";
|
||
m_szMyName += szIP;
|
||
|
||
char szBuf[256];
|
||
sprintf_s( szBuf, m_szMyName );
|
||
for( int i=0; i<(int)strlen(szBuf); i++ ) {
|
||
if( szBuf[i] == '.' ) szBuf[i] = ';';
|
||
}
|
||
m_szMyName = szBuf;
|
||
}
|
||
|
||
CPropSignalManager::~CPropSignalManager()
|
||
{
|
||
SAFE_DELETE_PVEC( m_pVecList );
|
||
}
|
||
|
||
void CPropSignalManager::GetHostName( CString &szHost )
|
||
{
|
||
char szStr[512] = { 0, };
|
||
|
||
WSADATA winsockInfo;
|
||
WSAStartup( MAKEWORD( 2, 2 ), &winsockInfo );
|
||
|
||
gethostname( szStr, 512 );
|
||
|
||
szHost = szStr;
|
||
|
||
WSACleanup();
|
||
}
|
||
|
||
void CPropSignalManager::GetLocalIP( CString &szIP )
|
||
{
|
||
char szStr[512] = { 0, };
|
||
|
||
WSADATA winsockInfo;
|
||
WSAStartup( MAKEWORD( 2, 2 ), &winsockInfo );
|
||
|
||
gethostname( szStr, 512 );
|
||
HOSTENT *p = NULL;
|
||
p = gethostbyname( szStr );
|
||
|
||
struct in_addr *ptr;
|
||
char **listptr = p->h_addr_list;
|
||
|
||
if( p->h_addrtype == AF_INET ) {
|
||
while( ( ptr = reinterpret_cast<struct in_addr*>(*listptr++) ) != NULL ) {
|
||
szIP = inet_ntoa( *ptr );
|
||
break;
|
||
}
|
||
}
|
||
|
||
WSACleanup();
|
||
}
|
||
|
||
|
||
bool CPropSignalManager::AddSignalItem( CPropSignalItem *pSignalItem )
|
||
{
|
||
if( pSignalItem->GetUniqueIndex() == -1 ) {
|
||
pSignalItem->SetUniqueIndex( s_nUniqueIndex );
|
||
s_nUniqueIndex++;
|
||
}
|
||
m_pVecList.push_back( pSignalItem );
|
||
return true;
|
||
}
|
||
|
||
bool CPropSignalManager::RemoveSignalItem( CString &szName )
|
||
{
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
if( m_pVecList[i]->GetName() == szName ) {
|
||
SAFE_DELETE( m_pVecList[i] );
|
||
m_pVecList.erase( m_pVecList.begin() + i );
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
DWORD CPropSignalManager::GetSignalItemCount()
|
||
{
|
||
return (DWORD)m_pVecList.size();
|
||
}
|
||
|
||
CPropSignalItem *CPropSignalManager::GetSignalItem( DWORD dwIndex )
|
||
{
|
||
if( dwIndex < 0 || dwIndex >= m_pVecList.size() ) return NULL;
|
||
return m_pVecList[dwIndex];
|
||
}
|
||
|
||
CPropSignalItem *CPropSignalManager::GetSignalItemFromName( CString &szSignalName )
|
||
{
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
if( m_pVecList[i]->GetName() == szSignalName ) return m_pVecList[i];
|
||
}
|
||
return NULL;
|
||
}
|
||
|
||
CPropSignalItem *CPropSignalManager::GetSignalItemFromUniqueIndex( int nValue )
|
||
{
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
if( m_pVecList[i]->GetUniqueIndex() == nValue ) return m_pVecList[i];
|
||
}
|
||
return NULL;
|
||
}
|
||
|
||
bool CPropSignalManager::IsExistSignalItem( CString &szName, CString &szOriginalName )
|
||
{
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
if( m_pVecList[i]->GetName() == szName ) {
|
||
if( !szOriginalName.IsEmpty() && m_pVecList[i]->GetName() == szOriginalName ) continue;
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
bool CPropSignalManager::IsUseUniqueIndex( CString &szOriginalName, int nIndex )
|
||
{
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
if( m_pVecList[i]->GetUniqueIndex() == nIndex ) {
|
||
if( !szOriginalName.IsEmpty() && m_pVecList[i]->GetName() == szOriginalName ) continue;
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
void CPropSignalManager::BackupFiles( CString szFileName )
|
||
{
|
||
char szFullName[2048];
|
||
char szBackupName[2048];
|
||
|
||
__time64_t ltime;
|
||
char szBuf[2048];
|
||
errno_t err;
|
||
|
||
time( <ime );
|
||
|
||
|
||
err = _ctime64_s( szBuf, 2048, <ime );
|
||
if( err != 0 ) return;
|
||
|
||
for( int i=0; i<(int)strlen(szBuf); i++ ) {
|
||
if( szBuf[i] == ' ' ) szBuf[i] = '_';
|
||
if( szBuf[i] == ':' ) szBuf[i] = ';';
|
||
}
|
||
szBuf[strlen(szBuf) - 1] = 0;
|
||
|
||
|
||
char szExt[256];
|
||
char szName[256];
|
||
_GetExt( szExt, _countof(szExt), szFileName );
|
||
_GetFileName( szName, _countof(szName), szFileName );
|
||
|
||
sprintf_s( szFullName, "%s\\%s", m_szWorkingFolder, szFileName );
|
||
sprintf_s( szBackupName, "%s\\%s_%s_%s.%s", m_szWorkingFolder, szName, szBuf, m_szMyName, szExt );
|
||
|
||
CopyFile( szFullName, szBackupName, FALSE );
|
||
}
|
||
|
||
bool CPropSignalManager::Save()
|
||
{
|
||
#ifndef NO_BACKUP
|
||
BackupFiles( m_szFileName );
|
||
#endif //NO_BACKUP
|
||
|
||
FILE *fp;
|
||
char szFullName[2048];
|
||
sprintf_s( szFullName, "%s\\%s", m_szWorkingFolder, m_szFileName );
|
||
fopen_s( &fp, szFullName, "wb" );
|
||
if( fp == NULL ) return false;
|
||
|
||
fwrite( &s_nUniqueIndex, sizeof(int), 1, fp );
|
||
|
||
bool bResult = true;
|
||
DWORD dwCount = (DWORD)m_pVecList.size();
|
||
fwrite( &dwCount, sizeof(int), 1, fp );
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
if( m_pVecList[i]->Save( fp ) == false ) {
|
||
bResult = false;
|
||
break;
|
||
}
|
||
}
|
||
fclose(fp);
|
||
|
||
return bResult;
|
||
}
|
||
|
||
bool CPropSignalManager::Load()
|
||
{
|
||
CGlobalValue::GetInstance().UpdateInitDesc( "Register Prop Signal Info.." );
|
||
FILE *fp;
|
||
char szFullName[2048];
|
||
sprintf_s( szFullName, "%s\\%s", m_szWorkingFolder, m_szFileName );
|
||
fopen_s( &fp, szFullName, "rb" );
|
||
if( fp == NULL ) return false;
|
||
|
||
fread( &s_nUniqueIndex, sizeof(int), 1, fp );
|
||
|
||
bool bResult = true;
|
||
DWORD dwCount;
|
||
fread( &dwCount, sizeof(int), 1, fp );
|
||
for( DWORD i=0; i<dwCount; i++ ) {
|
||
CPropSignalItem *pItem = new CPropSignalItem( -1 );
|
||
if( pItem->Load( fp ) == false ) {
|
||
SAFE_DELETE( pItem );
|
||
SAFE_DELETE_PVEC( m_pVecList );
|
||
bResult = false;
|
||
break;
|
||
}
|
||
m_pVecList.push_back( pItem );
|
||
}
|
||
fclose(fp);
|
||
|
||
return bResult;
|
||
}
|
||
|
||
void CPropSignalManager::ExportEnum( FILE *fp )
|
||
{
|
||
fprintf_s( fp, "#pragma once\n\n" );
|
||
fprintf_s( fp, "#pragma pack(push, 4)\n" );
|
||
fprintf_s( fp, "// <20><><EFBFBD><EFBFBD> Ÿ<><C5B8> <20><><EFBFBD><EFBFBD>\n" );
|
||
fprintf_s( fp, "enum PropTypeEnum {\n" );
|
||
CString szStr;
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
szStr.Format( " PTE_%s = %d,\n", m_pVecList[i]->GetName(), m_pVecList[i]->GetUniqueIndex() );
|
||
fprintf_s( fp, szStr );
|
||
}
|
||
fprintf_s( fp, "};\n\n" );
|
||
fprintf_s( fp, "// <20><><EFBFBD><EFBFBD> Ÿ<>Ժ<EFBFBD> <20><>Ʈ<EFBFBD><C6AE><EFBFBD><EFBFBD>\n" );
|
||
}
|
||
|
||
bool CPropSignalManager::ExportHeader( CString szFileName )
|
||
{
|
||
FILE *fp;
|
||
fopen_s( &fp, szFileName, "wt" );
|
||
if( fp == NULL ) return false;
|
||
|
||
std::map<int, int> VecDataUsingCount;
|
||
std::map<int, int> VecDataUsingCount64;
|
||
|
||
ExportEnum( fp );
|
||
bool bResult = true;
|
||
for( DWORD i=0; i<m_pVecList.size(); i++ ) {
|
||
int nUsingBuffCount = 0;
|
||
int nUsingBuffCount64 = 0;
|
||
if( m_pVecList[i]->ExportHeader( fp, nUsingBuffCount, nUsingBuffCount64 ) == false ) {
|
||
bResult = false;
|
||
break;
|
||
}
|
||
|
||
VecDataUsingCount.insert(std::make_pair(m_pVecList[i]->GetUniqueIndex(), nUsingBuffCount));
|
||
VecDataUsingCount64.insert(std::make_pair(m_pVecList[i]->GetUniqueIndex(), nUsingBuffCount64));
|
||
}
|
||
|
||
fprintf_s(fp, "static int GetPropSignalDataUsingCount(int signalIdx) {\n\tswitch(signalIdx) {\n");
|
||
std::map<int, int>::const_iterator iter = VecDataUsingCount.begin();
|
||
for (; iter != VecDataUsingCount.end(); ++iter)
|
||
{
|
||
fprintf_s(fp, "\tcase %d: return %d;\n", (*iter).first, (*iter).second);
|
||
}
|
||
fprintf_s(fp, "\t}\n\treturn 256;\n};\n");
|
||
|
||
fprintf_s(fp, "\nstatic int GetPropSignalDataUsingCount64(int signalIdx) {\n\tswitch(signalIdx) {\n");
|
||
iter = VecDataUsingCount64.begin();
|
||
for (; iter != VecDataUsingCount64.end(); ++iter)
|
||
{
|
||
fprintf_s(fp, "\tcase %d: return %d;\n", (*iter).first, (*iter).second);
|
||
}
|
||
fprintf_s(fp, "\t}\n\treturn 256;\n};\n");
|
||
|
||
|
||
fprintf_s( fp, "\n#pragma pack(pop)\n" );
|
||
fclose(fp);
|
||
|
||
return bResult;
|
||
}
|
||
|
||
void CPropSignalManager::ChangeWorkingFolder( CString szPath )
|
||
{
|
||
m_szWorkingFolder = szPath;
|
||
m_szWorkingFolder += "\\Resource\\PropInfo";
|
||
}
|
||
|
||
bool CPropSignalManager::IsCanCheckOut()
|
||
{
|
||
#ifdef NO_LOCK
|
||
return true;
|
||
#endif //NO_LOCK
|
||
CString szFullName;
|
||
szFullName.Format( "%s\\checkout.ini", m_szWorkingFolder, m_szFileName );
|
||
|
||
FILE *fp;
|
||
fopen_s( &fp, szFullName, "r" );
|
||
if( fp == NULL ) return true;
|
||
fclose(fp);
|
||
return false;
|
||
}
|
||
|
||
bool CPropSignalManager::CheckOut()
|
||
{
|
||
#ifdef NO_LOCK
|
||
return true;
|
||
#endif //NO_LOCK
|
||
if( IsCanCheckOut() == false ) return false;
|
||
CString szFullName;
|
||
szFullName.Format( "%s\\checkout.ini", m_szWorkingFolder, m_szFileName );
|
||
|
||
FILE *fp;
|
||
fopen_s( &fp, szFullName, "wt" );
|
||
fprintf_s( fp, m_szMyName );
|
||
fclose(fp);
|
||
|
||
return true;
|
||
}
|
||
|
||
bool CPropSignalManager::CheckIn()
|
||
{
|
||
#ifdef NO_LOCK
|
||
return true;
|
||
#endif //NO_LOCK
|
||
CString szFullName;
|
||
szFullName.Format( "%s\\checkout.ini", m_szWorkingFolder, m_szFileName );
|
||
|
||
DeleteFile( szFullName );
|
||
return true;
|
||
} |