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

简介: 简单的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;
}
目录
相关文章
|
6月前
|
存储 算法 调度
【复现】【遗传算法】考虑储能和可再生能源消纳责任制的售电公司购售电策略(Python代码实现)
【复现】【遗传算法】考虑储能和可再生能源消纳责任制的售电公司购售电策略(Python代码实现)
312 26
|
6月前
|
测试技术 开发者 Python
Python单元测试入门:3个核心断言方法,帮你快速定位代码bug
本文介绍Python单元测试基础,详解`unittest`框架中的三大核心断言方法:`assertEqual`验证值相等,`assertTrue`和`assertFalse`判断条件真假。通过实例演示其用法,帮助开发者自动化检测代码逻辑,提升测试效率与可靠性。
508 1
|
6月前
|
机器学习/深度学习 算法 调度
基于多动作深度强化学习的柔性车间调度研究(Python代码实现)
基于多动作深度强化学习的柔性车间调度研究(Python代码实现)
335 1
|
5月前
|
测试技术 Python
Python装饰器:为你的代码施展“魔法”
Python装饰器:为你的代码施展“魔法”
334 100
|
5月前
|
开发者 Python
Python列表推导式:一行代码的艺术与力量
Python列表推导式:一行代码的艺术与力量
504 95
|
6月前
|
Python
Python的简洁之道:5个让代码更优雅的技巧
Python的简洁之道:5个让代码更优雅的技巧
333 104
|
6月前
|
开发者 Python
Python神技:用列表推导式让你的代码更优雅
Python神技:用列表推导式让你的代码更优雅
594 99
|
5月前
|
缓存 Python
Python装饰器:为你的代码施展“魔法
Python装饰器:为你的代码施展“魔法
285 88
|
6月前
|
IDE 开发工具 开发者
Python类型注解:提升代码可读性与健壮性
Python类型注解:提升代码可读性与健壮性
326 102
|
5月前
|
监控 机器人 编译器
如何将python代码打包成exe文件---PyInstaller打包之神
PyInstaller可将Python程序打包为独立可执行文件,无需用户安装Python环境。它自动分析代码依赖,整合解释器、库及资源,支持一键生成exe,方便分发。使用pip安装后,通过简单命令即可完成打包,适合各类项目部署。
1011 68

推荐镜像

更多