//-----------------------------------------------------------------
//   ■小さなチャットサーバのクライアント側ソフト
//-----------------------------------------------------------------
// 最初、ホスト名を聞いてくるので, 同じコンピュータ内ならlocalhost,
// 遠隔ホストならIPアドレスを指定する。
// サーバから"Your Name?>"と聞いてくるので、名前を入力して通信開始。

#include "stdio.h"
#include "string.h"
#include "winsock2.h"
#include "conio.h"
#include "process.h"
#pragma comment(lib, "WS2_32.LIB")
SOCKADDR_IN addrServer;SOCKET soc;int icount;
void dspErr(char msg[]){//■エラー表示
	printf("\n** %s ** Errno %d\n",msg,WSAGetLastError());
}
void check2J(char INP[]){//■VT100用画面クリアメッセージを無視
	if(strlen(INP)>=4)
		if(INP[0]==0x1B && INP[1]=='[' && INP[2]=='2' && INP[3]=='J'){
			INP[0]='\n';int i=1;
			for(int j=4; INP[j]!=0;i++,j++) INP[i]=INP[j];
			INP[i]=0;
		}
}
int checkGoodBy(char INP[]){//■サーバからのGoodByメッセージの判定
	char GoodBy[]="\r\nGreet : Good by!"; int num=strlen(GoodBy);
	for(int i=0;i<num;i++) if(INP[i]!=GoodBy[i]) return false;
	return true;
}
void recDT(void *dt){//■サーバからの受信データを表示
	char buff[1024]; int nRecv;
	while(1){
		nRecv=recv(soc, buff, sizeof(buff),0); buff[nRecv]=0;
		if(nRecv==SOCKET_ERROR)	{dspErr("受信失敗"); break;}
		else if(checkGoodBy(buff)){	printf(buff);break;	}//サーバからのGood byで終わる
		else {	check2J(buff); printf(buff);}
		Sleep(100);
	}
	icount--;// 自スレッドが終わったことを親に報告
}
void client(void *dt){//■クライアントからの入力をサーバに送信
	char buff[512];char msg[516];int ch;
	while(scanf("%s",buff)){
		sprintf(msg,"%s\r\n",buff);
		if(send(soc,msg,strlen(msg),0)==SOCKET_ERROR) dspErr("送信失敗");
		if(strcmp(buff,"$by")==0)break;
		Sleep(100);
	}
	icount--;// 自スレッドが終わったことを親に報告
}
void endProc(){//■終了処理
	char msg[1024]; int nByte;//終わる前に"$by"を送る(ここは動いているよう)
	sprintf(msg,"$by\r\n");	send(soc,msg,strlen(msg),0);
	nByte=recv(soc,msg,sizeof(msg),0);//サーバからの受信待ちのつもり
	msg[nByte]=0; printf("%s",msg);   //(動かないでサーバ側が送信失敗になる）
	shutdown(soc,2);closesocket(soc); WSACleanup();	
}
int main(void){
	char str[512]; WSADATA DT; SOCKADDR_IN name ={AF_INET};
	int addrLen=sizeof(SOCKADDR); WSAStartup(2,&DT);
	if(atexit((void(*)(void))(endProc))){
		printf("atExit(endProc)失敗\n");
		return 1;
	}
	soc=socket(AF_INET,SOCK_STREAM, 0);
	if(soc==INVALID_SOCKET)dspErr("ソケット作成エラー");
	else {
		LPHOSTENT PH; unsigned long IPAddress;     //サーバのIPアドレスを取得
		printf("ホスト名入力>"); scanf("%s",str); IPAddress=inet_addr(str);
		if(IPAddress==-1) {
			if((PH=gethostbyname(str))!=NULL)IPAddress=*((unsigned long *)(PH->h_addr));
			else {
					dspErr("ホスト名取得失敗");
					shutdown(soc,2);closesocket(soc); WSACleanup();	return -1;
			}
		}
		addrServer.sin_family=AF_INET;             //サーバとの接続
		addrServer.sin_addr.S_un.S_addr=IPAddress;
		addrServer.sin_port=htons((unsigned short)2048);
		if(connect(soc,(LPSOCKADDR)&addrServer,sizeof(addrServer))==SOCKET_ERROR)
								dspErr("サーバへの接続失敗");
		else{	
			icount=2;                          //子スレッド開始(2個)
			_beginthread(recDT ,0,NULL); _beginthread(client,0,NULL);
			while(icount!=0)Sleep(100);        //子スレッドが終わるまで待つ
		}
	}
	shutdown(soc,2);closesocket(soc); WSACleanup();//終了処理
	return 0;
}