#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <WinSock2.h>
#include <ws2tcpip.h>
#include <process.h>
#include <conio.h>
#pragma comment(lib, "ws2_32.lib")
int TCP = 0; // 전역변수, TCP가 0이면 UDP 방식, 1이면 TCP 방식
// 에러 감지용
void error_handling(const char* message) {
fputs(message, stderr);
fputc('\n', stderr);
exit(1);
}
// 서버 thread
unsigned __stdcall server_thread(void* arg) {
int serv_sock; // 변수: 서버 소켓
// UDP 연결일경우
if (TCP == 0) {
serv_sock = socket(AF_INET, SOCK_DGRAM, 0);
if (serv_sock == -1)
error_handling("에러: 소켓에 문제가 발생했습니다.");
}
else // TCP 연결일경우
{
serv_sock = socket(AF_INET, SOCK_STREAM, 0);
if (serv_sock == -1)
error_handling("에러: 소켓에 문제가 발생했습니다.");
}
// 소켓 포트와 IP 지정
struct sockaddr_in serv_addr;
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_addr.sin_port = htons(8080);
// bind 지정
if (bind(serv_sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) == -1)
error_handling("에러: 접근에 문제가 발생했습니다.");
// TCP일 경우 listen 및 accept 사용
if (TCP == 1) {
if (listen(serv_sock, 5) == -1)
error_handling("에러: 듣기에 문제가 발생했습니다.");
struct sockaddr_in clnt_addr;
socklen_t clnt_addr_size = sizeof(clnt_addr);
int clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_addr, &clnt_addr_size);
if (clnt_sock == -1)
error_handling("에러: 승인에 문제가 발생했습니다.");
// TCP 방식으로 메시지 수신 및 응답
while (1) {
char buffer[1024];
int bytes_received = recv(clnt_sock, buffer, sizeof(buffer) - 1, 0);
if (buffer[0] == 'E' && buffer[1] == 'N' && buffer[2] == 'D' && buffer[3] == '\n') break;// 'END' 메시지 확인
if (bytes_received <= 0) {
error_handling("에러: 전송에 문제가 발생했습니다.");
}
buffer[bytes_received] = '\0';
// TCP식 응답 전송
if (send(clnt_sock, buffer, strlen(buffer), 0) == -1) {
error_handling("에러: 전송에 문제가 발생했습니다.");
}
}
closesocket(clnt_sock);
closesocket(serv_sock);//소켓 종료
}
else // UDP일 경우, listen과 accept를 사용하지 않음
{
struct sockaddr_in clnt_addr;
socklen_t clnt_addr_size = sizeof(clnt_addr);
// UDP 방식으로 메시지 수신 및 응답
while (1) {
char buffer[1024];
int bytes_received = recvfrom(serv_sock, buffer, sizeof(buffer) - 1, 0, (struct sockaddr*)&clnt_addr, &clnt_addr_size);
if (buffer[0] == 'E' && buffer[1] == 'N' && buffer[2] == 'D' && buffer[3] == '\n') break;// 'END' 메시지 확인
if (bytes_received <= 0) {
error_handling("에러: 전송에 문제가 발생했습니다.");
}
buffer[bytes_received] = '\0';
// UDP식 응답 전송
if (sendto(serv_sock, buffer, strlen(buffer), 0,
(struct sockaddr*)&clnt_addr, sizeof(clnt_addr)) == -1) {
error_handling("에러: 전송에 문제가 발생했습니다.");
}
}
closesocket(serv_sock); //소켓 종료
}
return 0;
}
// 클라이언트 thread
unsigned __stdcall client_thread(void* arg) {
int clnt_sock; // 클라이언트 소켓
// UDP 연결일경우
if (TCP == 0) {
clnt_sock = socket(AF_INET, SOCK_DGRAM, 0);
if (clnt_sock == -1)
error_handling("에러: 소켓에 문제가 발생했습니다.");
printf("UDP 환경으로 연결되었습니다.\n");
}
else // TCP 연결일경우
{
clnt_sock = socket(AF_INET, SOCK_STREAM, 0);
if (clnt_sock == -1)
error_handling("에러: 소켓에 문제가 발생했습니다.");
printf("TCP 환경으로 연결되었습니다.\n");
}
// 소켓 포트와 IP 지정
struct sockaddr_in serv_addr;
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1"); // local host 사용
serv_addr.sin_port = htons(8080);
if (connect(clnt_sock, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) == -1) {
error_handling("에러: 연결에 문제가 발생했습니다.");
}
// 메시지 입력받은 다음 그 내용을 서버에 전송함(반복)
while (1) {
int escape = 0; // 탈출용 변수
printf("server로 보낼 메시지 입력: ");
char input[1024];
fgets(input, sizeof(input), stdin);
if (input[0] == 'E'&& input[1] == 'N'&& input[2] == 'D'&& input[3] == '\n') { // 'END' 메시지 확인
printf("'END' 응답을 받았습니다.연결 종료 중...\n");
escape = 1;
}
// 클라이언트에서 서버로 메시지 전달
if (send(clnt_sock, input, strlen(input), 0) == -1 && escape == 0) {
error_handling("에러: 클라이언트 쪽 전송에 문제가 발생했습니다.");
}
if (escape == 1) break;
// 서버로부터 응답 수신
char responseFromServer[1024];
int bytes_received = recv(clnt_sock, responseFromServer, sizeof(responseFromServer) - 1, 0);
if (bytes_received > 0 && escape == 0) {
responseFromServer[bytes_received] = '\0';
printf("서버로부터 응답: %s\n\n", responseFromServer);
}
else if (bytes_received == -1 && escape == 0) {
error_handling("에러: 수신에 문제가 발생했습니다.");
}
}
closesocket(clnt_sock); // 소켓 종료
return 0;
}
int main() {
// Windows 소켓 구현에 대한 정보
WSADATA wsaData;
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
error_handling("에러: 빌드에 문제가 발생했습니다.");
}
int key = 0;
printf("연결 방식을 선택하십시오. (U : UDP / T : TCP)");
// 맨 처음에는 키보드 입력 방식으로 UDP / TCP 연결을 결정함
while (1) {
if (_kbhit())
{
key = _getch();
// 아스키 코드 117은 U
if (key == 117) {
TCP = 0;
system("cls");
break;
}
// 아스키 코드 116은 T
else if (key == 116) {
TCP = 1;
system("cls");
break;
}
}
}
// 서버 및 클라이언트 thread 실행
HANDLE hServerThread = (HANDLE)_beginthreadex(NULL, 0, server_thread, NULL, 0, NULL);
if (hServerThread == NULL) {
error_handling("에러: 서버 스레드 생성에 문제가 발생했습니다.");
}
HANDLE hClientThread = (HANDLE)_beginthreadex(NULL, 0, client_thread, NULL, 0, NULL);
if (hClientThread == NULL) {
error_handling("에러: 클라이언트 스레드 생성에 문제가 발생했습니다.");
}
// thread 종료때까지 대기
WaitForSingleObject(hServerThread, INFINITE);
WaitForSingleObject(hClientThread, INFINITE);
CloseHandle(hServerThread);
CloseHandle(hClientThread);
// Windows 소켓 종료
WSACleanup();
// 완전 종결
return 0;
}
코드는 이렇게 TCP 연결 위해서 만들었는데
local ip에 로컬호스트(127.0.0.1) 해놨는데도 인식 자체를 못함
댓글 0