[原创]Linux下使用Select模型的两个类Client And Server

终于搞定了所有的事情，一句话，无语，极度无语

IOCP的封装类[By Sodme]

大 | 中 | 小

2008/03/10 发表时间

11:46

3371

huzhangyou2002

开发语言 » 代码库

本来项目打算用这种方案的，后来推翻了，当时写的两个类就搁置了，不过感觉还是可以用的上，算是对博客这段时间关闭的道歉了。是模仿一个开源服务器开发的，并发2K没有问题的。

下载文件 (已下载 326 次)

点击这里下载文件: codes.rar

#include "client.h"
client::client(int fd, sockaddr_in sinaddr, int timeout){
    _fd = fd;
    _timeout = timeout;
    fcntl( _fd, F_SETFL, fcntl( _fd, F_GETFL ) | O_NONBLOCK );
    buffer = NULL;
    buffer = new char[BUFFERSIZE];
}
client::~client(){
    close(_fd);
    delete buffer;
    buffer = NULL;
}

bool client::update(){
    fd_set fdClient;
    
    FD_ZERO( &fdClient );
    FD_SET( _fd, &fdClient );
    
    struct timeval tv;
    
    tv.tv_sec = _timeout;
    tv.tv_usec = 0;
    
    
    if( select( _fd + 1, &fdClient, NULL, NULL, &tv ) == -1 ) {
        return true;
    }
    
    if( FD_ISSET( _fd, &fdClient ) ) {
        memset( buffer, 0, sizeof( char ) * BUFFERSIZE );
        
        int c = recv( _fd, buffer, BUFFERSIZE, 0 );
        
        if( c == -1 && errno != EWOULDBLOCK ) {
            if( errno != ECONNRESET )
                cout << "peer reset the connection" << endl;
            return true;
        }
        else if( c == 0 )
            return true;
        else if( c > 0 ) {
            _value = string( buffer, c );
            cout << _value << endl;
        }
        else {
            return true;
        }
    }
    return false;
}

#include "server.h"
server::server(int port = 10000){
    _kill = false;
    struct sockaddr_in sin;
    
    memset( &sin, 0, sizeof( sin ) );
    
    sin.sin_family = AF_INET;
    sin.sin_addr.s_addr = INADDR_ANY;
    
    sin.sin_port = htons(port);
    _fd = socket( PF_INET, SOCK_STREAM, 0 );
    
    int optval = 1;
    
    setsockopt( _fd, SOL_SOCKET, SO_REUSEADDR, (const void *)&optval, sizeof( int ) );
    if( bind( _fd, (struct sockaddr *)&sin, sizeof( sin ) ) == -1 ) {
        cout << "bind error" << endl;
    }
    
    // listen, queue length 8
    
    if( listen( _fd, 8 ) == -1 ) {
        cout << "listen error" << endl;
        //exit( 1 );
    }
}
server::~server(){
    close(_fd);
    
    for( vector<client *> :: iterator i = clients.begin( ); i != clients.end( ); i++ )
        delete *i;
    
    clients.clear( );
    
}

bool server::update(bool block){
    if( clients.size( ) < MAXCLIENTS ) {
        fd_set fdServer;
        
        FD_ZERO( &fdServer );
        FD_SET( _fd, &fdServer );
        
        struct timeval tv;
        
        if( block ) {
            // block for 100 ms to keep from eating up all cpu time
            
            tv.tv_sec = 0;
            tv.tv_usec = 100000;
        }
        else {
            tv.tv_sec = 0;
            tv.tv_usec = 0;
        }
        
        if( select( _fd + 1, &fdServer, NULL, NULL, &tv ) == -1 )
            
        {
            FD_ZERO( &fdServer );
            
            usleep( 100 );
        }
        
        if( FD_ISSET( _fd, &fdServer ) ) {
            struct sockaddr_in adrFrom;
            
            int iAddrLen = sizeof( adrFrom );
            
            int sckClient;
            
            if( ( sckClient = accept( _fd, (struct sockaddr *)&adrFrom, (socklen_t *)&iAddrLen ) ) == -1 )
                cout << "accept error" << endl;
            else{
                clients.push_back( new client( sckClient, adrFrom, 0) );
                cout << "clients comes" << endl;
            }
        }
    }
    else {
        // maximum connections reached
        
        usleep( 100 );
    }
    
    // process
    
    for( vector<client *> :: iterator i = clients.begin( ); i != clients.end( ); ) {
        if( (*i)->update( ) ) {
            delete *i;
            
            i = clients.erase( i );
            cout << "clients exit" << endl;
        }
        else
            i++;
    }
    return _kill;
}

dzl

2008/03/25 22:10

多谢回复。
仔细研究了个把小时，不是太明白。
能给个使用这两个类的例子么？服务端和客户端的。

huzhangyou2002 回复于 2008/03/25 23:00

明天给你贴出来不好意思忘记了贴

DZL

2008/03/25 03:24

client::client(int fd, sockaddr_in sinaddr, int timeout){
    _fd = fd;
    _timeout = timeout;
    fcntl( _fd, F_SETFL, fcntl( _fd, F_GETFL ) | O_NONBLOCK );
    buffer = NULL;
    buffer = new char[BUFFERSIZE];
}
sinaddr做什么用？

huzhangyou2002 回复于 2008/03/25 07:48

多谢兄台指点，的确并不需要该参数，而server类需要而已。不好意思，我的疏忽

分页： 1/1