/home/mrjunejune/zenbu: seobeo/snapshot

annotate seobeo/snapshot_creator.c @ 88:a3962e681490

Fixed routing issues.

author	June Park <parkjune1995@gmail.com>
date	Thu, 01 Jan 2026 14:29:36 -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 @ 88:a3962e681490