1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
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
193
194
|
// MIT License, Copyright (c) 2020 Marvin Borner
#include <conv.h>
#include <def.h>
#include <gfx.h>
#include <gui.h>
#include <html.h>
#include <input.h>
#include <mem.h>
#include <net.h>
#include <print.h>
#include <str.h>
#include <xml.h>
#define WIDTH 640
#define HEIGHT 400
#define FONT_HEIGHT 24
#define LABEL_WIDTH 36 // Thx Lars
static struct element *root;
static struct element *code_label;
static struct element *output;
u32 status_color(char *http_code)
{
u32 c = 0;
switch (http_code[0]) {
case '1': // Information response
c = COLOR_BLUE;
break;
case '2': // Successful response
c = COLOR_GREEN;
break;
case '3': // Redirects
c = COLOR_YELLOW;
break;
case '4': // Client error
c = COLOR_RED;
break;
case '5': // Server error
c = COLOR_MAGENTA;
break;
default:
c = COLOR_WHITE;
break;
}
return c;
}
void print_indent(char *buf, u32 n)
{
for (u32 i = 0; i < n; i++)
strcat(buf, "\t");
}
void parse(void *data, u32 len, char *out)
{
struct xml_token tokens[128];
struct xml parser;
xml_init(&parser);
void *buffer = data;
len = strlen(data);
out[0] = '\0';
enum xml_error err = xml_parse(&parser, buffer, len, tokens, 128);
if (err != XML_SUCCESS) {
printf("\nXML parse error: %d\n", err);
return;
}
u32 indent = 0;
char name[16] = { 0 };
for (u32 i = 0; i < parser.ntokens; i++) {
const struct xml_token *token = tokens + i;
name[0] = '\0';
switch (token->type) {
case XML_START_TAG:
memcpy(&name, (u8 *)buffer + token->start_pos,
token->end_pos - token->start_pos);
name[token->end_pos - token->start_pos] = '\0';
if (html_self_closing(name))
print_indent(out, indent);
else
print_indent(out, indent++);
strcat(out, name);
strcat(out, "\n");
break;
case XML_END_TAG:
print_indent(out, --indent);
memcpy(&name, (u8 *)buffer + token->start_pos,
token->end_pos - token->start_pos);
name[token->end_pos - token->start_pos] = '\0';
strcat(out, name);
strcat(out, "/\n");
break;
case XML_CHARACTER:
if (token->end_pos == token->start_pos + 2) {
const char *ptr = (char *)buffer + token->start_pos;
if (ptr[0] == '\r' && ptr[1] == '\n')
continue;
}
memcpy(&name, (u8 *)buffer + token->start_pos,
token->end_pos - token->start_pos);
name[token->end_pos - token->start_pos] = '\0';
char *clean_name = name;
for (u32 j = 0; j < strlen(name); j++) {
if (name[j] == ' ' || name[j] == '\n' || name[j] == '\r' ||
name[j] == '\t') {
clean_name++;
} else {
break;
}
}
if (!strlen(clean_name))
break;
print_indent(out, indent++);
strcat(out, clean_name);
strcat(out, "\n");
indent--;
break;
default:
break;
}
i += token->size;
}
}
void on_submit(void *event, struct element *box)
{
(void)event;
char *url = ((struct element_text_input *)box->data)->text;
u16 port = 80;
char *port_str = strchr(url, ':');
if (port_str) {
port_str[0] = '\0';
port_str++;
port = atoi(port_str);
}
char *path = strchr(url, '/');
if (path) {
path[0] = '\0';
path++;
}
char *query = http_query_get(url, path ? path : "/");
u32 ip = 0;
if (!ip_pton(url, &ip)) {
ip = dns_request(url);
}
struct element_text_box *l = output->data;
struct element_label *c = code_label->data;
struct socket *socket = net_open(S_TCP);
if (socket && net_connect(socket, ip, port, NET_TIMEOUT)) {
net_send(socket, query, strlen(query));
char buf[4096] = { 0 };
char parsed[4096] = { 0 };
if (!net_receive(socket, buf, 4096, NET_TIMEOUT))
return;
parse(http_data(buf), 4096, parsed);
l->text = parsed[0] ? parsed : http_data(buf);
c->text = http_code(buf);
c->color_fg = status_color(c->text);
} else {
l->text = strdup("Can't connect to server.");
c->text = strdup("000");
c->color_fg = COLOR_RED;
}
gui_sync(root, output);
gui_sync(root, code_label);
net_close(socket);
}
int main()
{
// TODO: Dynamic element positioning
root = gui_init("browser", WIDTH, HEIGHT, COLOR_BG);
code_label = gui_add_label(root, 0, 0, FONT_24, "000", COLOR_BLACK, COLOR_WHITE);
struct element *text_input =
gui_add_text_input(root, LABEL_WIDTH, 0, 100, FONT_24, COLOR_WHITE, COLOR_BLACK);
output = gui_add_text_box(root, 0, FONT_HEIGHT + 2, 100, 100, FONT_16,
"Enter URL and press Enter :)", COLOR_WHITE, COLOR_BLACK);
text_input->event.on_submit = on_submit;
gui_event_loop(root);
return 0;
}
|
