uint
O2Agent::
LaunchThread(void)
{
for (uint i = 0; i < Services.size(); i++) {
Services[i]->SetActive(true);
Services[i]->OnServiceStart();
}
Agent_GetGlobalIP->SetActive(true);
Agent_GetGlobalIP->OnServiceStart();
if (Logger)
Logger->AddLog(O2LT_INFO, MODULE, L"エージェント正常起動");
time_t prevconnecttime = time(NULL);
uint index = 0;
while (Active) {
index %= Services.size();
/*
* ウェイト処理
*/
if (NodeDB->Count() == 0) {
#if DEBUG_THREADLOOP && defined(_DEBUG)
TRACEA("■Node 0\n");
#endif
Sleep(1000);
continue;
}
uint pasttime_ms = ((uint)time(NULL) - (uint)prevconnecttime) * 1000;
if (pasttime_ms < Interval) {
uint sleeptime_ms = Interval - pasttime_ms;
#if DEBUG_THREADLOOP && defined(_DEBUG)
TRACEA("Wait interval\n");
#endif
if (Agent_GetGlobalIP->IsActive()) {
;
//グローバルIP未確定のときは急ぐ
}
else {
if (sleeptime_ms < 1000)
Sleep(sleeptime_ms);
else
Sleep(1000);
continue;
}
}
/*
* サービス起動
*/
O2AgentService *service = Services[index];
if (Agent_GetGlobalIP->IsActive()) //グローバルIP未決定のとき
service = Agent_GetGlobalIP; //強制的にAgent_GetGlobalIP起動
#if DEBUG_SESSION_COUNT && defined(_DEBUG)
wchar_t abc[256]; swprintf(abc, L"● count:%d limit:%d %s\n",
service->GetSessionCount(), service->GetSessionLimit(), service->GetName());
TRACEW(abc);
#endif
if (!service->IsActive()) {
index++;
continue;
}
if (service->GetSessionCount() >= service->GetSessionLimit()) {
Sleep(500);
index++;
continue;
}
O2SocketSession *ss = new O2SocketSession();
ss->service = (void*)service;
ulong ip;
ushort port;
wstring ipstr;
//接続先ノード決定
//接続 TODO:クラスタリング
if (service->HaveNodeDecision()) {
service->OnLaunch(ss);
if (!ss->IsActive()) {
delete ss;
index++;
continue;
}
ip = ss->sin.sin_addr.S_un.S_addr;
port = htons(ss->sin.sin_port);
}
else if (!NodeDB->GetNodeByPriority(ip, port)) {
if (Logger) {
Logger->AddLog(O2LT_WARNING, MODULE,
L"有効なノード情報が無いので接続不可(%s)",
service->GetName());
}
delete ss;
index++;
continue;
}
#if !defined(_DEBUG)
if (ip == Profile->GetIP() || ip == 0x0100007f) {
#if 0
struct in_addr addr;
addr.S_un.S_addr = ip;
if (Logger) {
Logger->AddLog(O2LT_WARNING, MODULE,
"自ノード情報破棄 IP: %s Port: %d",
inet_ntoa(addr), port);
}
#endif
NodeDB->DeleteNode(ip);
delete ss;
index++;
continue;
}
#endif
#if defined(_DEBUG)
ulong2ipstr(ip, ipstr);
#else
ip2e(ip, ipstr);
#endif
/*
* Socket生成
*/
ss->sock = socket(AF_INET, SOCK_STREAM, 0);
if (ss->sock == INVALID_SOCKET) {
if (Logger)
Logger->AddLog(O2LT_ERROR, MODULE, L"ソケット生成に失敗");
delete ss;
index++;
continue;
}
ss->sin.sin_family = AF_INET;
ss->sin.sin_port = htons(port);
ss->sin.sin_addr.S_un.S_addr = ip;
/*
* Connect
*/
int timeout = service == Agent_GetGlobalIP ? GIP_CONNECT_TIMEOUT_MS : CONNECT_TIMEOUT_MS;
if (connect2(ss->sock, (struct sockaddr*)&ss->sin, sizeof(ss->sin), timeout) != 0) {
//NodeDB->DeleteNode(ip);
NodeDB->IncrementErrorCount(ip, 1);
if (Logger) {
Logger->AddLog(O2LT_NET, MODULE,
L"connect失敗(%s %s:%d)",
service->GetName(), ipstr.c_str(), port);
}
closesocket(ss->sock);
delete ss;
index++;
continue;
}
if (hwndSetIconCallback) {
PostMessage(hwndSetIconCallback, msgSetIconCallback, 1, 0);
}
//launch
if (!service->HaveNodeDecision()) {
service->OnLaunch(ss);
if (!ss->IsActive()) {
closesocket(ss->sock);
delete ss;
index++;
continue;
}
}
ss->UpdateTimer();
service->IncrementCount();
Lock();
sss.push_back(ss);
Unlock();
SetEvent(SessionAvailableEvent);
NodeDB->AddStatusBit(ip, O2_NODESTATUS_LINKEDTO);
prevconnecttime = time(NULL);
index++;
#if DEBUG_SESSION_COUNT && defined(_DEBUG)
swprintf(abc, L"●● count:%d limit:%d %s\n",
service->GetSessionCount(), service->GetSessionLimit(), service->GetName());
TRACEW(abc);
#endif
if (Logger) {
Logger->AddLog(O2LT_NET, MODULE,
L"connect成功 (%s %s:%d)",
service->GetName(), ipstr.c_str(), port);
}
}
//end
Lock();
SetEvent(SessionAvailableEvent);
Unlock();
for (uint i = 0; i < Services.size(); i++) {
O2AgentService *service = Services[i];
if (service->IsActive()) {
service->SetActive(false);
service->OnServiceEnd();
}
}
if (Logger)
Logger->AddLog(O2LT_INFO, MODULE, L"エージェント正常終了");
return (0);
}
uint
O2Agent::
NetIOThread(void)
{
O2SocketSessionListIt ssit;
int lasterror;
while (Active) {
//setup FD
fd_set readfds;
fd_set writefds;
FD_ZERO(&readfds);
FD_ZERO(&writefds);
//wait
Lock();
uint n = sss.size();
Unlock();
if (n == 0) {
ResetEvent(SessionAvailableEvent);
}
WaitForSingleObject(SessionAvailableEvent, INFINITE);
if (!Active)
break;
//FD
Lock();
for (ssit = sss.begin(); ssit != sss.end(); ssit++) {
O2SocketSession *ss = *ssit;
if (!ss->sendcomplete)
FD_SET(ss->sock, &writefds);
else
FD_SET(ss->sock, &readfds);
}
Unlock();
//select
struct timeval tv;
tv.tv_sec = SELECT_TIMEOUT_MS/1000;
tv.tv_usec = SELECT_TIMEOUT_MS%1000;
select(0, &readfds, &writefds, NULL, &tv);
//
// 既存セッションとの送受信
//
Lock();
ssit = sss.begin();
Unlock();
while (1) {
bool loopend = false;
Lock();
if (ssit == sss.end())
loopend = true;
Unlock();
if (loopend)
break;
#if DEBUG_THREADLOOP && defined(_DEBUG)
TRACEA("[NetIO]\n");
#endif
O2SocketSession *ss = *ssit;
O2AgentService *service = (O2AgentService*)ss->service;
ushort port = htons(ss->sin.sin_port);
wstring ipstr;
#if defined (_DEBUG)
ulong2ipstr(ss->sin.sin_addr.S_un.S_addr, ipstr);
#else
ip2e(ss->sin.sin_addr.S_un.S_addr, ipstr);
#endif
//send
if (FD_ISSET(ss->sock, &writefds)) {
int len;
const char *buff = ss->GetSendBuffer(len);
if (len > 0) {
int n = send(ss->sock, buff, len, 0);
#if DEBUG_THREADLOOP && defined(_DEBUG)
TRACEA("[A]send\n");
#endif
if (n > 0) {
ss->UpdateSend(n);
NodeDB->IncrementSendByte_me2n(ss->sin.sin_addr.S_un.S_addr, n);
service->IncrementSendByte(n);
service->OnSend(ss);
ss->UpdateTimer();
if (hwndSetIconCallback) {
PostMessage(hwndSetIconCallback, msgSetIconCallback, 1, 0);
}
}
else if (n == 0) {
;
}
else if ((lasterror = WSAGetLastError()) != WSAEWOULDBLOCK) {
if (Logger) {
Logger->AddLog(O2LT_NET, MODULE,
L"送信エラー(%d) (%s %s:%d %dbyte)",
lasterror, service->GetName(), ipstr.c_str(), port, ss->sbuff.size());
}
ss->SetActive(false);
}
}
}
//recv
if (FD_ISSET(ss->sock, &readfds)) {
char buff[RECVBUFFSIZE];
int n = recv(ss->sock, buff, RECVBUFFSIZE, 0);
#if DEBUG_THREADLOOP && defined(_DEBUG)
TRACEA("[A]recv\n");
#endif
if (n > 0) {
ss->AppendRecv(buff, n);
NodeDB->IncrementRecvByte_me2n(ss->sin.sin_addr.S_un.S_addr, n);
service->IncrementRecvByte(n);
service->OnRecv(ss);
ss->UpdateTimer();
if (hwndSetIconCallback) {
PostMessage(hwndSetIconCallback, msgSetIconCallback, 0, 0);
}
}
else if (n == 0) {
if (Logger) {
Logger->AddLog(O2LT_NET, MODULE,
L"受信0 (%s %s:%d)",
service->GetName(), ipstr.c_str(), port);
}
ss->SetActive(false);
}
else if ((lasterror = WSAGetLastError()) != WSAEWOULDBLOCK) {
if (Logger) {
Logger->AddLog(O2LT_NET, MODULE,
L"受信エラー(%d) (%s %s:%d %dbyte)",
lasterror, service->GetName(), ipstr.c_str(), port, ss->rbuff.size());
}
ss->SetActive(false);
}
}
//delete
if (!ss->IsActive() || ss->GetPastTime() >= SOCKET_TIMEOUT_S) {
void *databk = ss->data;
ulong ip = ss->sin.sin_addr.S_un.S_addr;
closesocket(ss->sock);
service->OnClose(ss);
NodeDB->RemoveStatusBit(ip, O2_NODESTATUS_LINKEDTO);
NodeDB->AddStatusBit(ip, O2_NODESTATUS_PASTLINKEDTO);
service->DecrementCount();
if (!ss->rbuff.empty() && !ss->sbuff.empty()) {
NodeDB->UpdateLastModified(ip, 2);
service->IncrementTotalSessionCount();
}
else if (ss->rbuff.empty())
NodeDB->IncrementErrorCount(ip, 1);
if (!ss->rbuff.empty() && (!databk || ss->rbuff[0] != 'H')) {
NodeDB->DeleteNode(ip);
if (Logger) {
Logger->AddLog(O2LT_WARNING, MODULE,
L"不正データ受信 (ノード削除: %s:%d)",
ipstr.c_str(), htons(ss->sin.sin_port));
}
if (O2DEBUG) {
char filename[256];
sprintf_s(filename, 256, "dump\\A_%d.txt", rand());
FILE *fp;
fopen_s(&fp, filename, "wb");
fwrite(ss->rbuff.c_str(), ss->rbuff.size(), 1, fp);
fclose(fp);
}
}
#if DEBUG_SESSION_COUNT && defined(_DEBUG)
wchar_t abc[256]; swprintf(abc, L"■■削除■■ count:%d limit:%d %s\n",
service->GetSessionCount(), service->GetSessionLimit(), service->GetName());
TRACEW(abc);
#endif
Lock();
ssit = sss.erase(ssit);
Unlock();
delete ss;
continue;
}
Sleep(1);
ssit++;
}
}
//end
Lock();
for (ssit = sss.begin(); ssit != sss.end(); ssit++) {
O2SocketSession *ss = *ssit;
closesocket(ss->sock);
ss->sock = 0;
((O2AgentService*)((O2AgentService*)ss->service))->OnClose(ss);
delete ss;
}
sss.clear();
Unlock();
return (0);
}
void
O2Client::
NetIOThread(void)
{
O2SocketSessionPListIt ssit;
int lasterror;
while (Active) {
// Signal Off
SessionListLock.Lock();
{
if (sss.empty())
SessionExistSignal.Off();
}
SessionListLock.Unlock();
// Wait
SessionExistSignal.Wait();
if (!Active) break;
// Setup FD
fd_set readfds;
fd_set writefds;
FD_ZERO(&readfds);
FD_ZERO(&writefds);
SessionListLock.Lock();
{
for (ssit = sss.begin(); ssit != sss.end(); ssit++) {
O2SocketSession *ss = *ssit;
if (ss->can_send)
FD_SET(ss->sock, &writefds);
if (ss->can_recv)
FD_SET(ss->sock, &readfds);
}
}
SessionListLock.Unlock();
// select
struct timeval tv;
tv.tv_sec = SELECT_TIMEOUT_MS/1000;
tv.tv_usec = SELECT_TIMEOUT_MS%1000;
select(0, &readfds, &writefds, NULL, &tv);
if (!Active) break;
//
// 既存セッションとの送受信
//
SessionListLock.Lock();
ssit = sss.begin();
SessionListLock.Unlock();
while (Active) {
bool end = false;
SessionListLock.Lock();
{
if (ssit == sss.end())
end = true;
}
SessionListLock.Unlock();
if (end) break;
O2SocketSession *ss = *ssit;
// IP文字列
wstring ipstr;
if (O2DEBUG)
ulong2ipstr(ss->ip, ipstr);
else
ip2e(ss->ip, ipstr);
// Send
if (FD_ISSET(ss->sock, &writefds)) {
int len;
const char *buff = ss->GetNextSend(len);
if (len > 0) {
int n = send(ss->sock, buff, len, 0);
if (n > 0) {
ss->UpdateSend(n);
ss->UpdateTimer();
SendByte += n;
OnSend(ss);
if (hwndSetIconCallback) {
PostMessage(hwndSetIconCallback, msgSetIconCallback, 1, 0);
}
}
else if (n == 0) {
;
}
else if ((lasterror = WSAGetLastError()) != WSAEWOULDBLOCK) {
if (Logger) {
Logger->AddLog(O2LT_NETERR, ClientName.c_str(),
ss->ip, ss->port, L"送信エラー(%d)", lasterror);
}
ss->error = true;
ss->Deactivate();
}
}
}
// Recv
if (FD_ISSET(ss->sock, &readfds)) {
char buff[RECVBUFFSIZE];
int n = recv(ss->sock, buff, RECVBUFFSIZE, 0);
if (n > 0) {
ss->AppendRecv(buff, n);
ss->UpdateTimer();
RecvByte += n;
OnRecv(ss);
if (hwndSetIconCallback) {
PostMessage(hwndSetIconCallback, msgSetIconCallback, 0, 0);
}
}
else if (n == 0) {
/*if (Logger) {
Logger->AddLog(O2LT_NETERR, ClientName.c_str(),
ss->ip, ss->port, L"受信0");
}*/
ss->Deactivate();
}
else if ((lasterror = WSAGetLastError()) != WSAEWOULDBLOCK) {
if (Logger) {
Logger->AddLog(O2LT_NETERR, ClientName.c_str(),
ss->ip, ss->port, L"受信エラー(%d)", lasterror);
}
ss->error = true;
ss->Deactivate();
}
}
// Delete
if (ss->CanDelete()) {
bool timeout = ss->GetPastTime() >= ss->timeout_s ? true : false;
if (!ss->IsActive() || timeout) {
if (timeout) {
if (Logger) {
Logger->AddLog(O2LT_NETERR, ClientName.c_str(),
ss->ip, ss->port, L"timeout");
}
}
closesocket(ss->sock);
ss->sock = 0;
SessionListLock.Lock();
ssit = sss.erase(ssit);
SessionListLock.Unlock();
OnClose(ss);
continue;
}
}
Sleep(1);
SessionListLock.Lock();
ssit++;
SessionListLock.Unlock();
}
}
// End
SessionListLock.Lock();
for (ssit = sss.begin(); ssit != sss.end(); ssit++) {
O2SocketSession *ss = *ssit;
closesocket(ss->sock);
ss->sock = 0;
OnClose(ss);
}
sss.clear();
SessionListLock.Unlock();
}