C언어 기능별로 함수 나누는 것에 대한 질문~
텃골
소켓프로그램인데요.. 서버에서 텍스트파일을 받아서 클라이언트에 저장하는건데..서버랑 클라이언트 소스를 짰는데.. 잘돼는데.. 이걸 좀 바꾸니까 연결이 안돼더라구요 ㅜㅠ
이게 원래 짠 소스구요~
-----서버 ------
#include stdio.h
#include winsock2.h
#include stdlib.h
#include string.h
#define BUFSIZE 256
#pragma comment(lib,ws2_32.lib)
int main()
{
WSADATA wsa;
SOCKET SV_socket;
SOCKET CL_socket;
FILE* fp;
SOCKADDR_IN SV_addr;
SOCKADDR_IN CL_addr;
int sz_CLaddr;
int ms_len;
char message[BUFSIZE];
char input_port[BUFSIZE];
printf(서버가 사용할 포트번호를 입력하시오 :);
fgets(input_port, BUFSIZE, stdin);
fp=fopen(memo.txt,r);
if(fp==NULL)
{
printf(File open error!\n);
}
if(WSAStartup(MAKEWORD(2,2), &wsa) !=0)
return -1;
SV_socket = socket(AF_INET, SOCK_STREAM,0);
if(SV_socket==INVALID_SOCKET)
{
printf(socket error!\n);
}
memset(&SV_addr,0,sizeof(SV_addr));
SV_addr.sin_family = AF_INET;
SV_addr.sin_addr.s_addr=htonl(INADDR_ANY);
SV_addr.sin_port = htons((unsigned short)atoi(input_port)); if(bind(SV_socket,(SOCKADDR*)&SV_addr,sizeof(SV_addr))==SOCKET_ERROR)
{
printf(bind error!\n);
}
if(listen(SV_socket, 5)==SOCKET_ERROR)
{
printf(listen error!\n);
}
sz_CLaddr = sizeof(CL_addr);
CL_socket = accept(SV_socket, (SOCKADDR *)&CL_addr,&sz_CLaddr);
if(CL_socket==INVALID_SOCKET)
{
printf(accept error!\n);
} printf(\n클라이언트 접속 IP주소 = %s, 포트번호 = %d\n,inet_ntoa(CL_addr.sin_addr),ntohs(CL_addr.sin_port));
while(1)
{
ms_len=fread(message, sizeof(char), BUFSIZE, fp);
send(CL_socket, message, ms_len, 0);
if(feof(fp))
break;
}
if(shutdown(CL_socket, SD_SEND) == SOCKET_ERROR)
{
printf(shutdown error!\n);
}
ms_len = recv(CL_socket, message, BUFSIZE-1, 0);
message[ms_len]=0;
fputs(message, stdout);
fclose(fp);
closesocket(CL_socket);
WSACleanup();
return 0;
}
클라이언트
#include stdio.h
#include winsock2.h
#include stdlib.h
#include string.h
#define BUFSIZE 256
#pragma comment(lib,ws2_32.lib)
int main(int argc,char* argv[])
{
SOCKET h_socket;
WSADATA wsa;
char message[BUFSIZE];
char serv_addr[BUFSIZE];
char serv_port[BUFSIZE];
FILE *fp;
SOCKADDR_IN SV_addr;
int ms_len;
printf(연결할 서버 주소를 입력하세요 :);
fgets(serv_addr, BUFSIZE, stdin);
printf(연결할 포트 번호를 입력하세요 :);
fgets(serv_port, BUFSIZE, stdin);
fp=fopen(receive.txt,w+);
if(WSAStartup(MAKEWORD(2,2), &wsa) !=0)
{
printf(WSAStartup error!\n);
}
h_socket = socket(AF_INET, SOCK_STREAM,0);
if(h_socket==INVALID_SOCKET)
{
printf(socket error!\n);
}
memset(&SV_addr,0,sizeof(SV_addr));
SV_addr.sin_family = AF_INET;
SV_addr.sin_addr.s_addr=inet_addr(serv_addr);
SV_addr.sin_port = htons((unsigned short)atoi(serv_port));
if(connect(h_socket, (SOCKADDR*)&SV_addr, sizeof(SV_addr)) ==SOCKET_ERROR)
{
printf(connect error!\n);
}
while((ms_len=recv(h_socket,message, BUFSIZE,0))!=0)
{
fwrite(message, sizeof(char), ms_len,fp);
}
fclose(fp);
closesocket(h_socket);
WSACleanup();
return 0;
}
기능별로 함수로 나눠서 하고싶어서 바꿨는데..ㅜㅠ
--------서버----------
#include stdio.h
#include winsock2.h
#include stdlib.h
#include string.h
#define BUFSIZE 256
#pragma comment(lib,ws2_32.lib)
void file_open_send();
int check();
int main()
{
WSADATA wsa;
SOCKET SV_socket;
SOCKET CL_socket;
SOCKADDR_IN SV_addr;
SOCKADDR_IN CL_addr;
// FILE *fp;
int ms_len;
char message[BUFSIZE];
int sz_CLaddr;
char input_port[BUFSIZE];
int op;
op=check();
if(WSAStartup(MAKEWORD(2,2), &wsa) !=0)
return -1;
SV_socket = socket(AF_INET, SOCK_STREAM,0);
if(SV_socket==INVALID_SOCKET)
printf(socket error!\n);
memset(&SV_addr,0,sizeof(SV_addr));
SV_addr.sin_family = AF_INET;
SV_addr.sin_addr.s_addr=htonl(INADDR_ANY);
SV_addr.sin_port = htons((unsigned short)atoi(input_port));
if(bind(SV_socket,(SOCKADDR*)&SV_addr,sizeof(SV_addr))==SOCKET_ERROR)
printf(bind error!\n);
if(listen(SV_socket, 5)==SOCKET_ERROR)
printf(listen error!\n);
sz_CLaddr = sizeof(CL_addr);
CL_socket = accept(SV_socket, (SOCKADDR *)&CL_addr,&sz_CLaddr);
if(CL_socket==INVALID_SOCKET)
printf(accept error!\n);
//continue;
printf(\n클라이언트 접속);
if(shutdown(CL_socket, SD_SEND) == SOCKET_ERROR)
printf(shutdown error!\n);
ms_len = recv(CL_socket, message, BUFSIZE-1, 0);
message[ms_len]=0;
fputs(message, stdout);
//fclose(fp);
closesocket(CL_socket);
file_open_send();
closesocket(CL_socket);
WSACleanup();
return 0;
}
int check()
{
int op=0;
char input_port[BUFSIZE];
printf(서버가 사용할 포트번호를 입력하시오 :);
fgets(input_port, BUFSIZE, stdin);
return op;
}
void file_open_send()
{
FILE *fp;
SOCKET CL_socket=0;
int ms_len;
char message[BUFSIZE];
fp=fopen(memo.txt,r);
if(fp==NULL)
printf(file open error!\n);
while(1)
{
ms_len=fread(message, sizeof(char), BUFSIZE, fp);
send(CL_socket, message, ms_len, 0);
if(feof(fp))
break;
}
if(shutdown(CL_socket, SD_SEND) == SOCKET_ERROR)
printf(shutdown error!\n);
ms_len = recv(CL_socket, message, BUFSIZE-1, 0);
message[ms_len]=0;
fputs(message, stdout);
fclose(fp);
closesocket(CL_socket);
}
----------클라이언트-------------
#include stdio.h
#include winsock2.h
#include stdlib.h
#include string.h
#define BUFSIZE 1024
#pragma comment(lib,ws2_32.lib)
void Serch(void);
void Filedown();
int Connectsvr();
int
int Menuprint();
int main()
{
int option;
do
{
system(cls);
option=Menuprint();
switch(option)
{
case 1: Connectsvr(); break;
case 2: Filedown(); break;
case 3: Serch(); break;
case 9: break;
default:
{
printf(옵션을 잘못 선택 하셨습니다!\n);
printf(아무키나 누르세요...\n);
getchar();
getchar();
system(cls);
}
}
}while(option!=9);
}
int Menuprint()
{
int option=0;
printf(소켓 연결 프로그램 \n\n);
printf( 1. CONNECT\n 2. FILE DOWN\n 3. SERCH\n 9. 종료\n);
printf($);
fflush(stdin);
scanf(%d,&option);
return option;
}
int Connectsvr()
{
WSADATA wsa;
SOCKET h_socket;
char message[BUFSIZE];
int ms_len;
SOCKADDR_IN SV_addr;
char serv_addr[BUFSIZE];
char serv_port[BUFSIZE]; system(cls);
printf(====== 1. connect Mode ======\n);
printf(연결할 서버 주소를 입력하세요 :);
fgets(serv_addr, BUFSIZE, stdin);
getchar();
// fflush(stdin);
printf(연결할 포트 번호를 입력하세요 :);
fgets(serv_port, BUFSIZE, stdin);
getchar();
if(WSAStartup(MAKEWORD(2,2), &wsa) !=0)
printf(WSAStartup error!\n);
h_socket = socket(AF_INET, SOCK_STREAM,0);
if(h_socket==INVALID_SOCKET)
printf(socket error!\n);
memset(&SV_addr,0,sizeof(SV_addr));
SV_addr.sin_family = AF_INET;
SV_addr.sin_addr.s_addr=inet_addr(serv_addr);
SV_addr.sin_port = htons((unsigned short)atoi(serv_port));
if(connect(h_socket, (SOCKADDR*)&SV_addr, sizeof(SV_addr)) ==SOCKET_ERROR)
printf(connect error!\n);
system(cls);
closesocket(h_socket);
WSACleanup();
return 0;
}
void Filedown()
{
FILE *fp;
int ms_len;
SOCKET h_socket=0;
char message[BUFSIZE];
system(cls);
printf(====== 2. file down Mode ======\n);
fp=fopen(receive.txt,w+);
while((ms_len=recv(h_socket,message, BUFSIZE,0))!=0)
{
fwrite(message, sizeof(char), ms_len,fp);
}
system(cls);
printf(file down OK!\n);
fclose(fp);
closesocket(h_socket);
}
void Serch()
{
system(cls);
printf(====== 4. serch Mode ======\n);
getchar();
serchmodul(inf);
//printf(%s\n, message[BUFSIZE]);
}