/home/mrjunejune/zenbu: seobeo/snapshot

annotate seobeo/snapshot_creator.c @ 71:75de5903355c

Giagantic changes that update Dowa library to be more align with stb style array and hashmap. Updated Seobeo to be caching on server side instead of file level caching. Deleted bunch of things I don't really use.

author	June Park <parkjune1995@gmail.com>
date	Sun, 28 Dec 2025 20:34:22 -0800
parents	6626ec933933
children	e7899c93da77

rev	line source
67 6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	1 #include "seobeo/snapshot_creator.h"
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	2 #include <stdio.h>
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	3 #include <stdlib.h>
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	4 #include <string.h>
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	5 #include <sys/stat.h>
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	6 #include <unistd.h>
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	7
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	8 #define MAX_RESPONSE_SIZE (1024 * 1024)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	9
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	10 // Helper: Convert URL path to filename
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	11 static void path_to_filename(const char path, char filename, size_t max_len)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	12 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	13 if (strcmp(path, "/") == 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	14 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	15 snprintf(filename, max_len, "root.snapshot");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	16 return;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	17 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	18
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	19 const char *p = path;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	20 if (*p == '/')
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	21 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	22 p++;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	23 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	24
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	25 char *out = filename;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	26 size_t remaining = max_len - 1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	27
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	28 while (*p && remaining > 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	29 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	30 if (*p == '/')
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	31 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	32 *out++ = '_';
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	33 remaining--;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	34 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	35 else
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	36 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	37 out++ = p;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	38 remaining--;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	39 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	40 p++;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	41 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	42
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	43 snprintf(out, remaining, ".snapshot");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	44 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	45
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	46 // Helper: Create directory recursively
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	47 static int create_directory(const char *path)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	48 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	49 char tmp[1024];
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	50 char *p = NULL;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	51 size_t len;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	52
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	53 snprintf(tmp, sizeof(tmp), "%s", path);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	54 len = strlen(tmp);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	55 if (tmp[len - 1] == '/')
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	56 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	57 tmp[len - 1] = 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	58 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	59
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	60 for (p = tmp + 1; *p; p++)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	61 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	62 if (*p == '/')
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	63 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	64 *p = 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	65 mkdir(tmp, 0755);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	66 *p = '/';
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	67 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	68 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	69 return mkdir(tmp, 0755);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	70 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	71
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	72 // Helper: Generate HTTP GET request
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	73 static int generate_http_get(char buffer, size_t size, const char path, const char *host)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	74 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	75 return snprintf(
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	76 buffer, size,
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	77 "GET %s HTTP/1.1\r\n"
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	78 "Host: %s\r\n"
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	79 "Connection: close\r\n"
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	80 "Accept: text/html,application/xhtml+xml,application/xml;q=0.9,/;q=0.8\r\n"
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	81 "User-Agent: SeobeoSnapshotCreator/1.0\r\n"
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	82 "\r\n",
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	83 path, host
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	84 );
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	85 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	86
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	87 // Helper: Read full HTTP response
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	88 static char* read_full_response(Seobeo_Handle client, size_t len_out)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	89 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	90 char *response = malloc(MAX_RESPONSE_SIZE);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	91 if (!response)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	92 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	93 return NULL;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	94 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	95
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	96 size_t total = 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	97 int attempts = 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	98 const int max_attempts = 100;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	99
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	100 while (attempts++ < max_attempts && total < MAX_RESPONSE_SIZE - 1)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	101 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	102 int n = Seobeo_Handle_Read(client);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	103
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	104 if (n > 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	105 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	106 size_t to_copy = client->read_buffer_len;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	107 if (total + to_copy > MAX_RESPONSE_SIZE - 1)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	108 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	109 to_copy = MAX_RESPONSE_SIZE - 1 - total;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	110 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	111
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	112 memcpy(response + total, client->read_buffer, to_copy);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	113 total += to_copy;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	114 Seobeo_Handle_Consume(client, (uint32)to_copy);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	115 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	116 else if (n == -2)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	117 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	118 break;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	119 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	120 else if (n == 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	121 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	122 usleep(10000);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	123 continue;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	124 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	125 else
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	126 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	127 free(response);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	128 return NULL;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	129 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	130 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	131
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	132 response[total] = '\0';
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	133 *len_out = total;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	134 return response;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	135 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	136
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	137 // Helper: Write snapshot to file
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	138 static int write_snapshot(const char filepath, const char data, size_t size)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	139 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	140 // Create directory if needed
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	141 char dir_copy[1024];
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	142 snprintf(dir_copy, sizeof(dir_copy), "%s", filepath);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	143
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	144 char *last_slash = strrchr(dir_copy, '/');
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	145 if (last_slash)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	146 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	147 *last_slash = '\0';
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	148 create_directory(dir_copy);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	149 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	150
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	151 FILE *f = fopen(filepath, "wb");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	152 if (!f)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	153 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	154 perror("fopen");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	155 return -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	156 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	157
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	158 size_t written = fwrite(data, 1, size, f);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	159 fclose(f);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	160
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	161 return (written == size) ? 0 : -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	162 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	163
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	164 int Seobeo_Snapshot_Create(const SnapshotConfig *config)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	165 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	166 if (!config \|\| !config->path \|\| !config->snapshot_dir \|\| !config->host \|\| !config->port)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	167 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	168 fprintf(stderr, "Invalid snapshot config\n");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	169 return -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	170 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	171
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	172 printf("Creating snapshot: %s (expecting %d)\n", config->path, config->expected_status);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	173
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	174 // Connect to server
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	175 Seobeo_Handle *client = Seobeo_Stream_Handle_Client_Create(config->host, config->port, FALSE);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	176 if (!client \|\| client->socket < 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	177 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	178 fprintf(stderr, " ✗ Failed to connect to %s:%s\n", config->host, config->port);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	179 if (client)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	180 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	181 Seobeo_Handle_Destroy(client);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	182 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	183 return -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	184 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	185
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	186 // Send GET request
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	187 char request[4096];
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	188 int req_len = generate_http_get(request, sizeof(request), config->path, config->host);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	189 Seobeo_Handle_Queue(client, (uint8*)request, (uint32)req_len);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	190
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	191 if (Seobeo_Handle_Flush(client) < 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	192 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	193 fprintf(stderr, " ✗ Failed to send request\n");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	194 Seobeo_Handle_Destroy(client);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	195 return -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	196 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	197
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	198 // Read response
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	199 size_t response_len = 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	200 char *response = read_full_response(client, &response_len);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	201 Seobeo_Handle_Destroy(client);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	202
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	203 if (!response \|\| response_len == 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	204 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	205 fprintf(stderr, " ✗ Failed to read response\n");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	206 if (response)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	207 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	208 free(response);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	209 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	210 return -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	211 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	212
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	213 // Parse status code
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	214 int status = -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	215 const char *status_line = strstr(response, "HTTP/1.1 ");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	216 if (!status_line)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	217 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	218 status_line = strstr(response, "HTTP/1.0 ");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	219 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	220 if (status_line)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	221 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	222 sscanf(status_line + 9, "%d", &status);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	223 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	224
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	225 if (status != config->expected_status)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	226 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	227 fprintf(stderr, " ✗ Status mismatch: expected %d, got %d\n",
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	228 config->expected_status, status);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	229 free(response);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	230 return -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	231 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	232
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	233 printf(" ✓ Status: %d\n", status);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	234
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	235 // Generate snapshot filename
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	236 char filename[256];
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	237 path_to_filename(config->path, filename, sizeof(filename));
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	238
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	239 char filepath[1024];
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	240 snprintf(filepath, sizeof(filepath), "%s/%s", config->snapshot_dir, filename);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	241
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	242 // Write snapshot
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	243 if (write_snapshot(filepath, response, response_len) == 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	244 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	245 printf(" ✓ Snapshot saved: %s (%zu bytes)\n", filepath, response_len);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	246 free(response);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	247 return 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	248 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	249 else
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	250 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	251 fprintf(stderr, " ✗ Failed to write snapshot: %s\n", filepath);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	252 free(response);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	253 return -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	254 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	255 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	256
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	257 int Seobeo_Snapshots_Create_Batch(const SnapshotConfig configs[], int count)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	258 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	259 int failed = 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	260 int passed = 0;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	261
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	262 for (int i = 0; i < count; i++)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	263 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	264 if (Seobeo_Snapshot_Create(&configs[i]) == 0)
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	265 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	266 passed++;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	267 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	268 else
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	269 {
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	270 failed++;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	271 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	272 printf("\n");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	273 }
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	274
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	275 printf("=== Summary ===\n");
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	276 printf("Created: %d\n", passed);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	277 printf("Failed: %d\n", failed);
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	278
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	279 return (failed == 0) ? 0 : -1;
6626ec933933 [Seobeo] Separated out Client Server logic. Created test tools. June Park <parkjune1995@gmail.com> parents: diff changeset	280 }

Mercurial

annotate seobeo/snapshot_creator.c @ 71:75de5903355c