chatClient.h
#ifndef CHAT_CLIENT_H
#define CHAT_CLIENT_H
#include
#include
/*socket() connect() recv()*/
#include
#include
#include //struct sockaddr_in
#include //thread
#include //malloc()
#include //read()
#if 1
#include "node.h"
#include "protocol.h"
#include "file.h"
#include "error.h"
#endif
#if 0
#include "../link/node.h"
#include "../protocol/protocol.h"
#include "../file/file.h"
#include "../error/error.h"
#endif
enum FILE_OPE
{
ENUM_FILE_OPE_MIN = 0,
ENUM_FILE_LIST, //查看文件
ENUM_FILE_UPLOAD, //上传文件
ENUM_FILE_DOWNLOAD, //下载文件
ENUM_FILE_RENAME, //文件重命名
ENUM_FILE_OPE_MAX = 0x0fffffff
};
#define CHAECK_INPUT(i, a, b) ((i)>=(a)&&(i<=(b)))?1:0
void sendDataToServer(int sockfd, ChatPDU *pChatPDU);
ChatPDU *readDataFromServer(int sockfd);
int makeSocket();
int connectToServer(int sockfd);
int operateUI();
void interactWithServer(int sockfd);
#endif
chatClient.c
#include "chatClient.h"
Head *g_pHead = NULL;
void sendDataToServer(int sockfd, ChatPDU *pChatPDU)
{
if (NULL == pChatPDU)
{
return;
}
int ret = 0;
int iLeft = pChatPDU->iChatPDULen;
int iSend = 0;
ret = write(sockfd, pChatPDU, iLeft);
if (-1 == ret)
{
PRINT_ERROR("send");
return;
}
iLeft -= ret; //还有多少数据要发送
iSend += ret; //已发送的数据大小
while (ret > 0 && iLeft)
{
ret = write(sockfd, (char *)pChatPDU+iSend
, iLeft);
if (-1 == ret)
{
PRINT_ERROR("send");
return;
}
iLeft -= ret;
iSend += ret;
}
}
ChatPDU *readDataFromServer(int sockfd)
{
int ret = 0;
unsigned int iPDULen = 0;
ret = read(sockfd, &iPDULen, sizeof(int));
if (-1 == ret)
{
PRINT_ERROR("recv");
return;
}
ChatPDU *pChatPDU = makeChatPDU(iPDULen);
memset(pChatPDU, 0, iPDULen);
pChatPDU->iChatPDULen = iPDULen;
int iLeft = iPDULen - ret; //剩余的数据大小
int iRecv = ret; //已读的数据大小
while (ret > 0 && iLeft)
{
ret = read(sockfd, (char*)pChatPDU+iRecv
, iLeft);
if (-1 == ret)
{
PRINT_ERROR("recv");
return;
}
iRecv += ret;
iLeft -= ret;
}
return pChatPDU;
}
int makeSocket()
{
int sockfd = 0;
//AF_INET: ipv4
//SOCK_STREAM: tcp/ip
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (-1 == sockfd)
{
PRINT_ERROR("socket");
exit(-1);
}
return sockfd;
}
int connectToServer(int sockfd)
{
struct sockaddr_in servaddr;
//地址协议:ipv4
servaddr.sin_family = AF_INET;
//服务器端口
servaddr.sin_port = htons(8668);
//服务器ip
servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
bzero(&(servaddr.sin_zero), 8);
int ret = connect(sockfd
, (struct sockaddr *)&servaddr
, sizeof(struct sockaddr));
if (-1 == ret)
{
PRINT_ERROR("connect");
exit(-1);
}
return ret;
}
int operateUI()
{
printf("1,查看文件\t2,上传文件\n");
printf("3,下载文件\t4,文件重命名\n");
printf("ope:");
int iOpe = 0;
scanf("%d", &iOpe);
if (CHAECK_INPUT(iOpe, 1, 4))
{
return iOpe;
}
return -1;
}
static void getFileListFromPDU(const ChatPDU *pChatPDU)
{
if (NULL != pChatPDU)
{
int iNum = 0;
Node *pNode = NULL;
iNum = pChatPDU->iChatDataLen/FILE_NAME_LEN;
int i = 0;
for (; i < iNum; i++)
{
pNode = makeNode();
memcpy(pNode->caName
, pChatPDU->chatData+i*FILE_NAME_LEN
, FILE_NAME_LEN);
insertList(g_pHead, pNode);
}
}
}
static void handleAskFileList(int sockfd)
{
destoryList(g_pHead);
int iLen = sizeof(ChatPDU);
ChatPDU *pChatPDU = makeChatPDU(iLen);
if (NULL != pChatPDU)
{
pChatPDU->iChatPDULen = iLen;
pChatPDU->iType = ENUM_TYPE_ASK_FILE_INFO;
sendDataToServer(sockfd, pChatPDU);
freePDU(pChatPDU);
pChatPDU = NULL;
pChatPDU = readDataFromServer(sockfd);
getFileListFromPDU(pChatPDU);
freePDU(pChatPDU);
pChatPDU = NULL;
showList(g_pHead);
}
}
void readDataFromSTDIN(char caData[FILE_IO_MAX_LEN])
{
memset(caData, '\0', FILE_IO_MAX_LEN);
int ret = read(STDIN_FILENO, caData, FILE_IO_MAX_LEN);
if (-1 == ret)
{
perror("read");
return;
}
if ('\0' != caData[FILE_IO_MAX_LEN-1]
&& '\n' != caData[FILE_IO_MAX_LEN-1])
{
while ('\n' != getchar())
{}
}
caData[FILE_IO_MAX_LEN-1] = '\0';
}
static void handleUploadFile(int sockfd)
{
char caData[FILE_IO_MAX_LEN];
// 输入时以'/'结束
printf("请输入要上传的文件路径(以\'/\'结尾):\n");
readDataFromSTDIN(caData);
char caPath[64];
memset(caPath, '\0', 64);
sscanf(caData, "%s", caPath);
scanFileInfo(caPath);
printf("请输入要上传的文件名:\n");
readDataFromSTDIN(caData);
char caName[32] = {'\0'};
memset(caName, '\0', 32);
sscanf(caData, "%s", caName);
Node *pNode = makeNode();
initNode(pNode, caName);
insertList(g_pHead, pNode);
strcat(caPath, caName);
printf("name=%s\n", caPath);
int fd = openFile(caPath);
int ret = -1;
int iLen = sizeof(ChatPDU)-sizeof(int)+FILE_IO_MAX_LEN;
ChatPDU *pChatPDU = makeChatPDU(iLen);
pChatPDU->iChatPDULen = iLen;
pChatPDU->iType = ENUM_TYPE_UPLOAD_FILE;
strcpy(pChatPDU->caFileName, caName);
while (ret = read(fd, caData, FILE_IO_MAX_LEN))
{
pChatPDU->iChatDataLen = ret;
memcpy(pChatPDU->chatData, caData, ret);
sendDataToServer(sockfd, pChatPDU);
}
freePDU(pChatPDU);
}
void handleDownloadFile(int sockfd)
{
int iLen = sizeof(ChatPDU);
ChatPDU *pChatPDU = makeChatPDU(iLen);
if (NULL != pChatPDU)
{
char caData[FILE_IO_MAX_LEN];
char caPath[64];
pChatPDU->iChatPDULen = iLen;
pChatPDU->iType = ENUM_TYPE_ASK_DOWNLOAD_FILE;
showList(g_pHead);
printf("请输入要下载的文件名:\n");
readDataFromSTDIN(caData);
sscanf(caData, "%s", caPath);
strcpy(pChatPDU->caFileName, caPath);
sendDataToServer(sockfd, pChatPDU);
freePDU(pChatPDU);
pChatPDU = NULL;
unsigned int iReceivedFileSize = 0;
// 输入时以'/'结束
printf("请输入要保存的文件路径(以\'/\'结尾):\n");
readDataFromSTDIN(caData);
memset(caPath, '\0', 64);
sscanf(caData, "%s", caPath);
pChatPDU = readDataFromServer(sockfd);
strcat(caPath, pChatPDU->caFileName);
int fd = -1;
fd = openFile(caPath);
lseek(fd, 0, SEEK_END);
printf("file size = %d\n", pChatPDU->iFileSize);
while (1)
{
memset(caData, '\0', FILE_IO_MAX_LEN+1);
strncpy(caData, pChatPDU->chatData, pChatPDU->iChatDataLen);
printf("per size = %d\n", pChatPDU->iChatDataLen);
lseek(fd, 0, SEEK_END);
writeDataToFile(fd, caData, pChatPDU->iChatDataLen);
iReceivedFileSize += pChatPDU->iChatDataLen;
printf("rec=%d\n", iReceivedFileSize);
printf("total = %d\n", pChatPDU->iFileSize);
if (iReceivedFileSize == pChatPDU->iFileSize)
{
freePDU(pChatPDU);
pChatPDU = NULL;
break;
}
freePDU(pChatPDU);
pChatPDU = NULL;
pChatPDU = readDataFromServer(sockfd);
}
closeFile(fd);
printf("文件下载完成\n");
}
}
void interactWithServer(int sockfd)
{
g_pHead = makeList();
int ret = 0;
while (1)
{
ret = operateUI();
if (-1 == ret)
{
printf("input error\n");
continue;
}
switch (ret)
{
case ENUM_FILE_LIST:
handleAskFileList(sockfd);
break;
case ENUM_FILE_UPLOAD:
handleUploadFile(sockfd);
break;
case ENUM_FILE_DOWNLOAD:
handleDownloadFile(sockfd);
break;
case ENUM_FILE_RENAME:
break;
default:
break;
}
}
}
main.c(client)
#include "chatClient.h"
int main(void)
{
int sockfd = 0;
sockfd = makeSocket();
connectToServer(sockfd);
interactWithServer(sockfd);
return 0;
}
error.h
#ifndef ERROR_H
#define ERROR_H
#include
#include
#include
#define PRINT_ERROR(s) printf("%s: errno=%d,err=%s\n", s, errno, strerror(errno));
#endif
error.c
#include "error.h"
file.h
#ifndef FILE_H
#define FILE_H
#include
#include //perror strerror
#include //errno
/*open()*/
#include
#include
#include
#include //exit()
#include //read() write()
#include "node.h"
#include "error.h"
#define FILE_IO_MAX_LEN 4096
int openFile(const char *pFileName);
void readDataFromFile(int fd, char *pBuf);
void writeDataToFile(int fd, const char *pBuf, unsigned int iLen);
void closeFile(int fd);
void scanFileIntoList(Head *pHead, const char *pDirName);
void scanFileInfo(const char *pDirName);
#endif
file.c
#include "file.h"
#include
#include
int openFile(const char *pFileName)
{
int fd = -1;
if (NULL != pFileName)
{
fd = open(pFileName, O_RDWR | O_CREAT
, S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH);
if (-1 == fd)
{
PRINT_ERROR("open file");
exit(-1);
}
}
return fd;
}
void readDataFromFile(int fd, char *pBuf)
{
if (NULL != pBuf && fd >= 0)
{
int ret = 0;
int iLeft = FILE_IO_MAX_LEN;
int iReaded = 0;
ret = read(fd, pBuf, iLeft);
if (-1 == ret)
{
PRINT_ERROR("read");
exit(-1);
}
iLeft -= ret;
iReaded += ret;
while (ret > 0 && iLeft)
{
ret = read(fd, pBuf+iReaded, iLeft);
if (-1 == ret)
{
PRINT_ERROR("read");
exit(-1);
}
iLeft -= ret;
iReaded += ret;
}
}
}
void writeDataToFile(int fd, const char *pBuf, unsigned int iLen)
{
if (NULL != pBuf && fd >= 0)
{
int ret = 0;
int iLeft = iLen;
int iWrited = 0;
ret = write(fd, pBuf, iLeft);
if (-1 == ret)
{
PRINT_ERROR("write");
exit(-1);
}
iLeft -= ret;
iWrited += ret;
while (ret > 0 && iLeft)
{
ret = write(fd, pBuf+iWrited, iLeft);
if (-1 == ret)
{
PRINT_ERROR("write");
exit(-1);
}
iLeft -= ret;
iWrited += ret;
}
}
}
void closeFile(int fd)
{
if (fd >= 0)
{
int ret = 0;
ret = close(fd);
if (-1 == ret)
{
PRINT_ERROR("close");
exit(-1);
}
}
}
void scanFileIntoList(Head *pHead
, const char *pDirName)
{
if (NULL == pHead)
{
return;
}
DIR *pDir = NULL;
pDir = opendir(pDirName); //打开目录
if (NULL == pDir)
{
PRINT_ERROR("opendir");
exit(-1);
}
struct dirent *pDirent = NULL;
Node *pNode = NULL;
//循环遍历该目录下的文件,返回某个文件的信息
while (pDirent = readdir(pDir))
{
if (0 == strcmp(".", pDirent->d_name)
|| 0 == strcmp("..", pDirent->d_name))
{
continue;
}
pNode = makeNode();
//获得某个文件的名字并保存到节点中
strncpy(pNode->caName, pDirent->d_name
, FILE_NAME_LEN);
insertList(pHead, pNode);
}
}
void scanFileInfo(const char *pDirName)
{
DIR *pDir = NULL;
pDir = opendir(pDirName); //打开目录
if (NULL == pDir)
{
PRINT_ERROR("opendir");
exit(-1);
}
struct dirent *pDirent = NULL;
int i = 0;
//循环遍历该目录下的文件,返回某个文件的信息
while (pDirent = readdir(pDir))
{
if (0 == strcmp(".", pDirent->d_name)
|| 0 == strcmp("..", pDirent->d_name))
{
continue;
}
printf("%s ", pDirent->d_name);
i++;
if (0 == i%5)
{
printf("\n");
}
}
printf("\n");
}
node.h
#ifndef NODE_H
#define NODE_H
#include
#define FILE_NAME_LEN 32
typedef struct Node
{
char caName[FILE_NAME_LEN];
struct Node *pNext;
}Node;
typedef struct Head
{
int iNum;
Node *pFirstNode;
}Head;
Head *makeList();
Node *makeNode();
void initNode(Node *node, char *pFileName);
void insertList(Head *head, Node *node);
Node *findNode(Head *head, char *pFileName);
void deleteNode(Head *head, char *pFileName);
void showList(Head *head);
void destoryList(Head *head);
void showNode(Node *node);
#endif
node.c
#include "node.h"
#include "error.h"
#include
#include
Head *makeList()
{
Head *pHead = NULL;
pHead = (Head *)malloc(sizeof(Head));
if (NULL == pHead)
{
PRINT_ERROR("malloc head");
exit(-1);
}
pHead->iNum = 0;
pHead->pFirstNode = NULL;
return pHead;
}
Node *makeNode()
{
Node *pNode = NULL;
pNode = (Node *)malloc(sizeof(Node));
if (NULL == pNode)
{
PRINT_ERROR("malloc node");
exit(-1);
}
memset(pNode->caName, '\0', FILE_NAME_LEN);
pNode->pNext = NULL;
return pNode;
}
void initNode(Node *node, char *pFileName)
{
if (NULL != node && NULL != pFileName)
{
strcpy(node->caName, pFileName);
}
}
void insertList(Head *head, Node *node)
{
if (NULL == head || NULL == node)
{
return;
}
if (NULL == head->pFirstNode)
{
head->pFirstNode = node;
}
else
{
node->pNext = head->pFirstNode;
head->pFirstNode = node;
}
head->iNum++;
}
Node *findNode(Head *head, char *pFileName)
{
Node *pNode = NULL;
if (NULL != head && NULL != pFileName)
{
pNode = head->pFirstNode;
while (NULL != pNode)
{
if (0 == strcmp(pNode->caName, pFileName))
{
return pNode;
}
pNode = pNode->pNext;
}
}
return pNode;
}
void deleteNode(Head *head, char *pFileName)
{
if (NULL != head && NULL != pFileName)
{
Node *pPreNode = head->pFirstNode;
Node *pCurNode = pPreNode;
while (NULL != pCurNode)
{
if (0 == strcmp(pCurNode->caName
, pFileName))
{
if (pCurNode == pPreNode)
{
head->pFirstNode = pCurNode->pNext;
}
else
{
pPreNode->pNext = pCurNode->pNext;
}
break;
}
pPreNode = pCurNode;
pCurNode = pCurNode->pNext;
}
if (NULL != pCurNode)
{
free(pCurNode);
pCurNode = NULL;
head->iNum--;
}
}
}
void showList(Head *head)
{
if (NULL != head)
{
Node *pNode = head->pFirstNode;
while (NULL != pNode)
{
printf("name:%s\n", pNode->caName);
pNode = pNode->pNext;
}
}
}
void destoryList(Head *head)
{
if (NULL != head)
{
Node *pNode = head->pFirstNode;
while (NULL != pNode)
{
head->pFirstNode = pNode->pNext;
free(pNode);
head->iNum--;
pNode = head->pFirstNode;
}
}
}
void showNode(Node *node)
{
if (NULL != node)
{
printf("name:%s\n", node->caName);
}
}
protocol.h
#ifndef PROTOCOL_H
#define PROTOCOL_H
#include
enum TYPE
{
ENUM_TYPE_MIN = 0,
ENUM_TYPE_ASK_FILE_INFO, //客户端请求文件信息
ENUM_TYPE_RESP_FILE_INFO, //服务器返回文件信息
ENUM_TYPE_UPLOAD_FILE, //客户端上传文件
ENUM_TYPE_ASK_DOWNLOAD_FILE, //客户端请求下载文件
ENUM_TYPE_RESP_DOWNLOAD_FILE, //服务器发送文件
ENUM_TYPE_RENAME_FILE, //客户端重命名文件
ENUM_TYPE_DELETE_FILE, //客户端删除文件
ENUM_TYPE_MAX = 0x0fffffff
};
#define SERVER 0
typedef struct ChatPDU
{
unsigned int iChatPDULen; //数据单元总大小
unsigned int iType; //消息类型
unsigned int iChatDataLen; //消息大小
char caFileName[32]; //文件名字
unsigned int iFileSize; //文件大小
char chatData[4]; //发送的消息
}ChatPDU;
ChatPDU *makeChatPDU(int len);
int parsePDU(ChatPDU *pChatPDU);
void freePDU(ChatPDU *pChatPDU);
#endif
protocol.c
#if 1
#include "protocol.h"
#include "error.h"
#endif
#include
#include
ChatPDU *makeChatPDU(int len)
{
if (len <= 0)
{
return NULL;
}
ChatPDU *pChatPDU = (ChatPDU *)malloc(len);
if (NULL == pChatPDU)
{
PRINT_ERROR("malloc PDU");
exit(-1);
}
memset(pChatPDU, 0, len);
return pChatPDU;
}
int parsePDU(ChatPDU *pChatPDU)
{
if (NULL == pChatPDU)
{
return ENUM_TYPE_MIN;
}
return pChatPDU->iType;
}
void freePDU(ChatPDU *pChatPDU)
{
if (NULL != pChatPDU)
{
free(pChatPDU);
pChatPDU = NULL;
}
}
chatServer.h
#ifndef CHAT_SERVER_H
#define CHAT_SERVER_H
#include
#include
#include
/*socket() send()*/
#include
#include
#include
#include //fork()
/*wait()*/
#include
#include
#include //signal()
#include //thread
#include "node.h"
#include "protocol.h"
#include "error.h"
#include "file.h"
#include //exit()
int makeSocket();
void bindSocketAddr(int sockfd);
void listenClient(int sockfd);
void handle(int sockfd);
#endif
chatServer.c
#include "chatServer.h"
Head *g_pHead = NULL;
#define FILE_PATH "file/"
static void sendDataToClient(int sockfd
, const ChatPDU *pChatPDU)
{
if (NULL == pChatPDU)
{
return;
}
int ret = 0;
int iLeft = pChatPDU->iChatPDULen;
int iSend = 0;
ret = write(sockfd, pChatPDU, iLeft);
iLeft -= ret; //还有多少数据要发送
iSend += ret; //已发送的数据大小
while (ret > 0 && iLeft)
{
ret = write(sockfd, (char *)pChatPDU+iSend, iLeft);
iLeft -= ret;
iSend += ret;
}
}
static ChatPDU *readDataFromClient(int sockfd)
{
int ret = 0;
unsigned int iPDULen = 0;
ret = read(sockfd, &iPDULen, sizeof(int));
if (-1 == ret)
{
PRINT_ERROR("recv");
return NULL;
}
printf("pdulen:%d\n", iPDULen);
ChatPDU *pChatPDU = makeChatPDU(iPDULen);
pChatPDU->iChatPDULen = iPDULen;
int iLeft = iPDULen - ret; //剩余的数据大小
int iRecv = ret; //已读的数据大小
while (ret > 0 && iLeft)
{
ret = read(sockfd, (char*)pChatPDU+iRecv, iLeft);
iRecv += ret;
iLeft -= ret;
}
printf("datalen:%d\n", pChatPDU->iChatDataLen);
return pChatPDU;
}
void sendFileListToClient(int sockfd)
{
if (NULL == g_pHead)
{
return;
}
destoryList(g_pHead);
scanFileIntoList(g_pHead, FILE_PATH);
unsigned int iLen = sizeof(ChatPDU) + (g_pHead->iNum) * FILE_NAME_LEN - sizeof(int);
ChatPDU *pChatPDU = NULL;
pChatPDU = makeChatPDU(iLen);
Node *pNode = g_pHead->pFirstNode;
unsigned int i = 0;
while (NULL != pNode)
{
memcpy(pChatPDU->chatData+i*FILE_NAME_LEN
, pNode->caName, FILE_NAME_LEN);
i++;
pNode = pNode->pNext;
}
pChatPDU->iChatPDULen = iLen;
pChatPDU->iChatDataLen = g_pHead->iNum * FILE_NAME_LEN;
pChatPDU->iType = ENUM_TYPE_RESP_FILE_INFO;
sendDataToClient(sockfd, pChatPDU);
freePDU(pChatPDU);
pChatPDU = NULL;
}
static void handleClientUploadFile(ChatPDU *pChatPDU)
{
if (NULL == pChatPDU)
{
return;
}
char caPath[64] = FILE_PATH;
strcat(caPath, pChatPDU->caFileName);
int fd = openFile(caPath);
char caBuf[FILE_IO_MAX_LEN+1] = {'\0'};
memset(caBuf, '\0', FILE_IO_MAX_LEN+1);
strncpy(caBuf, pChatPDU->chatData, pChatPDU->iChatDataLen);
lseek(fd, 0, SEEK_END);
writeDataToFile(fd, caBuf, pChatPDU->iChatDataLen);
closeFile(fd);
}
static void handleClientDownloadFile(int sockfd, ChatPDU *pChatPDU)
{
if (NULL != pChatPDU)
{
char caPath[64] = FILE_PATH;
strcat(caPath, pChatPDU->caFileName);
int fd = openFile(caPath);
int iFileSize = lseek(fd, 0, SEEK_END);
int ret = -1;
int iLen = sizeof(ChatPDU)-sizeof(int)+FILE_IO_MAX_LEN;
ChatPDU *pChatPDUSend = makeChatPDU(iLen);
pChatPDUSend->iChatPDULen = iLen;
pChatPDUSend->iType = ENUM_TYPE_RESP_DOWNLOAD_FILE;
strcpy(pChatPDUSend->caFileName, pChatPDU->caFileName);
pChatPDUSend->iFileSize = iFileSize;
printf("file size = %d\n", pChatPDUSend->iFileSize);
char caData[FILE_IO_MAX_LEN];
lseek(fd, 0, SEEK_SET);
while (ret = read(fd, caData, FILE_IO_MAX_LEN))
{
pChatPDUSend->iChatDataLen = ret;
memcpy(pChatPDUSend->chatData, caData, ret);
printf("data len = %d\n", ret);
sendDataToClient(sockfd, pChatPDUSend);
}
freePDU(pChatPDU);
freePDU(pChatPDUSend);
}
}
static void *handleClient(void *arg)
{
int sockfd = (int)arg;
ChatPDU *pChatPDU = NULL;
int iMsgType = 0;
while (1)
{
/*获得PDU*/
pChatPDU = readDataFromClient(sockfd);
if (NULL != pChatPDU)
{
/*解析PDU*/
iMsgType = parsePDU(pChatPDU);
switch (iMsgType)
{
//客户端请求查看所有文件
case ENUM_TYPE_ASK_FILE_INFO:
freePDU(pChatPDU);
pChatPDU = NULL;
sendFileListToClient(sockfd);
break;
//客户端上传文件
case ENUM_TYPE_UPLOAD_FILE:
handleClientUploadFile(pChatPDU);
freePDU(pChatPDU);
pChatPDU = NULL;
break;
//客户端请求下载文件
case ENUM_TYPE_ASK_DOWNLOAD_FILE:
handleClientDownloadFile(sockfd, pChatPDU);
break;
//客户端修改文件名
case ENUM_TYPE_RENAME_FILE:
break;
//客户端删除文件
case ENUM_TYPE_DELETE_FILE:
break;
default:
break;
}
}
}
}
int makeSocket()
{
int sockfd = 0;
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (-1 == sockfd)
{
PRINT_ERROR("socket");
exit(-1);
}
return sockfd;
}
void bindSocketAddr(int sockfd)
{
struct sockaddr_in servaddr;
//地址协议:ipv4
servaddr.sin_family = AF_INET;
//服务器端口
servaddr.sin_port = htons(8668);
//服务器ip
servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
bzero(&(servaddr.sin_zero), 8);
int ret = bind(sockfd
, (struct sockaddr *)&servaddr
, sizeof(struct sockaddr));
if (-1 == ret)
{
PRINT_ERROR("bind");
exit(-1);
}
}
void listenClient(int sockfd)
{
int ret = listen(sockfd, 10);
if (-1 == ret)
{
PRINT_ERROR("listen");
exit(-1);
}
}
static void acceptClient(int sockfd)
{
struct sockaddr_in clientaddr;
int iLen = sizeof(struct sockaddr);
int sockClient = 0;
pthread_t pt;
int ret = 0;
while (1)
{
printf("accepting client connect...\n");
//服务器通过sockClient和客户端进行通信
//clientaddr:用来保存客户端地址等信息
//accept():阻塞等待客户端的连接
sockClient = accept(sockfd
, (struct sockaddr *)&clientaddr
, &iLen);
printf("after accepted client\n");
ret = pthread_create(&pt, NULL, handleClient
, (void *)sockClient);
if (0 != ret)
{
PRINT_ERROR("pthread_create");
exit(-1);
}
// pthread_detach(pt);
}
}
void handle(int sockfd)
{
g_pHead = makeList();
//scanFileIntoList(g_pHead, FILE_PATH);
showList(g_pHead);
acceptClient(sockfd);
}
main(server)
#include "chatServer.h"
int main(void)
{
int sockfd = -1;
sockfd = makeSocket();
bindSocketAddr(sockfd);
listenClient(sockfd);
handle(sockfd);
return 0;
}