95 lines
1.9 KiB
C
95 lines
1.9 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <curl/curl.h>
|
|
|
|
#define DEFAULT_PAGE_LEN 4
|
|
|
|
int pageChunkLen = DEFAULT_PAGE_LEN;
|
|
char** pageChunk;
|
|
|
|
typedef struct UserConfig {
|
|
int port;
|
|
char* addr;
|
|
} UserConfig;
|
|
|
|
UserConfig config = {
|
|
.port = 443,
|
|
.addr = "https://github.com/CaptainBoulbi/webpage",
|
|
};
|
|
|
|
void getUserConfig(int argc, char* argv[]){
|
|
for (int i=1; i<argc; i++){
|
|
if (strcmp(argv[i], "-p") == 0){
|
|
if (argc <= (i+1)){
|
|
perror("ERROR: Port not defined by user");
|
|
exit(1);
|
|
}
|
|
config.port = atoi(argv[++i]);
|
|
} else {
|
|
config.addr = argv[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
size_t save_chunk(char* buffer, size_t itemsize, size_t nitems, void* ingore){
|
|
size_t bytes = itemsize * nitems;
|
|
static int nbchuck = 0;
|
|
|
|
if (nbchuck >= pageChunkLen){
|
|
pageChunkLen = (nbchuck+1) * 1.5;
|
|
pageChunk = realloc(pageChunk, sizeof(*pageChunk)*pageChunkLen);
|
|
if (pageChunk == NULL){
|
|
perror("ERROR: Buy more ram.\n");
|
|
exit(1);
|
|
}
|
|
}
|
|
pageChunk[nbchuck] = malloc(bytes);
|
|
|
|
strncpy(pageChunk[nbchuck], buffer, bytes);
|
|
pageChunk[nbchuck][bytes] = '\0';
|
|
nbchuck++;
|
|
|
|
return bytes;
|
|
}
|
|
|
|
void getPage(void){
|
|
CURL* curl = curl_easy_init();
|
|
if (curl == NULL){
|
|
perror("ERROR: curl init failed.\n");
|
|
exit(1);
|
|
}
|
|
|
|
curl_easy_setopt(curl, CURLOPT_URL, config.addr);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, save_chunk);
|
|
|
|
CURLcode result = curl_easy_perform(curl);
|
|
if (result != CURLE_OK){
|
|
fprintf(stderr, "ERROR: %s\n", curl_easy_strerror(result));
|
|
}
|
|
|
|
curl_easy_cleanup(curl);
|
|
}
|
|
|
|
void printPage(void){
|
|
int i=0;
|
|
for (i=0; i<pageChunkLen && pageChunk[i] != NULL; i++){
|
|
printf("%s", pageChunk[i]);
|
|
}
|
|
printf("%d chunks\n", i);
|
|
}
|
|
|
|
int main(int argc, char* argv[]){
|
|
pageChunk = malloc(sizeof(*pageChunk) * DEFAULT_PAGE_LEN);
|
|
|
|
getUserConfig(argc, argv);
|
|
|
|
getPage();
|
|
|
|
printPage();
|
|
|
|
printf("chunk len = %d\n", pageChunkLen);
|
|
|
|
return 0;
|
|
}
|