branches/RB-1.1/ftptype.c

/*

    libftp 1.0 (stable): this is an FTP library to simplify the work to a Developer
    who want to work with FTP servers (RFC 959).
    Copyright (C) 2007/2008  omnipresent - omnipresent[at]email.it

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.

*/

#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <netinet/in.h>
#include <netdb.h>
#include <fcntl.h>

#include "ftp_main.h"

#define         ASCIITYPE       1       //ASCII Type
#define         NPRINTYPE       2       //Non-Print Type
#define         IMAGETYPE       3       //Image Type
#define         EBCDIC          4       //EBCDIC Type
#define         TELNET          5       //Telnet Type

/***
        ftp_type:
        - int sck      => Socket Descriptor
        - int t_type => What kind of TYPE? ASCII(1), IMAGE(3)...
        - int verbose  => Verbose?
                . 0 Do not Print
                . 1 Print
        - ret          => Success?
                . 0 Success
                .-1 Error
***/

int ftp_type(int sck, int t_type, int verbose) {


        char cmd_type[10]={'\0'};
        char *buf;
        int n = 0;
        buf = (char *)malloc(1024*sizeof(char));

                
        switch(t_type) {

        case ASCIITYPE:
        sprintf(cmd_type, "TYPE A\n");
        break;

        case NPRINTYPE:
        sprintf(cmd_type, "TYPE N\n");
        break;

        case IMAGETYPE:
        sprintf(cmd_type, "TYPE I\n");
        break;

        case EBCDIC:
        sprintf(cmd_type, "TYPE E\n");
        break;

        case TELNET:
        sprintf(cmd_type, "TYPE T\n");
        break;

        default:
        free(buf);

        return -1;
        break;

        }

        memset(buf, 0x0, 1024);

        
    if (write(sck, cmd_type, strlen(cmd_type)) == -1) {

                if (verbose) {
                        if (errno == EBADF) {
                                fprintf(stderr, "Socket Descriptor Not Valid!\n");
                                free(buf);
                                
                                return -1;
                        } else if (errno == EIO) {
                                fprintf(stderr, "I/O Error!\n");
                                free(buf);
                                
                                return -1;
                        } else if (errno == EINTR) {
                                fprintf(stderr, "Signal Interrupted The write() Function\n");
                                free(buf);
                                
                                return -1;
                        }

                } else {

                        if (errno == EBADF) {
                                free(buf);
                                
                                return -1;
                        } else if (errno == EIO) {
                                free(buf);
                                
                                return -1;
                        } else if (errno == EINTR) {
                                free(buf);
                                
                                return -1;
                        }

                }

                        free(buf);
                        
                        return -1;

                        }

        struct timeval tm; //tv_sec - tv_usec
        fd_set readfds;
        tm.tv_sec = 3;
        tm.tv_usec = 0;

        FD_ZERO(&readfds);
        
        FD_SET(sck, &readfds);

        if (select(sck+1, &readfds, NULL, NULL, &tm) < 0 ) {
                if (verbose) {
                perror("select()");
                }
                free(buf);
                
                return -1;
        } else {

                if (FD_ISSET(sck, &readfds)) {

                        if ( (n = recv(sck, buf, 1022, 0)) < 0)  {
                        if (verbose) {
                        perror("recv()");
                        }
                        free(buf);
                
                        return -1;
                        }

                        if (verbose)
                        printf("[Server] %s\n", buf);
                        
                }
                else {
                        if (verbose)
                        printf("[-] Timeout\n");
                        
                        return -1;
                }
        }

        buf[n] = '\0';

        memset(buf, 0x0, 1024);

        free(buf);


    return 0; //0 returned; operation sucessfull!

}