#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <errno.h>
#include <signal.h>
#define IPLEN 40
#define NUM 1024
#define RCVPORT 1990
#define NAME 45
#define EOF (-1)
int my_open(int sd,char *p);
int my_rdwr(int fd,int stat_st_size,int sd);
int main(int argc,char * argv[])
{
int fd;
int i=1;
int sockedfd;
int newsockedfd;
int stop = EOF;
char buf[20];
size_t len;
struct sockaddr_in local_addr;
struct sockaddr_in peer_addr;
socklen_t addr_len;
char ipstr[IPLEN];
pid_t fpid;
sockedfd = socket(PF_INET, SOCK_STREAM, 0);
if(sockedfd< 0){
perror("socket()");
exit(1);
}
setsockopt(sockedfd,SOL_SOCKET,SO_REUSEADDR,&i,sizeof(i) );
local_addr.sin_family = PF_INET;//ipv4
local_addr.sin_port = htons(RCVPORT);
inet_pton(PF_INET, "0.0.0.0", &local_addr.sin_addr);//邦定端口
if(bind(sockedfd, (struct sockaddr *)&local_addr, sizeof(local_addr)) < 0){
perror("bind()");
exit(1);
}
if(listen(sockedfd, 200) < 0){
perror("listen()");
exit(1);
}
while(1)
{
addr_len = sizeof(peer_addr);
newsockedfd = accept(sockedfd,(void*)&peer_addr, &addr_len);
if(newsockedfd < 0)
{
perror("accept()");
exit(1);
}
i=1;
char n='0';
#if 1
fpid=fork();
if(0 ==fpid )
{
close(sockedfd);//关闭父进程监听描述符
#endif
while(1)
{
len = read(newsockedfd,buf,1);
if(0 > len )
{
perror("read()");
return -1;
}
if( 0 ==len )
break;
if ( '0'==n )
{
n=buf[0];
printf("n=%c, i=%d \n",n,i);
if(n<'4'&&n>='1')
{
snprintf(buf,sizeof(buf),"music/%c.wma",n);
printf("%s \n",buf);
my_fopen_rdwr(newsockedfd,buf);
}
else
{
printf("ERROR()\n");
len=write(newsockedfd,&stop,4);
if(len<0)
{
perror("write()");
printf("writr end fill");
}
n='0';
break;
}
}
else
{
if( NAME == i )
{//图片张数
len=write(newsockedfd,&stop,4);//结束写19
if(len<0)
{
perror("write()");
printf("writr end fill");
}
break;
}
snprintf(buf,sizeof(buf),"jpg%c/%d.jpg",n,i++);
printf("%s \n",buf);
my_fopen_rdwr(newsockedfd,buf);
}
}
close(newsockedfd);
exit(0);
}
else if (0 < fpid)
{
close(newsockedfd);//父进程关闭新套接字
signal(SIGCLD, SIG_IGN);//通知内核回收进程，该进程不是父进程回收
}
else
{
printf("this is fork error\n");
}
//printf("*******%s %s %d****pid=%d****\n",__FILE__,__FUNCTION__,__LINE__,fpid);
}
close(sockedfd);
return 0;
}
int my_fopen_rdwr(int sd,char *p)
{
struct stat stat_init;
size_t ret;
int pos,pos1;
char buf[NUM];
int j=0;
FILE * fd;
fd=fopen(p,"r");
if(fd<0)
{
perror("fopen()");
return -1;
}
if(stat(p,&stat_init)<0)
{
perror("stat()");
return -1;
}
#if 1
ret=write(sd,&stat_init.st_size,4);
pos1+=ret;
if(ret<0)
{
perror("write()");
return -1;
}
#endif
while(stat_init.st_size>0)
{
pos=0;
pos1=0;
ret=fread(buf+pos,sizeof(char),NUM,fd);
pos+=ret;
if(ret<0){
perror("fread()");
return -1;
}
ret=write(sd,buf+pos1,ret);
pos1+=ret;
if(ret<0){
perror("write()");
return -1;
}
stat_init.st_size-=ret;
}
printf("is over*******\n");
fclose(fd);
return 0;
}
本服务端代码测试通过 ，cpu占用率低，该服务端是采用多进程方式通讯，有客户端连接到服务端时，父进程会fork个子进程，子进程来连接客户，父进程继续监听端口，子进程下载完毕后自动退出，这样做的目的是在没有客户连接时，只有父进程一个处于监听状态，节约内存，提高性能，要是有多个客户连接，父进程只管fork子进程，节约了时间，同时可以多个客户端下载，本代码出错处理不够完善，后期继续跟新，如有建议和本人写的不足之处敬请给位多多指教。
|
