io: io_set_alloc()

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>