123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353 |
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <usb.h>
- #include <unistd.h>
- #include <getopt.h>
- #include "opendevice.h"
- #include "qdips.h"
- #define _16MBIT 0x200000
- #define HIROM_BANK_SIZE_SHIFT 16
- #define LOROM_BANK_SIZE_SHIFT 15
- #define HIROM_BANK_COUNT_SHIFT 32
- #define LOROM_BANK_COUNT_SHIFT 64
- #define IPS_MAGIC "PATCH"
- #define EOF_TAG 0x454f46
- #define PATCH_END 1
- #define PATCH_ERR 2
- #define PATCH_CHUNKS_MAX 2048
- #define RLE_LEN 0
- #define RLE_VAL 1
- #define QD16_MAX_TX_LEN 128
- #define QD16_CMD_TIMEOUT 5000
- #define QD16_SEND_DATA(__addr_hi,__addr_lo,__src,__len)\
- usb_control_msg(handle,\
- USB_TYPE_VENDOR|USB_RECIP_DEVICE|USB_ENDPOINT_OUT,\
- USB_BULK_UPLOAD_NEXT,\
- __addr_hi, __addr_lo,\
- (char*)__src, __len,\
- QD16_CMD_TIMEOUT);
- #define QD16_SEND_ADDR(__addr_hi,__addr_lo,__src,__len)\
- usb_control_msg(handle,\
- USB_TYPE_VENDOR|USB_RECIP_DEVICE|USB_ENDPOINT_OUT,\
- USB_BULK_UPLOAD_ADDR,\
- __addr_hi,__addr_lo,\
- (char*)__src,__len,\
- QD16_CMD_TIMEOUT);
- typedef struct _ips_tag {
- uint32_t ofs; /* offset in file to patch (3 bytes (24bit address) in file!) */
- uint16_t len; /* number of bytes in this chunk (if len==0 : data[0] = counter / data[1] = fill byte*/
- uint8_t* data; /* chunk data */
- } ips_tag_t;
- int qd16_transfer_data(int addr_hi, int addr_lo, uint8_t* data, int len);
- int qd16_apply_ips_chunk(ips_tag_t* ips_tag);
- int ips_read_next_tag(FILE* patch_file, ips_tag_t* tag);
- int qd16_init(void);
- void qd16_finish(void);
- void usage(void);
- usb_dev_handle *handle = NULL;
- const unsigned char rawVid[2] = { USB_CFG_VENDOR_ID }, rawPid[2] = { USB_CFG_DEVICE_ID};
- char vendor[] = { USB_CFG_VENDOR_NAME, 0 }, product[] = { USB_CFG_DEVICE_NAME, 0};
- int cnt, vid, pid;
- int hirom = 0;
- int quiet = 0;
- uint8_t dummy[128];
- void usage(void)
- {
- fprintf(stderr, "Usage: qdips [option] <patchfile>\n\n\
- Options:\n\
- -q\tBe quiet\n\
- -h\tForce HiROM mode\n\n");
- exit(1);
- }
- int main(int argc, char * argv[])
- {
- FILE* patch_file;
- char * fname = 0;
- int opt = 0;
- char magic[6];
- ips_tag_t ips_tags[PATCH_CHUNKS_MAX];
- int chunk_index = 0;
- if (argc==1)
- usage();
- while ((opt = getopt(argc,argv,"qh")) != -1){
- switch (opt) {
- case 'h':
- hirom = 1;
- break;
- case 'q':
- quiet = 1;
- break;
- default:
- usage();
- }
- }
- if (optind >= argc) {
- usage();
- }
- int ret = qd16_init();
- if (ret != 0){
- fprintf(stderr,"Error during init. Exiting!\n");
- exit(1);
- }
- fname = argv[optind];
- patch_file = fopen(fname,"rb");
- if (patch_file == 0) {
- fprintf(stderr,"Failed to open file '%s'!\n",fname);
- exit(1);
- }
- if (fgets(magic,strlen(IPS_MAGIC)+1,patch_file) == NULL) {
- fprintf(stderr,"Error reading from file %s\n",fname);
- fclose(patch_file);
- exit(1);
- }
- if (strcmp(magic,IPS_MAGIC) != 0){
- fprintf(stderr,"Patch is not in ips format!\n");
- }
- int patch_error = 0;
- while (!feof(patch_file) && chunk_index < PATCH_CHUNKS_MAX) {
- int ret = ips_read_next_tag(patch_file,&ips_tags[chunk_index]);
- if (ret == PATCH_END)
- break;
- else if (ret == PATCH_ERR) {
- patch_error = 1;
- break;
- }
- chunk_index++;
- }
- fclose(patch_file);
- if (patch_error){
- printf("error at patch chunk #%d!\n",chunk_index);
- }
- int errors=0;
- for (int i=0; i<chunk_index; i++){
- if (!quiet)
- printf("uploading chunk %d ..\r",i+1);
- fflush(stdout);
- int ret = qd16_apply_ips_chunk(&ips_tags[i]);
- if (ret!=0){
- printf("\nfailed applying chunk %d !\n",i);
- errors++;
- }
- }
- if (!quiet)
- printf("\nsuccessfully applied %d chunks\n",chunk_index-errors);
- qd16_finish();
- return 0;
- }
- int qd16_transfer_data(int addr_hi, int addr_lo, uint8_t* data, int len)
- {
- int addr = (addr_hi<<16) | (addr_lo&0xffff);
- int pos = 0;
- int ret;
- int bytes_left = len;
- int txlen;
- while (bytes_left>0) {
- txlen = (bytes_left>QD16_MAX_TX_LEN?QD16_MAX_TX_LEN:bytes_left);
- ret = QD16_SEND_ADDR( ((addr>>16)&0xff), (addr&0xffff), data+pos, txlen);
- if (ret<0)
- break;
- addr += txlen;
- pos += txlen;
- bytes_left -= txlen;
- }
- if (bytes_left){
- printf("%s\n",usb_strerror());
- return 1;
- }
- return 0;
- }
- int qd16_apply_ips_chunk(ips_tag_t* ips_tag)
- {
- int addr_hi = (ips_tag->ofs)>>16;
- int addr_lo = (ips_tag->ofs)&0xffff;
- if (ips_tag->len > 0){
- int ret = qd16_transfer_data(addr_hi,addr_lo,ips_tag->data,ips_tag->len);
- if (ret!=0)
- printf("error transfering data to qd16\n");
- } else {
- uint16_t len = (uint16_t)ips_tag->data[RLE_LEN];
- uint8_t val = ips_tag->data[RLE_VAL];
- // prepare patch buffer
- uint8_t *buff = malloc(len);
- memset(buff,val,len);
- // transfer
- int ret = QD16_SEND_ADDR(addr_hi,addr_lo,buff,len);
- if (ret<0)
- printf("error executing command: %s\n",usb_strerror());
- }
- return 0;
- }
- void qd16_finish(void)
- {
- cnt = usb_control_msg(handle,
- USB_TYPE_VENDOR | USB_RECIP_DEVICE |
- USB_ENDPOINT_OUT, USB_BULK_UPLOAD_END, 0, 0, NULL,
- 0, 5000);
- if (cnt<0) {
- printf("USB_BULK_UPLOAD_END failed: %s\n",usb_strerror());
- }
- cnt = usb_control_msg(handle,
- USB_TYPE_VENDOR | USB_RECIP_DEVICE |
- USB_ENDPOINT_OUT, USB_SET_LOADER, 1, 1, NULL,
- 0, 5000);
- cnt = usb_control_msg(handle,
- USB_TYPE_VENDOR | USB_RECIP_DEVICE |
- USB_ENDPOINT_OUT, USB_MODE_SNES, 0, 0, NULL,
- 0, 5000);
- if (cnt<0) {
- printf("USB_MODE_SNES failed: %s\n",usb_strerror());
- }
- }
- int qd16_init(void)
- {
- usb_init();
- vid = rawVid[1] * 256 + rawVid[0];
- pid = rawPid[1] * 256 + rawPid[0];
- if (usbOpenDevice(&handle, vid, vendor, pid, product, NULL, NULL, NULL) != 0) {
- fprintf(stderr, "Could not find USB device \"%s\" with vid=0x%x pid=0x%x\n", product, vid, pid);
- return 1;
- }
- cnt = usb_control_msg(handle,
- USB_TYPE_VENDOR | USB_RECIP_DEVICE |
- USB_ENDPOINT_OUT, USB_SET_LOADER, 0, 0, NULL,
- 0, 5000);
- if (cnt<0) {
- printf("USB_SET_LOADER failed: %s\n",usb_strerror());
- }
- cnt = usb_control_msg(handle,
- USB_TYPE_VENDOR | USB_RECIP_DEVICE |
- USB_ENDPOINT_OUT, USB_MODE_AVR, 0, 0, NULL,
- 0, 5000);
- if (cnt<0) {
- printf("USB_MODE_AVR failed: %s\n",usb_strerror());
- }
- usleep(500000);
- cnt = usb_control_msg(handle,
- USB_TYPE_VENDOR | USB_RECIP_DEVICE | USB_ENDPOINT_OUT,
- USB_BULK_UPLOAD_INIT,
- (hirom ? HIROM_BANK_SIZE_SHIFT : LOROM_BANK_SIZE_SHIFT) ,
- /*(hirom ? HIROM_BANK_COUNT_SHIFT : LOROM_BANK_COUNT_SHIFT), */ // ???
- 5,
- NULL, 0, 5000);
- if (cnt<0) {
- printf("USB_BULK_UPLOAD_INIT failed: %s\n",usb_strerror());
- }
- return 0;
- }
- int ips_read_next_tag(FILE* patch_file, ips_tag_t* tag)
- {
- tag->ofs = getc(patch_file)<<16 | getc(patch_file)<<8 | getc(patch_file);
- if (tag->ofs == EOF_TAG)
- return PATCH_END;
- if (tag->ofs > _16MBIT)
- return PATCH_ERR;
- tag->len = getc(patch_file)<<8 | getc(patch_file);
- // if the length field is > 0 then it is
- // a standard replacement tag
- //
- if (tag->len>0) {
- tag->data = malloc(tag->len);
- size_t ret = fread(tag->data,1,tag->len,patch_file);
- if (ret != tag->len)
- return PATCH_ERR;
- } else {
- // if the length tag is 0 then it is
- // a RLE tag: read 3 bytes from patch: 2bytes: count / 1 byte: value
- tag->data = malloc(sizeof(3));
- tag->data[RLE_LEN] = (uint16_t)(getc(patch_file)<<8 | getc(patch_file));
- tag->data[RLE_VAL] = getc(patch_file);
- }
- return 0;
- }
|