Source

sks-keyserver-with-async / add_mail.ml

Full commit
(***********************************************************************)
(* add_mail.ml - Executable: interprets stdin as mail message and      *)
(*               posts content to specified HTTP address               *)
(*                                                                     *)
(* Copyright (C) 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010, *)
(*               2011, 2012, 2013  Yaron Minsky and Contributors       *)
(*                                                                     *)
(* This file is part of SKS.  SKS 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.    *)
(*                                                                     *)
(* This program 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 this program; if not, write to the Free Software         *)
(* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *)
(* USA or see <http://www.gnu.org/licenses/>.                          *)
(***********************************************************************)

open Core.Std

(** Argument parsing *)

let anonymous = ref []

let usage_string =
  Sys.argv.(0) ^ " sks_directory_name"

let anon_options option =
  anonymous := option::!anonymous

let parse_spec = [ ]

let dirname =
  Arg.parse parse_spec anon_options usage_string;
  match !anonymous with
  | [x] -> Filename.concat x "messages"
  | _ ->
    printf "Wrong number (%d) of arguments given.  %s\n"
      (List.length !anonymous)
      usage_string;
    exit (-1)

(** dumps contents of one file into another *)
let pipe_file =
  let blocksize = 100 * 1024 in
  let buf = String.create blocksize in
  let rec pipe_file file1 file2 =
    let bytes_read = input file1 buf 0 blocksize in
    if bytes_read <> 0 then (
      output file2 buf 0 bytes_read;
      pipe_file file1 file2
    )
  in
  pipe_file

let run () =
  (* create the directory if it doesn't exist *)
  begin match Sys.file_exists dirname with
  | `No -> Unix.mkdir dirname ~perm:0o700;
  | `Yes -> ()
  | `Unknown ->
    failwithf "existence check of directory <%s> failed" dirname ()
  end;
  let fname = sprintf "msg-%08d" (Random.int 100000000) in
  let fname = Filename.concat dirname fname in
  let f = open_out fname in
  pipe_file stdin f;
  Out_channel.close f;
  Sys.rename fname (fname ^ ".ready")

let () =
  Random.self_init ();
  run ()