课程学习--网络--第1讲--socket

client

#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <iostream>
#include <opencv2/highgui/highgui.hpp>
#include <fstream>
#include <thread>
#define MAXLINE 4096
using namespace std;
using namespace cv;


int capture_image()
{
    VideoCapture capture(0);
    if ( !capture.isOpened())
    {
        std::cout << "capture camera error! " << std::endl;
    }
    cv::namedWindow("capture frame", CV_WINDOW_NORMAL);
    std::cout << "Press C to save the current frame\n to quit the frame fetching process by Q\n" << std::endl;
    Mat frame;
    int count = 0;
    char filename[64];

    while (true)
    {
        capture >> frame;

        if (frame.empty()) { break; }
        cv::imshow("capture frame", frame);

        if ( (char)waitKey(10) == 'C')
        {
            count++;
            //filename = "calib" + to_string(count) + ".jpg";
            sprintf(filename, "%d.jpg", count);
            cout << filename << endl;
            imwrite(filename, frame);
            std::cout << "Successfully retrieve the current frame and use the file name" << filename << "saved..." << std::endl;
            std::cout << "Press C to save the current frame...\n to quit the frame fetching process by Q...\n" << std::endl;
        }
        else if ((char)waitKey(10) == 'Q')
        {
            std::cout << "Saved images" << std::endl;
            std::cout << "include" << count << "picture(s)..." << std::endl;
            break;
        }
    }
    capture.release(); //释放对象
    return 0;
}

int main(int argc, char** argv){

    std::thread t1(capture_image);
    t1.join();
           
    int   sockfd, len;
    char  buffer[MAXLINE];
    struct sockaddr_in  servaddr;
    FILE *fq;

    if( argc != 2){
        printf("usage: ./client <ipaddress>\n");
        return 0;
    }
 
    if( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0){
        printf("create socket error: %s(errno: %d)\n", strerror(errno),errno);
        return 0;
    }
 
    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(8000);

    if( inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0){
        printf("inet_pton error for %s\n",argv[1]);
        return 0;
    }
 
    if( connect(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0){
        printf("connect error: %s(errno: %d)\n",strerror(errno),errno);
        return 0;
    }
  
    if(( fq = fopen("1.jpg","rb") ) == NULL ){
        printf("File can not open.\n");
        close(sockfd);
        exit(1);
    }
  
    bzero(buffer,sizeof(buffer));

    while(!feof(fq)){

        len = fread(buffer, 1, sizeof(buffer), fq);

        if(len != write(sockfd, buffer, len)){

            printf("write.\n");

            break;
        }
    }
    close(sockfd);
    fclose(fq);
 
    return 0;
}

server

 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <errno.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <unistd.h>
 #include <assert.h>
 #define MAXLINE 4096

int main(int argc, char** argv){
     int  listenfd, connfd;
     int  count=1;
     char filename[16] = {0};
     struct sockaddr_in  servaddr;
     char  buff[4096];
     FILE *fp;
     int  n;
 
     if( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1 ){
         printf("create socket error: %s(errno: %d)\n",strerror(errno),errno);
         return 0;
     }
     printf("----init socket----\n");
 
     memset(&servaddr, 0, sizeof(servaddr));
     servaddr.sin_family = AF_INET;
     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
     servaddr.sin_port = htons(8000);
     //设置端口可重用

     int contain;
     setsockopt(listenfd,SOL_SOCKET, SO_REUSEADDR, &contain, sizeof(int));
 
     if( bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1){
         printf("bind socket error: %s(errno: %d)\n",strerror(errno),errno);
         return 0;
     }
     printf("----bind sucess----\n");
 
     if( listen(listenfd, 10) == -1){
         printf("listen socket error: %s(errno: %d)\n",strerror(errno),errno);
         return 0;
     }
     
 
     printf("======waiting for client's request======\n");
     while(1){
         struct sockaddr_in client_addr;
         socklen_t size=sizeof(client_addr);
         if( (connfd = accept(listenfd, (struct sockaddr*)&client_addr, &size)) == -1){
             printf("accept socket error: %s(errno: %d)",strerror(errno),errno);
             continue;
         }

        //filename = "calib" + to_string(count) + ".jpg";
        sprintf(filename, "%d.jpg", count);
        if(0 == access(filename, F_OK))
        {
            /*如果文件已经存在 先删*/
            if( 0 != remove(filename) )
            {
                printf("####L(%d) rm file:%s err！\n", __LINE__, filename);
            }
        }

        if((fp = fopen(filename,"ab") ) == NULL )
        {
            printf("File.\n");
            close(listenfd);
            exit(1);
        }
         
         while(1){
            n = read(connfd, buff, MAXLINE);
            if(n == 0){
                break;
            }
            fwrite(buff, 1, n, fp);

        }
        count++;
        //buff[n] = '\0';
        printf("recv msg from client: %s\n", filename);
        close(connfd);
        fclose(fp);
     }
     close(listenfd);
     return 0;
 }