Linux下cs简单通讯(socket)

Server：

#include<iostream>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<unistd.h>
#include <errno.h>
#include<string.h>

#define PORT 10004

using namespace std;

void showError()
{
    cout<<"Error : "<<strerror(errno)<<" Error code : "<<errno<<endl;
}

int main()
{
    int ss,sc;
    struct sockaddr_in server_addr,client_addr,peer_addr;
    char conn_ip[INET_ADDRSTRLEN];
    int err;
    ss=socket(AF_INET, SOCK_STREAM ,0);
    if(ss<0){
        showError();
        return -1;
    }
    cout<<"Init Socket"<<endl;

    bzero(&server_addr,sizeof(server_addr));
    server_addr.sin_family=AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port=htons(PORT);
    
    err=bind(ss,(struct sockaddr *)&server_addr,sizeof(server_addr));
    if(err<0){
        showError();
        return -1;
    }
    cout<<"Bind"<<endl;
    
    err=listen(ss,2);
    if(err<0){
        showError();
        return -1;
    }
    cout<<"Waiting for connect..."<<endl;
    
    while(true){
        socklen_t addrlen=sizeof(struct sockaddr);
        sc=accept(ss,(struct sockaddr*)&client_addr,&addrlen);
        if(sc==-1){
            showError();
            continue;
        } else {
            int peer_len=sizeof(peer_addr);
            getpeername(sc,(struct sockaddr *)&peer_addr,(socklen_t *)&peer_len);
            cout<<"Connect from "<<inet_ntop(AF_INET,&peer_addr.sin_addr,conn_ip,sizeof(conn_ip))<<". : "<<ntohs(peer_addr.sin_port)<<endl;
            ssize_t size=0;
            char buffer[1024];
            while(true){
                size=read(sc,buffer,1024);
                if(size<=0){
                    showError();
                    continue;
                }
                cout<<"Get message: "<<buffer<<endl;
                cout<<"Send : "<<endl;
                cin>>buffer;
                write(sc,buffer,strlen(buffer)+1);
            }
        }
    }
    return 0;
}

Client：

#include<iostream>
#include<string.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<unistd.h>

#define PORT 10004

using namespace std;

int main(int argc,char *argv[])
{
    int s;
    struct sockaddr_in server_addr;
    
    s=socket(AF_INET,SOCK_STREAM,0);
    if(s<0){
        cout<<"Init socket error"<<endl;
        return -1;
    }
    bzero(&server_addr,sizeof(server_addr));
    server_addr.sin_family=AF_INET;
    server_addr.sin_port=htons(PORT);
    inet_pton(AF_INET, argv[1], &server_addr.sin_addr);
    
    cout<<"Waiting for connect to "<<argv[1]<<"."<<endl;

    while(true){
        int err=connect(s,(struct sockaddr*)&server_addr,sizeof(server_addr));
        if(err==-1){
            cout<<"Conn Error."<<endl;
        } else {
            ssize_t size=0;
            char buffer[1024];
            while(true){
                cout<<"Send msg:"<<endl;
                cin>>buffer;
                write(s,buffer,strlen(buffer)+1);
                size=read(s,buffer,1024);
                cout<<"Message from server: "<<buffer<<endl;
            } 
        }
    }
    return 0;
}