summaryrefslogtreecommitdiff
blob: 32eb7ca0dc225b16a8d4055e7752184ade515dd3 (plain)
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
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#ifndef WIN32
#  include <unistd.h>
#endif
#include <curl/multi.h>

#include <stdio.h>
#include <stdlib.h>
#include <fstream>
#include <iostream>
#include <json/json.h>
#include "segment.cpp"

int max_segment_size=1000*500;

using namespace std;

typedef unsigned int uint;

class Tdistfile{
public:
  string *url_list;
  unsigned int url_num;
  unsigned int segment_num;
  unsigned int segments_count;
  Tsegment *dn_segments;
  string name;
  int size;
  string RMD160;
  string SHA1;
  string SHA256;
  uint url_count;
  unsigned int segment_size;
  Tdistfile(): url_list(0),url_num(0),segment_num(0),segments_count(0),dn_segments(0),name(""),size(0),
	       RMD160(""),SHA1(""),SHA256(""),url_count(0),segment_size(max_segment_size){};
  Tdistfile(const Tdistfile &L);             // copy constructor
  Tdistfile & operator=(const Tdistfile &L);
  ~Tdistfile();
  void load_distfile_from_json(json_object* json_obj_distfile);
  void load_url_list(json_object* json_array_distfile_urllist);
  void split_into_segments();
  Tsegment* provide_segment(CURLM* cm, uint con_num, uint seg_num);
};

void Tdistfile::load_url_list(json_object* json_array_distfile_urllist){
	url_count=json_object_array_length(json_array_distfile_urllist);
	url_list= new string [url_count];
	for(uint urllist_array_item_num=0;urllist_array_item_num<url_count;urllist_array_item_num++){
		url_list[urllist_array_item_num]=json_object_get_string(json_object_array_get_idx(json_array_distfile_urllist,urllist_array_item_num));
	}
}

void Tdistfile::load_distfile_from_json(json_object* json_obj_distfile){
	name=json_object_get_string(json_object_object_get(json_obj_distfile,"name"));
	size=atoi(json_object_to_json_string(json_object_object_get(json_obj_distfile,"size")));
	split_into_segments();
	RMD160=json_object_get_string(json_object_object_get(json_obj_distfile,"RMD160"));
	SHA1=json_object_get_string(json_object_object_get(json_obj_distfile,"SHA1"));
	SHA256=json_object_get_string(json_object_object_get(json_obj_distfile,"SHA256"));
	load_url_list(json_object_object_get(json_obj_distfile,"url_list"));
}

void Tdistfile::split_into_segments(){
  int range_end;
  if (size>max_segment_size){
    segments_count=size/max_segment_size;
    // set segments to have equal size, only last segment may be somewhat large (up to segment_count bytes large)
    segment_size=size/segments_count;	  
  }
  else{
    segment_size=size;
    segments_count=1;
  };
  dn_segments = new Tsegment [segments_count];
  //none downloaded yet
  for (uint segment_num=0; segment_num<segments_count; segment_num++){
    if (segment_num == segments_count)
      range_end=size;
    else
      range_end=(segment_num+1)*segment_size-1;
    dn_segments[segment_num].set_segment(this, segment_num, name, segment_size, toString(segment_num*segment_size)+"-"+toString(range_end));
  }
}


Tdistfile::~Tdistfile(){
	delete [] url_list;
}

Tsegment* Tdistfile::provide_segment(CURLM* cm, uint con_num, uint seg_num)
{
  //  cout << "=Seg #"<<seg_num<< " of "<<segments_count<< " from " << "url:"<<url_list[url_num]<<"\n";
  url_num++;
  if (url_num >= url_count) 
    url_num=0;
  dn_segments[seg_num].prepare_for_connection(cm, con_num, url_list[url_num]);
  return &dn_segments[seg_num];
}