Python版DNW

准备全线迁到Linux下做实验,虚拟机里编译太慢了。DNW是个问题。其实DNW到底是个什么东西我并不清楚,厂家提供的bootloader就是用DNW通过USB线接受PC传来数据。记得两年前就见过网友Linux版的DNW。那时没开始接触内核,觉得写这个东西的人真NB。现在把那个代码找出来看看,原来蛮简单的。一个内核态的驱动,一个用户态的程序。通过看用户态的程序,知道了DNW传输的格式,即4字节的下载地址+4字节的传输总长度+传输内容+2字节的checksum。checksum就是把前面三个部分以unsigned char的形式全部加起来。

后来找到了无需内核态驱动的DNW,借助libusb纯在用户态做。这个蛮不错,昨晚准备编一个试一下,结果装了libusb一还干了别的事,到头到也没试。今天又找到了一个xdnw,它用QT给DNW做了个UI,还是利用libusb纯在用户态下实现。为了试一下它还特意去装了QT4,看了下界面,很简陋。没有具体使用,不过应该问题不大。

想了半天,还是决定看能不能用Python实现一个。首先要找到能操纵USB的Python模块。有个叫PyUSB的东西一搜就搜到了。大致扫了一眼它的文档,觉得蛮简单的,于是开始着手实现。Python的一些东西还是很不熟悉,搞了几个小时,总算是搞出来了。不到40行代码,有效代码可能还没有30行,但确实折腾了两三个小时。偷了不少懒,错误处理都没做,而且直接假定第一个configuration的第一个interface的第一个setting的第二个endpoint是用来输出的。跟USB相关的代码只有四行。测试了一下,有时候是正常,有时候传输到的内存地址有误,比如要传到0x32000000结果终端上显示传到0x30000000。

有凑合用吧,以后再慢慢完善。用Python写的好处在于便是扩展。比如想加个界面,或者作为模块融入到更大的代码中,都是很容易的。

  1. #!/usr/bin/python   
  2. import usb.core  
  3. import StringIO  
  4. import struct  
  5. import sys  
  6.   
  7. if len(sys.argv) == 2 :  
  8.         addr = 0x32000000  
  9. elif len(sys.argv) == 3:  
  10.         addr = int(sys.argv[2], 16)  
  11. else:  
  12.         print 'Usage: dnw filename [address]'  
  13.         quit()  
  14.   
  15. def checksum(data):  
  16.         s = 0  
  17.         for i in data:  
  18.                 s += struct.unpack('B', i)[0];  
  19.         return s&0xffff  
  20.   
  21. content = open(sys.argv[1]).read()  
  22. address = struct.pack('L', addr)  
  23. length = struct.pack('L', len(content) + 10)  
  24. check = struct.pack('H', checksum(address + length + content))  
  25.   
  26. inp = StringIO.StringIO(address + length + content + check)  
  27.   
  28. dev = usb.core.find(idVendor=0x5345, idProduct=0x1234)  
  29. dev.set_configuration()  
  30. ep = dev[0][(0,0)][1]  
  31.   
  32. data = inp.read(512)  
  33. while data:  
  34.         ep.write(data)  
  35.         data = inp.read(512)  

相关推荐