(* Copyright 2001, 2002 b8_bavard, b8_fee_carabine, INRIA *)
(*
    This file is part of mldonkey.

    mldonkey is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    mldonkey is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with mldonkey; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*)

(*
  The jobs of swarmers are :
  * select what data to ask from each uploader
  * merge data coming from uploaders, potentially from different
  networks, into a single Unix32 backend.

                            OVERALL SCHEMA

Each network frontend can have a different (fixed) chunk size
t1 +--------+--------+--------+--------+--------+--------+-------- chunks
t2 +------+------+------+------+------+------+------+------+------ chunks

 each block is contained in at most /\ chunk_of_block
         one chunk, for any network ||                            mappings 
                                    \/ blocks_of_chunk
swarmer
   +------+-+----+---+--+------+------++-----+--+---+----+-+------ blocks
    |      | |    |   |  |      |    ...                      variable size
    v      v v    v   v  v      v
    r<>r   r r    r   r  r<>r   r<>r                               ranges
                            ^                      one dbl linked list/block
                            |                    encoding missing data ranges
uploaders physically     uploader
reference ranges
*)

open Int64ops
open Options
open Printf2
  
open CommonOptions
open CommonTypes


let debug_all = false

type strategy =
  LinearStrategy    (* one after the other one *)
| AdvancedStrategy

exception VerifierNotReady

type intervals =
  AvailableIntervals of (int64 * int64) list
| AvailableBitv of Bitv.t

type verification =
  NoVerification
| VerificationNotAvailable
| ForceVerification
| Verification of uid_type array
  
let exit_on_error = ref false

let log_prefix = "[cSw]"

let lprintf_nl fmt =
  lprintf_nl2 log_prefix fmt

let lprintf_n fmt =
  lprintf2 log_prefix fmt

open CommonTypes

open CommonFile
open CommonTypes
open CommonClient

module VB = VerificationBitmap

(* If we want to become 'multinet', we should:
* there shouldn't be any block_size, instead blocks should correspond
  to the largest blocks which are completely included in one chunk for
  every network.
* a completed block should not be verified until all the blocks are
   completed for all networks.
* swarmers should be loaded before other files, so that they can be
used and shared by different files.

* Verification:
  - must_verify_chunk

TODO: s_last_seen is useless, only t_last_seen is useful, at least in the
  first version.

*)

(* network "frontend"/"view"/... to a swarmer *)
(* glossary:
   network frontend use "chunks" of data,
   swarmer use "blocks" of data *)
type t = {
    t_num : int;
    mutable t_primary : bool;
    t_file : file;
    mutable t_s : swarmer;
    t_chunk_size : int64;

    t_nchunks : int;
    mutable t_converted_verified_bitmap : VerificationBitmap.t;
    mutable t_last_seen : int array;
    mutable t_ncomplete_chunks : int;
    mutable t_nverified_chunks : int;

    mutable t_verifier : verification; (* information available to
					  check data correctness *)
    mutable t_verified : (int -> int -> unit); (* function to call
						  when a chunk is verified;
						  receives the number
						  of verified chunks,
						  and the index of the
						  chunk just verified *)

(* mapping from network chunks to swarmer blocks *)
    mutable t_blocks_of_chunk : int list array;
(* mapping from swarmer blocks to network chunks *)
    mutable t_chunk_of_block : int array;
  }

and swarmer = {
    s_num : int;
    s_filename : string;
    s_size : int64;
    s_disk_allocation_block_size : int64;

    mutable s_networks : t list; (** list of frontends, primary at head 
				     t.t_s = s <=> t in s.s_networks *)
    mutable s_strategy : strategy;

    mutable s_verified_bitmap : VerificationBitmap.t;
    mutable s_disk_allocated : Bitv.t;
    mutable s_availability : int array;
    mutable s_nuploading : int array;
(*    mutable s_last_seen : int array; *)

    mutable s_blocks : block_v array;
    mutable s_block_pos : int64 array; (** offset of the beginning of
					   each block *)
  }

and block_v =
  EmptyBlock
| PartialBlock of block
| CompleteBlock
| VerifiedBlock

and block = {
    block_s : swarmer;
    mutable block_num : int;
    mutable block_begin : Int64.t;
    mutable block_end : Int64.t;
    mutable block_ranges : range; (** [range] of the double-linked
				      list of ranges associated to the
				      [block] 
				      what about using a standard list
				      instead ? 
				      or a balanced tree ? *)
    mutable block_remaining : int64; (* amount of bytes missing. *)
    mutable block_unselected_remaining : int64; (* same, less ranges
						   selected for uploaders. *)
  }

and range = {
    mutable range_block : block;
    mutable range_begin : Int64.t;
    mutable range_end : Int64.t;
    mutable range_prev : range option;
    mutable range_next : range option;
    mutable range_nuploading : int; (* number of uploaders currently
				       referencing that range *)
  }

and uploader = {
    up_t : t;
    up_client : client;

    mutable up_declared : bool;

    mutable up_intervals : intervals;
    mutable up_complete_blocks : int array; (** block numbers *)
    mutable up_ncomplete : int; (** number of blocks not yet handled,
				    at the beginning of
				    up_complete_blocks *)

    mutable up_partial_blocks : (int * int64 * int64) array; (** block
								 number,
								 begin_pos,
								 end_pos
								 *)
    mutable up_npartial : int; (** number of blocks not yet handled,
				   at the beginning of
				   up_partial_blocks *)

    mutable up_blocks : uploader_block list;

    mutable up_ranges : (int64 * int64 * range) list; (* ranges referenced by
							 that uploader, see
							 range_nuploading *)
  }
and uploader_block = {
  up_block : block;
  up_block_begin : int64;
  up_block_end : int64;
}

(* range invariants: 
   Ranges represent "holes" of missing data in a block; Data is
   missing between offsets range_begin and range_end.

   [block]'s [block_ranges] reference the first (smallest offsets) of
   the [range]s associated with it. 

   [range]s are double-linked together thru [range_prev] and
   [range_next]:

   r.range_next.range_prev = r.range_prev.range_next = r
   ( when links are different from None )

   [range]s have a backlink to their "owner" [block]:

   b.block_ranges.{range_next}*.{range_prev}*.range_block = b

   ranges offsets are all within their block's offsets limits, do not
   overlap, and are sorted in increasing offsets order:

   b.block_begin <= b.block_ranges.block_begin ... <=
   r.range_prev.range_end <= r.range_begin <=
   r.range_end <= r.range_next.range_begin <= ...
   <= b.block_end *)
   
(* range owners are only used thru uploaders.up_ranges. blocks could be
   saved in [uploaders]' [up_ranges] along range, but would
   need updating when the swarmer is splitted.

   Removing [range] from [up_ranges] and [range_nuploading] from
   [range] could be good too, because they're not correctly updated
   when the swarmer is splitted. Again, getting rid of them is a
   problem of performance.
*)

(* block invariants
   Data missing for a block is the sum of the "sizes" of its ranges.

   b.block_remaining = sum (r.range_end - r.range_begin) b.block_ranges
*)

(* swarmer invariants ?
   s.s_verified_bitmap.[i] = State_missing <=> s_blocks.[i] = EmptyBlock
   s.s_verified_bitmap.[i] = State_partial <=> s_blocks.[i] = PartialBlock _
   s.s_verified_bitmap.[i] = State_complete <=> s_blocks.[i] = CompletedBlock
   s.s_verified_bitmap.[i] = State_verified <=> s_blocks.[i] = VerifiedBlock
   If so, why not drop s_verified_bitmap, and replace it by some
   verified_bitmap s i and verified_bitmap_all s functions ?
*)

(* frontend invariants ?
   t_ncomplete_chunks = 
   List.length (List.filter (fun x -> x >= State_complete) t_converted_verified_bitmap)
   t_nverified_chunks = 
   List.length (List.filter (fun x -> x = State_verified) t_converted_verified_bitmap)

   hence t_ncomplete_chunks >= t_nverified_chunks

   All chunks are [t_chunk_size] bytes in size, and first start at
   offset 0. This is assumed in [create], [associate], [verify_chunk],
   [duplicate_chunks], maybe more.
*)

(* uploaders invariants ?
   uploader block numbers are stored in reverse order in
   up_complete_blocks and up_partial_blocks (first blocks at the end
   of arrays), then array is processed from end to begin.

   0 <= up_ncomplete < Array.length up_complete_blocks 
   0 <= up.up_npartial < Array.length up_partial_blocks

   When a block has been selected, it's pushed out of the first
   up_ncomplete first elements of up_complete_blocks by swapping it
   with the #(up_ncomplete-1) element, then up_ncomplete is
   decreased. (and similarly with s/complete/partial/ ?)

   The question is now, aren't there better datastructures than
   arrays for the job ? ;)
*)

(*************************************************************************)
(*                                                                       *)
(*                         Global values                                 *)
(*                                                                       *)
(*************************************************************************)

module HS = Weak.Make(struct
      type t = swarmer
      let hash file = Hashtbl.hash file.s_filename

      let equal x y  = x.s_filename = y.s_filename
    end)

let swarmers_by_name = HS.create 31

module HT = Weak.Make(struct
  (* since type declarations are recursive, One can't write type t = t *)
  (* this is a workaround; a bit ugly, but works *)
      type frontend = t
      type t = frontend
      let hash t = t.t_num

      let equal x y  = x.t_num = y.t_num
    end)

let frontends_by_num = HT.create 31

module HU = Weak.Make(struct
      type t = uploader
      let hash u = Hashtbl.hash (client_num u.up_client)

      let equal x y  =  (client_num x.up_client) =  (client_num y.up_client)
    end)

let uploaders_by_num = HU.create 113

let frontend_counter = ref 0
let swarmer_counter = ref 0

(** sets [t.t_last_seen] of the verified blocks to current time, and 
    associated file's [t.t_s.s_file] last seen value to the oldest of the
    remaining last seen values *)

let compute_last_seen t =
  let last_seen_total = ref (BasicSocket.last_time ()) in
  VB.iteri (fun i c -> 
    if c = VB.State_verified then
      t.t_last_seen.(i) <- BasicSocket.last_time ()
    else
      last_seen_total := min !last_seen_total t.t_last_seen.(i)
  ) t.t_converted_verified_bitmap;
  set_file_last_seen t.t_file !last_seen_total;
  t.t_last_seen

(** (internal) return a 0 sized range at offset [pos], and assigned to
    block [b] *)

let void_range b pos =
  let r = {
      range_prev = None;
      range_next = None;
      range_begin = pos;
      range_end = pos;
      range_block = b;
      range_nuploading = 0;
    }
  in
  r

let compute_range_size r =
  r.range_end -- r.range_begin

let rec ranges_iter f r =
  f r;
  match r.range_next with
  | None -> ()
  | Some r ->
      ranges_iter f r

let rec ranges_fold f acc r =
  let acc = f acc r in
  match r.range_next with
  | None -> acc
  | Some rr -> ranges_fold f acc rr

let rec ranges_for_all p r =
  p r && 
    (match r.range_next with
     | None -> true
     | Some r -> ranges_for_all p r)

let block_ranges_for_all p b =
  ranges_for_all p b.block_ranges

let block_ranges_fold f acc b =
  ranges_fold f acc b.block_ranges

(** (internal) assigns range [r], and all other ranges along
    [range_next] links, to block [b] *)

let rec own_ranges b r =
  ranges_iter (fun r -> r.range_block <- b) r

(** (internal) 
    Find ranges that are after [cut_pos] offset, unlink them from r
    double-linked list of ranges, set their owner to [b] and return
    the first of the removed ranges.

    If all ranges are before [cut_pos] return a 0-sized range.

    If [cut_pos] is within one of the ranges, that range is cut in
    two at [cut_pos] offset, and link each half to its side.

    Also, what do to if range_nuploading is not 0 ?
    => [cut_ranges_after] is being called from [split_blocks] that
    does not preserve [s_nuploading] for blocks either
*)

let cut_ranges_after b r cut_pos =
  let rec iter r =
    (* after the cut position already ? *)
    if r.range_begin >= cut_pos then begin
      (match r.range_prev with
      | None -> 
	  let b1 = r.range_block in
	  b1.block_ranges <- void_range b1 cut_pos
      | Some rp ->
          rp.range_next <- None;
          r.range_prev <- None);
      r
    end 
      (* still before the cut position ? *)
    else if r.range_end <= cut_pos then
      match r.range_next with
      | None -> void_range b cut_pos
      | Some r -> iter r
    else
      (* across cut position, must split a range *)
      (* "right" half *)
      let split_r = { r with
	range_begin = cut_pos;
	range_prev = None;
      } in
      (match split_r.range_next with
      | None -> ()
      | Some rr -> rr.range_prev <- Some split_r);

      (* "left" half *)
      r.range_end <- cut_pos;
      r.range_next <- None;

      if r.range_nuploading <> 0 then
	lprintf_nl "WARNING: Splitting a range currently being uploaded, don't know what to do with range_nuploading :/";

      split_r in
  let cut_ranges = iter r in
  own_ranges b cut_ranges;
  cut_ranges
  
(** (internal) return the offset of the end of the [i]th block of
    swarmer [s] *)

let compute_block_end s i =
  let b = s.s_block_pos in
  if Array.length b = i + 1 then
    s.s_size
  else
    b.(i+1)

(** (internal) return the offset of the beginning of the [i]th block
    of swarmer [s] *)

let compute_block_begin s i =
  let b = s.s_block_pos in
  b.(i)

(** Finds the number of the block containing [chunk_pos] offset, using
    dichotomy. Blocks are half opened [block_begin, block_end[ *)

(* 0 <= chunk_pos < s.s_size *)
let compute_block_num s chunk_pos =
  assert (0L <= chunk_pos && chunk_pos < s.s_size);
  let b = s.s_block_pos in
(* invariants:
   0 <= min <= max <= Array.length b - 1
   compute_block_begin s min <= chunk_pos < compute_block_end s max *)

  let rec iter min max =
    if min = max then min
    else (* from now on, min < max *)
      let medium = (min + max) / 2 in
      (* Euclide        => 2*medium <= min + max <= 2*medium + 1 *)
      (* min < max      => 2*min < min + max < 2*max
                        => min <= medium < max                   *)
      if min < medium then
        if chunk_pos < b.(medium) then
          iter min (medium - 1)
        else
          iter medium max
      else
        (* min = medium < max  => 2*min < min + max <= 2*min + 1
                              <=>   min <       max <=   min + 1
                              <=> min + 1 = max                  *)
        if chunk_pos < b.(max) then
          min else max
  in
  let i = iter 0 (Array.length b - 1) in
  if debug_all then
    lprintf_nl "%Ld is block %d [%Ld-%Ld]" chunk_pos i
      (compute_block_begin s i) (compute_block_end s i);
  i

(** Return true if ranges fully "cover" their block
    ("the block is made of holes") *)

let block_is_empty b =
  let rec iter begin_pos r =
    r.range_begin = begin_pos &&
    (match r.range_next with
     | Some rr -> iter r.range_end rr
     | None -> r.range_end = b.block_end)
  in
  iter b.block_begin b.block_ranges
    
let block_is_full b =
  let r = b.block_ranges in
  r.range_next = None && r.range_begin = r.range_end
      
(** iter function [f] over all the blocks contained in the list of [intervals]

    [f] will receive block number, block beginning and ending offsets,
    and overlapping interval beginning and ending offsets.

    If an interval starts halfway of a block, iteration starts on the
    next block, with interval_begin < block_begin indicating where the
    interval really started.

    If an interval ends halfway of a block, iteration ends on that
    block, with interval_end < block_end indicating where the interval
    really ended.
*)

let iter_intervals s f intervals =
  let nchunks = Array.length s.s_blocks in
  List.iter (fun (interval_begin, interval_end) ->
    let interval_begin = min interval_begin s.s_size in
    let interval_end = min interval_end s.s_size in
(*  lprintf "apply on %Ld-%Ld\n" interval_begin interval_end; *)
    if interval_begin < interval_end then
      let i0 = compute_block_num s interval_begin in
      let block_begin = compute_block_begin s i0 in
      let rec iter_blocks i block_begin interval_begin =
(*      lprintf "iter_blocks %d %Ld %Ld\n" i block_begin interval_begin; *)
        if i < nchunks && block_begin < interval_end then
          let block_end = compute_block_end s i in
          let current_end =  min block_end interval_end in
	  
          if debug_all then
            lprintf_nl "Apply: %d %Ld-%Ld %Ld-%Ld"
              i block_begin block_end interval_begin current_end;
	  
          f i block_begin block_end interval_begin current_end;
          iter_blocks (i+1) block_end block_end
      in
      iter_blocks i0 block_begin interval_begin
  ) intervals

(*************************************************************************)
(*                                                                       *)
(*                              Swarmers                                 *)
(*                                                                       *)
(*************************************************************************)

let dummy_swarmer = {
    s_num = 0;
    s_filename = "";
    s_size = zero;
    s_disk_allocation_block_size = zero;
    s_networks = [];
    s_strategy = AdvancedStrategy;
    s_verified_bitmap = VB.create 0 VB.State_missing;
    s_disk_allocated = Bitv.create 0 false;
    s_blocks = [||];
    s_block_pos = [||];
    s_availability = [||];
    s_nuploading = [||];
  }

(** if a swarmer is already associated with that [file_name], return it;
    Otherwise create a new one with default values, that will be fixed
    by the first frontend association *)

let create_swarmer file_name file_size =
  try
    HS.find swarmers_by_name
      { dummy_swarmer with
	s_filename = file_name
      }
  with Not_found ->
    incr swarmer_counter;

    let nblocks = 1 in
    (* to avoid extreme disk fragmentation, space is pre-allocated in
       blocks of this size *)
    let disk_allocation_block_size = 
      min (megabytes 10) 
	(round_up64 (max 1L (file_size // 200L)) (megabytes 1)) in
    let ndiskblocks =
      1 + Int64.to_int (Int64.pred file_size // disk_allocation_block_size) in
    let rec s = {

      s_num = !swarmer_counter;
      s_filename = file_name;
      s_size = file_size;
      s_disk_allocation_block_size = disk_allocation_block_size;

      s_networks = [];

      s_strategy = AdvancedStrategy;

      s_verified_bitmap = VB.create nblocks VB.State_missing;
      s_disk_allocated = Bitv.create ndiskblocks false;
      s_blocks = Array.create nblocks EmptyBlock ;
      s_block_pos = Array.create nblocks zero;
      s_availability = Array.create nblocks 0;
      s_nuploading = Array.create nblocks 0;
(*      s_last_seen = Array.create nblocks 0; *)
    }
    in
    HS.add swarmers_by_name s;
    s

(** Split swarmer existing blocks in at [chunk_size] boundaries *)
let split_blocks s chunk_size =

  let size = s.s_size in

  let nblocks = Array.length s.s_blocks in
  (* Split existing blocks at [chunk_size] boundaries
     invariants:
     [index_s] is the index of the existing block being analysed
     [chunk_begin] is the offset of the beginning of the current containing chunk
     [new_blocks] is the list of new blocks already splitted, in
     reverse order. 
     List contains tuples: block, beginning offset, verified status char *)
  let rec iter index_s chunk_begin new_blocks =
(*    lprintf "iter (1) %d/%d %Ld\n" index_s nblocks chunk_begin; *)
    if index_s = nblocks then List.rev new_blocks else

    (* existing block *)
    let block_begin = compute_block_begin s index_s in
    let block_end = compute_block_end s index_s in

    (* current chunk *)
    let chunk_end = chunk_begin ++ chunk_size in
    let chunk_end = min chunk_end size in

    if chunk_end > block_end then
        let new_blocks = (
            s.s_blocks.(index_s),
            block_begin,
	    VB.get s.s_verified_bitmap index_s
          ) :: new_blocks in
        iter (index_s+1) chunk_begin new_blocks

    else if chunk_end = block_end then
        let new_blocks =  (
            s.s_blocks.(index_s),
            block_begin,
            VB.get s.s_verified_bitmap index_s
          ) :: new_blocks in
        iter (index_s+1) chunk_end new_blocks

    else begin
      (* chunk_end < block_end
	 We need to split this block in two parts *)
        s.s_block_pos.(index_s) <- chunk_end;
        match s.s_blocks.(index_s) with
	| EmptyBlock | CompleteBlock | VerifiedBlock ->

(* s.s_blocks.(index_s) will appear twice in the result list *)
            let new_blocks =  (
                s.s_blocks.(index_s),
                block_begin,
                VB.get s.s_verified_bitmap index_s
              ) :: new_blocks in
            iter index_s chunk_end new_blocks

        | PartialBlock b1 ->
	    (* split b1 in two; b2 is the part after [chunk_end] offset *)
            let b2 = {
                block_s = s;

                block_begin = chunk_end;
                block_end = b1.block_end;
                block_ranges = b1.block_ranges; (* fixed below *)
                block_num = 0; (* fixed below *)
                block_remaining = zero; (* fixed below *)
		block_unselected_remaining = zero; (* fixed below *)
              } in
	    b2.block_ranges <- cut_ranges_after b2 b1.block_ranges chunk_end;
            b1.block_end <- chunk_end;

            let new_blocks =
              (if block_is_full b1 then
(* lprintf "Partial block b1 should become CompleteBlock\n"; *)
		(
		  CompleteBlock,
		  block_begin,
		  VB.State_complete
		) else if block_is_empty b1 then
(* lprintf "Partial block b1 should become EmptyBlock\n"; *)
                  (
                    EmptyBlock,
                    block_begin,
                    VB.State_missing
                  ) else (
                    PartialBlock b1,
                    block_begin,
                    VB.get s.s_verified_bitmap index_s
                  ))
              :: new_blocks in

	    if block_is_full b2 then begin
	      (* lprintf "Partial block b2 should become CompleteBlock\n"; *)
	      s.s_blocks.(index_s) <- CompleteBlock;
	      VB.set s.s_verified_bitmap index_s VB.State_complete
	    end
            else if block_is_empty b2 then begin
	      (* lprintf "Partial block b2 should become EmptyBlock\n"; *)
              s.s_blocks.(index_s) <- EmptyBlock;
              VB.set s.s_verified_bitmap index_s VB.State_missing;
            end 
	    else
              s.s_blocks.(index_s) <- PartialBlock b2;
            iter index_s chunk_end new_blocks
      end
  in
  let blocks = iter 0 zero [] in

(* blocks have been splitted, now rebuild swarmer *)
  let nblocks = List.length blocks in
(*  lprintf "%d blocks to generate\n" nblocks; *)

  if Array2.exists ((<>) 0) s.s_availability then
    lprintf_nl "WARNING: splitting swarmer discarded availability counters";
  if Array2.exists ((<>) 0) s.s_nuploading then
    lprintf_nl "WARNING: splitting a swarmer beging uploaded to";

  s.s_blocks <- Array.create nblocks EmptyBlock;
  s.s_verified_bitmap <- VB.create nblocks VB.State_missing;
  s.s_block_pos <- Array.create nblocks zero;
  s.s_availability <- Array.create nblocks 0; (* not preserved ? *)
  s.s_nuploading <- Array.create nblocks 0; (* not preserved ? *)
(*  s.s_last_seen <- Array.create nblocks 0; *)

  let rec iter i list =
    match list with
    | [] -> ()
    | (b, pos, c) :: tail ->
        begin
          match b with
	  | PartialBlock b -> 
	      begin
		b.block_num <- i;
		let block_size = compute_block_end s i --
		  compute_block_begin s i in
		let remaining, unselected_remaining =
		  block_ranges_fold (fun (racc, uracc) r ->
		    let range_size = compute_range_size r in
		    (racc -- range_size,
		    if r.range_nuploading = 0 then 
		      uracc -- range_size else uracc)
		  ) (block_size, block_size) b in
		b.block_remaining <- remaining;
		b.block_unselected_remaining <- unselected_remaining;
	      end
          | EmptyBlock | CompleteBlock | VerifiedBlock -> ()
        end;
        s.s_blocks.(i) <- b;
        VB.set s.s_verified_bitmap i c;
        s.s_block_pos.(i) <- pos;

        iter (i+1) tail
  in
  iter 0 blocks

(** Associate a(n additional) frontend to a swarmer *)

let associate is_primary t s =
(* a swarmer cannot be associated more than once to a network *)
  if not (List.memq t s.s_networks) then
  let size = file_size t.t_file in

(* what about raising an exception instead ? *)
  if s.s_size <> size then begin
    lprintf_nl "file_size for %s does not match: swarmer %Ld / real %Ld" s.s_filename s.s_size size;
    exit 2
  end;

  t.t_s <- s;
  t.t_converted_verified_bitmap <- VB.create t.t_nchunks VB.State_missing;
  t.t_last_seen <- Array.create t.t_nchunks 0;
  t.t_chunk_of_block <- [||];
  t.t_blocks_of_chunk <- Array.create t.t_nchunks [];

(* invariant: primary frontend is at the head of swarmer's
   [s_networks], and is the first associated with the swarmer *)
  if is_primary then begin
    t.t_primary <- true;
    assert(s.s_networks = []);
    s.s_networks <- [t]
    (* was   s.s_networks <- t :: s.s_networks; *)
  end else begin
    match s.s_networks with
    | tprim :: _ ->
	assert(tprim.t_primary);
	if file_disk_name t.t_file <> file_disk_name tprim.t_file then
	  (* TODO: transfer data into swarmer instead of discarding it *)
	  Unix32.remove (file_fd t.t_file);
	t.t_primary <- false;
	s.s_networks <- s.s_networks @ [t];
    | [] -> assert false
  end;

  (match s.s_networks with
   | t :: tail ->
       assert(t.t_primary);
       List.iter (fun tt -> assert(not tt.t_primary)) tail
   | [] -> assert false);

(* at this point, we are supposed to split the blocks in the swarmer
in smaller blocks depending on the block_size of this network, and compute
the t_chunk_of_block and t_blocks_of_chunk fields. *)

  let chunk_size = t.t_chunk_size in
  split_blocks s chunk_size;

  let nblocks = Array.length s.s_blocks in
  (* For all networks, adjust the chunks and mappings *)
  List.iter (fun t ->
      let nchunks = VB.length t.t_converted_verified_bitmap in
      t.t_chunk_of_block <- Array.create nblocks 0;
      t.t_blocks_of_chunk <- Array.create nchunks [];

      let chunk_size = t.t_chunk_size in
      for i = 0 to nblocks - 1 do
        let block_begin = compute_block_begin s i in
        let chunk = Int64.to_int (block_begin // chunk_size) in
        t.t_chunk_of_block.(i) <- chunk;
        t.t_blocks_of_chunk.(chunk) <- i :: t.t_blocks_of_chunk.(chunk)
      done
    ) s.s_networks;

(* TODO: If not primary, set_file_downloaded should be called *)
  if not is_primary then
    add_file_downloaded t.t_file (zero -- file_downloaded t.t_file);

(* check that all frontends use the primary's file backend *)
  (match s.s_networks with
   | t :: tail when is_primary ->
       List.iter (fun tt ->
         set_file_fd tt.t_file (file_fd t.t_file)
       ) tail
   | tprim :: tail ->
       set_file_fd t.t_file (file_fd tprim.t_file)
   | [] -> assert false)

(** Create a primary frontend and its swarmer *)

let create ss file chunk_size =

  incr frontend_counter;
  let size = file_size file in
  (* wrong if size is a multiple of chunk_size, or on purpose ? *)
  let nchunks =
    1 + Int64.to_int (Int64.pred size // chunk_size) in

  let t = {
      t_num = !frontend_counter;
      t_s = ss;
      t_primary = true;
      t_file = file;

      t_nchunks = nchunks;
      t_chunk_size = chunk_size;

      t_ncomplete_chunks = 0;
      t_nverified_chunks = 0;

      t_converted_verified_bitmap = VB.create nchunks VB.State_missing;
      t_last_seen = Array.create nchunks 0;

      t_verifier = NoVerification;
      t_verified = (fun _ _ -> ());

      t_chunk_of_block = [||];
      t_blocks_of_chunk = Array.create nchunks [];
    }
  in
  HT.add frontends_by_num t;
  associate true t ss;
  t

(* copy the disk block over itself; Safer than overwriting it with zeroes *)

let iter_disk_space f t interval_begin interval_end =

  (* 0 <= chunk_pos < s.s_size *)
  let compute_disk_block_num s chunk_pos =
    assert (0L <= chunk_pos && chunk_pos < s.s_size);
    Int64.to_int (chunk_pos // s.s_disk_allocation_block_size) in

  if interval_begin < interval_end then
    let s = t.t_s in
    let fd = file_fd t.t_file in
    let first_disk_block = compute_disk_block_num s interval_begin in
    let last_disk_block = compute_disk_block_num s (Int64.pred interval_end) in
    for disk_block = first_disk_block to last_disk_block do
      f s fd disk_block
    done

exception Not_preallocated_block_found

let is_fully_preallocated t interval_begin interval_end =
  try
    iter_disk_space (fun s fd disk_block ->
      if not(Bitv.get s.s_disk_allocated disk_block) then
	raise Not_preallocated_block_found
    ) t interval_begin interval_end;
    true
  with Not_preallocated_block_found -> false
  
let preallocate_disk_space t interval_begin interval_end =
  iter_disk_space (fun s fd disk_block ->
    if not(Bitv.get s.s_disk_allocated disk_block) then begin
      let pos = (Int64.of_int disk_block) ** s.s_disk_allocation_block_size in
      let size = min (file_size t.t_file -- pos) s.s_disk_allocation_block_size in
      if !verbose then lprintf_nl "preallocating %Ld bytes for %s" size (file_best_name t.t_file);
      Unix32.copy_chunk fd fd pos pos (Int64.to_int size)
    end;
    Bitv.set s.s_disk_allocated disk_block true
  ) t interval_begin interval_end

let mark_disk_space_preallocated t interval_begin interval_end =
  iter_disk_space (fun s fd disk_block ->
    Bitv.set s.s_disk_allocated disk_block true
  ) t interval_begin interval_end

let check_finished t =
  let file = t.t_file in
  match file_state file with
  | FileNew 
  | FileCancelled 
  | FileAborted _ 
  | FileShared 
  | FileDownloaded 
  | FileQueued 
  | FilePaused -> 
      false
  | FileDownloading ->
      if VB.existsi (fun i c -> c <> VB.State_verified)
	t.t_converted_verified_bitmap then false
      else begin
        if file_size file <> file_downloaded t.t_file then
          lprintf_nl "Downloaded size differs after complete verification";
        true
      end

(** (debug) output a [swarmer] to current log *)

let print_s str s =
  lprintf_nl "Ranges after %s:" str;

  let rec iter r =
    lprintf_n " %Ld-%Ld(%d)"
      r.range_begin r.range_end r.range_nuploading;
    match r.range_next with
    | None -> lprint_newline ()
    | Some r -> iter r in

  Array.iteri (fun i b ->
      lprintf_n "   %d: " i;
      let block_begin = compute_block_begin s i in
      let block_end = compute_block_end s i in
      lprintf "%Ld - %Ld [%Ld] %c " block_begin block_end
        (block_end -- block_begin) 
	(VB.state_to_char 
	  (VB.get s.s_verified_bitmap i));
      List.iter (fun t ->
          let j = t.t_chunk_of_block.(i) in
          lprintf "(b %d %c [" j 
	    (VB.state_to_char 
	      (VB.get t.t_converted_verified_bitmap j));
          List.iter (fun ii -> lprintf "%d " ii) t.t_blocks_of_chunk.(j);
          lprintf "]";
      ) s.s_networks;

      match b with
      | PartialBlock b ->
          lprintf " [%Ld .. %Ld] --> "
            b.block_begin b.block_end;
          iter b.block_ranges
      | EmptyBlock -> lprintf_nl "_"
      | CompleteBlock -> lprintf_nl "C"
      | VerifiedBlock -> lprintf_nl "V"
  ) s.s_blocks;

  lprintf_nl "Files:";
  List.iter (fun t ->
    lprintf_nl "  File num: %d" (file_num t.t_file);
    lprintf_nl "  %s" (if t.t_primary then "primary" else "secondary");
    lprintf_nl "  Downloaded: %Ld" (file_downloaded t.t_file);
    lprintf_nl "  Bitmap: %s" (VB.to_string t.t_converted_verified_bitmap)
  ) s.s_networks

(** iter function f over all the ranges of a block *)

let iter_block_ranges f b =
  let rec iter_range f r =
    let next = r.range_next in (* keep next range in case f mutates it *)
    f r;
    match next with
    | None -> ()
    | Some rr -> iter_range f rr
  in
  iter_range f b.block_ranges

(** (debug) output a [block] to current log *)

let print_block b =
  lprintf_n "Block %d: %Ld-%Ld"
    b.block_num b.block_begin b.block_end;
  lprint_newline ();
  lprintf_nl "  ranges:";
  iter_block_ranges (fun r ->
    lprintf_nl "   %Ld-%Ld" r.range_begin r.range_end) b;
  lprint_newline ()

(** (shadows CommonFile.add_file_downloaded) 
    increments amount downloaded of the primary frontend of the swarmer,
    and of maybe_t, if provided, and if it's different from the primary. *)

let add_file_downloaded maybe_t s size =
(*  lprintf "add_file_downloaded %Ld\n" size; *)
  match s.s_networks with
  | t :: _ ->
      assert(t.t_primary);
      add_file_downloaded t.t_file size;
      (match maybe_t with
       | None -> ()
       | Some tt ->
	   if t.t_num <> tt.t_num then
             add_file_downloaded tt.t_file size);
      if file_downloaded t.t_file < zero then
        lprintf_nl "ERROR: file_downloaded < zero!";
      
  | _ -> assert false

(** Close all the ranges of a block, adding their size to the
    downloaded amount *)

let close_block_ranges maybe_t s b =
  iter_block_ranges (fun r ->
    let added = compute_range_size r in
    add_file_downloaded maybe_t s added;
    b.block_remaining <- b.block_remaining -- added;
    if r.range_nuploading = 0 then
      b.block_unselected_remaining <- b.block_unselected_remaining -- added;
    r.range_begin <- r.range_end;
    r.range_prev <- None;
    r.range_next <- None) b;
  if b.block_remaining <> 0L then
    lprintf_nl "WARNING: block_remaining should be 0 after close_block_ranges";
  if b.block_unselected_remaining <> 0L then
    lprintf_nl "WARNING: block_unselected_remaining should be 0 after close_block_ranges"

(*************************************************************************)
(*                                                                       *)
(*                         swarmers verified bitmaps                     *)
(*                                                                       *)
(*************************************************************************)

(* For every swarmer, there is a "primary" verifier, and secondary
   verifiers.  When a block is downloaded, it is tagged State_complete
   in the verified_bitmap, and this State_complete is propagated to
   the primary bitmap if possible (if all sub-blocks are also
   State_complete). 

   If the primary chunk becomes State_complete, then a verification is
   needed on the primary. If the verification works, the
   verified_bitmap becomes State_verified, and the secondary verifiers
   are tagged with State_complete (if they use a different
   verification scheme) or State_verified (if no verification scheme
   or a verification scheme that has already been used). 
*)

(* corruption has been detected, and the block has been reset to 
   State_missing *)

let set_swarmer_state_missing s i =
  let current_state = VB.get s.s_verified_bitmap i in
  match current_state with 
  | VB.State_complete | VB.State_verified ->
      (VB.set s.s_verified_bitmap i VB.State_missing;
       List.iter (fun t ->
          let j = t.t_chunk_of_block.(i) in
	  match VB.get t.t_converted_verified_bitmap j with
	  | VB.State_missing -> ()
	  | VB.State_partial ->
	      if List.for_all (fun i -> VB.get s.s_verified_bitmap i = VB.State_missing)
		t.t_blocks_of_chunk.(j) then
		  VB.set t.t_converted_verified_bitmap j VB.State_missing
	  | VB.State_complete -> 
	      lprintf_nl "set_swarmer_state_missing: invalidating a block within a completed chunk?"
	  | VB.State_verified -> 
	      lprintf_nl "set_swarmer_state_missing: invalidating a block within a verified chunk?"
      ) s.s_networks)
  | VB.State_missing -> ()
  | VB.State_partial ->
      lprintf_nl "set_swarmer_state_missing: invalidating a partial block ?"

(* we have started downloading this block, so mark all containing chunks
  also as started. *)
let set_swarmer_state_partial s i =
  match VB.get s.s_verified_bitmap i with
  | VB.State_missing ->
      VB.set s.s_verified_bitmap i VB.State_partial;
      List.iter (fun t ->
        let j = t.t_chunk_of_block.(i) in
        match VB.get t.t_converted_verified_bitmap j with
	| VB.State_missing -> 
	    VB.set t.t_converted_verified_bitmap j VB.State_partial
	| VB.State_partial -> ()
	| VB.State_complete -> 
	    lprintf_nl "set_swarmer_state_partial: partial block within a completed chunk?"
	| VB.State_verified -> 
	    lprintf_nl "set_swarmer_state_partial: partial block within a verified chunk?"
      ) s.s_networks
  | VB.State_partial -> ()
  | VB.State_complete -> 
      lprintf_nl "set_swarmer_state_partial: trying to demote a completed block?"
  | VB.State_verified -> 
      lprintf_nl "set_swarmer_state_partial: trying to demote a verified block?"
	

(* we finished this block, trying to escalate to primary frontend
   verification bitmap *)
let set_swarmer_state_complete s i =
  match VB.get s.s_verified_bitmap i with
  | VB.State_missing | VB.State_partial ->
      (VB.set s.s_verified_bitmap i VB.State_complete;
      match s.s_networks with
      | t :: _ ->
	  assert (t.t_primary);
          let j = t.t_chunk_of_block.(i) in
	  (match VB.get t.t_converted_verified_bitmap j with
	  | VB.State_missing | VB.State_partial -> 
	      if List.for_all (fun i -> VB.get s.s_verified_bitmap i = VB.State_complete)
		t.t_blocks_of_chunk.(j) then begin
		  t.t_ncomplete_chunks <- t.t_ncomplete_chunks + 1;
		  VB.set t.t_converted_verified_bitmap j VB.State_complete
		end
	  | VB.State_complete -> ()
	  | VB.State_verified -> 
	      (* lprintf_nl "set_swarmer_state_complete: trying to demote a verified block? (1)" *)
	      ())
      | [] -> assert false)
  | VB.State_complete -> ()
  | VB.State_verified -> 
(* lprintf_nl "set_swarmer_state_complete: trying to demote a verified block? (2)" *)
      VB.set s.s_verified_bitmap i VB.State_complete;
      match s.s_networks with
      | t :: _ ->
	  assert (t.t_primary);
          let j = t.t_chunk_of_block.(i) in
	  (match VB.get t.t_converted_verified_bitmap j with
	   | VB.State_verified ->
	       VB.set t.t_converted_verified_bitmap j VB.State_complete;
	       t.t_nverified_chunks <- t.t_nverified_chunks - 1;
	   | VB.State_complete -> ()
	   | VB.State_missing | VB.State_partial ->
	       lprintf_nl "BUG: set_swarmer_state_complete: demoting a verified block from an incomplete chunk")
      | [] -> assert false


(* the primary verifier has worked, so let ask secondary ones for
   verification too *)
let set_swarmer_state_verified s i =
  match VB.get s.s_verified_bitmap i with
  | VB.State_missing | VB.State_partial | VB.State_complete ->
      (VB.set s.s_verified_bitmap i VB.State_verified;
(*      lprintf "set_swarmer_state_verified %d done\n" i; *)
      match s.s_networks with
      | [] -> assert false
      | tprim :: secondaries ->
	  assert (tprim.t_primary);
	  (* that test is somewhat redundant, since only primary
	     frontends with verification can have merged secondary
	     frontends; See merge *)
	  match tprim.t_verifier with
	  | NoVerification | VerificationNotAvailable -> ()
	  | Verification _ | ForceVerification ->
	      let jprim = tprim.t_chunk_of_block.(i) in
	      assert (VB.get tprim.t_converted_verified_bitmap jprim = VB.State_verified);
	      List.iter (fun t ->
		assert (not t.t_primary);
		let j = t.t_chunk_of_block.(i) in
		if List.for_all (fun i -> VB.get s.s_verified_bitmap i = VB.State_verified)
                  t.t_blocks_of_chunk.(j) then
		    match t.t_verifier with
		    | NoVerification | VerificationNotAvailable ->
			(* we have no way to check data integrity
			   for this network, assume other(s) know
			   better *)
			(match VB.get t.t_converted_verified_bitmap j with
			 | VB.State_missing | VB.State_partial ->
			     VB.set t.t_converted_verified_bitmap j VB.State_verified;
			     t.t_ncomplete_chunks <- t.t_ncomplete_chunks + 1;
			     t.t_nverified_chunks <- t.t_nverified_chunks + 1
			 | VB.State_complete ->
			     VB.set t.t_converted_verified_bitmap j VB.State_verified;
			     t.t_nverified_chunks <- t.t_nverified_chunks + 1
			 | VB.State_verified -> ())
		    | ForceVerification
		    | Verification _ ->
			(* all chunks are verified, so set
			   converted_verified_bitmap to State_complete,
			   probably to trigger data verification later.
			   
			   Is that code necessary at all ? *)
			(match VB.get t.t_converted_verified_bitmap j with
			 | VB.State_missing | VB.State_partial ->
			     VB.set t.t_converted_verified_bitmap j VB.State_complete;
			     t.t_ncomplete_chunks <- t.t_ncomplete_chunks + 1
			 | VB.State_complete -> ()
			 | VB.State_verified -> 
			     lprintf_nl "set_swarmer_state_verified: trying to demote a verified block in another frontend?")
	      ) secondaries)
  | VB.State_verified -> ()

(** set block as completed, closing all remaining ranges, and
    incrementing amount downloaded by their total size.
    If the block was empty its whole size is added *)

let set_completed_block maybe_t s i =
  let mark_completed () =
    set_swarmer_state_complete s i;
    s.s_blocks.(i) <- CompleteBlock in
  match s.s_blocks.(i) with
  | CompleteBlock | VerifiedBlock -> ()
  | EmptyBlock ->
      let block_begin = compute_block_begin s i in
      let block_end = compute_block_end s i in
      add_file_downloaded maybe_t s (block_end -- block_begin);
      mark_completed ()
  | PartialBlock b ->
      close_block_ranges maybe_t s b;
      mark_completed ()

(** set block as verified, closing all remaining ranges, and
    incrementing amount downloaded by their total size.
    If the block was empty its whole size is added 

    (is it normal that no maybe_t can be provided ? my guess is that
    this function is always called on behalf of a primary frontend) *)

let set_verified_block s j =
  match s.s_blocks.(j) with
  | VerifiedBlock -> ()
  | _ ->
      set_completed_block None s j;
      s.s_blocks.(j) <- VerifiedBlock;
      set_swarmer_state_verified s j

(*************************************************************************)
(*                                                                       *)
(*                        frontends verified bitmaps                     *)
(*                                                                       *)
(*************************************************************************)

(* We've seen how swarmer verification propagates to the frontend(s)
   verifications, now let's see the reverse *)
	
let set_frontend_state_missing t j =
  assert(VB.get t.t_converted_verified_bitmap j = VB.State_complete);
  let s = t.t_s in
  assert(List.for_all (fun i -> VB.get s.s_verified_bitmap i <> VB.State_verified) t.t_blocks_of_chunk.(j));
  t.t_ncomplete_chunks <- t.t_ncomplete_chunks - 1;
  if List.for_all (fun i -> VB.get s.s_verified_bitmap i = VB.State_complete) t.t_blocks_of_chunk.(j) then begin
      if !verbose_swarming || !verbose then
        lprintf_nl "Complete block %d/%d of %s failed verification, reloading..."
          (j + 1) t.t_nchunks (file_best_name t.t_file);
      
      VB.set t.t_converted_verified_bitmap j VB.State_missing;
      List.iter (fun i ->
        match s.s_blocks.(i) with
	| EmptyBlock -> set_swarmer_state_missing s i
        | PartialBlock _ ->  set_swarmer_state_partial s i
        | CompleteBlock ->
            let block_begin = compute_block_begin s i in
            let block_end = compute_block_end s i in
	    (* negative *)
            add_file_downloaded None s (block_begin -- block_end);
	      
            s.s_blocks.(i) <- EmptyBlock;
            set_swarmer_state_missing s i
	      
        | VerifiedBlock -> assert false
      ) t.t_blocks_of_chunk.(j)
    end
  else begin
    (* afaiu not supposed to happen, so this code is for debugging ? *)
    if !verbose_swarming then begin
      let nsub = ref 0 in
      lprintf_n "  Swarmer was incomplete: ";
      List.iter (fun i ->
        lprintf "%c" (VB.state_to_char (VB.get s.s_verified_bitmap i));
        if VB.get s.s_verified_bitmap i = VB.State_complete then incr nsub;
      ) t.t_blocks_of_chunk.(j);
      lprintf_nl "   = %d/%d" !nsub (List.length t.t_blocks_of_chunk.(j))
    end;
    VB.set t.t_converted_verified_bitmap j VB.State_partial
  end

(* aka set_completed_chunk (internal) *)
let set_frontend_state_complete t j =
  match VB.get t.t_converted_verified_bitmap j with
  | VB.State_missing | VB.State_partial ->
      if !verbose_swarming || !verbose then
	lprintf_nl "Completed block %d/%d of %s"
          (j + 1) t.t_nchunks (file_best_name t.t_file);
      let s = t.t_s in
      List.iter (fun i -> set_completed_block None s i)
	t.t_blocks_of_chunk.(j)
  | VB.State_complete | VB.State_verified -> ()

(* aka set_verified_chunk (internal) *)
let set_frontend_state_verified t j =
  let mark_verified () =
    VB.set t.t_converted_verified_bitmap j VB.State_verified;
    if (not !CommonGlobals.is_startup_phase) && (!verbose_swarming || !verbose) then
      lprintf_nl "Verified block %d/%d of %s"
        (j + 1) t.t_nchunks (file_best_name t.t_file);
    if t.t_primary then begin
      let s = t.t_s in
      (* The primary is supposed to propagate verified chunks to the file *)
      List.iter (fun i -> set_verified_block s i) t.t_blocks_of_chunk.(j);
      if !verbose_swarming then
	print_s "VERIFIED" s
    end;
    t.t_verified t.t_nverified_chunks j in
    if j = 0 && !Autoconf.magic_works then check_magic t.t_file;
  match VB.get t.t_converted_verified_bitmap j with
  | VB.State_missing | VB.State_partial ->
      t.t_ncomplete_chunks <- t.t_ncomplete_chunks + 1;
      t.t_nverified_chunks <- t.t_nverified_chunks + 1;
      mark_verified ()
  | VB.State_complete ->
      t.t_nverified_chunks <- t.t_nverified_chunks + 1;
      mark_verified ()
  | VB.State_verified -> ()

let set_chunks_verified_bitmap t bitmap =
  VB.iteri (fun j c ->
    match c with
    | VB.State_missing | VB.State_partial -> 
	()
    | VB.State_complete ->
	set_frontend_state_complete t j
    | VB.State_verified ->
	set_frontend_state_verified t j;
        if VB.get t.t_converted_verified_bitmap j <> VB.State_verified then
          lprintf_nl "FIELD AS BEEN CLEARED"
  ) bitmap

let chunks_verified_bitmap t = 
    t.t_converted_verified_bitmap

(** Check the equality of the hash of [t]'s data between offsets
    [begin_pos] and [end_pos] against the value of [uid] *)

(*************************************************************************)
(*                                                                       *)
(*                         verify_chunk (internal)                       *)
(*                                                                       *)
(*************************************************************************)

let verify_chunk t j =
  let verify t uid begin_pos end_pos =
    file_verify t.t_file uid begin_pos end_pos in

  if VB.get t.t_converted_verified_bitmap j = VB.State_complete then
    let nchunks = VB.length t.t_converted_verified_bitmap in
    match t.t_verifier with
    | NoVerification
    | VerificationNotAvailable -> ()

    | ForceVerification ->
        set_frontend_state_verified t j

    | Verification uids when Array.length uids = nchunks ->

        (try
          let s = t.t_s in
          let chunk_begin = t.t_chunk_size *.. j in
          let chunk_end = chunk_begin ++ t.t_chunk_size in
          let chunk_end = min chunk_end s.s_size in
          if verify t uids.(j) chunk_begin chunk_end then
            set_frontend_state_verified t j
	  else
	    set_frontend_state_missing t j
        with VerifierNotReady -> ())

    | Verification chunks ->
	(* network only provides a hash for the whole file ? *)
        assert (Array.length chunks = 1);
(*        let nchunks = String.length t.t_converted_verified_bitmap in *)

	if VB.for_all (function
	  | VB.State_missing | VB.State_partial -> false
	  | VB.State_complete | VB.State_verified -> true) t.t_converted_verified_bitmap then
          try
            let s = t.t_s in
            if verify t chunks.(0) zero s.s_size then
	      VB.iteri (fun j _ ->
		set_frontend_state_verified t j
	      ) t.t_converted_verified_bitmap
	    else
	      VB.iteri (fun j c ->
		if c = VB.State_complete then set_frontend_state_missing t j
	      ) t.t_converted_verified_bitmap
          with VerifierNotReady -> ()


(** mark a block as completed, ready for verification *)

let must_verify_block s i =
  set_swarmer_state_complete s i

(** mark all blocks as completed, ready for verification *)

let verify_all_chunks t =
  let s = t.t_s in
  VB.iteri (fun i state -> 
    match state with
    | VB.State_missing | VB.State_verified ->
	must_verify_block s i
    | VB.State_partial | VB.State_complete -> ()
  ) s.s_verified_bitmap

(** same, and synchronously calls the verification of all chunks *)

let verify_all_chunks_immediately t =
  verify_all_chunks t;
  VB.iteri (fun i _ -> verify_chunk t i) t.t_converted_verified_bitmap
    

(** synchronously verify all completed chunks not yet verified *)

let compute_bitmap t =
  if t.t_ncomplete_chunks > t.t_nverified_chunks then 
    VB.iteri (fun i c -> 
      if c = VB.State_complete then verify_chunk t i) t.t_converted_verified_bitmap


(** Replaces the ith block of the swarmer with a PartialBlock
    ranges are created with s_range_size size *)

let new_block s i =

  let block_begin = compute_block_begin s i in
  let block_end = compute_block_end s i in
  let block_size = block_end -- block_begin in
  let rec b = {
      block_s = s;

      block_begin = block_begin;
      block_end = block_end;
      block_ranges = range;
      block_num = i;
      block_remaining = block_size;
      block_unselected_remaining = block_size;
    }

  and range = {
      range_prev = None;
      range_next = None;
      range_begin = block_begin;
      range_end = block_end;
      range_block = b;
      range_nuploading = 0;
    }
  in
(*  lprintf "New block %Ld-%Ld\n" block_begin block_end; *)

  s.s_blocks.(i) <- PartialBlock b;
  if VB.get s.s_verified_bitmap i = VB.State_missing then
    set_swarmer_state_partial s i;
  if debug_all then lprintf_nl "NB[%s]" (VB.to_string s.s_verified_bitmap);
  b

(** Remove an interval from the beginning of a range, adding the size
    of the removed part to the downloaded amount
    Closed ranges are removed
    When last range is removed, mark the block for verification *)

let range_received maybe_t r interval_begin interval_end =
(*  lprintf "   range_received: %Ld-%Ld for %Ld-%Ld\n"
    interval_begin interval_end r.range_begin r.range_end;  *)
  (* interval overlap with the beginning of range ? *)
  (* was: r.range_begin < interval_end && r.range_end > interval_begin *)
  if r.range_begin >= interval_begin && 
    r.range_begin < interval_end then begin
(*      lprintf "... entered\n"; *)
    let new_begin =
      max (min interval_end r.range_end) r.range_begin in
    let downloaded = new_begin -- r.range_begin in
    let b = r.range_block in
    let s = b.block_s in
    add_file_downloaded maybe_t s downloaded;
    b.block_remaining <- b.block_remaining -- downloaded;
    if r.range_nuploading = 0 then
      b.block_unselected_remaining <- b.block_unselected_remaining -- downloaded;
    r.range_begin <- new_begin;
    if r.range_begin = r.range_end then begin
      (* range completed, unlink it *)
      (match r.range_next with
       | Some rr -> rr.range_prev <- r.range_prev
       | None -> ());
      (match r.range_prev with
       | Some rr -> rr.range_next <- r.range_next
       | None ->
	   (* that was the first range of the block *)
           match r.range_next with
           | Some rr -> (* fix block's first range *)
	       b.block_ranges <- rr
           | None -> (* that was the last remaining range of the block *)
	       match s.s_blocks.(b.block_num) with
	       | PartialBlock _ | EmptyBlock ->
		   (match s.s_networks with
		   | t :: _ -> 
		       assert(t.t_primary);
		       (match t.t_verifier with
		       | NoVerification ->
			   set_verified_block s b.block_num
		       | _ ->
			   set_completed_block (Some t) s b.block_num;
			   must_verify_block s b.block_num)
		   | [] -> assert false)
	       | _ -> () );
      r.range_next <- None;
      r.range_prev <- None;
    end (* else begin
           lprintf " ... new range %Ld-%Ld\n" r.range_begin r.range_end;
           end *)
  end

(** Split a range at [cut_pos] offset, if needed;
    ranges stay linked together *)

let rec split_range r cut_pos =
(*  lprintf "   split_range: cut_pos %Ld\n" cut_pos; *)
  if r.range_begin < cut_pos && r.range_end > cut_pos then
    (* "right" half *)
    let split_r = {
      range_block = r.range_block;
      range_nuploading = 0;
      range_next = r.range_next;
      range_prev = Some r;
      range_begin = cut_pos;
      range_end = r.range_end;
    } in
    (match r.range_next with
     | None -> ()
     | Some old_next_range ->
         old_next_range.range_prev <- Some split_r);
    (* "left" half *)
    r.range_next <- Some split_r;
    r.range_end <- cut_pos
(*    lprintf "      NEW RANGE: %Ld- OLD RANGE: %Ld-%Ld\n"
      split_r.range_begin r.range_begin r.range_end; *)

(** Remove an interval from the ranges of a block, calling
    range_received over all of them

    Assumption: we never download ranges from the middle, so present
    intervals can only overlap the beginning of ranges

    A (double linked) list is definitely not the most efficient
    datastructure for this operation... *)

let set_present_block b interval_begin interval_end =
  let interval_size = interval_end -- interval_begin in
  let old_remaining = b.block_remaining in
  (* download can only happen at the beginning of ranges, so we must
     first split at each interval beginning *)
  iter_block_ranges (fun r ->
    split_range r interval_begin) b;
  iter_block_ranges (fun r ->
    range_received None r interval_begin interval_end) b;
  let new_present = old_remaining -- b.block_remaining in
  if new_present <> interval_size then
    lprintf_nl "set_present_block: %Ld added <> %Ld effectively added"
      interval_size new_present

(** Remove a list of intervals from the ranges of a swarmer *)

let set_present s intervals =
  iter_intervals s (fun i block_begin block_end interval_begin interval_end ->
(*      lprintf "interval: %Ld-%Ld in block %d [%Ld-%Ld]\n"
      interval_begin interval_end i block_begin block_end;  *)
    match s.s_blocks.(i) with
    | EmptyBlock ->
(*          lprintf "  EmptyBlock"; *)
        if block_begin >= interval_begin && block_end <= interval_end
	then begin
(*              lprintf " --> CompleteBlock\n"; *)
          s.s_blocks.(i) <- CompleteBlock;
          must_verify_block s i;
          add_file_downloaded None s (block_end -- block_begin)
        end
        else
          let b = new_block s i in
(*          lprintf " ... set_present_block\n";  *)
          set_present_block b interval_begin interval_end
    | PartialBlock b ->
(*          lprintf "  PartialBlock\n"; *)
        set_present_block b interval_begin interval_end
    | CompleteBlock | VerifiedBlock ->
(*          lprintf "  Other\n"; *)
        ()
  ) intervals;
  match s.s_networks with
  | tprim :: _ ->
      List.iter (fun (interval_begin, interval_end) ->
	mark_disk_space_preallocated tprim interval_begin interval_end;
      ) intervals
  | [] -> assert false

(** reverse absent/present in the list and call set_present *)

let set_absent s list_absent =
(** Build the complementary list of intervals of [intervals] in 
    [set_begin, set_end[ *)
  let rec complementary acc set_begin set_end intervals =
    match intervals with
    | [] ->
	let acc =
          if set_begin = set_end then acc else
            (set_begin, set_end) :: acc
	in
	List.rev acc
    | (interval_begin, interval_end) :: other_intervals ->
	let acc =
          if set_begin = interval_begin then acc
          else (set_begin, interval_begin) :: acc
	in
	complementary acc interval_end set_end other_intervals in
  let list_present = complementary [] Int64.zero s.s_size list_absent in
  set_present s list_present

let intervals_to_string s intervals =
  match intervals with
  | AvailableIntervals intervals ->
      let st = VB.create (Array.length s.s_blocks) VB.State_missing in
      iter_intervals s (fun i _ _ _ _ -> VB.set st i VB.State_partial) intervals;
      VB.to_string st
  | AvailableBitv b -> Bitv.to_string b

(*************************************************************************)
(*                                                                       *)
(*                               Uploaders                               *)
(*                                                                       *)
(*************************************************************************)

(** (debug) output an [uploader] to current log *)

let print_uploader up =
  lprintf_n "  interesting complete_blocks: %d\n     " up.up_ncomplete;
  Array.iter (fun i -> lprintf " %d " i) up.up_complete_blocks;
  lprint_newline ();
  lprintf_n "  interesting partial_blocks: %d\n     " up.up_npartial;
  Array.iter (fun (i, begin_pos, end_pos) ->
      lprintf " %d[%Ld...%Ld] " i begin_pos end_pos
  ) up.up_partial_blocks;
  lprint_newline ()

(** if not [up_declared], 
    sets [up_intervals], [up_complete_blocks], [up_ncomplete],
    [up_partial_blocks], [up_npartial] according to [intervals],
    resets [up_blocks], and calls
    [client_has_bitmap] on associated client.

    My feeling is that if all those fields only make sense when
    up_declared is true, they should be regrouped in a record option.
*)

let set_uploader_intervals up intervals =
  if up.up_declared then
    lprintf_nl "set_uploader_intervals: called on an already declared uploader\n"
  else
    let t = up.up_t in
    let s = t.t_s in
    (* INVARIANT: complete_blocks must be in reverse order *)

  let complete_blocks = ref [] in
  let partial_blocks = ref [] in

  let incr_availability s i =
    s.s_availability.(i) <- s.s_availability.(i) + 1 in

  (match intervals with
   | AvailableIntervals intervals ->
       iter_intervals s (fun i block_begin block_end interval_begin interval_end ->
(*              lprintf "iter_intervals %d %Ld-%Ld %Ld-%Ld\n"
                i block_begin block_end interval_begin interval_end; *)
         incr_availability s i;

         match s.s_blocks.(i) with
	 | CompleteBlock | VerifiedBlock -> ()
         | EmptyBlock | PartialBlock _ ->
             if block_begin = interval_begin && block_end = interval_end then
               complete_blocks := i :: !complete_blocks
             else
               partial_blocks :=
                 (i, interval_begin, interval_end) :: !partial_blocks
       ) intervals

   | AvailableBitv bitmap ->
       Bitv.iteri_true (fun i ->
         List.iter (fun j ->
           incr_availability s j;
           complete_blocks := j :: !complete_blocks
         ) t.t_blocks_of_chunk.(i)
       ) bitmap
  );

  List.iter (fun i ->
(*        s.s_last_seen.(i) <- BasicSocket.last_time (); *)
    let i = t.t_chunk_of_block.(i) in
    t.t_last_seen.(i) <- BasicSocket.last_time ()
  ) !complete_blocks;

  let complete_blocks = Array.of_list !complete_blocks in
  let partial_blocks = Array.of_list !partial_blocks in
  up.up_intervals <- intervals;

  up.up_complete_blocks <- complete_blocks;
  up.up_ncomplete <- Array.length complete_blocks;
  
  if Array.length partial_blocks > 0 then
    lprintf_nl "WARNING: partial_blocks = %d" (Array.length partial_blocks);
  up.up_partial_blocks <- partial_blocks;
  up.up_npartial <- Array.length partial_blocks;

  up.up_blocks <- [];

  up.up_declared <- true;
  
  let bm = intervals_to_string s intervals in
  client_has_bitmap up.up_client up.up_t.t_file bm;

  if debug_all then print_uploader up

(*************************************************************************)
(*                                                                       *)
(*                         register_uploader                             *)
(*                                                                       *)
(*************************************************************************)

let register_uploader t client intervals =
  let up =
    {
      up_t = t;
      up_client = client;

      up_declared = false;
      up_intervals = intervals;

      up_complete_blocks = [||];
      up_ncomplete = 0;

      up_partial_blocks = [||];
      up_npartial = 0;

      up_blocks = [];

      up_ranges = [];
    }
  in
  HU.add uploaders_by_num up;
  set_uploader_intervals up intervals;
  up

(*************************************************************************)
(*                                                                       *)
(*                         unregister_uploader                           *)
(*                                                                       *)
(*************************************************************************)

let clear_uploader_ranges up =
  List.iter (fun (_,_,r) ->
    if r.range_nuploading > 0 then
      r.range_nuploading <- r.range_nuploading - 1
    else
      lprintf_nl "clear_uploader_ranges: some range_nuploading was about to become negative\n";
    if r.range_nuploading = 0 then
      let b = r.range_block in
      b.block_unselected_remaining <-
	b.block_unselected_remaining ++ (compute_range_size r);
      if b.block_unselected_remaining > b.block_remaining then
	lprintf_nl "clear_uploader_ranges: block_unselected_remaining larger than block_remaining!";
  ) up.up_ranges;
  up.up_ranges <- []

let clear_uploader_blocks up =
  List.iter (fun b ->
    let num = b.up_block.block_num in
    let t = up.up_t in
    let s = t.t_s in
    if debug_all then
      lprintf_nl "Client %d unselect %d" (client_num up.up_client) num;
    if s.s_nuploading.(num) > 0 then
      s.s_nuploading.(num) <- s.s_nuploading.(num) - 1
    else
      lprintf_nl "clear_uploader_blocks: some s_nuploading was about to
  become negative\n";
  ) up.up_blocks;
  up.up_blocks <- []

let clear_uploader_intervals up =
  if up.up_declared then
    let decr_availability s i =
      if s.s_availability.(i) > 0 then
	s.s_availability.(i) <- s.s_availability.(i) - 1 
      else 
	lprintf_nl "clear_uploader_intervals: some s_availability was about to become negative\n" in
(*          lprintf "clean_uploader_chunks:\n"; *)
    let t = up.up_t in
    let s = t.t_s in
    Array.iter (decr_availability s) up.up_complete_blocks;
    up.up_complete_blocks <- [||];
    up.up_ncomplete <- 0;
    Array.iter (fun (b,_,_) -> decr_availability s b) up.up_partial_blocks;
    up.up_partial_blocks <- [||];
    up.up_npartial <- 0;
    clear_uploader_blocks up;
    up.up_declared <- false

let update_uploader_intervals up intervals =
  clear_uploader_intervals up;
  set_uploader_intervals up intervals

let unregister_uploader up =
  clear_uploader_intervals up;
  clear_uploader_ranges up

(** (debug) output the uploaders of a swarmer to current log *)

let print_uploaders s =
  Array.iteri (fun i b ->
    match b with
    | EmptyBlock -> lprintf "_"
    | CompleteBlock -> lprintf "C"
    | VerifiedBlock -> lprintf "V"
    | PartialBlock b ->
        if s.s_nuploading.(i) > 9 then
          lprintf "X"
        else
          lprintf "%d" s.s_nuploading.(i)
  ) s.s_blocks;
  lprint_newline ();
  Array.iteri (fun i b ->
    match b with
    | EmptyBlock -> lprintf "_"
    | CompleteBlock -> lprintf "C"
    | VerifiedBlock -> lprintf "V"
    | PartialBlock b ->
        lprintf "{ %d : %d=" b.block_num
          s.s_nuploading.(b.block_num);
	iter_block_ranges (fun r -> 
          lprintf "(%d)" r.range_nuploading
	) b;
        lprintf " }";
  ) s.s_blocks;
  lprint_newline ()

(** (see uploaders invariants above)
    Drop the [n]th element from the [up.up_ncomplete] first elements
    of [up.complete_blocks] by swapping it with the
    ([up.up_ncomplete]-1)th element, then decrease [up.up_ncomplete];
    Then return that element, after converting associated block to
    PartialBlock if necessary.
*)

let permute_and_return up n =
  assert (n <= up.up_ncomplete-1);
  let b = up.up_complete_blocks.(n) in
  if debug_all then lprintf_nl "permute_and_return %d <> %d" n b;
  if n < up.up_ncomplete then begin
    up.up_complete_blocks.(n) <- up.up_complete_blocks.(up.up_ncomplete-1);
    up.up_complete_blocks.(up.up_ncomplete-1) <- b
  end;
  up.up_ncomplete <- up.up_ncomplete - 1;
  let t = up.up_t in
  let s = t.t_s in
  match s.s_blocks.(b) with
  | EmptyBlock ->
      let b = new_block s b in
      { 
	up_block = b;
	up_block_begin = b.block_begin;
	up_block_end = b.block_end }
  | PartialBlock b ->
      {
	up_block = b;
	up_block_begin = b.block_begin;
	up_block_end = b.block_end }
  | VerifiedBlock ->
      lprintf_nl "ERROR: verified block in permute_and_return %d\n" b;
      assert false
  | CompleteBlock ->
      lprintf_nl "ERROR: complete block in permute_and_return %d\n" b;
      assert false

(** find a block in up_complete_blocks that's not already
    CompleteBlock or VerifiedBlock.
    If none can be found, do the same with up_partial_blocks.
    If none can be found still, raise Not_found exception 
    
    up_ncomplete and up_npartial are used as in the same way as in
    permute_and_return, but no element is ever permuted.

    Since set_uploader_intervals puts the blocks with the lowest
    offsets at the end of up_complete_blocks and up_partial_blocks,
    this also selects the blocks in increasing offsets order.
*)

let linear_select_blocks up =
  let rec iter_partial up =
    let n = up.up_npartial in
    if n = 0 then raise Not_found;
    let b, block_begin, block_end = up.up_partial_blocks.(n-1) in
    up.up_npartial <- n-1;
    let t = up.up_t in
    let s = t.t_s in
    (* priority bitmap <> 0 here ? *)
    let chunk = t.t_chunk_of_block.(b) in
    match s.s_blocks.(b) with
    | CompleteBlock | VerifiedBlock ->
        iter_partial up
    | PartialBlock b ->
	chunk,
        [{ 
	  up_block = b; 
	  up_block_begin = block_begin; 
	  up_block_end = block_end }]
    | EmptyBlock ->
        let b = new_block s b in
	chunk, 
        [{
	  up_block = b;
	  up_block_begin = block_begin;
	  up_block_end = block_end }] in
  let rec iter_complete up =
    let n = up.up_ncomplete in
    if n = 0 then iter_partial up
    else
      let b = up.up_complete_blocks.(n-1) in
      up.up_ncomplete <- n-1;
      let t = up.up_t in
      let s = t.t_s in
      (* priority bitmap <> 0 here ? *)
      let chunk = t.t_chunk_of_block.(b) in
      match s.s_blocks.(b) with
      | CompleteBlock | VerifiedBlock ->
          iter_complete up
      | PartialBlock b ->
	  chunk,
          [{
	    up_block = b;
	    up_block_begin = b.block_begin;
	    up_block_end = b.block_end }]
      | EmptyBlock ->
          let b = new_block s b in
	  chunk,
          [{
	    up_block = b;
	    up_block_begin = b.block_begin;
	    up_block_end = b.block_end }]
  in
  iter_complete up

(** Check whether block [n] of swarmer [s] is not already downloaded
    and verified.

    Chunk verification may be called if the block is completed but not
    verified.
*)

let should_download_block s n =
  (*  lprintf "should_download_block %d\n" n; *)
  let result =
    match VB.get s.s_verified_bitmap n with
    | VB.State_missing | VB.State_partial -> true
    | VB.State_complete ->
        (match s.s_networks with
         | t :: _ ->
	     assert(t.t_primary);
             (try
               let n = t.t_chunk_of_block.(n) in
               if VB.get t.t_converted_verified_bitmap n = VB.State_complete then
                 verify_chunk t n
             with VerifierNotReady -> ());
	 | [] -> assert false);
        (match VB.get s.s_verified_bitmap n with
         | VB.State_missing | VB.State_partial ->  true
	 | VB.State_complete | VB.State_verified -> false)
    | VB.State_verified -> false
  in
(*  if result then
    lprintf "should_download_block %d\n" n; *)
  result

(*************************************************************************)
(*                                                                       *)
(*                         select_blocks (internal)                      *)
(*                                                                       *)
(*************************************************************************)

type choice = {
  choice_num : int;
  choice_block : int;
  choice_user_priority : int;
  choice_remaining : int64;
  choice_unselected_remaining : int64;
  choice_preallocated : bool;
}

let dummy_choice = {
  choice_num = 0;
  choice_block = 0;
  choice_user_priority = 0;
  choice_remaining = 0L;
  choice_unselected_remaining = 0L;
  choice_preallocated = false;
}

(* Return the best list of blocks to ask from an uploader
   All the blocks from the list must come from the same chunk *)

let select_blocks up =
  let t = up.up_t in
  let s = t.t_s in
  try
    match s.s_strategy with
    | LinearStrategy ->
        linear_select_blocks up
    | _ ->
        if up.up_ncomplete = 0 && up.up_npartial = 0 then raise Not_found;

(* to evaluate the relative rarity of a block, we must compare it to
   the availability of *all* blocks, not only those available from
   that uploader *)
	let sum_availability = Array.fold_left (+) 0 s.s_availability in
	let mean_availability = sum_availability / Array.length s.s_blocks in

	let my_t = if t.t_verifier <> NoVerification then t else 
	  match s.s_networks with
	    | tprim :: _ ->
		assert(tprim.t_primary);
		tprim
	    | [] -> assert false in
	let verification_available = my_t.t_verifier <> NoVerification in

	let several_frontends = List.length s.s_networks > 1 in
	(* compute the number of missing or partial blocks in the same
	   chunks (for all networks) as a given block *)
	(* memoize some results *)
	let memoization_calls = ref 0 in
	let memoization_hits = ref 0 in
	let debug_memoization = false in
	let memoize h f p =
	  incr memoization_calls;
	  try
	    let result = Hashtbl.find h p in
	    incr memoization_hits;
	    if debug_memoization then
	      (* defeats the purpose of memoization, only enable for
		 debugging *)
	      let recomputed_result = f p in
	      if result <> recomputed_result then begin
		lprintf_nl "memoization failure";
		recomputed_result
	      end else result
	    else result
	  with Not_found ->
	    let result = f p in
	    Hashtbl.add h p result;
	    result in
	let memoize_remaining_blocks_in_chunk = Hashtbl.create 17 in
	let memoize_remaining_blocks_in_chunks = Hashtbl.create 17 in

 	let remaining_blocks_in_chunks i = 
	  assert (i >= 0 && i < VB.length s.s_verified_bitmap);
	  memoize memoize_remaining_blocks_in_chunks
	    (fun i ->
 	      List.fold_left (fun acc t ->
		acc + 
		  (memoize memoize_remaining_blocks_in_chunk
		    (fun (tnum, i) ->
 		      let chunk = t.t_chunk_of_block.(i) in
 		      List.fold_left (fun acc b ->
 			if b <> i &&
 			  (match VB.get s.s_verified_bitmap b with
			  | VB.State_missing | VB.State_partial -> true
			  | VB.State_complete | VB.State_verified -> false) then acc + 1
 			else acc) 0 t.t_blocks_of_chunk.(chunk)) (t.t_num, i))
 	      ) 0 s.s_networks) i in

	let preview_beginning = 10000000L in
	let preview_end = (s.s_size ** 98L) // 100L in

	(* sources_per_chunk was initially for edonkey only *)
	let data_per_source = 9728000L // (Int64.of_int !!sources_per_chunk) in
	
	let need_to_complete_some_blocks_quickly = 
	  verification_available && t.t_nverified_chunks < 2
        in

	let create_choice n b =
	  let block_begin = compute_block_begin s b in
	  let block_end = compute_block_end s b in
	  let size = block_end -- block_begin in
	  let remaining, unselected_remaining = match s.s_blocks.(b) with
	    | EmptyBlock -> size, size
	    | PartialBlock b -> b.block_remaining, b.block_unselected_remaining
	    | CompleteBlock | VerifiedBlock -> 0L, 0L in
	  {
	    choice_num = n;
	    choice_block = b;
	    choice_user_priority = (* priority bitmap here instead ? *)
	      if block_begin < preview_beginning then 3 else
		if block_end > preview_end then 2 else 1;
	    choice_remaining = remaining;
            choice_preallocated = is_fully_preallocated t block_begin block_end;
	    choice_unselected_remaining = unselected_remaining;
	  } in

	(* accessors *)
	let choice_num choice =
	  choice.choice_num in

	let _choice_block choice =
	  choice.choice_block in

	let choice_user_priority choice =
	  choice.choice_user_priority in

	let choice_remaining choice =
	  choice.choice_remaining in

	let choice_unselected_remaining choice =
	  choice.choice_unselected_remaining in

	let choice_nuploaders choice =
	  s.s_nuploading.(choice.choice_block) in

	let choice_remaining_per_uploader choice =
	  choice.choice_unselected_remaining //
	    (Int64.of_int (choice_nuploaders choice + 1)) (* planned *) in

	(* has enough uploaders *)
	let choice_saturated choice =
	  choice.choice_unselected_remaining <= 
	    Int64.of_int (choice_nuploaders choice) ** data_per_source in

	let choice_availability choice =
	  s.s_availability.(choice.choice_block) in

	(* remaining blocks in the same chunk, for all frontends *)
	let choice_other_remaining choice =
	  remaining_blocks_in_chunks (choice.choice_block) in

	let choice_preallocated choice =
	  choice.choice_preallocated in
	
	let print_choice c =
	  lprintf_nl "choice %d:%d priority:%d nup:%d rem:%Ld rmu:%Ld rpu:%Ld sat:%B sib:%d av:%d pre:%b" 
	    (choice_num c) up.up_complete_blocks.(choice_num c)
	    (choice_user_priority c)
	    (choice_nuploaders c)
	    (choice_remaining c)
	    (choice_unselected_remaining c)
	    (choice_remaining_per_uploader c)
	    (choice_saturated c)
	    (choice_other_remaining c) 
	    (choice_availability c) 
	    (choice_preallocated c) in

	let compare_choices c1 c2 =
	  (* "RULES" *)
	  (* Avoid stepping on each other's feet *)
	  let cmp =
	    match choice_unselected_remaining c1,
	    choice_unselected_remaining c2 with
	    | 0L, 0L -> 0
	    | _, 0L -> 1
	    | 0L, _ -> -1
	    | _, _ -> 0 in
	  if cmp <> 0 then cmp else

          (* avoid overly unbalanced situations *)
          let cmp =
            match choice_saturated c1, choice_saturated c2 with
            | false, false -> 0
            | false, true -> 1
            | true, false -> -1
            | true, true -> 0 in
          if cmp <> 0 then cmp else

          (* "WISHES" *)
	  (* Do what Master asked for *)
	    let cmp = compare (choice_user_priority c1)
	      (choice_user_priority c2) in
	  if cmp <> 0 then cmp else

          (* "OPTIMIZATIONS" *)
	    (* Pick really rare gems: if average availability of all
	       blocks is higher than 5 connected sources, pick in
	       priority blocks present in at most 3 connected sources;
	       is that too restrictive ? *)
            let cmp =
              if not need_to_complete_some_blocks_quickly &&
		mean_availability > 5 &&
		(choice_availability c1 <= 3 || 
		 choice_availability c2 <= 3) then
		  compare (choice_availability c2)
		    (choice_availability c1)
              else 0 in
            if cmp <> 0 then cmp else
  
	  (* try to quickly complete (and validate) chunks; 
	     if there's only one frontend, each chunk has only one
	     block, and looking at siblings make no sense *)
	  let cmp = 
	    if verification_available && several_frontends then 
	      compare (choice_other_remaining c2)
		(choice_other_remaining c1)
	    else 0 in
	  if cmp <> 0 then cmp else

	  (* try to quickly complete blocks *)
	  let cmp = 
	    compare (choice_unselected_remaining c2)
	      (choice_unselected_remaining c1) in
	  if cmp <> 0 then cmp else

	  (* pick blocks that won't require allocating more disk space *)
	  let cmp =
	    match choice_preallocated c1, choice_preallocated c2 with
	    | true, false -> 1
	    | false, true -> -1
	    | _ -> 0 in
	  if cmp <> 0 then cmp else

	  (* "DEFAULT" *)
	    (* Can't tell *)
	    0 in

	(* compare a new chunk against a list of best choices numbers (and a
	   specimen of best choice) *)
	let keep_best_chunks chunk_blocks_indexes best_choices specimen =
	  match chunk_blocks_indexes with 
	  | [] -> best_choices, specimen
	  | h :: q ->
	      let this_chunk_specimen = List.fold_left (fun acc n ->
		let choice = create_choice n up.up_complete_blocks.(n) in
		let cmp = compare_choices choice acc in
		if cmp <= 0 then acc
		else choice) (create_choice h up.up_complete_blocks.(h)) q in
	      let cmp = compare_choices this_chunk_specimen specimen in
	      if cmp < 0 then best_choices, specimen
	      else if cmp > 0 then [chunk_blocks_indexes], this_chunk_specimen
	      else chunk_blocks_indexes :: best_choices, specimen in
	
	let current_chunk_num, current_chunk_blocks_indexes, 
	  best_choices, specimen =
	  Array2.subarray_fold_lefti (fun 
	    ((current_chunk_num, current_chunk_blocks_indexes, 
	    best_choices, specimen) as acc) n b ->
	    if not (should_download_block s b) then acc 
	    else
	      let chunk_num = t.t_chunk_of_block.(b) in
	      if chunk_num = current_chunk_num then
		(current_chunk_num, n :: current_chunk_blocks_indexes,
		best_choices, specimen)
	      else 
		(* different chunk *)
		match current_chunk_blocks_indexes with
		| [] -> 
		    (* no previous chunk *)
		    (chunk_num, [n], best_choices, specimen)
		| h :: q ->
		    let new_best_choices, new_specimen =
		      keep_best_chunks current_chunk_blocks_indexes best_choices specimen in
		    (chunk_num, [n], new_best_choices, new_specimen)
	  ) (-1, [], [], dummy_choice) up.up_complete_blocks 0
	    (up.up_ncomplete -1) in

	(* last chunk *)
	let best_choices, specimen = 
	  keep_best_chunks current_chunk_blocks_indexes best_choices specimen in
	(* what about up_partial_blocks ? 
	   currently they're taken care of by linear_select_block
	   fallback below *)

	if debug_all then print_choice specimen;

	try
	  let blocks = 
	    match best_choices with
	      | [] -> raise Not_found
	      | [choice] -> choice
	      | _::_ -> 
		  let nchoices = List.length best_choices in
		  List.nth best_choices (Random.int nchoices) in

	  let chunk =
	    match blocks with
	    | [] -> assert false
	    | b :: _ -> t.t_chunk_of_block.(up.up_complete_blocks.(b)) in

          if debug_all || !verbose_swarming then begin
	    lprintf_nl "\nBlocksFound in %d: %d" chunk (List.length blocks);
	    List.iter (fun n ->
              lprintf_n " %d" up.up_complete_blocks.(n)
	    ) blocks;
	    lprint_newline ()
	  end;

(*
          (* DEBUG *)
	  let probably_buggy =
	    List.for_all (fun n ->
	      let block_num = up.up_complete_blocks.(n) in
	      match s.s_blocks.(block_num) with
	      | EmptyBlock -> false
	      | PartialBlock b ->
		  block_ranges_for_all (fun r ->
		    r.range_nuploading > 0) b
	      | CompleteBlock | VerifiedBlock ->
		  true) blocks in
	  if probably_buggy then begin
            lprintf_nl "Probably buggy choice (%d):" chunk;
	    Array2.subarray_fold_lefti (fun () n b ->
	      if should_download_block s b then
		let this_choice = create_choice n b in
		if List.mem n blocks then lprintf "** "
		else if List.exists (List.mem n) best_choices then
		  lprintf "-- "
		else lprintf "   ";
		print_choice this_choice;
		match s.s_blocks.(b) with
		| EmptyBlock | CompleteBlock | VerifiedBlock -> ()
		| PartialBlock b ->
		    let total_uploading =
		      block_ranges_fold	(fun acc r ->
			lprintf "%d " r.range_nuploading;
			  acc + r.range_nuploading) 0 b in
		    lprintf "total=%d" total_uploading;
		    lprint_newline ()
	    ) () up.up_complete_blocks 0 (up.up_ncomplete - 1);
	  end;
	  (* /DEBUG *)
*)

	  chunk, List.map (fun n -> permute_and_return up n) blocks
	with Not_found ->
	  if !verbose_swarming then
	    lprintf_nl "select_block: fallback to linear strategy";
	  linear_select_blocks up
  with Not_found ->

    (* print_s "NO BLOCK FOUND" s; *)
    raise Not_found

(** If uploader is associated to a file being downloaded,
    clear previously selected block (in any) and select best available
    block, according to block selection strategy 
    @param up the uploader *)

let find_blocks up =
  try
    if debug_all then begin
      lprintf "C: ";
      for i = 0 to up.up_ncomplete - 1 do
        lprintf "%d " up.up_complete_blocks.(i)
      done;
    end;

    let t = up.up_t in
    let s = t.t_s in
    match file_state t.t_file with
    | FilePaused
    | FileAborted _
    | FileCancelled 
    | FileShared
    | FileNew 
    | FileDownloaded -> 
	raise Not_found
    | FileDownloading
    | FileQueued ->
	List.iter (fun b ->
          let num = b.up_block.block_num in
	  if debug_all then
	    lprintf_nl "Client %d unselected %d" (client_num up.up_client) num;
	  if s.s_nuploading.(num) > 0 then
            s.s_nuploading.(num) <- s.s_nuploading.(num) - 1
	  else
	    lprintf_nl "find_blocks: s_nuploading was about to become  negative"
	) up.up_blocks;
        up.up_blocks <- [];

        let chunk, blocks = select_blocks up in
	List.iter (fun b ->
          let num = b.up_block.block_num in
	  if debug_all then
	    lprintf_nl "Client %d selected %d" (client_num up.up_client) num;
          s.s_nuploading.(num) <- s.s_nuploading.(num) + 1;
	) blocks;
	up.up_blocks <- blocks;
        chunk, blocks
  with e ->
    if debug_all then lprintf_nl "Exception %s in find_blocks" (Printexc2.to_string e);
    raise e

(** Remove completed ranges from an uploader's range list, and
    decrease their reference counter *)

let remove_completed_uploader_ranges up =
  let not_completed_ranges, completed_ranges = 
    List.partition (fun (_,_,r) -> 
      r.range_begin < r.range_end) up.up_ranges in
  up.up_ranges <- not_completed_ranges;
  List.iter (fun (_,_,r) -> 
    if r.range_nuploading > 0 then
      r.range_nuploading <- r.range_nuploading - 1
    else
      lprintf_nl "remove_completed_uploader_ranges: range_nuploading
  was about to become negative!";
    if r.range_nuploading = 0 then
      let b = r.range_block in
      b.block_unselected_remaining <-
	b.block_unselected_remaining ++ (compute_range_size r);
      if b.block_unselected_remaining > b.block_remaining then
	lprintf_nl "remove_completed_uploader_ranges: block_unselected_remaining is larger than block_remaining";
    ) completed_ranges

(** uploader accessors *)

let current_ranges up =  up.up_ranges

let current_blocks up =
  match up.up_blocks with
  | [] -> raise Not_found
  | bl -> bl

(** Check whether a range is in a list *)

let in_uploader_ranges r list =
  List.exists (fun (_,_,r') -> r' == r) list

(*************************************************************************)
(*                                                                       *)
(*                         find_range                                    *)
(*                                                                       *)
(*************************************************************************)

let uploader_ranges_fold_left f acc l =
  let rec aux acc l =
    match l with
    | [] -> acc
    | h :: q -> aux (f acc h) q
  in aux acc l

(** Find a range to upload from [up], that is at most [range_size]
    bytes long (split some range if necessary) *)

(* Is merging at all useful ? Once next range starts downloading, they
   can no longer be merged, so it should be rare... *)
let allow_merge_ranges = true

type ranges_cluster = {
  cluster_ranges: range list;
  cluster_nuploading: int;
  cluster_size: Int64.t
}    

let dummy_ranges_cluster = {
  cluster_ranges = [];
  cluster_nuploading = 0;
  cluster_size = 0L
}

let is_dummy_cluster cluster = 
  cluster.cluster_ranges = []

let find_range up range_size =

  (** merge two consecutive ranges in the first, if possible;
      Return true if successful *)
  let merge_ranges r r2 =
    match r.range_next with
    | None -> false
    | Some rr ->
	if rr != r2 ||
	  r.range_end < r2.range_begin ||
	  r2.range_nuploading > 0 then false
	else begin
	  r.range_end <- r2.range_end;
	  r.range_next <- r2.range_next;
	  (match r.range_next with
	  | None -> ()
	  | Some r3 ->
	      r3.range_prev <- Some r);
	  true
	end in

  remove_completed_uploader_ranges up;

  let b, more_blocks =
    match up.up_blocks with
    | [] -> 
	if debug_all then
	  lprintf_nl "find_range: uploader had no block selected";
	raise Not_found
    | b :: more_blocks -> b, more_blocks
  in
  let t = up.up_t in
  match file_state t.t_file with
  | FilePaused
  | FileAborted _
  | FileCancelled 
  | FileShared
  | FileNew
  | FileDownloaded -> 
      lprintf_nl "find_range: file %s in bad state %s" 
	t.t_s.s_filename (string_of_state (file_state t.t_file));
      raise Not_found
  | FileDownloading
  | FileQueued ->
      if debug_all then
	lprintf_nl "find_range: is there a range of size %Ld in %s for %d ?" 
	  range_size 
	  (String.concat " " (List.map (fun b ->
	    Printf.sprintf "[%Ld-%Ld]" b.up_block_begin b.up_block_end
	  ) up.up_blocks))
	  (client_num up.up_client);
      (* pick the first correct cluster with fewest uploaders 
	 We're not trying to get a range that's at least as big as
	 [range_size] bytes - that would prevent partially downloaded
	 ranges from being completed first *)
      let rec iter acc r b more_blocks =
	let correct_range r =
	  not (in_uploader_ranges r up.up_ranges) &&
	    r.range_begin < r.range_end &&
	    r.range_begin >= b.up_block_begin &&
	    r.range_begin < b.up_block_end in
	let best_cluster =
	  if not (correct_range r) then acc
	  else 
	    (* find if there are ranges to merge ahead *)
	    let rec iter_cluster r cluster =
	      let cluster = { cluster with
		cluster_ranges = r :: cluster.cluster_ranges;
		cluster_size = cluster.cluster_size ++ 
		  (compute_range_size r)
	      } in
	      if not allow_merge_ranges ||
		cluster.cluster_size >= range_size then cluster 
	      else
		match r.range_next with
		| None -> cluster
		| Some rr ->
		    if rr.range_begin = r.range_end &&
		      correct_range rr && rr.range_nuploading = 0 then 
		      iter_cluster rr cluster
		    else cluster in

	    let cluster = 
	      iter_cluster r { dummy_ranges_cluster with
		cluster_nuploading = r.range_nuploading } in
	    if debug_all then
	      lprint_newline ();
	    if is_dummy_cluster acc then cluster
	    else
	      (* find a range with as few uploaders as possible *)
	      let cmp = compare acc.cluster_nuploading
		cluster.cluster_nuploading in
	      if cmp < 0 then acc
	      else cluster in

	(* fast exit, and why I didn't use an iterator :/ 
	   Could have used an exception, but I don't like that ;) *)
	if not (is_dummy_cluster best_cluster) &&
	  best_cluster.cluster_nuploading = 0 then b, best_cluster
	else
	  match r.range_next with
	  | Some rr -> iter best_cluster rr b more_blocks
	  | None -> 
	      match more_blocks with
	      | [] -> b, best_cluster
	      | b :: more_blocks -> 
		  if debug_all || !verbose then
		    lprintf_nl "find_range: Client %d, next block" 
		      (client_num up.up_client);
		  iter best_cluster b.up_block.block_ranges b more_blocks in

      let best_block, best_cluster = 
	iter dummy_ranges_cluster b.up_block.block_ranges b more_blocks in
      if not (is_dummy_cluster best_cluster) &&
	best_cluster.cluster_nuploading > 0 &&
	(file_downloaded t.t_file < file_size t.t_file ** 98L // 100L) then begin
	(* it seems they're only sucky choices left on that block, is
	   there really nothing else better elsewhere ? *)
	let s = b.up_block.block_s in
	for i = 0 to up.up_ncomplete - 1 do
	  let block = up.up_complete_blocks.(i) in
	  if not (List.exists (fun b -> b.up_block.block_num = block
			      ) up.up_blocks) then
	    if should_download_block s block then (* priority bitmap <> 0 here ? *)
	      let partial_found = match s.s_blocks.(block) with
		| EmptyBlock -> true
		| CompleteBlock | VerifiedBlock -> false
		| PartialBlock b -> b.block_unselected_remaining > 0L in
	      if partial_found then begin
(*
		if debug_all || !verbose then
		  lprintf_nl "find_range: Client %d better switch cluster now!" 
		    (client_num up.up_client);
*)
		raise Not_found
	      end
	done
      end;
      match List.rev best_cluster.cluster_ranges with
      | [] -> 
	  if debug_all then
	    lprintf_nl "find_range: no correct range found!";
	  raise Not_found
      | r :: q ->
	  if not (List.for_all (merge_ranges r) q) then
	    lprintf_nl "find_range: ranges did not merge as well as planned";
	  split_range r (min (r.range_begin ++ range_size)
	   best_block.up_block_end);
	  if debug_all then begin
	    lprintf "=> ";
	    iter_block_ranges (fun rr ->
	      let selected = if rr == r then "*" else "" in
	      lprintf " %s[%Ld-%Ld]:%d%s" selected 
		rr.range_begin rr.range_end rr.range_nuploading
		selected
	    ) best_block.up_block;
	    lprint_newline ();
	  end;
	  let key = r.range_begin, r.range_end, r in
	  up.up_ranges <- up.up_ranges @ [key];
	  if r.range_nuploading = 0 then begin
	    let b = r.range_block in
	    b.block_unselected_remaining <-
	      b.block_unselected_remaining -- (compute_range_size r);
	    if b.block_unselected_remaining < 0L then
	      lprintf_nl "find_range: block_unselected_remaining is negative!";
	  end;
	  r.range_nuploading <- r.range_nuploading + 1;
	  key

(** range accessor(s) *)

let range_range r = (r.range_begin, r.range_end)

(** Data has been received from uploader [up]. Transfer data to file
    and update uploader ranges.
    Data = String.sub [str] [string_begin] [string_len] *)

let received up file_begin str string_begin string_len =
  assert (string_begin >= 0);
  assert (string_len >= 0);
  assert (string_begin + string_len <= String.length str);

  let t = up.up_t in
  let s = t.t_s in

  let debug_bad_write unexpected_intervals =
    if !verbose_swarming then begin
      lprintf "Dismiss unwanted data from %d for %s:" 
	(client_num up.up_client) (file_best_name t.t_file);
      List.iter (fun (i_begin, i_end) -> 
	lprintf " %Ld-%Ld" i_begin i_end) unexpected_intervals;
      lprint_newline ();
      lprintf_nl "  received: file_pos:%Ld string:%d %d"
        file_begin string_begin string_len;
      lprintf_nl "  ranges:";
      List.iter (fun (_,_,r) ->
        lprintf_n "     range: %Ld-%Ld"
          r.range_begin
          r.range_end;
        (match r.range_next with
          | None -> ()
          | Some rr ->
              lprintf "  next: %Ld" rr.range_begin);
        (match r.range_prev with
          | None -> ()
          | Some rr ->
              lprintf "  prev: %Ld" rr.range_begin);
        lprint_newline ();
        let b = r.range_block in
        lprintf_n "        block: %d[%c] %Ld-%Ld [%s]"
          b.block_num
          (VB.state_to_char (VB.get s.s_verified_bitmap b.block_num))
          b.block_begin b.block_end
          (match s.s_blocks.(b.block_num) with
            | EmptyBlock -> "empty"
            | PartialBlock _ -> "partial"
            | CompleteBlock -> "complete"
            | VerifiedBlock -> "verified"
          );
        let br = b.block_ranges in
        lprintf " first range: %Ld(%Ld)"
          br.range_begin
          (compute_range_size br);
        lprint_newline ();
      ) up.up_ranges
    end;
  if !exit_on_error then exit 2 in

  if string_len > 0 then
    let file_end = file_begin ++ (Int64.of_int string_len) in
    if !verbose_swarming then
      lprintf_nl "received on %Ld-%Ld" file_begin file_end;

    (* DEBUG *)
    let intervals_out = 
      let remove_interval intervals interval_begin interval_end =
	let rec remove acc intervals =
	  match intervals with 
	  | [] -> List.rev acc
	  | ((fi_begin, fi_end) as first_interval) :: others ->
	      if fi_begin >= interval_end then
		List.rev_append acc intervals 
	      else if fi_end <= interval_begin then
		remove (first_interval :: acc) others
	      else 
		remove (
		  let acc = if fi_begin < interval_begin then
		    (fi_begin, interval_begin) :: acc else acc in
		  let acc = if fi_end > interval_end then
		    (interval_end, fi_end) :: acc else acc in
		  acc) others in
	remove [] intervals in
      List.fold_left (fun acc (_, _, r) ->
	remove_interval acc r.range_begin r.range_end
      ) [(file_begin, file_end)] up.up_ranges in
    if intervals_out <> [] then
      debug_bad_write intervals_out;
    
    let file_end = min file_end s.s_size in

    match file_state t.t_file with
    | FilePaused
    | FileAborted _
    | FileCancelled 
    | FileShared
    | FileNew
    | FileDownloaded -> 
	if !verbose then
	  lprintf_nl "received: wrong file state %s for %s" (string_of_state (file_state t.t_file)) s.s_filename;
	()
    | FileDownloading
    | FileQueued ->
	try
	  List.iter (fun (_,_,r) ->
    (* was: r.range_begin < file_end && r.range_end > file_begin *)
            if r.range_begin >= file_begin &&
              r.range_begin < file_end then 
		let file_end = min file_end r.range_end in
		let written_len = file_end -- r.range_begin in
                let string_pos = string_begin +
                  Int64.to_int (r.range_begin -- file_begin) in
                let string_length = Int64.to_int written_len in
                  if string_length > 0 then
		    match s.s_networks with
                    | [] -> assert false
		    | tprim :: _ ->
			assert (tprim.t_primary);
			(try
			  preallocate_disk_space tprim 
			    r.range_begin file_end
			with
			  End_of_file -> ()
			| e ->
			  lprintf_nl "Exception %s while preallocating disk space [%Ld-%Ld] for %s"
			    (Printexc2.to_string e) 
			    r.range_begin file_end
			    (file_best_name t.t_file));
                        file_write tprim.t_file
                          r.range_begin
                          str string_pos string_length;
			range_received (Some t) r r.range_begin file_end;
	  ) up.up_ranges;
	  remove_completed_uploader_ranges up
	with e ->
	  lprintf_nl "Exception %s while receiving data"
	    (Printexc2.to_string e);
	  remove_completed_uploader_ranges up;
          raise e

(** compute the list of present intervals of a swarmer *)

let present_intervals s =
  (* intervals is a reversed list of intervals *)
  let append_interval ((interval_begin, interval_end) as interval) intervals =
    (* remove void intervals *)
    if interval_begin = interval_end then intervals
    else
      match intervals with
      | [] -> [interval]
      | (last_interval_begin, last_interval_end) :: other_intervals  ->
	  if last_interval_end < interval_begin then
	    interval :: intervals
	  else 
	    (* coalescing intervals *)
	    (last_interval_begin, interval_end) :: other_intervals in

  List.rev (
    Array2.fold_lefti (fun acc i b -> 
      match s.s_blocks.(i) with
      | EmptyBlock -> acc
      | CompleteBlock | VerifiedBlock ->
	  append_interval (compute_block_begin s i, compute_block_end s i) acc
      | PartialBlock b ->
	  let acc, last_interval_end =
	    block_ranges_fold (fun (acc,  lie) r ->
	      (append_interval (lie, r.range_begin) acc, r.range_end)
	    ) (acc, compute_block_begin s i) b in
	  append_interval (last_interval_end, compute_block_end s i) acc
    ) [] s.s_blocks)

(*************************************************************************)
(*                                                                       *)
(*                         propagate_chunk                               *)
(*                                                                       *)
(*************************************************************************)

type chunk = {
    chunk_uid : uid_type;
    chunk_size : int64;
  }

type chunk_occurrence = t * int * Int64.t (* frontend, chunk number, offset *)

type chunk_occurrences = {
  mutable occurrence_present : chunk_occurrence list;
  mutable occurrence_missing : chunk_occurrence list;
}

let propagate_chunk t1 pos1 size destinations =
  List.iter (fun (t2, j2, pos2) ->
    if t1.t_num <> t2.t_num || pos1 <> pos2 then begin
      if !verbose then lprintf_nl "Should propagate chunk from %s %Ld to %s %Ld [%Ld]"
        (file_best_name t1.t_file) pos1
        (file_best_name t2.t_file) pos2 size;
      Unix32.copy_chunk (file_fd t1.t_file)  (file_fd t2.t_file)
	pos1 pos2 (Int64.to_int size);
      set_frontend_state_complete t2 j2
    end
  ) destinations

let dummy_chunk_occurrences () = 
  { occurrence_present = []; occurrence_missing = [] }

let duplicate_chunks () =
  let chunks = Hashtbl.create 100 in
  HS.iter (fun s ->
    List.iter (fun t ->
      let nchunks = VB.length t.t_converted_verified_bitmap in
      match t.t_verifier with
      | Verification uids when Array.length uids = nchunks ->
	  let rec iter j len pos =
            if j < len then
              let c = {
		chunk_uid = uids.(j);
		chunk_size = min (s.s_size -- pos) t.t_chunk_size;
              } in
	      let occurrences = 
		try
		  Hashtbl.find chunks c
		with Not_found ->
                  let occurrences = dummy_chunk_occurrences () in
                  Hashtbl.add chunks c occurrences;
                  occurrences in
	      (match VB.get t.t_converted_verified_bitmap j with
	      | VB.State_missing | VB.State_partial ->
		occurrences.occurrence_missing <- 
		  (t, j, pos) :: occurrences.occurrence_missing
	      | VB.State_complete -> ()
	      | VB.State_verified ->
		occurrences.occurrence_present <- 
		  (t, j, pos) :: occurrences.occurrence_present);
              iter (j+1) len (pos ++ t.t_chunk_size)
	  in
	  iter 0 (VB.length t.t_converted_verified_bitmap) zero
      | _ -> ()
    ) s.s_networks
  ) swarmers_by_name;
  Hashtbl.iter (fun c occurrences ->
    (* we need a verified chunk to copy over the others *)
    match occurrences.occurrence_present, occurrences.occurrence_missing with
    | _ , []
    | [], _ -> ()
    | (t, _, pos) :: _, missing ->
        propagate_chunk t pos c.chunk_size missing
  ) chunks

let set_verifier t f =
  t.t_verifier <- f;
(* TODO: check that false as t_primary is a good value to start with *)
  set_chunks_verified_bitmap t t.t_converted_verified_bitmap

let set_verified t f =
  t.t_verified <- f

let downloaded t = file_downloaded t.t_file

let block_chunk_num t b =
  t.t_chunk_of_block.(b.block_num)

let partition_size t = VB.length t.t_converted_verified_bitmap

let uploader_swarmer up = up.up_t

(** Return the availability of the chunks of [t] as a string *)

let chunks_availability t =
  let s = t.t_s in
  String2.init (partition_size t) (fun i ->
    char_of_int (
      let v = List2.min
        (List.map (fun i -> s.s_availability.(i)) t.t_blocks_of_chunk.(i)) in
      if v < 0 then 0 else
	if v > 200 then 200 else v))

let is_interesting up =
  up.up_ncomplete > 0 || up.up_npartial > 0


(*************************************************************************)
(*                                                                       *)
(*                         value_to_int64_pair (internal)                *)
(*                                                                       *)
(*************************************************************************)

let value_to_int64_pair v =
  match v with
  | List [v1;v2] | SmallList [v1;v2] ->
      (value_to_int64 v1, value_to_int64 v2)
  | _ ->
      failwith "Options: Not an int32 pair"

(*************************************************************************)
(*                                                                       *)
(*                         WRAPPERS                                      *)
(*                                                                       *)
(*************************************************************************)

let set_present t = set_present t.t_s
let set_absent t = set_absent t.t_s
let present_intervals t = present_intervals t.t_s
let print_t str t = print_s str t.t_s
let print_uploaders t = print_uploaders t.t_s

(*************************************************************************)
(*                                                                       *)
(*                         value_to_frontend                             *)
(*                                                                       *)
(*************************************************************************)

let value_to_frontend t assocs =

  let debug_wrong_downloaded t present d =
    lprintf_nl "ERROR: stored downloaded value not restored  !!! (%Ld/%Ld)" (downloaded t) d;
    lprintf_nl "ERROR: present:";
    List.iter (fun (x,y) ->
      lprintf_nl "     (%Ld,%Ld);" x y
    ) present;
    
    let p = present_intervals t in
    lprintf_nl "ERROR: present now:";
    
    let total = 
      List.fold_left (fun acc (x,y) ->
	lprintf_nl "     (%Ld,%Ld);" x y;
	acc ++ (y -- x)
      ) zero p in
    
    lprintf_nl "ERROR: total %Ld" total;
    if p = present then begin
      lprintf_nl "ERROR: both appear to be the same!";
    end;
    if !exit_on_error then exit 2 in

  let get_value name conv = conv (List.assoc name assocs) in

  let primary =
    try get_value "file_primary" value_to_bool with _ -> true in

  (try
    let file_name = get_value "file_swarmer" value_to_string in
    let s = HS.find swarmers_by_name 
	{ dummy_swarmer with s_filename = file_name } in
    associate primary t s
    (* TODO: make as many checks as possible to ensure the file and the swarmers
       are correctly associed. *)
  with Not_found -> ());

  let _ =
    let mtime = try file_mtime t.t_file with _ -> 0. in
    let old_mtime =
      try
        value_to_float (List.assoc "file_mtime" assocs)
      with Not_found -> mtime
    in
    old_mtime = mtime
  in
  
  (try
    try
      set_chunks_verified_bitmap t
        (VB.of_string (get_value  "file_chunks" value_to_string))
    with Not_found ->
      set_chunks_verified_bitmap t
        (VB.of_string (get_value  "file_all_chunks" value_to_string))
	
  with e ->
    lprintf_nl "Exception %s while loading bitmap"
      (Printexc2.to_string e);
    (* force everything to be checked ASAP ? *)
    set_chunks_verified_bitmap t (VB.create (partition_size t) VB.State_complete)
  );

  (*
  lprintf "set_verified_bitmap: t = %s\n" t.t_converted_verified_bitmap;
  lprintf "set_verified_bitmap: s = %s\n" t.t_s.s_verified_bitmap;
*)

  if primary then begin
    if !verbose_swarming then lprintf_nl "Loading present...";
    let present = try
      let present =
        (get_value "file_present_chunks"
          (value_to_list value_to_int64_pair))
      in
      set_present t present;
      present
    with e ->
      lprintf_nl "Exception %s while set present"
        (Printexc2.to_string e);
      verify_all_chunks t;
      []
    in
    if !verbose_swarming then lprintf_nl "Downloaded after present %Ld" (downloaded t);

    (try
      let d = get_value "file_downloaded" value_to_int64 in
      if d <> downloaded t && !verbose then
	debug_wrong_downloaded t present d
    with Not_found -> ());
  end;

(* TODO re-implement this
  (try
      let last_seen = get_value "file_chunks_age"
          (value_to_list value_to_int) in
      t.t_last_seen <- Array.of_list last_seen
    with _ -> ());
*)

  ()

(*************************************************************************)
(*                                                                       *)
(*                         frontend_to_value                             *)
(*                                                                       *)
(*************************************************************************)

let frontend_to_value t other_vals =
  [("file_primary", bool_to_value t.t_primary);
   ("file_swarmer", string_to_value t.t_s.s_filename);
   ("file_mtime", float_to_value (try file_mtime t.t_file with _ -> 0.));
   ("file_chunks", string_to_value (VB.to_string (chunks_verified_bitmap t)))] @
  (if t.t_primary then
    [("file_present_chunks", List
      (List.map (fun (i1,i2) ->
        SmallList [int64_to_value i1; int64_to_value i2])
	(present_intervals t)))] 
   else []) @
  [("file_downloaded", int64_to_value (downloaded t));
   ("file_chunks_age", List (Array.to_list
      (Array.map int_to_value t.t_last_seen)))] @
  other_vals

(** Verify one chunk of swarmer [s], if any frontend of that swarmer
    has a chunk to verify *)

let verify_one_chunk s =
  (*  lprintf "verify_one_chunk: %d networks\n" (List.length s.s_networks);  *)
  List.exists (fun t ->
(*      lprintf "verify_one_chunk of file %d\n" (file_num t.t_file); *)
    VB.existsi (fun i c ->
      if c = VB.State_complete then verify_chunk t i;
      c = VB.State_complete) t.t_converted_verified_bitmap
  ) s.s_networks
(*  lprintf "verify_one_chunk: nothing done\n"; *)

(** Verify one chunk of each swarmer that needs it *)

let verify_some_chunks () =
  HS.iter (fun s ->
    try
      ignore(verify_one_chunk s)
    with _ -> ()
  ) swarmers_by_name

(** Verify one chunk of the swarmer associated with [t], if needed *)

let verify_one_chunk t =
  ignore(verify_one_chunk t.t_s)

(** Merge a second frontend [f2] to a first one [f1], so they share
    the same swarmer.

    First swarmer [f1] must support some hashing scheme.
    Data of the second swarmer [f2] is currently lost during merging, so
    you'd better merge in swarmers quickly.
    Merging is denied if any of the two frontends is being used, so it
    may be necessary to pause them first, to get rid of any downloads.
*)

let merge f1 f2 =

  let s1 = HS.find swarmers_by_name { dummy_swarmer with 
    s_filename = file_disk_name f1 } in
  let s2 = HS.find swarmers_by_name { dummy_swarmer with 
    s_filename = file_disk_name f2 } in

  if s1.s_filename = s2.s_filename then
    failwith "Files are already sharing their swarmer";

  if s1.s_size <> s2.s_size then
    failwith "Files don't have the same size";

  let t2 = 
    match s2.s_networks with
    | [t] -> t
    | list ->
        lprintf_nl "s_networks: %d files" (List.length list);
        failwith "Second file is already merged with other files"
  in

  let t1 =
    match s1.s_networks with
    | [] -> assert false
    | t1 :: _ ->
        match t1.t_verifier with
	| NoVerification | VerificationNotAvailable ->
            failwith "Cannot use first file as a primary for swarming (no verification scheme)"
        | Verification _ | ForceVerification -> t1
  in

  List.iter (fun (s, filename) ->
    Array.iteri (fun i nuploading ->
      if nuploading > 0 then
        failwith (Printf.sprintf "%s is currently being downloaded" filename)
    ) s.s_nuploading
  ) [
    s1, "First file";
    s2, "Second file"];

(* replace T2 swarmer *)
  associate false t2 t1.t_s

(*************************************************************************)
(*                                                                       *)
(*                         has_secondaries                               *)
(*                                                                       *)
(*************************************************************************)

let has_secondaries t =
  t.t_primary && List.length t.t_s.s_networks > 1

(*************************************************************************)
(*                                                                       *)
(*                         Remove swarmer                                *)
(*                                                                       *)
(*************************************************************************)

let remove_swarmer file_swarmer =
  match file_swarmer with 
  | None -> () 
  | Some sw -> if not (has_secondaries sw)
                then HS.remove swarmers_by_name sw.t_s
                else lprintf_nl "Tried to remove swarmer with secondaries"

(*************************************************************************)
(*                                                                       *)
(*                         subfiles                                      *)
(*                                                                       *)
(*************************************************************************)

let subfiles t =
  List.map (fun t -> t.t_file) t.t_s.s_networks

(*************************************************************************)
(*                                                                       *)
(*                         SwarmerOption                                 *)
(*                                                                       *)
(*************************************************************************)

module SwarmerOption = struct

    let value_to_swarmer v =
      match v with
      | Module assocs ->
          let get_value name conv = conv (List.assoc name assocs) in
          let file_size = get_value "file_size" value_to_int64 in
          let file_name = get_value "file_name" value_to_string in
          let s = create_swarmer file_name file_size in
	  (try
	    let bitmap = Bitv.of_string (get_value "file_disk_allocation_bitmap"
	      value_to_string) in
	    if Bitv.length bitmap = Bitv.length s.s_disk_allocated then
	      s.s_disk_allocated <- bitmap
	  with _ -> ());
	  (* s_disk_allocated missing or inconsistent ? 
	     set_present will fix it *)
          let block_sizes = get_value "file_chunk_sizes"
              (value_to_list value_to_int64) in
          List.iter (fun bsize ->
              split_blocks s bsize
          ) block_sizes;
          s

      | _ -> assert false

    let swarmer_to_value s =
      Module [
        ("file_size", int64_to_value s.s_size);
        ("file_name", string_to_value s.s_filename);
	("file_disk_allocation_bitmap", string_to_value
	  (Bitv.to_string s.s_disk_allocated));
        ("file_chunk_sizes", list_to_value int64_to_value
            (List.map (fun t -> t.t_chunk_size) s.s_networks));
        ]

    let t =
      define_option_class "Swarmer" value_to_swarmer swarmer_to_value

  end

(** Checks most variants of a swarmer, nobably verification bitmaps
    consistency; Raise an exception if a problem is found *)

let check_swarmer s =
  try
    match s.s_networks with
    | [] -> lprintf_nl "found unused swarmer %s, discarding" s.s_filename;
    | tprim :: tail ->
        assert(tprim.t_primary);

	VB.iteri (fun i c ->
	    if c = VB.State_verified then begin
	      if List.exists (fun j -> VB.get s.s_verified_bitmap j <> VB.State_verified) 
		tprim.t_blocks_of_chunk.(i) then
                  failwith "Bad propagation of State_verified from primary to swarmer";
            end
            else if List.exists (fun j -> VB.get s.s_verified_bitmap j = VB.State_verified)
	      tprim.t_blocks_of_chunk.(i) then
                failwith "Swarmer has State_verified not coming from primary";
	) tprim.t_converted_verified_bitmap;

        let fd = file_fd tprim.t_file in

        List.iter (fun t ->
          assert (not t.t_primary);
          assert (file_fd t.t_file == fd);
	  
	  VB.iteri (fun i c ->
	    if c = VB.State_verified then begin
	      if List.exists (fun j -> VB.get s.s_verified_bitmap j <> VB.State_verified)
		t.t_blocks_of_chunk.(i) then
                  failwith "State_verified in secondary without State_verified in primary"
	    end 
	    else if c = VB.State_complete then begin
	      if List.exists (fun j -> VB.get s.s_verified_bitmap j <> VB.State_verified)
		t.t_blocks_of_chunk.(i) then
                  failwith "State_complete in secondary without State_verified in primary"
	    end 
	  ) t.t_converted_verified_bitmap
        ) tail
  with e ->
    print_s "ERROR" s;
    raise e

(*************************************************************************)
(*                                                                       *)
(*                         Option swarmers                               *)
(*                                                                       *)
(*************************************************************************)

let swarmers =
  define_option CommonComplexOptions.swarmers_section
    ["swarmers"] "All the swarmers used" (list_option SwarmerOption.t) []

(*************************************************************************)
(*                                                                       *)
(*                         Options hooks                                 *)
(*                                                                       *)
(*************************************************************************)

let _ =
  set_after_save_hook files_ini (fun _ -> swarmers =:= []);
  set_before_save_hook files_ini (fun _ ->
    let list = ref [] in
    HS.iter (fun s ->
      if s.s_networks <> [] then
        list := s :: !list) swarmers_by_name;
    swarmers =:= !list;
    (* put primary frontends to the head, so that swarmers' invariants
       can be verified while downloads are being restored from ini files *)
    let primary_files, secondary_files = 
      List.partition (fun file -> 
	match file_files file with
	| primary_file :: _ when primary_file == file -> true
	| _ -> false) !!CommonComplexOptions.files in
    CommonComplexOptions.files =:= primary_files @ secondary_files
  );
  set_after_load_hook files_ini (fun _ ->
    List.iter (fun s ->
      check_swarmer s;
    ) !!swarmers;
    swarmers =:= []
  )

(*************************************************************************)
(*                                                                       *)
(*                         MAIN                                          *)
(*                                                                       *)
(*************************************************************************)

(* Compute an approximation of the storage used by this module *)

let _ =
  BasicSocket.add_infinite_timer 300. duplicate_chunks;
  Heap.add_memstat "CommonSwarming" (fun level buf ->
      let counter = ref 0 in
      let nchunks = ref 0 in
      let nblocks = ref 0 in
      let nranges = ref 0 in
      HS.iter (fun s ->
          let n = VB.length s.s_verified_bitmap in
          nchunks := !nchunks + n;

          Array.iter (fun b ->
              match b with
              | PartialBlock b ->
                  incr nblocks;
                  iter_block_ranges (fun _ -> incr nranges) b
              | _ -> ()
          ) s.s_blocks;

          incr counter
      ) swarmers_by_name;
      Printf.bprintf buf "  Swarmers: %d\n" !counter;
      Printf.bprintf buf "    nchunks: %d nblocks: %d nranges: %d\n"
        !nchunks !nblocks !nranges;
      Printf.bprintf buf "  Storage (without blocks): %d bytes\n"
        ( !counter * 108 +
          !nchunks * 17 +
          !nblocks * 64 +
          !nranges * 84);

      let counter = ref 0 in
      let storage = ref 0 in
      HU.iter (fun up ->
          storage := !storage + 76 +
            Array.length up.up_complete_blocks * 4 +
            List.length up.up_ranges * (12 + 16 + 12 + 12 +  4) +
            Array.length up.up_partial_blocks * (16 + 12 + 12) +
            (8 + match up.up_intervals with
            | AvailableIntervals list -> List.length list * (12 + 12 + 12 + 12)
            | AvailableBitv b -> let ws = Sys.word_size in (ws/8) + ((ws / 8) * (Bitv.length b / (ws - 2))) 
          ) ;
          incr counter;
      ) uploaders_by_num;
      Printf.bprintf buf "  Uploaders: %d\n" !counter;
      Printf.bprintf buf "  Storage: %d bytes\n" !storage;
  )

(* using compute_block_num outside of swarming code is probably
   broken, networks supports are aware of chunks, not blocks 
   maybe other block-related functions should be censored in the same
   way ?
   tag block numbers are chunk numbers so they're not inadvertedly
   mistaken for each other ?
*)
let compute_block_num = ()
