CGI配置程序开发总结

呱牛
2025-03-03 17:56:08
编程语言
Linux网络编程 C

CGI的后端，简单理解就是C、或者python写的后端服务，接收前端HTML form表单的输入，然后进行相应的操作。

CGI上传程序花的时间相对多些，在deepseek生成代码的基础上，结合抓包对form表单的分析，确实效率是提升了不少。但deepseek不能解决所有的问题，就跟网上说的一样，碰到程序员的问题，给出的答案就开始胡说，哈哈哈。

呱牛笔记

其中上传程序的CGI代码参考：

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/stat.h> 
#include <sys/types.h> 
#include <signal.h>
#include <fcntl.h> 
 
#define UPLOAD_DIR "/tmp/bms/" // 上传文件保存目录
#define BUF_SIZE 4096
 
// 创建上传目录（如果不存在）
void create_upload_dir() {
    if (access(UPLOAD_DIR, F_OK)) {
        mkdir(UPLOAD_DIR, 0777);
    }
}
   
 
// 安全释放指针并将指针置空
#define SAFE_FREE(ptr) do { free(ptr); (ptr) = NULL; } while (0)
 
 
/*
 * 在内存中找出一段字符串，返回出现的首地址
 * m 缓冲区地址
 * len 缓冲区长度
 * str 带有结束符的字符串
 */
void *memstr(unsigned char *m, int len, const char *str)
{
    unsigned char *sub = (unsigned char *)str;
    unsigned char *s, *d;
    int i = 0;
    int slen = len - strlen(str) + 1;
    if ((m == NULL) || (slen < 1)) {
        return NULL;
    }
     
    while (i < slen) {
        s = m + i; d = (unsigned char *)str;
        while(*s == *d) {
            s += 1;
            d += 1;
        };
        if (*d == 0) {
            return m + i;
        }
        i += 1;
    }
    return NULL;
}
 
// 解析 multipart/form-data 数据
void parse_multipart_data(const char *boundary) {
    char *line = NULL;
    size_t line_len = 0;
    FILE *file = NULL;
    char filepath[256];
    char buffer[BUF_SIZE];
    int start_flag = 0;
 
    struct timeval tv; 
    gettimeofday(&tv, NULL);  
    int curr_system_time = tv.tv_sec*1000+tv.tv_usec/1000;//ms 
    // 逐行读取输入
    while (1) {
        getline(&line, &line_len, stdin);
        // 1. 查找文件名
        if (!file && strstr(line, "filename=\"")) {
            char *filename_start = strstr(line, "filename=\"") + 10;
            char *filename_end = strchr(filename_start, '\"');
            if (!filename_end) continue; // 格式错误
             
            *filename_end = '\0'; // 截断文件名
            //snprintf(filepath, sizeof(filepath), "%s/%d_%s.tar.gz", UPLOAD_DIR, curr_system_time, filename_end);
             
            snprintf(filepath, sizeof(filepath), "%s/%s", UPLOAD_DIR, filename_start);
 
             
            sprintf(buffer, "rm %s -rf", filepath);
            system(buffer);  
 
            // 打开文件
            if (!(file = fopen(filepath, "wb"))) {
                printf("Status: 500\r\nContent-Type: application/json\r\n\r\n"
                       "{\"code\":1, \"msg\":\"Failed to open file\"}");
                SAFE_FREE(line);
                return;
            }
 
            // 跳过头部剩余部分直到空行 
        while (getline(&line, &line_len, stdin) != -1 && strcmp(line, "\r\n") != 0) {
            free(line);
            line = NULL;
        }
            SAFE_FREE(line);
            break;
        } 
        SAFE_FREE(line);   
    }
     
         
    int file_len=atoi(getenv("CONTENT_LENGTH"));
    char *http_buf=(char *)malloc(file_len+1);
    if (http_buf == NULL) {
        printf("malloc fail.\r\n");
        return;
    }
    memset((char *)http_buf,0x0,file_len+1);
    char *tmp = http_buf;
    int readlen = 0; 
    int ret = 0;
    int temp_len = file_len;
    int read_count = 0;
    while (temp_len) {
        readlen = (temp_len > 1024) ? 1024 : temp_len;
        ret =fread(tmp, 1, readlen, stdin); 
        if(readlen != ret) {
            read_count += ret;
        //printf("failed to read len %d size %d\r\n", readlen, ret);
        break;
        }
        read_count += ret;
        tmp += readlen;
        temp_len -= readlen;
    }
    tmp = http_buf;
    #if 1
    read_count -= 2 + strlen(boundary) + 2;
        fwrite(tmp, 1,  read_count, file);
    #else
    if (tmp[0] = '\n'){
        tmp += 1;
        read_count -= 1;
    }else if (tmp[0] = '\r' && tmp[1] == '\n'){
        tmp += 2;
        read_count -= 2;
    } 
        char *end = memstr(tmp, read_count , boundary);
        if (end == NULL){       
        fwrite(tmp, 1,  read_count, file);
        }else{
       read_count = end - tmp + 1;
         
        fwrite(tmp, 1,  read_count, file);
    }
    #endif
         
    printf("Status: 200 OK\r\n");
    printf("Content-Type: application/json\r\n\r\n");
    printf("{\"code\": 0, \"boundary\":\"%s\", \"msg\": \"File uploaded successfully\"}", boundary);
    free(http_buf);
 
    // 清理残留
    if (file) fclose(file);
    SAFE_FREE(line);
}
 
int main() {
    // 创建上传目录
    create_upload_dir();
 
    // 获取 Content-Type 中的 boundary
    char *content_type = getenv("CONTENT_TYPE");
    if (!content_type || !strstr(content_type, "multipart/form-data")) {
        printf("Status: 400 Bad Request\r\n");
        printf("Content-Type: application/json\r\n\r\n");
        printf("{\"code\": 1, \"msg\": \"Invalid content type ,%s\"}", content_type);
        return 1;
    }
 
    char *boundary_start = strstr(content_type, "boundary=");
    if (!boundary_start) {
        printf("Status: 400 Bad Request\r\n");
        printf("Content-Type: application/json\r\n\r\n");
        printf("{\"code\": 1, \"msg\": \"No boundary found\"}");
        return 1;
    }
 
    boundary_start += 9; // 跳过 "boundary="
    char boundary[1024];
    snprintf(boundary, sizeof(boundary), "--%s--", boundary_start);
 
    // 解析上传数据
    parse_multipart_data(boundary);
 
    return 0;
}