libusb in windows

libusb in windows

最早接触 libusb 当然是在linux下面。 原来玩2440开发板的时候，网上有人写了一个基于libusb的工具，2440工作在device 模式下，我记得命令是usb slave.

然后通过这个工具将电脑上的镜像传到板子上，非常方便。

公司的项目正好有usb boot mode.这对裸机来说非常有帮助。 于是把握手机制加入以后，居然能够成功的上传镜像。

后来，我发现libusb有windows的版本，于是就把这个工具稍作修改，并且在vs2010上编译通过，并且也能够工作。

下面是代码：

#include <stdio.h>
#include <string.h>

#include <conio.h>

//#include <usb.h>
#include <errno.h>
#include <sys/stat.h>
#include <fcntl.h>
//#include "dbt.h"

#include "lusb0_usb.h"

#define DEVICE_SECBULK_IDVENDOR  0x3252
#define DEVICE_SECBULK_IDPRODUCT 0x0004

struct usb_device_a6
{
 usb_dev_handle *hdev;
 struct usb_device *dev;
 char *filebuf;
 int filelen;
 char usb_name[MAX_PATH];

};
struct usb_device_a6 a6_usb;

char buf_pr[512];

int usb_init_search()
{
 struct usb_bus *busses, *bus;
 struct usb_device *dev;
 usb_init();
 usb_find_busses();
 usb_find_devices();
 for (bus = usb_get_busses(); bus; bus = bus->next)
 {
   for (dev = bus->devices; dev; dev = dev->next)
   {
    printf("idVendor:0x%x\t,ipProduct:0x%x\n",
      dev->descriptor.idVendor, dev->descriptor.idProduct);

    if (
     ((DEVICE_SECBULK_IDVENDOR == dev->descriptor.idVendor
     && DEVICE_SECBULK_IDPRODUCT == dev->descriptor.idProduct) )
    )
    {
     a6_usb.dev = dev;
     printf("Target usb device found!\n");
     return 0;
    }

   }
 }
 printf("Target usb device not found!\n");
 return -1;
}

int usb_open_port()
{
 a6_usb.hdev = usb_open(a6_usb.dev);
 if (!a6_usb.hdev)
 {
  return -1;
 }
 else
 {
  if (0 != usb_claim_interface(a6_usb.hdev, 0))
  {
   perror("Cannot claim interface");
   usb_close(a6_usb.hdev);
   a6_usb.hdev = NULL;
   return -1;
  }
 }
 return 0;
}

char* prepare_write_buf(char *filename, unsigned int *len) {
 char *write_buf = NULL;
 int ret, ret2=0;
 FILE *pfile;
 int sz;

 pfile=fopen(filename,"rb");
 if(!pfile) {
  printf("open file err\n");
  return NULL;
 }
 fseek(pfile, 0L, SEEK_END);
 sz = ftell(pfile);
 fseek(pfile, 0L, SEEK_SET);

 write_buf = (char*) malloc(sz);
 if (NULL == write_buf) {
  perror("malloc failed");
  goto error;
 }

 ret = fread(write_buf, 1, sz, pfile);
 printf("---Reading file  %d\n", ret);
 
 while(ret > 0) {
  ret2 += ret;
  ret=fread(write_buf+ret2,1, sz, pfile);
  printf("---Reading file  %d %d\n", ret,ret2);
 }
 if (sz != ret2) {
  printf("---Reading file failed %d %d\n", sz, ret2);
  goto error;
 }

 printf("Filename : %s\n", filename);
 printf("Filesize : %d bytes\n", (int)sz);
 *len = sz;
 return write_buf;

error:
  fclose(pfile);
    if (NULL != write_buf)
     free(write_buf);
    sz = 0;
    return NULL;
}

int usb_read_device(int *len)
{
    char readbuf[32];
 unsigned int *magic=(unsigned int *)readbuf;
 int i;
 int wantread;
 memset(readbuf, 0 , sizeof(readbuf));
 wantread = usb_bulk_read(a6_usb.hdev, 0x82, readbuf, 16, 5000);

 for(i=0;i<16;i++)
  printf("%02x ", readbuf[i]);
 printf("\n0x%04x\n",*magic);

 if(*magic == 0x22c3b2a1) {
  *len=*(magic+2);
  printf("\n0x%04x\n",*(magic+2));

  return 0;
 }
 else
  return -1;

}

int usb_write_device(unsigned int writelen, char* buf )
{
 unsigned int remain = writelen;
 unsigned int towrite;
 while (remain)
 {
  towrite = remain > 256 ? 256 : remain;
  if (towrite != usb_bulk_write(a6_usb.hdev, 0x2, buf + (writelen - remain),
     towrite, 3000)) {
   perror("usb_bulk_write failed");
   break;
  }
  remain -= towrite;
  printf("\r%d%\t %d bytes     ", (writelen - remain) * 100 / writelen, writelen-remain);
  fflush(stdout);
 }

 return remain;

}

int loadfile_A6(char *f)
{
 unsigned int remain;
 unsigned int writed=0;
 unsigned int retwrite;
 int asklen = 0;
 unsigned int len;
 char *buf;
 //todo: add the logic here for getting file path and name
 a6_usb.filebuf = prepare_write_buf(f, &len);
 a6_usb.filelen = len;
 
 if(!a6_usb.filebuf)
 {
  return -1;
 }
 
 buf = a6_usb.filebuf;
 remain = a6_usb.filelen;

 for(;;) {
  Sleep(200);
  if(writed == a6_usb.filelen)
  {
   printf("Done!\n");
   return 0;
  }

  if(usb_read_device(&asklen) == 0) {
   if(asklen > 0) {
    retwrite=usb_write_device(asklen, a6_usb.filebuf+writed);
    if(retwrite > 0)
     break;
    writed += asklen;
   }
  }

 }
 return -1;
}

int scan_enable_a6()
{
 if(usb_init_search() < 0)
  return -1;
 if(usb_open_port() < 0)
  return -1;
 return 0;
}

int main(int argc, char *argv[])
{
 if(scan_enable_a6() == 0)
 loadfile_A6(argv[1]);
 
}