src/process_url.h


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

#ifndef PROCESS_URL_H_
#define PROCESS_URL_H_

#include <curl/curl.h>
#include <limits.h>
#include <stdbool.h>

#include "utils.h"

#define MAX_THREAD 6
#define MAX_THREAD_SIZE 10485760

#define ERRTOSTRING(err) curl_easy_strerror(err)
#define logerr(X)                                                              \
  _Generic((X), CURLcode                                                       \
           : logerr_b, CURLHcode                                               \
           : logerr_h, CURLUcode                                               \
           : logerr_u)(X)

typedef struct curl_conf {
  curl_off_t dlnow_per_thrd[MAX_THREAD];
  curl_off_t dlnow;
  curl_off_t dltotal;
  unsigned char success_thrd;
  unsigned char total_thrd;
  char *URL;
  char outfn[USHRT_MAX];
  str_array_t partfn;
  FILE *fplist[MAX_THREAD];
} curl_conf_t;

typedef struct thrd_info {
  unsigned char no;
  curl_conf_t *curl_c;
  char range[UCHAR_MAX];
} thrd_info_t;

typedef struct status {
  curl_off_t cur;
  curl_off_t total;
  bool is_done;
} status_t;

typedef struct str_data {
  char *string;
  size_t len;
} str_data_t;

void curl_init();

void curl_cleanup(status_t *);

void poll_status(status_t *);

int get(const char *, char **);

void add_url(const char *, const char *, const char *, const char *);

#endif