pikapython/package/requests/_requests.c

#include "_requests.h"
#include "PikaObj.h"
#include <ctype.h>
#include "_requests_Response.h"
#include "webclient.h"
#include "random.h"

#if !PIKASCRIPT_VERSION_REQUIRE_MINIMUN(1, 11, 9)
#error "This library requires PikaScript version 1.11.9 or higher"
#endif

/* 标准输出函数 */
#define RQ_print(fmt, ...) __platform_printf(fmt, ##__VA_ARGS__)
#define RQ_cli(fmt, ...) __platform_printf(fmt, ##__VA_ARGS__)
#define RQ_debug(fmt, ...)   {}
/* #define RQ_debug(fmt, ...) __platform_printf("[RQ]:**|"fmt"|**\n", ##__VA_ARGS__) */
#define RQ_error_pointer(...) \
    RQ_print("[%s]: Checking NULL pointer of {" #__VA_ARGS__ "}.\n", __fun__)
#define RQ_error_value(str, ...)                                            \
    RQ_print("[%s]: Checking error value of {" #__VA_ARGS__ ":" str "}.\n", \
             __fun__, ##__VA_ARGS__)
#define RQ_err(fmt, ...) RQ_print("[%s]:" fmt "\n", __VA_ARGS__)

/* 补充常见宏定义 */
#ifndef unlikely
#define unlikely(x) __builtin_expect(!!(x), 0)
#endif
#ifndef likely
#define likely(x) __builtin_expect(!!(x), 1)
#endif

/**
 *  现在支持可选的额外参数
 *      params: 用于填充url参数，即get方法通过url传递数据
 *      headers: 用于响应头信息填充
 *      data: 负载内容
 *      json: 负载内容
 *      files: 负载内容
 *      timeout: 超时设置
 * 实际上支持有限
 */
int _requests_Response_request(PikaObj* self,
                               char* method,
                               char* url,
                               pika_float timeout,
                               char* data) {
    size_t data_len, resp_len; /* 长度信息 */
    void* resp_data;           /* 返回的负载内容 */
    int32_t ret;               /* 返回值 */
    struct webclient_session* session;

    session = (struct webclient_session*)obj_getPtr(self, "session_address");
    if (unlikely(session == NULL)) {
        RQ_cli("Sorry, can not operate NULL session object.\n");
        return -1;
    }
    if (unlikely(url == NULL)) {
        RQ_cli("Sorry, please give a specified url.\n");
        return -1;
    }

    RQ_debug("header length:%d", session->header->length);
    RQ_debug("header buffer:%.4096s", session->header->buffer);
    RQ_debug("header url:%.1024s", url);

    /* 进行post或者get操作 */
    if (strEqu(method, "GET")) {
        /* Get之后，header->buffer缓冲区内容会被清空 */
        /* FIXME: 保存一下header->buffer内容 */
        if (webclient_get2(session, url) != 200) {
            return -1;
        }
        ret = webclient_response(session, &resp_data, &resp_len);
        if (ret < 0) {
            return -1;
        }
        /* 正常得到了数据 */
        obj_setInt(self, "status_code", session->resp_status);
        obj_setInt(self, "content_length", resp_len);
        obj_setStr(self, "headers", session->header->buffer);
        /* 释放申请的缓冲区内存 */
        if (resp_data != NULL) {
            obj_setStr(self, "text", (char*)resp_data);
            web_free(resp_data);
        } else {
            obj_setStr(self, "text", "");
        }
    } else if (strEqu(method, "POST")) {
        if (data == NULL) {
            data_len = 0;
        } else {
            data_len = strlen(data);
        }
        /* FIXME: 默认二进制数据 */
        if (strstr(session->header->buffer, "Content-Length") == RT_NULL) {
            ret = webclient_header_fields_add(session, "Content-Length: %d\r\n",
                                              data_len);
            if (ret < 0) {
                return -1;
            }
        }
        if (strstr(session->header->buffer, "Content-Type") == RT_NULL) {
            /* 二进制数据流 */
            ret = webclient_header_fields_add(
                session, "Content-Type: application/octet-stream\r\n");
            if (ret < 0) {
                return -1;
            }
        }
        RQ_debug("header buffer:%.4096s", session->header->buffer);
        if (webclient_post(session, url, data, data_len) != 200) {
            return -1;
        }
        ret = webclient_response(session, &resp_data, &resp_len);
        if (ret < 0) {
            return -1;
        }
        /* 正常得到了数据 */
        obj_setInt(self, "status_code", session->resp_status);
        obj_setInt(self, "content_length", resp_len);
        obj_setStr(self, "headers", session->header->buffer);
        /* 释放申请的缓冲区内存 */
        if (resp_data != NULL) {
            obj_setStr(self, "text", (char*)resp_data);
            web_free(resp_data);
        } else {
            obj_setStr(self, "text", "");
        }
    } else {
        return -1;
    }

    return 1;
}

int _requests_Response_header_write(PikaObj* self, char* header, char* value) {
    struct webclient_session* session;

    session = (struct webclient_session*)obj_getPtr(self, "session_address");
    if (unlikely(session == NULL)) {
        RQ_cli("Sorry, can not operate NULL session object.\n");
        return -1;
    }
    RQ_debug("Add HTTP fileds:");
    /* 写入请求初始内容 */
    if (webclient_header_fields_add(session, "%s:%s\r\n", header, value) < 0) {
        RQ_cli("Sorry, request header too long.\n");
        return -1;
    }
    RQ_debug("buffer:%p.", session->header->buffer);
    return 1;
}

int _requests_Response_proto_write(PikaObj* self, char* proto) {
    struct webclient_session* session;

    session = (struct webclient_session*)obj_getPtr(self, "session_address");
    if (unlikely(session == NULL)) {
        RQ_cli("Sorry, can not operate NULL session object.\n");
        return -1;
    }
    /* TODO: 写入协议头之前可以保存 */
    RQ_debug("Add proto content:");
    /* 写入请求初始内容 */
    if (proto != NULL && *proto != 0) {
        if (webclient_header_fields_add(session, " %s\r\n", proto) < 0) {
            RQ_cli("Sorry, request header too long.\n");
            return -1;
        }
    } else {
        if (webclient_header_fields_add(session, " HTTP/1.1\r\n") < 0) {
            RQ_cli("Sorry, request header too long.\n");
            return -1;
        }
    }
    RQ_debug("buffer:%p.", session->header->buffer);
    return 1;
}

char to_hex(char code) {
    static char hex[] = "0123456789abcdef";
    return hex[code & 15];
}

int _requests_Response_urlencode_write(PikaObj* self,
                                       char* s1,
                                       char* s2,
                                       char* start,
                                       char* connect) {
    struct webclient_session* session;
    char *url_address, *p, *s;
    int32_t length, header_length;

    session = (struct webclient_session*)obj_getPtr(self, "session_address");
    if (unlikely(session == NULL)) {
        RQ_cli("Sorry, can not operate NULL session object.\n");
        return -1;
    }
    header_length = session->header->length;
    url_address = session->header->buffer + header_length;
    p = url_address;
    RQ_debug("Add url content:");
    RQ_debug("buffer:%p,len:%d.", session->header->buffer, header_length);
    RQ_debug("%s%s%s%s", start, s1, connect, s2);
    
    if (*start == 0) {
        while(*s1)
        {
            *p++ = *s1++;
        }
        goto end;
    }

    if (start != NULL) {
        /* 写入前置符号 */
        s = (char*)start;
        while (*s) {
            *p++ = *s++;
        }
    }

    s = s1;
    while (*s) {
        if (isalnum(*s) || *s == '-' || *s == '_' || *s == '.' || *s == '~') {
            *p++ = *s;
        } else if (*s == ' ') {
            *p++ = '+';
        } else {
            *p++ = '%';
            *p++ = to_hex(*s >> 4);
            *p++ = to_hex(*s & 0xf);
        }
        s++;
    }
    if (connect != NULL) {
        /* 写入连接符号 */
        s = (char*)connect;
        while (*s) {
            *p++ = *s++;
        }
    }
    s = (char*)s2;
    if (s != NULL) {
        while (*s) {
            if (isalnum(*s) || *s == '-' || *s == '_' || *s == '.' ||
                *s == '~') {
                *p++ = *s;
            } else if (*s == ' ') {
                *p++ = '+';
            } else {
                *p++ = '%';
                *p++ = to_hex(*s >> 4);
                *p++ = to_hex(*s & 0xf);
            }
            s++;
        }
    }

end:
    *p = '\0';
    length = p - url_address;
    length += header_length;
    if (length > session->header->size) {
        RQ_cli("Error, header buffer has overflowed: {%d}.\n", length);
        return -1;
    }
    /* 更新长度信息 */
    session->header->length = length;
    RQ_debug("context:%.1024s.", session->header->buffer);
    RQ_debug("url length add:%d.", length);
    return 1;
}

int _requests_Response_request_init(PikaObj* self, char* method) {
    /* 创建会话对象，header长度固定 */
    struct webclient_session* session;
    char* temp;

    if (unlikely((strEqu(method, "GET") || strEqu(method, "POST")) == 0)) {
        /* 目前仅支持两种方法 */
        RQ_cli("Sorry, now only support method: GET and POST.\n");
        return -1;
    }

    session = webclient_session_create(WEBCLIENT_HEADER_BUFSZ);
    if (session == RT_NULL) {
        RQ_cli("Sorry, memeory is not enough.\n");
        obj_setPtr(self, "session_address", NULL);
        return -1;
    } else {
        /* 写入请求初始内容 */
        if (webclient_header_fields_add(session, "%s ", method) < 0) {
            _requests_Response_request_del(self);
            RQ_cli("Sorry, request header too long.\n");
            return -1;
        }
        temp = session->header->buffer + session->header->length;
        RQ_debug("buffer address:%p.", session->header->buffer);
        RQ_debug("url address:%p.", temp);
        RQ_debug("headers start:%.10s.", session->header->buffer);
        obj_setPtr(self, "url_address", temp);
        obj_setPtr(self, "session_address", session);
    }
    RQ_debug("Response deal session memory.");
    return 1;
}

PikaObj* _requests_Response_request_del(PikaObj* self) {
    struct webclient_session* session;
    session = (struct webclient_session*)obj_getPtr(self, "session_address");
    if (session) {
        webclient_close(session);
        RQ_debug("Response free session memory.");
    }
    RQ_debug("Response set variables None.");
    return NULL;
}

void _requests_Response___del__(PikaObj* self) {
    _requests_Response_request_del(self);
    RQ_debug("Response class deleted.");
}

void _requests_Response___init__(PikaObj* self) {
    /* 初始化 */
    obj_setNone(self, "url");
    obj_setNone(self, "text");
    obj_setNone(self, "headers");
    obj_setPtr(self, "session_address", NULL);
    RQ_debug("Response class initialized.");
}