src/networks/donkey/donkeyStats.ml

   1 (* Copyright 2001, 2002 b8_bavard, b8_fee_carabine, INRIA *)
   2 (*
   3     This file is part of mldonkey.
   4
   5     mldonkey is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     mldonkey is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with mldonkey; if not, write to the Free Software
  17     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18 *)
  19
  20 open Int64ops
  21 open Printf2
  22 open Options
  23 open CommonTypes
  24 open CommonOptions
  25 open CommonGlobals
  26 open CommonNetwork
  27 open CommonInteractive
  28 open CommonStats
  29
  30 open DonkeyOptions
  31 open DonkeyTypes
  32 open DonkeyGlobals
  33 open DonkeyComplexOptions
  34
  35 let start_session = ref (BasicSocket.last_time ())
  36
  37 let stats_array = Array.init brand_count (fun _ ->
  38     { dummy_stats with brand_seen = 0 }
  39   )
  40 let stats_mod_array = Array.init brand_mod_count (fun _ ->
  41     { dummy_stats with brand_seen = 0 }
  42   )
  43
  44 let count_seen c =
  45   let i = brand_to_int c.client_brand in
  46   stats_array.(i).brand_seen <- stats_array.(i).brand_seen + 1;
  47   !!gstats_array.(i).brand_seen <- !!gstats_array.(i).brand_seen + 1;
  48   check_client_country_code c;
  49   CommonStats.country_seen c.client_country_code;
  50
  51   if !!emule_mods_count then begin
  52     let i = brand_mod_to_int c.client_brand_mod in
  53     stats_mod_array.(i).brand_seen <- stats_mod_array.(i).brand_seen + 1;
  54     !!gstats_mod_array.(i).brand_seen <- !!gstats_mod_array.(i).brand_seen + 1
  55   end
  56
  57 let count_banned c =
  58   let i = brand_to_int c.client_brand in
  59   stats_array.(i).brand_banned <- stats_array.(i).brand_banned + 1;
  60   !!gstats_array.(i).brand_banned <- !!gstats_array.(i).brand_banned + 1;
  61
  62   if !!emule_mods_count then begin
  63     let i = brand_mod_to_int c.client_brand_mod in
  64     stats_mod_array.(i).brand_banned <- stats_mod_array.(i).brand_banned + 1;
  65     !!gstats_mod_array.(i).brand_banned <- !!gstats_mod_array.(i).brand_banned + 1
  66   end
  67
  68 let count_filerequest c =
  69   let i = brand_to_int c.client_brand in
  70   stats_array.(i).brand_filerequest <- stats_array.(i).brand_filerequest + 1;
  71   !!gstats_array.(i).brand_filerequest <- !!gstats_array.(i).brand_filerequest + 1;
  72
  73   if !!emule_mods_count then begin
  74     let i = brand_mod_to_int c.client_brand_mod in
  75     stats_mod_array.(i).brand_filerequest <- stats_mod_array.(i).brand_filerequest + 1;
  76     !!gstats_mod_array.(i).brand_filerequest <- !!gstats_mod_array.(i).brand_filerequest + 1
  77   end
  78
  79 let count_download c v =
  80   let i = brand_to_int c.client_brand in
  81   stats_array.(i).brand_download <- stats_array.(i).brand_download ++ v;
  82   !!gstats_array.(i).brand_download <- !!gstats_array.(i).brand_download ++ v;
  83
  84   if !!emule_mods_count then begin
  85     let i = brand_mod_to_int c.client_brand_mod in
  86     stats_mod_array.(i).brand_download <- stats_mod_array.(i).brand_download ++ v;
  87     !!gstats_mod_array.(i).brand_download <- !!gstats_mod_array.(i).brand_download ++ v;
  88   end;
  89
  90   c.client_total_downloaded <- c.client_total_downloaded ++ v;
  91   c.client_session_downloaded <- c.client_session_downloaded ++ v;
  92   donkey_download_counter := !donkey_download_counter ++ v;
  93   check_client_country_code c;
  94   global_count_download network c.client_country_code v
  95
  96 let count_upload c v =
  97   let i = brand_to_int c.client_brand in
  98   stats_array.(i).brand_upload <- stats_array.(i).brand_upload ++ v;
  99   !!gstats_array.(i).brand_upload <- !!gstats_array.(i).brand_upload ++ v;
 100
 101   if !!emule_mods_count then begin
 102     let i = brand_mod_to_int c.client_brand_mod in
 103     stats_mod_array.(i).brand_upload <- stats_mod_array.(i).brand_upload ++ v;
 104     !!gstats_mod_array.(i).brand_upload <- !!gstats_mod_array.(i).brand_upload ++ v;
 105   end;
 106
 107   c.client_total_uploaded <- c.client_total_uploaded ++ v;
 108   c.client_session_uploaded <- c.client_session_uploaded ++ v;
 109   donkey_upload_counter := !donkey_upload_counter ++ v;
 110   check_client_country_code c;
 111   global_count_upload network c.client_country_code v
 112
 113 let print_stats_mods o style =
 114   let buf = o.conn_buf in
 115
 116   let u1 = BasicSocket.last_time () - !start_session in
 117   let u2 = (guptime() + u1) in
 118
 119   let t1 = build_title "eMule Mods" "Session" u1 in
 120   let t2 = build_title "eMule Mods" "Total" u2 in
 121
 122   let l = brand_mod_list in
 123
 124   match style with
 125     | Old -> print_stats_old buf stats_mod_array l t1 u1
 126     | _ ->
 127       if !!emule_mods_count then begin
 128
 129         if use_html_mods o then begin
 130           print_stats_html_mods buf stats_mod_array l t1 u1;
 131           print_stats_html_mods buf !!gstats_mod_array l t2 u2
 132         end else begin
 133           print_stats_ascii buf stats_mod_array l t1 u1;
 134           print_stats_ascii buf !!gstats_mod_array l t2 u2
 135         end
 136
 137       end else
 138       begin
 139
 140         if use_html_mods o then begin
 141           Printf.bprintf buf "\\<div class=\\\"cs\\\"\\>";
 142           html_mods_table_header buf "emodsTable" "cs" [];
 143           Printf.bprintf buf "\\<tr class=\\\"dl-1\\\"\\>";
 144           html_mods_td buf [
 145               ("", "sr", "eMule mods statistics are disabled, to activate set emule_mods_count true in the \\<a href=\\\"/submit?q=voo+8\\\"\\>misc options\\</a\\> tab." );
 146           ];
 147           Printf.bprintf buf "\\</tr\\>\\</table\\>\\</div\\>\\</div\\>\n"
 148         end else
 149           Printf.bprintf buf "eMule mods statistics are disabled, to activate set emule_mods_count true \n"
 150
 151       end
 152
 153 let print_stats o style mods =
 154   let buf = o.conn_buf in
 155
 156   let u1 = BasicSocket.last_time () - !start_session in
 157   let u2 = (guptime() + u1) in
 158
 159   let t1 = build_title "eDonkey" "Session" u1 in
 160   let t2 = build_title "eDonkey" "Total" u2 in
 161
 162   let l = brand_list in
 163
 164   if mods then print_stats_mods o style else
 165     match style with
 166     | Old -> print_stats_old buf stats_array l t1 u1
 167     | _ ->
 168           if use_html_mods o then begin
 169             print_stats_html_mods buf stats_array l t1 u1;
 170             print_stats_html_mods buf !!gstats_array l t2 u2
 171           end else begin
 172             print_stats_ascii buf stats_array l t1 u1;
 173             print_stats_ascii buf !!gstats_array l t2 u2
 174           end
 175
 176 let _ =
 177   network.op_network_display_stats <- (fun o -> print_stats o New false);
 178
 179   network.op_network_stat_info_list <- (fun _ ->
 180     let r = ref [] in
 181     let l1 = stats_list brand_list stats_array in
 182     let l2 = stats_list brand_list !!gstats_array in
 183     let u1 = BasicSocket.last_time () - !start_session in
 184     let u2 = (guptime() + u1) in
 185     r := [("Session clients", u1, l1); ("Global clients", u2, l2)];
 186     if !!emule_mods_count then begin
 187       let l3 = stats_list brand_mod_list stats_mod_array in
 188       let l4 = stats_list brand_mod_list !!gstats_mod_array in
 189       r := !r @ [ ("Session mods", u1, l3); ("Global mods", u2, l4)]
 190     end;
 191     !r
 192   );
 193
 194   register_commands
 195     [
 196     "client_stats", "Network/Donkey",Arg_none (fun o ->
 197         print_stats o Old false;
 198         ""
 199     ), ":\t\t\t\tshow breakdown of download/upload by clients brand";
 200
 201     "cs", "Network/Donkey",Arg_none (fun o ->
 202         print_stats o New false;
 203         ""
 204     ), ":\t\t\t\t\tshow table of download/upload by ED2K clients brand";
 205
 206     "csm", "Network/Donkey",Arg_none (fun o ->
 207         print_stats o New true;
 208         ""
 209     ), ":\t\t\t\t\tshow table of download/upload by eMule MODs";
 210     "reset_stats", "Network/Donkey",Arg_none (fun o ->
 211
 212        Array.iteri  (fun x _ ->
 213          stats_array.(x).brand_seen <- 0;
 214          stats_array.(x).brand_banned <- 0;
 215          stats_array.(x).brand_filerequest <- 0;
 216          stats_array.(x).brand_download <- 0L;
 217          stats_array.(x).brand_upload <- 0L;
 218        ) stats_array;
 219
 220        Array.iteri (fun x _ ->
 221          stats_mod_array.(x).brand_seen <- 0;
 222          stats_mod_array.(x).brand_banned <- 0;
 223          stats_mod_array.(x).brand_filerequest <- 0;
 224          stats_mod_array.(x).brand_download <- Int64.zero;
 225          stats_mod_array.(x).brand_upload <- Int64.zero
 226        ) stats_mod_array;
 227
 228        start_session := BasicSocket.last_time ();
 229
 230        "done"
 231     ), ":\t\t\t\treset session statistics";
 232   ]
 233
 234 (*
 235 let save_download_history file =
 236
 237   let buf = Buffer.create 100 in
 238
 239 (* Some opcode for edonkey downloads *)
 240   buf_int8 buf 153;         (* opcode = stats *)
 241   buf_md4 buf !!client_md4; (* client md4, and NOT IP *)
 242
 243   let time = Unix.time () in
 244   let time = Unix.localtime time in
 245   (* date on 5 bytes *)
 246   buf_int8 buf time.Unix.tm_hour;
 247   buf_int8 buf time.Unix.tm_mday;
 248   buf_int8 buf time.Unix.tm_mon;
 249   buf_int16 buf time.Unix.tm_year;
 250 (* ANONYMISED Informations on the downloads: *)
 251 (*   Send the SHA1 hash of the MD4+size, so that they cannot be recovered,
 252           but they can be used to compare downloads. *)
 253   let m = Printf.sprintf "%s%Ld"
 254       (Md4.Md4.direct_to_string file.file_md4) (file_size file) in
 255   let m = Md4.Sha1.string m in (* compute SHA1 of the string *)
 256 (* SENT: 20 bytes *)
 257   Buffer.add_string buf (Md4.Sha1.direct_to_string m);
 258
 259 (* Send the magnitude of the size (power of 10) *)
 260   let size = ref (file_size file) in
 261   let m = ref 0 in
 262   while !size <> Int64.zero do
 263     incr m;
 264     size := !size // 10L
 265   done;
 266 (* SENT: 1 byte *)
 267   buf_int8 buf !m;
 268
 269   let current = ref [] in
 270
 271   Intmap.iter (fun _ c ->
 272       let location =
 273         match c.client_kind with
 274           Indirect_location (name, md4) ->
 275             Printf.sprintf "%s%s" name
 276               (Md4.Md4.direct_to_string md4)
 277         | Known_location (ip,port) ->
 278             Printf.sprintf "%s%d"
 279               (Ip.to_string ip) port
 280       in
 281
 282 (* ANONYMISATION of the source: we compute the Sha1 digest of the source,
 283   which cannot be recovered from this information *)
 284       let location = Md4.Sha1.string location in (* compute SHA1 of the string *)
 285       current := location :: !current;
 286   ) file.file_locations;
 287
 288   buf_list (fun buf s ->
 289       Buffer.add_string buf (Md4.Sha1.direct_to_string s)
 290   ) buf !current;
 291
 292
 293   let file_history = "downloads.stats" in
 294   let oc = append_out file_history in
 295   output_string oc (Buffer.contents buf);
 296   close_out oc
 297 *)