简单的Python文件服务器和HTTP POST上传文件C代码

代码都是网上收集的,已经打包提供下载。

  • 文件服务器:
#!/usr/bin/env python  

"""Simple HTTP Server With Upload. 

This module builds on BaseHTTPServer by implementing the standard GET 
and HEAD requests in a fairly straightforward manner. 

"""  


__version__ = "0.1"  
__all__ = ["SimpleHTTPRequestHandler"]  
__author__ = "bones7456"  
__home_page__ = "http://luy.li/"  

import os  
import posixpath  
import BaseHTTPServer  
import urllib  
import cgi  
import shutil  
import mimetypes  
import re  
try:  
    from cStringIO import StringIO  
except ImportError:  
    from StringIO import StringIO  


class SimpleHTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):  

    """Simple HTTP request handler with GET/HEAD/POST commands. 

    This serves files from the current directory and any of its 
    subdirectories.  The MIME type for files is determined by 
    calling the .guess_type() method. And can reveive file uploaded 
    by client. 

    The GET/HEAD/POST requests are identical except that the HEAD 
    request omits the actual contents of the file. 

    """  

    server_version = "SimpleHTTPWithUpload/" + __version__  

    def do_GET(self):  
        """Serve a GET request."""  
        f = self.send_head()  
        if f:  
            self.copyfile(f, self.wfile)  
            f.close()  

    def do_HEAD(self):  
        """Serve a HEAD request."""  
        f = self.send_head()  
        if f:  
            f.close()  

    def do_POST(self):  
        """Serve a POST request."""  
        r, info = self.deal_post_data()  
        print r, info, "by: ", self.client_address  
        f = StringIO()  
        f.write('<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN">')  
        f.write("<html>\n<title>Upload Result Page</title>\n")  
        f.write("<body>\n<h2>Upload Result Page</h2>\n")  
        f.write("<hr>\n")  
        if r:  
            f.write("<strong>Success:</strong>")  
        else:  
            f.write("<strong>Failed:</strong>")  
        f.write(info)  
        f.write("<br><a href=\"%s\">back</a>")
# % self.headers['referer'])  
        f.write("<hr><small>Powered By: bones7456, check new version at ")  
        f.write("<a href=\"http://luy.li/?s=SimpleHTTPServerWithUpload\">")  
        f.write("here</a>.</small></body>\n</html>\n")  
        length = f.tell()  
        f.seek(0)  
        self.send_response(200)  
        self.send_header("Content-type", "text/html")  
        self.send_header("Content-Length", str(length))  
        self.end_headers()  
        if f:  
            self.copyfile(f, self.wfile)  
            f.close()  

    def deal_post_data(self):  
        boundary = self.headers.plisttext.split("=")[1]  
        remainbytes = int(self.headers['content-length'])

        line = self.rfile.readline()  
        remainbytes -= len(line)  
        if not boundary in line:  
            return (False, "Content NOT begin with boundary")  

        # may have two Content-Disposition headers. and name= ?
        line = self.rfile.readline()  
        remainbytes -= len(line)  
        fn = re.findall(r'Content-Disposition.*name="file"; filename="(.*)"', line)  
        if not fn:
            return (False, "Can't find out file name...")  
            
        path = self.translate_path(self.path)  

        fn = os.path.join(path, fn[0])  
        #while os.path.exists(fn):  
        #    fn += "_"  
        line = self.rfile.readline()  
        remainbytes -= len(line)  
        line = self.rfile.readline()  
        remainbytes -= len(line)  
        try:  
            out = open(fn, 'wb')  
        except IOError:  
            return (False, "Can't create file to write, do you have permission to write?")  

        preline = self.rfile.readline()  
        remainbytes -= len(preline)  
        while remainbytes > 0:  
            line = self.rfile.readline()  
            remainbytes -= len(line)  
            if boundary in line:  
                preline = preline[0:-1]  
                if preline.endswith('\r'):  
                    preline = preline[0:-1]  
                out.write(preline)  
                out.close()  
                return (True, "File '%s' upload success!" % fn)  
            else:  
                out.write(preline)  
                preline = line  
        return (False, "Unexpect Ends of data.")  

    def send_head(self):  
        """Common code for GET and HEAD commands. 

        This sends the response code and MIME headers. 

        Return value is either a file object (which has to be copied 
        to the outputfile by the caller unless the command was HEAD, 
        and must be closed by the caller under all circumstances), or 
        None, in which case the caller has nothing further to do. 

        """  
        path = self.translate_path(self.path)  
        f = None  
        if os.path.isdir(path):  
            if not self.path.endswith('/'):  
                # redirect browser - doing basically what apache does  
                self.send_response(301)  
                self.send_header("Location", self.path + "/")  
                self.end_headers()  
                return None  
            for index in "index.html", "index.htm":  
                index = os.path.join(path, index)  
                if os.path.exists(index):  
                    path = index  
                    break  
            else:  
                return self.list_directory(path)  
        ctype = self.guess_type(path)  
        try:  
            # Always read in binary mode. Opening files in text mode may cause  
            # newline translations, making the actual size of the content  
            # transmitted *less* than the content-length!  
            f = open(path, 'rb')  
        except IOError:  
            self.send_error(404, "File not found")  
            return None  
        self.send_response(200)  
        self.send_header("Content-type", ctype)  
        fs = os.fstat(f.fileno())  
        self.send_header("Content-Length", str(fs[6]))  
        self.send_header("Last-Modified", self.date_time_string(fs.st_mtime))  
        self.end_headers()  
        return f  

    def list_directory(self, path):  
        """Helper to produce a directory listing (absent index.html). 

        Return value is either a file object, or None (indicating an 
        error).  In either case, the headers are sent, making the 
        interface the same as for send_head(). 

        """  
        try:  
            list = os.listdir(path)  
        except os.error:  
            self.send_error(404, "No permission to list directory")  
            return None  
        list.sort(key=lambda a: a.lower())  
        f = StringIO()  
        displaypath = cgi.escape(urllib.unquote(self.path))  
        f.write('<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN">')  
        f.write("<html>\n<title>Directory listing for %s</title>\n" % displaypath)  
        f.write("<body>\n<h2>Directory listing for %s</h2>\n" % displaypath)  
        f.write("<hr>\n")  
        f.write("<form ENCTYPE=\"multipart/form-data\" method=\"post\">")  
        f.write("<input name=\"file\" type=\"file\"/>")  
        f.write("<input type=\"submit\" value=\"upload\"/></form>\n")  
        f.write("<hr>\n<ul>\n")  
        for name in list:  
            fullname = os.path.join(path, name)  
            displayname = linkname = name  
            # Append / for directories or @ for symbolic links  
            if os.path.isdir(fullname):  
                displayname = name + "/"  
                linkname = name + "/"  
            if os.path.islink(fullname):  
                displayname = name + "@"  
                # Note: a link to a directory displays with @ and links with /  
            f.write('<li><a href="%s">%s</a>\n'  
                    % (urllib.quote(linkname), cgi.escape(displayname)))  
        f.write("</ul>\n<hr>\n</body>\n</html>\n")  
        length = f.tell()  
        f.seek(0)  
        self.send_response(200)  
        self.send_header("Content-type", "text/html")  
        self.send_header("Content-Length", str(length))  
        self.end_headers()  
        return f  

    def translate_path(self, path):  
        """Translate a /-separated PATH to the local filename syntax. 

        Components that mean special things to the local file system 
        (e.g. drive or directory names) are ignored.  (XXX They should 
        probably be diagnosed.) 

        """  
        # abandon query parameters  
        path = path.split('?',1)[0]  
        path = path.split('#',1)[0]  
        path = posixpath.normpath(urllib.unquote(path))  
        words = path.split('/')  
        words = filter(None, words)  
        path = os.getcwd()  
        for word in words:  
            drive, word = os.path.splitdrive(word)  
            head, word = os.path.split(word)  
            if word in (os.curdir, os.pardir): continue  
            path = os.path.join(path, word)  
        return path  

    def copyfile(self, source, outputfile):  
        """Copy all data between two file objects. 

        The SOURCE argument is a file object open for reading 
        (or anything with a read() method) and the DESTINATION 
        argument is a file object open for writing (or 
        anything with a write() method). 

        The only reason for overriding this would be to change 
        the block size or perhaps to replace newlines by CRLF 
        -- note however that this the default server uses this 
        to copy binary data as well. 

        """  
        shutil.copyfileobj(source, outputfile)  

    def guess_type(self, path):  
        """Guess the type of a file. 

        Argument is a PATH (a filename). 

        Return value is a string of the form type/subtype, 
        usable for a MIME Content-type header. 

        The default implementation looks the file's extension 
        up in the table self.extensions_map, using application/octet-stream 
        as a default; however it would be permissible (if 
        slow) to look inside the data to make a better guess. 

        """  

        base, ext = posixpath.splitext(path)  
        if ext in self.extensions_map:  
            return self.extensions_map[ext]  
        ext = ext.lower()  
        if ext in self.extensions_map:  
            return self.extensions_map[ext]  
        else:  
            return self.extensions_map['']  

    if not mimetypes.inited:  
        mimetypes.init() # try to read system mime.types  
    extensions_map = mimetypes.types_map.copy()  
    extensions_map.update({  
        '': 'application/octet-stream', # Default  
        '.py': 'text/plain',  
        '.c': 'text/plain',  
        '.h': 'text/plain',  
        })  


def test(HandlerClass = SimpleHTTPRequestHandler,  
         ServerClass = BaseHTTPServer.HTTPServer):  
    BaseHTTPServer.test(HandlerClass, ServerClass)  

if __name__ == '__main__':  
    test()
  • tcpclient.h
#ifndef __TCP_CLIENT_H__
#define __TCP_CLIENT_H__
 
#include <netinet/in.h>
#include <sys/socket.h>
 
typedef struct _tcpclient{
    int socket;
    int remote_port;
    char remote_ip[16];
    struct sockaddr_in _addr;
    int connected;
} tcpclient;
 
int tcpclient_create(tcpclient *,const char *host, int port);
int tcpclient_conn(tcpclient *);
int tcpclient_recv(tcpclient *,char **lpbuff,int size);
int tcpclient_send(tcpclient *,char *buff,int size);
int tcpclient_close(tcpclient *);
 
#endif
  • tcpclient.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <string.h>

#include "tcpclient.h"
 
#define BUFFER_SIZE 1024
 
 
int tcpclient_create(tcpclient *pclient,const char *host, int port){
    struct hostent *he;
 
    if(pclient == NULL) return -1;
    memset(pclient,0,sizeof(tcpclient));
 
    if((he = gethostbyname(host))==NULL){
        return -2;
    }
 
    pclient->remote_port = port;
    strcpy(pclient->remote_ip,inet_ntoa( *((struct in_addr *)he->h_addr) ));
 
    pclient->_addr.sin_family = AF_INET;
    pclient->_addr.sin_port = htons(pclient->remote_port);
    pclient->_addr.sin_addr = *((struct in_addr *)he->h_addr);
 
    if((pclient->socket = socket(AF_INET,SOCK_STREAM,0))==-1){
        return -3;
    }
 
    /*TODO:是否应该释放内存呢?*/
 
    return 0;
}
 
int tcpclient_conn(tcpclient *pclient){
    if(pclient->connected)
        return 1;
 
    if(connect(pclient->socket, (struct sockaddr *)&pclient->_addr,sizeof(struct sockaddr))==-1){
        return -1;
    }
 
    pclient->connected = 1;
 
    return 0;
}
 
int tcpclient_recv(tcpclient *pclient,char **lpbuff,int size){
    int recvnum=0,tmpres=0;
    char buff[BUFFER_SIZE];
 
    *lpbuff = NULL;
 
    while(recvnum < size || size==0){
        tmpres = recv(pclient->socket, buff,BUFFER_SIZE,0);
        if(tmpres <= 0)
            break;
        recvnum += tmpres;
 
        if(*lpbuff == NULL){
            *lpbuff = (char*)malloc(recvnum);
            if(*lpbuff == NULL)
                return -2;
        }else{
            *lpbuff = (char*)realloc(*lpbuff,recvnum);
            if(*lpbuff == NULL)
                return -2;
        }
 
        memcpy(*lpbuff+recvnum-tmpres,buff,tmpres);
    }
 
    return recvnum;
}
 
int tcpclient_send(tcpclient *pclient,char *buff,int size){
    int sent=0,tmpres=0;
 
    while(sent < size){
        tmpres = send(pclient->socket,buff+sent,size-sent,0);
        if(tmpres == -1){
            return -1;
        }
        sent += tmpres;
    }
    return sent;
}
 
int tcpclient_close(tcpclient *pclient){
    close(pclient->socket);
    pclient->connected = 0;
    return 0;
}
  • httpost.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <sys/stat.h>
#include "tcpclient.h"


int http_post_file(tcpclient *pclient, const char *page, const char *filepath,char **response){
 
	//check if the file is valid or not
	struct stat stat_buf;
	if(lstat(filepath,&stat_buf)<0){
		printf("lstat %s fail", filepath);
		return -1;
	}
	if(!S_ISREG(stat_buf.st_mode)){
		printf("%s is not a regular file!",filepath);
		return  -1;
	}
 
	char *filename;
	filename=strrchr(filepath,'/');
	if(filename==NULL){
 
	}
	filename+=1;
	if(filename>=filepath+strlen(filepath)){
		//'/' is the last character
		printf("%s is not a correct file!",filepath);
		return  -1;
	}
 
	printf("filepath=%s,filename=%s",filepath,filename);
 
	char content_type[4096];
	memset(content_type, 0, 4096);
 
	char post[512],host[256],content_len[256];
	char *lpbuf,*ptmp;
	int len=0;
 
	lpbuf = NULL;
	const char *header2="User-Agent: Is Http 1.1\r\nCache-Control: no-cache\r\nAccept: */*\r\n";
 
	sprintf(post,"POST %s HTTP/1.1\r\n",page);
	sprintf(host,"HOST: %s:%d\r\n",pclient->remote_ip,pclient->remote_port);
	strcpy(content_type,post);
	strcat(content_type,host);
 
 
	char *boundary = (char *)"-----------------------7d9ab1c50098";
 
	strcat(content_type, "Content-Type: multipart/form-data; boundary=");
	strcat(content_type, boundary);
	strcat(content_type, "\r\n");

	//--Construct request data {filePath, file}
	char content_before[8192];
	memset(content_before, 0, 8192);
	strcat(content_before, "--");
	strcat(content_before, boundary);
	strcat(content_before, "\r\n");


        /*

        //附加数据。
        char* message_json = "{\"password\":\"051784\",\"activated_time\":1544098669817,\"message_class\":1,\"phone_number\":\"15252450001\",\"message_id\":5}";

	strcat(content_before, "Content-Disposition: form-data; name=\"warning_message\"\r\n\r\n");
	strcat(content_before, message_json);
	strcat(content_before, "\r\n");
	strcat(content_before, "--");
	strcat(content_before, boundary);
	strcat(content_before, "\r\n");
        */

	strcat(content_before, "Content-Disposition: attachment; name=\"file\"; filename=\"");
	strcat(content_before, filename);
	strcat(content_before, "\"\r\n");
	strcat(content_before, "Content-Type: image/jpeg\r\n\r\n");


	char content_end[2048];
	memset(content_end, 0, 2048);
	strcat(content_end, "\r\n");
	strcat(content_end, "--");
	strcat(content_end, boundary);
	strcat(content_end, "--\r\n");

	int max_cont_len=5*1024*1024;
	char content[max_cont_len];
	int fd;
	fd=open(filepath,O_RDONLY,0666);
	if(!fd){
		printf("fail to open file : %s",filepath);
		return -1;
	}
	len=read(fd,content,max_cont_len);
	close(fd);

	char *lenstr;
	lenstr = (char*)malloc(256);
	sprintf(lenstr, "%d", (int)(strlen(content_before)+len+strlen(content_end)));
 
	strcat(content_type, "Content-Length: ");
	strcat(content_type, lenstr);
	strcat(content_type, "\r\n\r\n");

	//send
	if(!pclient->connected){
		tcpclient_conn(pclient);
	}
 
	//content-type
	tcpclient_send(pclient,content_type,strlen(content_type));
 
	//content-before
	tcpclient_send(pclient,content_before,strlen(content_before));
 
	//content
	tcpclient_send(pclient,content,len);
 
    //content-end
	tcpclient_send(pclient,content_end,strlen(content_end));

	/*it's time to recv from server*/
	if(tcpclient_recv(pclient,&lpbuf,0) <= 0){
		if(lpbuf) free(lpbuf);
		return -2;
	}

	printf("接收响应:\n%s",lpbuf);
 
	    /*响应代码,|HTTP/1.0 200 OK|
	     *从第10个字符开始,第3位
	     * */
	    memset(post,0,sizeof(post));
	    strncpy(post,lpbuf+9,3);
	    if(atoi(post)!=200){
	        if(lpbuf) free(lpbuf);
	        return atoi(post);
	    }
 
 
	    ptmp = (char*)strstr(lpbuf,"\r\n\r\n");
	    if(ptmp == NULL){
	        free(lpbuf);
	        return -3;
	    }
	    ptmp += 4;/*跳过\r\n*/
 
	    len = strlen(ptmp)+1;
	    *response=(char*)malloc(len);
	    if(*response == NULL){
	        if(lpbuf) free(lpbuf);
	        return -1;
	    }
	    memset(*response,0,len);
	    memcpy(*response,ptmp,len-1);
 
	    /*从头域找到内容长度,如果没有找到则不处理*/
	    ptmp = (char*)strstr(lpbuf,"Content-Length:");
	    if(ptmp != NULL){
	        char *ptmp2;
	        ptmp += 15;
	        ptmp2 = (char*)strstr(ptmp,"\r\n");
	        if(ptmp2 != NULL){
	            memset(post,0,sizeof(post));
	            strncpy(post,ptmp,ptmp2-ptmp);
	            if(atoi(post)<len)
	                (*response)[atoi(post)] = '\0';
	        }
	    }
 
	    if(lpbuf) free(lpbuf);
 
	    return 0;
 
}
 
int http_post(tcpclient *pclient,char *page,char *request,char **response){
 
    char post[300],host[100],content_len[100];
    char *lpbuf,*ptmp;
    int len=0;
 
    lpbuf = NULL;
    const char *header2="User-Agent: Grandhonor Http 0.1\r\nCache-Control: no-cache\r\nContent-Type: application/x-www-form-urlencoded\r\nAccept: */*\r\n";
 
    sprintf(post,"POST %s HTTP/1.0\r\n",page);
    sprintf(host,"HOST: %s:%u\r\n", pclient->remote_ip, pclient->remote_port);
    sprintf(content_len,"Content-Length: %zu\r\n\r\n",strlen(request));
 
    len = strlen(post)+strlen(host)+strlen(header2)+strlen(content_len)+strlen(request)+1;
    lpbuf = (char*)malloc(len);
    if(lpbuf==NULL){
        return -1;
    }
 
    strcpy(lpbuf,post);
    strcat(lpbuf,host);
    strcat(lpbuf,header2);
    strcat(lpbuf,content_len);
    strcat(lpbuf,request);
 
    if(!pclient->connected){
        tcpclient_conn(pclient);
    }
 
    if(tcpclient_send(pclient,lpbuf,len)<0){
        return -1;
    }
    //printf("发送请求:\n%s",lpbuf);
 
    /*释放内存*/
    if(lpbuf != NULL) free(lpbuf);
    lpbuf = NULL;
 
    /*it's time to recv from server*/
    if(tcpclient_recv(pclient,&lpbuf,0) <= 0){
        if(lpbuf) free(lpbuf);
        return -2;
    }
    //printf("接收响应:\n%s",lpbuf);
 
    /*响应代码,|HTTP/1.0 200 OK|
     *从第10个字符开始,第3位
     * */
    memset(post,0,sizeof(post));
    strncpy(post,lpbuf+9,3);
    if(atoi(post)!=200){
        if(lpbuf) free(lpbuf);
        return atoi(post);
    }
 

    ptmp = (char*)strstr(lpbuf,"\r\n\r\n");
    if(ptmp == NULL){
        free(lpbuf);
        return -3;
    }
    ptmp += 4;/*跳过\r\n*/
 
    len = strlen(ptmp)+1;
    *response=(char*)malloc(len);
    if(*response == NULL){
        if(lpbuf) free(lpbuf);
        return -1;
    }
    memset(*response,0,len);
    memcpy(*response,ptmp,len-1);
 
    /*从头域找到内容长度,如果没有找到则不处理*/
    ptmp = (char*)strstr(lpbuf,"Content-Length:");
    if(ptmp != NULL){
        char *ptmp2;
        ptmp += 15;
        ptmp2 = (char*)strstr(ptmp,"\r\n");
        if(ptmp2 != NULL){
            memset(post,0,sizeof(post));
            strncpy(post,ptmp,ptmp2-ptmp);
            if(atoi(post)<len)
                (*response)[atoi(post)] = '\0';
        }
    }
 
    if(lpbuf) free(lpbuf);
 
    return 0;
}

int http_upload_file(const char* pHost, const int nPort, const char* pServerPath,
                     const char* pFile)
{
 
    tcpclient client;
 
    char *response = NULL;
    printf("开始组包%s", pFile);
    tcpclient_create(&client, pHost, nPort);
 
//    if(http_post(&client,"/recv_file.php","f1=hello",&response)){
//        printf("失败!");
//        exit(2);
//    }
    http_post_file(&client, pServerPath, pFile, &response);
 
    printf("responsed %zu:%s", strlen(response), response);
 
    free(response);
    return 0;
}

int main()
{
    http_upload_file("127.0.0.1", 8000, "/","/home/quantum6/index.html");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/quantum7/article/details/84870398