test.ml

module M = Mpl_stdlib
open Openflow_packet

let path_of_switch name = Printf.sprintf "/var/run/openvswitch/%s.mgmt" name

let with_switch name f =
    let fd = Unix.socket Unix.PF_UNIX Unix.SOCK_STREAM 0 in
    Pervasiveext.finally 
	(fun () ->
	     let sa = Unix.ADDR_UNIX (path_of_switch "xenbr0") in
	     Unix.connect fd sa;
	     f fd
	)
    	(fun () -> Unix.close fd)


let hello env = 
	let (_: HELLO.o) = HELLO.t ~version:1 ~xid:0l ~data:(`Str "") env in ()

let vendor env = 
	let xensource_oui = 0x163el in (* an example *)
	let (_: VENDOR.o) = VENDOR.t ~version:1 ~xid:0l ~vendor:xensource_oui ~data:(`Str "") env in ()

let features_request env = 
	let (_: FEATURES_REQUEST.o) = FEATURES_REQUEST.t ~version:1 ~xid:0l ~data:(`Str "") env in ()

let get_config_request env = 
	let (_: GET_CONFIG_REQUEST.o) = GET_CONFIG_REQUEST.t ~version:1 ~xid:0l env in ()

let set_config env = 
	let (_: SET_CONFIG.o) = SET_CONFIG.t ~version:1 ~xid:0l ~reasm:0 ~drop:0 ~miss_send_len:0 env in ()

let stats_request env = 
	let (_: STATS_REQUEST.o) = STATS_REQUEST.t ~version:1 ~xid:0l ~req_ty:`DESCR env in ()

let features_reply x = match x with 
|`FEATURES_REPLY o ->
	FEATURES_REPLY.prettyprint o;
	M.fold_env o#data_env
		(fun env () ->
			let o = Openflow_phy_port.unmarshal env in
			Openflow_phy_port.prettyprint o;
		) ()

| _ -> failwith "Not a FEATURES_REPLY"

let stats_reply x = match x with 
|`STATS_REPLY o ->
	STATS_REPLY.prettyprint o;
	M.fold_env o#data_env
		(fun env () ->
			let o = Openflow_desc_stats.unmarshal env in
			Openflow_desc_stats.prettyprint o;
		) ()
| _ -> failwith "Not a STATS_REPLY"

let send f env fd = 
      M.reset env;     
      Printf.printf "send()\n"; flush stdout;
      f env;
      M.flush env fd;
      Printf.printf "  OK\n"; flush stdout

open Openflow_packet
let recv env fd handler = 
      Printf.printf "recv()\n"; flush stdout;
      M.reset env;
      M.fill env fd;
      let o = unmarshal env in
	begin
		Printf.printf "  %s\n" (match o with `ERROR _ -> "ERROR" | _ -> "OK");
		flush stdout;
	end;
      handler o;
      o

let _ =
    let env = M.new_env (String.make 16384 '\000') in
    let senv = M.new_env (String.make 1000 '\000') in

    with_switch "xenbr0"
    (fun fd ->
      Printf.printf "HELLO\n"; flush stdout;
      send hello senv fd;
      ignore(recv env fd prettyprint);
      Printf.printf "VENDOR\n"; flush stdout;
      send vendor senv fd;
      ignore(recv env fd prettyprint);
      Printf.printf "FEATURES_REQUEST\n"; flush stdout;
      send features_request senv fd;
      ignore(recv env fd features_reply);
      Printf.printf "GET_CONFIG_REQUEST\n"; flush stdout;
      send get_config_request senv fd;
      ignore(recv env fd prettyprint);
      Printf.printf "STATS_REQUEST\n"; flush stdout;
      send stats_request senv fd;
      ignore(recv env fd stats_reply);
      Printf.printf "SET_CONFIG\n"; flush stdout;
      send set_config senv fd;
      (* SET_CONFIG has no response. We make sure the connection hasn't lost
	sync by sending any message which elicits a response. *)
      send stats_request senv fd;
      ignore(recv env fd stats_reply);
      Printf.printf "Done\n";
    )