#include <netinet/in.h> // for sockaddr_in
#include <sys/types.h> // for socket
#include <sys/socket.h> // for socket
#include <stdio.h> // for printf
#include <stdlib.h> // for exit
#include <string.h> // for bzero
#include <pthread.h>
#include<unistd.h>
#include "queue.h"
#include <sys/errno.h> // for errno
#define SERVER_PORT 6666
#define BUFFER_SIZE 1024
#define WRITE_LEN 10
pthread_mutex_t mut;
MyQueue *qu;
void *main_recvfrom_loop(void)
{
char buf[BUFFER_SIZE];
int sockfd; /* 数据端口 */
struct sockaddr_in my_addr; /* 自身的地址信息 */
struct sockaddr_in their_addr; /* 连接对方的地址信息 */
int sin_size, retval;
if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
perror("socket");
exit(1);
}
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(SERVER_PORT); /* 网络字节顺序 */
my_addr.sin_addr.s_addr = INADDR_ANY; /* 自动填本机IP */
bzero(&(my_addr.sin_zero), 8); /* 其余部分置0 */
if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr)) == -1)
{
perror("bind");
exit(1);
}
while(1)
{
recvfrom(sockfd, buf, BUFFER_SIZE, 0, (struct sockaddr *)&their_addr, &sin_size);
//printf("%d\n",strlen(buf));
pthread_mutex_lock(&mut);
enqueue(qu,buf);
pthread_mutex_unlock(&mut);
usleep(100);
//printf("%s\n",buf);
bzero(buf, BUFFER_SIZE);
}
}
void *write2file(void *data)
{
char *filepath=(char *)data;
while(1)
{
if(GetLength(qu)>WRITE_LEN)
{
printf("write2file\n");
pthread_mutex_lock(&mut);
if(do_writefile(filepath,WRITE_LEN))
{
printf("write error\n");
}
pthread_mutex_unlock(&mut);
usleep(100);
}
usleep(10000);
}
}
int do_writefile(char *path,int num)
{
int i;
FILE *fp;
fp=fopen(path,"a+");
for(i=0;i<num;i++)
{
char *p=qu->front->buf;
fwrite(p,strlen(p)+1,1,fp);
dequeue(qu);
}
fclose(fp);
return 0;
}
int main(int argc,char **argv)
{
if (argc != 2)
{
printf("Usage: %s Filename\n",argv[0]);
exit(1);
}
qu=CreatMyQueue();
pthread_mutex_init(&mut,NULL);
int i;
pthread_t thread_write;
char *filepath;
filepath=argv[1];
pthread_attr_t write2file_attr;
pthread_attr_init(&write2file_attr);
pthread_attr_setdetachstate(&write2file_attr, PTHREAD_CREATE_DETACHED);
pthread_create(&thread_write, &write2file_attr, write2file, (void *)filepath);
pthread_t child_thread;
pthread_attr_t child_thread_attr;
pthread_attr_init(&child_thread_attr);
pthread_attr_setdetachstate(&child_thread_attr, PTHREAD_CREATE_DETACHED);
pthread_create(&child_thread, &child_thread_attr, main_recvfrom_loop, NULL);
while(1)
{
sleep(1000);
}
}
