tests/test_ocaml_errorcodes_plugin.ml

   1 (* nbdkit
   2  * Copyright Red Hat
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  * * Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *
  11  * * Redistributions in binary form must reproduce the above copyright
  12  * notice, this list of conditions and the following disclaimer in the
  13  * documentation and/or other materials provided with the distribution.
  14  *
  15  * * Neither the name of Red Hat nor the names of its contributors may be
  16  * used to endorse or promote products derived from this software without
  17  * specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY RED HAT AND CONTRIBUTORS ''AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  21  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  22  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RED HAT OR
  23  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  26  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  27  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  28  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  29  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  *)
  32
  33 open Unix
  34
  35 let sector_size = 512
  36
  37 let open_connection _ = ()
  38
  39 let get_size () = Int64.of_int (6 * sector_size)
  40
  41 let pread () count offset _ =
  42   (* Depending on the sector requested (offset), return a different
  43    * error code.
  44    *)
  45   match (Int64.to_int offset) / sector_size with
  46   | 0 -> (* good, return data *) String.make count '\000'
  47   | 1 -> NBDKit.set_error EPERM;     failwith "EPERM"
  48   | 2 -> NBDKit.set_error EIO;       failwith "EIO"
  49   | 3 -> NBDKit.set_error ENOMEM;    failwith "ENOMEM"
  50   | 4 -> NBDKit.set_error ESHUTDOWN; failwith "ESHUTDOWN"
  51   | 5 -> NBDKit.set_error EINVAL;    failwith "EINVAL"
  52   | _ -> assert false
  53
  54 let () =
  55   NBDKit.register_plugin
  56     ~name:    "test-ocaml-errorcodes"
  57     ~version: (NBDKit.version ())
  58
  59     ~open_connection
  60     ~get_size
  61     ~pread
  62     ()