tcp

/******************server.c*************/

 #include<stdio.h>
 #include<strings.h>
 #include<unistd.h>
 #include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>

#define PORT 1234
#define BACKLOG 1
#define MAXDATASIZE 100

struct ARG
{
 int connfd;
 struct sockaddr_in client;
};
struct RECEIVE
{
 int connfd;
 char *buf;
 struct sockaddr_in client;
};

struct RETMSG
{
        int client_fd;
 char bEnableInject;
 char fileName[125];
} retInjectMsg;

struct SC_MSG //服务器和客户端的消息结构
{
        int sock_id;// 套接字id
 char bRetMsg;//本线程是否需要启回消息的函数
 
} scMsg[1024]; //最多1024 线程

void *send_func(void *arg );
void *recive_func(void *arg );
int main()
{
  pthread_t tid;
  int listenfd, connectfd;
  struct sockaddr_in server;
  struct sockaddr_in client;
  socklen_t addrlen;
  char buf[MAXDATASIZE];
  struct ARG *arg, in;
  arg = &in;
  struct RECEIVE *rec, recevie;
  rec = &recevie;
       
 
 
 if((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
 {
  perror("socket() error.");
  exit(1);
 }
 
 int opt = 1;
 setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
 
 bzero(&server, sizeof(server));
 server.sin_family = AF_INET;
 server.sin_port = htons(PORT);
 server.sin_addr.s_addr = htonl(INADDR_ANY);
 if(bind(listenfd, (struct sockaddr *)&server, sizeof(server)) == -1)
 {
  perror("bind() error.");
  exit(1);
 } 
  if(listen(listenfd, BACKLOG) == -1)
  { 
    perror("listen() error.");
    exit(1);
  }
 
 addrlen = sizeof(client);
 while(1)//服务器程序永远不退出
 {
  printf("start while(1) /n");
  if((connectfd = accept(listenfd, (struct sockaddr *)&client, &addrlen )) == -1)
  {
    perror("accept() error.");
    exit(1);
  }
  printf("You got a connection from client ip is: %s,port is:%d/n",inet_ntoa(client.sin_addr), htons(client.sin_port));
 

 //arg = (struct ARG *)malloc(sizeof(struct ARG));
 arg->connfd = connectfd;
 arg->client = client;
 
  if(pthread_create(&tid, NULL, send_func, (void *)arg))
  {
    perror("error pthread_create");
    exit(1);
  }
  printf("thread send id=%d/n",tid);
  rec->buf=buf;
  rec->connfd = connectfd;
  rec->client = client;
  if(pthread_create(&tid, NULL, recive_func, (void *)rec))
  {
    perror("error pthread_create");
    exit(1);
  }
    printf("thread  recieve id=%d/n",tid);
  continue;
  close(connectfd);
  close(listenfd);
 }
}
void *recive_func(void *rec )
{
  struct RECEIVE *r;
  int ret=0;
 // r =  (struct RECEIVE *)rec;
 r=(struct RECEIVE *)malloc(sizeof(struct RECEIVE));
 memcpy(r,(struct RECEIVE *)rec,sizeof(struct RECEIVE));
  // 不能用 strncpy cqx
  while(1)
  {
  printf("server start recv..... /n");
    ret=recv(r->connfd, r->buf, MAXDATASIZE,0);
  
  printf("the client =%d..... /n",r->connfd);
    printf("the client says: %s/n",r->buf);
    printf("recv ret =%d..... /n",ret);
    if(ret<=0)
    {
   pthread_exit(NULL);
   break;
  }
  if(0==strncmp(r->buf,"inject -n test.txt",sizeof(r->buf)))
  {
         retInjectMsg.client_fd= r->connfd;
                retInjectMsg.bEnableInject=1;
   strcpy(retInjectMsg.fileName,strrpos(r->buf,''));//保存文件名，以便发回相应的客户端
  }

  if (0==scMsg.sock_id)

  }
}
void *send_func(void *arg)
{
  char buf[MAXDATASIZE];
  struct ARG *info;
  int sendLen=0；
  //info =  (struct ARG *)arg;
  info=(struct ARG *)malloc(sizeof(struct ARG));
  memcpy(info,(struct ARG *)arg,sizeof(struct ARG));
  while(1)
  {
  if(1==retInjectMsg.bEnableInject)
  {
   sendLen=send(retInjectMsg.client_fd, retInjectMsg.fileName, MAXDATASIZE, 0);
                      memset(&retInjectMsg);
  }

                
        if(scMsg.sock_id)
 
   //fgets(buf, MAXDATASIZE, stdin);
   
#if 0
  if(buf[0]%2 ==0)
     info->connfd=4;
  else
   info->connfd=5;
#endif

  //分析是法给哪一个客户
  //printf("info->connfd =%d ",info->connfd);
   //sendLen=send(info->connfd, buf, MAXDATASIZE, 0);
   
   }
}
/////////////////////////////////////////////////////////////////////////

 

//////////////////////////////////client.c///////////////////////////////////////////
/********************注入进程************************/
#include<stdio.h>
#include<unistd.h>
#include<strings.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netdb.h>

#define PORT 1234
#define MAXDATASIZE 100

#define fileTalSize  1024 //先规定大小为1024字节

void *func_get(void *arg);
void *func_send(void *arg);

static char bStartInjectTast=0;//未启动注入任务
int fd=NULL;
int fileStart=0;

int main ( int argc, char *argv[])
{
  pthread_t tid;
  int sockfd, num;
  char buf[MAXDATASIZE] ="123456789";
  struct hostent *he;
  struct sockaddr_in server;

  if(argc!=2)
  {
    printf("usage %s<ip address>/n",argv[0]);
    exit(1);
  }
  if((he = gethostbyname(argv[1])) == NULL)//根据ip找host
  {
    printf("gethostbyname error/n");
   exit(1);
  }
  if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
  {
    printf("socket() error /n");
    exit(1);
  }
  bzero(&server, sizeof(server));
  server.sin_family = AF_INET;
  server.sin_port = htons(PORT);
  server.sin_addr = *((struct in_addr *)he->h_addr);

  if(connect(sockfd, (struct sockaddr *)&server, sizeof(server)) == -1)
  {
    printf("connetc() error/n");
    exit(1);
  }
 
  if(pthread_create(&tid, NULL, func_send, (void *)&sockfd))
  {
    perror("error pthread_create");
    return 2;

  }
  printf("func_send sockfd=%d /n",sockfd);
  if(pthread_create(&tid, NULL, func_get, (void *)&sockfd))
  {
    perror("error pthread_create");
    return 2;

  }
  printf("func_send sockfd=%d /n",sockfd);
       
 // close(sockfd); 
}

//inject_enable -n test.txt //返回的允许注入的字符串
void *func_get(void *arg)
{
  int *sockfd;
  char buf[MAXDATASIZE];
  sockfd = (int *)arg;
  while(1)
  {
   num= recv(sockfd, buf, MAXDATASIZE, 0);
  printf(" get buf:%s /n",buf);

  if(0==strncmp(buf,"inject_enable -n test.txt",sizeof(buf)))
  {
     printf("server message: %s/n",buf);
   fd=open("test.txt",O_RDONLY );//后续直接窃取最后一段字符名
   if(fd>0)
    bStartInjectTast=1;
  }
  //判断是否可以注入内容的应答

  }
}
// inject -n test.txt
void *func_send(void *arg)
{
  int *sockfd;
  char *data;
  char buf[MAXDATASIZE];
  int len=0;
  int offset=0;
  sockfd = (int *)arg;
 
  while(1)
  {
       
        if(0==bStartInjectTast)
   {
    fgets(buf, MAXDATASIZE, stdin); //阻塞状态
    printf(" send buf:%s /n",buf);
         }
  else// 通过socket开始把数据连续发送
  {
  
   //test.txt
   if(fileStart +offset >=fileTalSize)
   {//文件已经发送完毕
    bStartInjectTast =0;
    fileStart=0;
    offset=0;
    close(fd);
   }
   else
    {
     fileStart +=offset;
            offset=send(*sockfd, fileStart, MAXDATASIZE, 0);
   }
  }
  }
}