#
# This file is part of ruby-ffi.
# For licensing, see LICENSE.SPECS
#

require File.expand_path(File.join(File.dirname(__FILE__), "spec_helper"))
describe "Custom type definitions" do
  it "attach_function with custom typedef" do
    module CustomTypedef
      extend FFI::Library
      ffi_lib TestLibrary::PATH
      typedef :uint, :fubar_t
      attach_function :ret_u32, [ :fubar_t ], :fubar_t
    end
    expect(CustomTypedef.ret_u32(0x12345678)).to eq(0x12345678)
  end

  it "variadic invoker with custom typedef" do
    module VariadicCustomTypedef
      extend FFI::Library
      ffi_lib TestLibrary::PATH
      typedef :uint, :fubar_t
      attach_function :pack_varargs, [ :buffer_out, :string, :varargs ], :void
    end
    buf = FFI::Buffer.new :uint, 10
    VariadicCustomTypedef.pack_varargs(buf, "i", :fubar_t, 0x12345678)
    expect(buf.get_int64(0)).to eq(0x12345678)
  end

  it "Callback with custom typedef parameter" do
    module CallbackCustomTypedef
      extend FFI::Library
      ffi_lib TestLibrary::PATH
      typedef :uint, :fubar3_t
      callback :cbIrV, [ :fubar3_t ], :void
      attach_function :testCallbackU32rV, :testClosureIrV, [ :cbIrV, :fubar3_t ], :void
    end
    i = 0
    CallbackCustomTypedef.testCallbackU32rV(0xdeadbeef) { |v| i = v }
    expect(i).to eq(0xdeadbeef)
  end
    module StructCustomTypedef
      extend FFI::Library
      ffi_lib TestLibrary::PATH
      typedef :uint, :fubar3_t
      class S < FFI::Struct
        layout :a, :fubar3_t
      end
    end

  it "Struct with custom typedef field" do
    s = StructCustomTypedef::S.new
    s[:a] = 0x12345678
    expect(s.pointer.get_uint(0)).to eq(0x12345678)
  end

  it "attach_function after a typedef should not reject normal types" do
    expect do
      Module.new do
        extend FFI::Library
        # enum() will insert a custom typedef called :foo for the enum
        enum :foo, [ :a, :b ]
        typedef :int, :bar

        ffi_lib TestLibrary::PATH
        begin
          attach_function :ptr_ret_int32_t, [ :string, :foo ], :bar
        rescue FFI::NotFoundError
          # NetBSD uses #define instead of typedef for these
          attach_function :ptr_ret_int32_t, :ptr_ret___int32_t, [ :string, :foo ], :bar
        end
      end
    end.not_to raise_error
  end

  it "detects the correct type for size_t" do
    expect do
      Module.new do
        extend FFI::Library
        ffi_lib "c"
        if FFI::Platform.windows?
          # _read() is a function of msvcrt.dll
          attach_function :_read, [:int, :pointer, :uint], :int
        else
          # read(2) is a standard UNIX function
          attach_function :read, [:int, :pointer, :size_t], :ssize_t
        end
      end
    end.not_to raise_error
  end
end
