Tag: <span>DBM 서비스 구현</span>

DbmSVC를 구현합시다. EH 메신저 솔루션에 DbmSVC 이름의 콘솔 응용 프로젝트를 만드세요. 그리고 이 책에서는 DbmSVC를 구현한 이후에 테스트 프로젝트를 만들고 테스트하는 부분은 다루지 않을게요.

DbmSVC 프로젝트에 Program.cpp 파일을 추가하여 소스를 구현합시다.

#include "..\\Common\\EHMessenger.h"
#pragma comment(lib,"ws2_32")
#pragma comment(lib,"..\\Debug\\EHPacketLib")
#pragma comment(lib,"..\\Debug\\DbmLib")
#pragma comment(lib,"..\\Debug\\EHWrapSocketLib")
#include <Windows.h>
#include <iostream>
using namespace std;
#pragma warning(disable:4996)

Dbm 서비스에서는 계정 정보를 관리해야 합니다. 여기에서는 파일 매핑을 이용할게요.

#define DBFNAME          TEXT("EHDB.dat")

Dbm 서비스에서 관리할 계정 정보를 구조체로 정의합시다.

#define MAX_ID_LEN     256
#define MAX_PW_LEN     256
#define MAX_NAME_LEN   256
#define MAX_ELEMENT    100
struct DBElem
{
    char id[MAX_ID_LEN];
    char pw[MAX_PW_LEN];
    char name[MAX_NAME_LEN];
    int status;
};

Dbm 서비스에서 계정 정보를 관리하기 위해 파일 매핑을 사용합시다. 이를 위해 파일 핸들, 파일 매핑 핸들과 매핑 시작 주소와 현재까지 추가한 계정 수를 위한 변수를 선언하세요.

HANDLE hFile;
HANDLE hMapping;
DBElem *base;
int ecnt;

진입점 main 함수에서는 윈속을 초기화한 후 Dbm 서버를 가동하고 윈속을 해제합니다.

void StartDbmServer();
int main()
{
    WSADATA wsadata;
    WSAStartup(MAKEWORD(2,2),&wsadata);
    StartDbmServer();
    WSACleanup();
    return 0;
}

Dbm 서버를 가동하는 함수를 작성합시다.

void InitializeDB();
DWORD WINAPI DoIt(LPVOID pin);
void StartDbmServer()
{

먼저 DB를 초기화합니다. 이 부분은 별도의 함수로 작성합시다.

    InitializeDB();

DBM 서버 소켓을 설정합니다.

    SOCKET sock = EHWrapSocket::CreateTCPServer(DBM_PORT,5);
    SOCKADDR_IN clientaddr;
    int len = sizeof(clientaddr);
    SOCKET dosock;
    DWORD ThreadID;

DBM 서버 소켓이 연결을 수락하면 이에 관한 처리는 별도의 쓰레드에서 수행하게 합시다.

    while(1)
    {
        dosock = accept(sock,(SOCKADDR *)&clientaddr, &len);
        CloseHandle(CreateThread(0,0,DoIt,(LPVOID)dosock,0,&ThreadID));
    } 
}

DB를 초기화하는 함수를 작성합시다.

void MakeDB();
void InitializeDB()
{

먼저 DB 파일을 열기 시도합니다.

    hFile = CreateFile(DBFNAME,
        GENERIC_READ|GENERIC_WRITE,
        FILE_SHARE_READ|FILE_SHARE_WRITE,0,OPEN_EXISTING,0,0);

만약 유효하지 않은 파일 핸들이면 DB 파일이 없다는 것이므로 비어있는 DB를 만들어야 합니다. DB를 만드는 부분은 별도의 함수로 작성합시다. 그리고 DB 파일을 만든 후에 파일 열기하세요.

    if(hFile == INVALID_HANDLE_VALUE)
    {
        MakeDB();
        hFile = CreateFile(DBFNAME,
                GENERIC_READ|GENERIC_WRITE,
                FILE_SHARE_READ|FILE_SHARE_WRITE,0,OPEN_EXISTING,0,0);
    }

연 파일 핸들로 파일 매핑 핸들을 생성합니다.

    hMapping = CreateFileMapping(hFile,0,PAGE_READWRITE,0,0,0);

파일 매핑 핸들로 전체 영역을 매핑합니다.

    base = (DBElem *)MapViewOfFile(hMapping, FILE_MAP_READ|FILE_MAP_WRITE, 0,0,0);

순차적으로 이동하면서 현재 DB에 보관한 계정 개수를 파악합니다.

    for(ecnt = 0; ecnt<MAX_ELEMENT; ++ecnt)
    {

처음으로 상태가 STS_UNREG인 곳을 만나면 빈 곳을 만난 것이므로 반복문을 탈출합니다.

        if(base[ecnt].status == STS_UNREG)
        {
            break;
        }
    }
}

비어있는 DB를 만드는 함수를 작성합시다.

void MakeDB()
{
    DBElem emptybase[MAX_ELEMENT]={0};

DB 파일을 생성하여 빈 레코드를 파일에 기록합니다.

    hFile = ::CreateFile(DBFNAME,GENERIC_WRITE,0,0,CREATE_ALWAYS,0,0);
    DWORD dw;
    WriteFile(hFile,emptybase,sizeof(emptybase),&dw,0);
}

Dbm 서버 소켓이 연결 수락하였을 때 처리하는 쓰레드 진입점 함수를 작성합시다.

void IdExistProc(SOCKET sock,EHPacket *ep);
void AddUserInfoProc(SOCKET sock,EHPacket *ep);
void UserStsProc(SOCKET sock,EHPacket *ep);
void IsCorrectProc(SOCKET sock,EHPacket *ep);
void ChangeStsProc(SOCKET sock,EHPacket *ep);
void RemoveUserProc(SOCKET sock,EHPacket *ep);
DWORD WINAPI DoIt(LPVOID pin)
{

쓰레드 입력 인자로 전달받은 소켓을 형식 변환합니다.

    SOCKET sock = (SOCKET)pin;

메시지를 수신합니다.

    EHPacket ep(sock);

메시지 종류에 따라 처리하는 프로시져를 호출합니다.

    switch(ep.GetMsgId())
    {
    case MID_IDEXIST: IdExistProc(sock,&ep); break;
    case MID_ADDUSERINFO: AddUserInfoProc(sock,&ep); break;
    case MID_USERSTS: UserStsProc(sock,&ep); break;
    case MID_ISCORRECT: IsCorrectProc(sock,&ep); break;
    case MID_CHANGESTS: ChangeStsProc(sock,&ep); break;
    case MID_REMOVEUSER: RemoveUserProc(sock,&ep); break;
    }
    closesocket(sock);
    return 0;
}

아이디 존재 요청 확인 메시지를 수신하였을 때 처리하는 함수를 작성합시다.

DBElem *FindDBElem(string id);
void IdExistProc(SOCKET sock,EHPacket *ep)
{
    cout<<"아이디 존재 요청 확인 메시지 수신"<<endl;

수신한 메시지를 IDExist 형식으로 변환합니다.

    IDExist ie(ep);

DB 파일에서 수신한 메시지의 아이디와 같은 레코드를 찾습니다. 이 부분은 별도의 함수로 작성합시다.

    DBElem *elem = FindDBElem(ie.GetId());
    int result = RES_ID_NOTEXIST; 

검색 결과가 유효하면 아이디가 존재하는 것입니다.

    if(elem)
    {
        cout<<ie.GetId()<<" 아이디 존재"<<endl;
        result = RES_ID_EXIST;
    }

그렇지 않으면 아이디가 존재하지 않는 것입니다.

    else
    {
        cout<<ie.GetId()<<" 아이디 존재하지 않음"<<endl;
    }

아이디 존재 요청 응답 메시지를 만들어 전송합니다.

    IDExistAck iea(result);
    cout<<"아이디 존재 요청 응답 메시지 전송"<<endl;
    iea.Serialize(sock);
}

DB 파일에서 아이디로 레코드를 찾는 함수를 작성합시다.

DBElem *FindDBElem(string id)
{

순차적으로 이동하면서 아이디가 같은 부분을 찾습니다.

    for(int i = 0; i<ecnt; ++i)
    {
        if(strcmp(base[i].id,id.c_str())==0)
        {
            return base+i;
        }
    }
    return 0;
}

계정 추가 메시지를 수신하였을 때 처리하는 함수를 작성합시다.

void AddUserInfoProc(SOCKET sock,EHPacket *ep)
{

수신한 메시지를 계정 추가 메시지로 변환하고 마지막 레코드 뒤에 수신한 메시지 정보를 기록합니다.

    AddUserInfo aui(ep);
    cout<<aui.GetId()<<" 계정 추가 메시지 수신"<<endl;
    strcpy(base[ecnt].id,aui.GetId().c_str());
    strcpy(base[ecnt].pw,aui.GetPW().c_str());
    strcpy(base[ecnt].name,aui.GetName().c_str());
    base[ecnt].status = STS_REG;
    ecnt++;
}

계정 상태 요청 메시지를 수신하였을 때 처리하는 함수를 작성합시다.

void UserStsProc(SOCKET sock,EHPacket *ep)
{
    cout<<"계정 상태 요청 메시지 수신"<<endl;

수신한 메시지를 계정 상태 요청 메시지로 변환합니다.

    UserSts usts(ep);

수신한 메시지의 아이디가 있는 레코드를 찾습니다.

    DBElem *elem = FindDBElem(usts.GetId());

레코드를 찾았으면 레코드의 상태로 결과를 변경합니다.

    int result = STS_UNREG; 
    if(elem)
    {
        cout<<usts.GetId()<<" 계정 상태 확인"<<endl;
        result = elem->status;
    }
    else
    {
        cout<<usts.GetId()<<" 계정을 찾을 수 없음"<<endl;
    }

계정 상태 요청 응답 메시지를 생성하여 전송합니다.

    UserStsAck ustsa(result);
    cout<<"계정 상태 요청 응답 메시지 전송"<<endl;
    ustsa.Serialize(sock);
}

비밀 번호 일치 확인 요청 메시지를 수신하였을 때 처리하는 함수를 작성합시다.

void IsCorrectProc(SOCKET sock,EHPacket *ep)
{
    cout<<"비밀번호 일치 확인 요청 메시지 수신"<<endl;

수신한 메시지를 비밀번호 일치 확인 요청 메시지로 변환합니다.

    IsCorrect ic(ep);

수신한 메시지의 아이디와 같은 레코드를 찾습니다.

    DBElem *elem = FindDBElem(ic.GetId());
    int result = IDPW_NOTCORRECT; 
    if(elem)
    {

찾은 레코드의 비밀 번호와 수신한 메시지의 비밀번호가 일치하는지 확인합니다.

        if(strcmp(elem->pw,ic.GetPW().c_str())==0)
        {
            cout<<"일치"<<endl;
            result = IDPW_CORRECT;
        }
        else
        {
            cout<<"불일치"<<endl;
        }
    }
    else
    {
        cout<<"오류!!! "<<ic.GetId()<<" 계정을 찾을 수 없음"<<endl;
    }

비밀 번호 일치 확인 요청 응답 메시지를 생성하여 전송합니다.

    IsCorrectAck ica(result);
    cout<<"비밀 번호 일치 확인 요청 응답 메시지 전송"<<endl;
    ica.Serialize(sock);
}

계정 상태 변경 요청 메시지를 수신하였을 때 처리하는 함수를 작성합시다.

void ChangeStsProc(SOCKET sock,EHPacket *ep)
{

수신한 메시지를 계정 상태 변경 요청 메시지로 변환합니다.

    ChangeSts csts(ep);
    cout<<csts.GetId()<<" 계정 상태 변경 요청 메시지 수신"<<endl;

수신한 메시지의 아이디와 일치하는 레코드를 찾습니다.

    DBElem *elem = FindDBElem(csts.GetId());
    if(elem)
    {

찾은 레코드의 상태를 수신함 메시지의 상태로 변경합니다.

        elem->status = csts.GetStatus();
        cout<<csts.GetId()<<" 계정 상태 변경"<<endl;
    }

수신한 메시지의 아이디와 일치하는 레코드를 찾지 못하는 것은 오류가 있는 것입니다.

    else
    {
        cout<<"오류!!! "<<csts.GetId()<<" 계정을 찾을 수 없음"<<endl;
    }
}

계정 삭제 요청 메시지를 수신하였을 때 처리하는 함수를 작성합시다.

void RemoveUserProc(SOCKET sock,EHPacket *ep)
{

수신한 메시지를 계정 삭제 요청 메시지로 변환합니다.

    RemoveUser ru(ep);
    cout<<ru.GetId()<<" 계정 삭제 요청 메시지 수신"<<endl;

수신한 메시지의 아이디와 일치하는 레코드를 찾습니다.

    DBElem *elem = FindDBElem(ru.GetId());
    if(elem)
    {

레코드를 찾았으면 계정 수를 감소합니다.

        ecnt--;

맨 뒤에 있던 레코드를 삭제할 레코드에 복사합니다.

        memcpy(elem,base+ecnt,sizeof(DBElem));

맨 뒤에 있는 레코드를 초기화합니다.

        memset(base+ecnt,0,sizeof(DBElem));
    }

만약 레코드를 찾지 못했으면 오류입니다.

    else
    {
        cout<<"오류!!! "<<ru.GetId()<<" 계정을 찾을 수 없음"<<endl;
    }
}