프로그래밍

3206,

9/161

매혹

http://omaggo.blog.me

행님들 버퍼문제 좀 도와주이소

http://www.hackerschool.org/HS_Boards/zboard.php?id=QNA_programming&no=3330 [복사]

[op_client.c]
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define BUF_SIZE 1024
void error_handling( char *message );

int main( int argc, char *argv[] )
{
        int sock;
        int result, cnt, num, op_cnt;

        char operator;
        struct sockaddr_in serv_adr;

        if( argc != 3 ) {
                printf( "Usage : %s <IP> <port> \n", argv[0] );
                exit(1);
        }

        sock = socket( PF_INET, SOCK_STREAM, 0 );
        if( sock == -1 )
                error_handling( "socket() error" );

        memset( &serv_adr, 0, sizeof(serv_adr) );
        serv_adr.sin_family = AF_INET;
        serv_adr.sin_addr.s_addr = inet_addr( argv[1] );
        serv_adr.sin_port = htons( atoi(argv[2]) );

        if( connect(sock, (struct sockaddr*) &serv_adr, sizeof(serv_adr)) == -1 )
                error_handling( "connect() error!" );
        else
                puts( "Connected.........." );

        {
                fputs( "Operand count: ", stdout );
                scanf( "%d", &op_cnt );

                write( sock, &op_cnt, sizeof(op_cnt) );

                cnt = 0;
                while( cnt++ < op_cnt ) {
                        printf( "Operand %d: ", cnt );
                        scanf( "%d", &num );

                        write( sock, &num, sizeof(num) );
                }

                fputs( "Operator: ", stdout );
                scanf( "%c", &operator );

                write( sock, &operator, sizeof(operator) );
                read( sock, &result, sizeof(result) );

                printf( "Operation result: %d \n", result );
        }
        close(sock);

        return 0;
}

void error_handling( char *message )
{
        fputs( message, stderr );
        fputc( '\n', stderr );
        exit(1);
}

[op_server.c]
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>

#define BUF_SIZE 1024
void error_handling( char *message );

int main( int argc, char *argv[] )
{
        int serv_sock, clnt_sock;
        int array[BUF_SIZE];

        int i, cnt, result, op_cnt;
        char buffer[BUF_SIZE], operator;

        struct sockaddr_in serv_adr, clnt_adr;
        socklen_t clnt_adr_sz;

        if( argc != 2 ) {
                printf( "Usage : %s <port> \n", argv[0] );
                exit(1);
        }

        serv_sock = socket( PF_INET, SOCK_STREAM, 0 );
        if( serv_sock == -1 )
                error_handling( "socket() error" );

        memset( &serv_adr, 0, sizeof(serv_adr) );
        serv_adr.sin_family = AF_INET;
        serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
        serv_adr.sin_port = htons( atoi(argv[1]) );

        if( bind(serv_sock, (struct sockaddr*) &serv_adr, sizeof(serv_adr) ) == -1 )
                error_handling( "listen() error" );

        if( listen(serv_sock, 5) == -1 )
                error_handling( "listen() error" );

        clnt_adr_sz = sizeof(clnt_adr);

        for( i=0; i<5; i++ )
        {
                clnt_sock = accept( serv_sock, (struct sockaddr*) &clnt_adr, &clnt_adr_sz );
                if( clnt_sock == -1 )
                        error_handling( "accept() error" );

                read( clnt_sock, &op_cnt, BUF_SIZE );

                cnt = 0;
                while( cnt < op_cnt ) {
                        read( clnt_sock, &array[cnt], sizeof(array[cnt]) );
                        cnt++;
                }

                read( clnt_sock, buffer, sizeof(buffer) );
                operator = buffer[0];

                cnt = 0;
                result = array[cnt];
                switch( operator ) {
                        case '+':
                                while( ++cnt < op_cnt )
                                        result += array[cnt];
                                break;
                        case '-':
                                while( ++cnt < op_cnt )
                                        result -= array[cnt];
                                break;
                        case '*':
                                while( ++cnt < op_cnt )
                                        result *= array[cnt];
                                break;
                }
                write( clnt_sock, &result, sizeof(int) );

                close(clnt_sock);
        }
        close(serv_sock);
        return 0;
}

void error_handling( char *message )
{
        fputs( message, stderr );
        fputc( '\n', stderr );
        exit(1);
}

[실행결과]
Connected..........
Operand count: 3
Operand 1: 12
Operand 2: 24
Operand 3: 36
Operator: Operation result: 12

마지막에 Operator: 에서 값을입력받지 않고 바로 훅 건너뛰는데
아무래도 버퍼에 문제가 있는듯 하오.
짧은 소인의 의견으로는 아마 \n이 문제이지 않을까 하는데..
해결해주이소~~

Hit : 4371 Date : 2012/03/05 08:39