Octave下操作CH341

#include <octave/oct.h>
#include <windows.h>
#include <cstdint>
#include <fstream>
#include <dMatrix.h>
#include <int32NDArray.h>

// CH341DLL Functions Prototype
typedef HANDLE WINAPI(*CH341OpenDevice_fn)(unsigned int iIndex);

typedef void (*CH341CloseDevice_fn)(unsigned int iIndex);

typedef BOOL WINAPI(*CH341SetStream_fn)(
    unsigned int     iIndex,
    unsigned int     iMode);

typedef BOOL WINAPI(*CH341StreamI2C_fn)(
    unsigned int    iIndex,
    unsigned int    iWriteLength,
    void *            iWriteBuffer,
    unsigned int    iReadLength,
    void *            oReadBuffer);

// Octave function declaration
DEFUN_DLD(i2c_rd, args, nargout, "USB I2C Read")
{
    int32NDArray bytes;
    size_t bytes_read;

    if(args.length() < 2) {
        octave_stdout << "
  --i2c_rd I2C_ADR REG_ADR [Bytes to Read]
" << std::endl;
        return octave_value_list();
    } else if(args.length() == 3) {
        bytes = args(2).matrix_value();
        bytes_read = bytes(0, 0);
    } else {
        bytes(0, 0) = 1;
        bytes_read = 1;
    }
    
    // Load CH341DLL to Current Process
    HMODULE hdll = LoadLibrary("CH341DLL.DLL");

    if(NULL == hdll) {
        octave_stdout <<"USB_I2C CH341DLL Load Failed!" << std::endl;
        return octave_value_list();
    }

    // Get CH341 Functions Handle
    CH341OpenDevice_fn  USBIO_OpenDevice    = (CH341OpenDevice_fn)    GetProcAddress(hdll, "CH341OpenDevice");
    CH341CloseDevice_fn USBIO_CloseDevice    = (CH341CloseDevice_fn)    GetProcAddress(hdll, "CH341CloseDevice");
    CH341SetStream_fn   USBIO_SetStream        = (CH341SetStream_fn)    GetProcAddress(hdll, "CH341SetStream");
    CH341StreamI2C_fn   USBIO_StreamI2C        = (CH341StreamI2C_fn)    GetProcAddress(hdll, "CH341StreamI2C");

    int32NDArray i2c_adr = args(0).int32_array_value();
    int32NDArray reg_adr = args(1).int32_array_value();

    Matrix result(1, bytes_read);
    uint8_t IBuf[bytes_read];
    uint8_t OBuf[2] = {uint8_t(i2c_adr(0, 0)), uint8_t(reg_adr(0, 0))};

    // Open CH341 Device
    HANDLE devh = USBIO_OpenDevice(0);

    if(devh == NULL) {
        octave_stdout << "CH341 Open Failed!" << std::endl;
        FreeLibrary(hdll);
        return octave_value_list();
    }
    
    if(!USBIO_SetStream(0, 0x82)) {
        octave_stdout << "CH341 SetStream Failed!" << std::endl;
        USBIO_CloseDevice(0);
        FreeLibrary(hdll);
        return octave_value_list();
    }

    // If number of bytes to read is not given.
    if(!USBIO_StreamI2C(0, 2, OBuf, bytes_read, IBuf)) {
        octave_stdout << "CH341 Read Failed!" << std::endl;    
    } else {
        for(size_t i = 0; i < bytes_read; ++i) {
            result(0, i) = double(IBuf[i]);
        }
    }
    
    USBIO_CloseDevice(0);
    FreeLibrary(hdll);
    return octave_value(result);
}