src/jtag/drivers/usbprog.c

   1 /* SPDX-License-Identifier: GPL-2.0-or-later */
   2
   3 /***************************************************************************
   4  *   Copyright (C) 2007 by Benedikt Sauter                                 *
   5  *   sauter@ixbat.de                                                       *
   6  ***************************************************************************/
   7
   8 /*
   9  * This file is based on Dominic Rath's amt_jtagaccel.c.
  10  *
  11  * usbprog is a free programming adapter. You can easily install
  12  * different firmware versions from an "online pool" over USB.
  13  * The adapter can be used for programming and debugging AVR and ARM
  14  * processors, as USB to RS232 converter, as JTAG interface or as
  15  * simple I/O interface (5 lines).
  16  *
  17  * http://www.embedded-projects.net/usbprog
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include "config.h"
  22 #endif
  23
  24 #include <jtag/interface.h>
  25 #include <jtag/commands.h>
  26 #include "libusb_helper.h"
  27
  28 #define VID 0x1781
  29 #define PID 0x0c63
  30
  31 /* Pins at usbprog */
  32 #define TDO_BIT                 0
  33 #define TDI_BIT                 3
  34 #define TCK_BIT                 2
  35 #define TMS_BIT                 1
  36
  37 static void usbprog_end_state(tap_state_t state);
  38 static void usbprog_state_move(void);
  39 static void usbprog_path_move(struct pathmove_command *cmd);
  40 static void usbprog_runtest(int num_cycles);
  41 static void usbprog_scan(bool ir_scan, enum scan_type type, uint8_t *buffer, int scan_size);
  42
  43 #define UNKNOWN_COMMAND 0x00
  44 #define PORT_DIRECTION  0x01
  45 #define PORT_SET                0x02
  46 #define PORT_GET                0x03
  47 #define PORT_SETBIT             0x04
  48 #define PORT_GETBIT             0x05
  49 #define WRITE_TDI               0x06
  50 #define READ_TDO                0x07
  51 #define WRITE_AND_READ  0x08
  52 #define WRITE_TMS               0x09
  53 #define WRITE_TMS_CHAIN 0x0A
  54
  55 struct usbprog_jtag {
  56         struct libusb_device_handle *usb_handle;
  57 };
  58
  59 static struct usbprog_jtag *usbprog_jtag_handle;
  60
  61 static struct usbprog_jtag *usbprog_jtag_open(void);
  62 /* static void usbprog_jtag_close(struct usbprog_jtag *usbprog_jtag); */
  63 static void usbprog_jtag_init(struct usbprog_jtag *usbprog_jtag);
  64 static unsigned char usbprog_jtag_message(struct usbprog_jtag *usbprog_jtag, char *msg, int msglen);
  65
  66 static void usbprog_jtag_read_tdo(struct usbprog_jtag *usbprog_jtag, char *buffer, int size);
  67 static void usbprog_jtag_write_tdi(struct usbprog_jtag *usbprog_jtag, char *buffer, int size);
  68 static void usbprog_jtag_write_and_read(struct usbprog_jtag *usbprog_jtag, char *buffer, int size);
  69 static void usbprog_jtag_write_tms(struct usbprog_jtag *usbprog_jtag, char tms_scan);
  70
  71 static char tms_chain[64];
  72 static int tms_chain_index;
  73
  74 static void usbprog_jtag_tms_collect(char tms_scan);
  75 static void usbprog_jtag_tms_send(struct usbprog_jtag *usbprog_jtag);
  76
  77 static void usbprog_write(int tck, int tms, int tdi);
  78 static void usbprog_reset(int trst, int srst);
  79
  80 static void usbprog_jtag_set_direction(struct usbprog_jtag *usbprog_jtag, unsigned char direction);
  81 static void usbprog_jtag_write_slice(struct usbprog_jtag *usbprog_jtag, unsigned char value);
  82 /* static unsigned char usbprog_jtag_get_port(struct usbprog_jtag *usbprog_jtag); */
  83 static void usbprog_jtag_set_bit(struct usbprog_jtag *usbprog_jtag, int bit, int value);
  84 /* static int usbprog_jtag_get_bit(struct usbprog_jtag *usbprog_jtag, int bit); */
  85
  86 static int usbprog_execute_queue(void)
  87 {
  88         struct jtag_command *cmd = jtag_command_queue;  /* currently processed command */
  89         int scan_size;
  90         enum scan_type type;
  91         uint8_t *buffer;
  92
  93         while (cmd) {
  94                 switch (cmd->type) {
  95                 case JTAG_RESET:
  96                         LOG_DEBUG_IO("reset trst: %i srst %i",
  97                                         cmd->cmd.reset->trst,
  98                                         cmd->cmd.reset->srst);
  99                         if (cmd->cmd.reset->trst == 1)
 100                                 tap_set_state(TAP_RESET);
 101                         usbprog_reset(cmd->cmd.reset->trst, cmd->cmd.reset->srst);
 102                         break;
 103                 case JTAG_RUNTEST:
 104                         LOG_DEBUG_IO("runtest %i cycles, end in %i",
 105                                         cmd->cmd.runtest->num_cycles,
 106                                         cmd->cmd.runtest->end_state);
 107                         usbprog_end_state(cmd->cmd.runtest->end_state);
 108                         usbprog_runtest(cmd->cmd.runtest->num_cycles);
 109                         break;
 110                 case JTAG_TLR_RESET:
 111                         LOG_DEBUG_IO("statemove end in %i", cmd->cmd.statemove->end_state);
 112                         usbprog_end_state(cmd->cmd.statemove->end_state);
 113                         usbprog_state_move();
 114                         break;
 115                 case JTAG_PATHMOVE:
 116                         LOG_DEBUG_IO("pathmove: %i states, end in %i",
 117                                         cmd->cmd.pathmove->num_states,
 118                                         cmd->cmd.pathmove->path[cmd->cmd.pathmove->num_states - 1]);
 119                         usbprog_path_move(cmd->cmd.pathmove);
 120                         break;
 121                 case JTAG_SCAN:
 122                         LOG_DEBUG_IO("scan end in %i", cmd->cmd.scan->end_state);
 123                         usbprog_end_state(cmd->cmd.scan->end_state);
 124                         scan_size = jtag_build_buffer(cmd->cmd.scan, &buffer);
 125                         type = jtag_scan_type(cmd->cmd.scan);
 126                         usbprog_scan(cmd->cmd.scan->ir_scan, type, buffer, scan_size);
 127                         if (jtag_read_buffer(buffer, cmd->cmd.scan) != ERROR_OK)
 128                                 return ERROR_JTAG_QUEUE_FAILED;
 129                         free(buffer);
 130                         break;
 131                 case JTAG_SLEEP:
 132                         LOG_DEBUG_IO("sleep %" PRIu32, cmd->cmd.sleep->us);
 133                         jtag_sleep(cmd->cmd.sleep->us);
 134                         break;
 135                 default:
 136                         LOG_ERROR("BUG: unknown JTAG command type encountered");
 137                         exit(-1);
 138                 }
 139
 140                 cmd = cmd->next;
 141         }
 142
 143         return ERROR_OK;
 144 }
 145
 146 static int usbprog_init(void)
 147 {
 148         usbprog_jtag_handle = usbprog_jtag_open();
 149
 150         tms_chain_index = 0;
 151         if (usbprog_jtag_handle == 0) {
 152                 LOG_ERROR("Can't find USB JTAG Interface! Please check connection and permissions.");
 153                 return ERROR_JTAG_INIT_FAILED;
 154         }
 155
 156         LOG_INFO("USB JTAG Interface ready!");
 157
 158         usbprog_jtag_init(usbprog_jtag_handle);
 159         usbprog_reset(0, 0);
 160         usbprog_write(0, 0, 0);
 161
 162         return ERROR_OK;
 163 }
 164
 165 static int usbprog_quit(void)
 166 {
 167         return ERROR_OK;
 168 }
 169
 170 /*************** jtag execute commands **********************/
 171 static void usbprog_end_state(tap_state_t state)
 172 {
 173         if (tap_is_state_stable(state))
 174                 tap_set_end_state(state);
 175         else {
 176                 LOG_ERROR("BUG: %i is not a valid end state", state);
 177                 exit(-1);
 178         }
 179 }
 180
 181 static void usbprog_state_move(void)
 182 {
 183         uint8_t tms_scan = tap_get_tms_path(tap_get_state(), tap_get_end_state());
 184
 185         usbprog_jtag_write_tms(usbprog_jtag_handle, (char)tms_scan);
 186
 187         tap_set_state(tap_get_end_state());
 188 }
 189
 190 static void usbprog_path_move(struct pathmove_command *cmd)
 191 {
 192         int num_states = cmd->num_states;
 193         int state_count;
 194
 195         /* There may be queued transitions, and before following a specified
 196            path, we must flush those queued transitions */
 197         usbprog_jtag_tms_send(usbprog_jtag_handle);
 198
 199         state_count = 0;
 200         while (num_states) {
 201                 if (tap_state_transition(tap_get_state(), false) == cmd->path[state_count]) {
 202                         /* LOG_INFO("1"); */
 203                         usbprog_write(0, 0, 0);
 204                         usbprog_write(1, 0, 0);
 205                 } else if (tap_state_transition(tap_get_state(),
 206                                    true) == cmd->path[state_count]) {
 207                         /* LOG_INFO("2"); */
 208                         usbprog_write(0, 1, 0);
 209                         usbprog_write(1, 1, 0);
 210                 } else {
 211                         LOG_ERROR("BUG: %s -> %s isn't a valid TAP transition",
 212                                 tap_state_name(tap_get_state()),
 213                                 tap_state_name(cmd->path[state_count]));
 214                         exit(-1);
 215                 }
 216
 217                 tap_set_state(cmd->path[state_count]);
 218                 state_count++;
 219                 num_states--;
 220         }
 221
 222         tap_set_end_state(tap_get_state());
 223 }
 224
 225 static void usbprog_runtest(int num_cycles)
 226 {
 227         int i;
 228
 229         /* only do a state_move when we're not already in IDLE */
 230         if (tap_get_state() != TAP_IDLE) {
 231                 usbprog_end_state(TAP_IDLE);
 232                 usbprog_state_move();
 233         }
 234
 235         /* execute num_cycles */
 236         if (num_cycles > 0) {
 237                 usbprog_jtag_tms_send(usbprog_jtag_handle);
 238                 usbprog_write(0, 0, 0);
 239         } else {
 240                 usbprog_jtag_tms_send(usbprog_jtag_handle);
 241                 /* LOG_INFO("NUM CYCLES %i",num_cycles); */
 242         }
 243
 244         for (i = 0; i < num_cycles; i++) {
 245                 usbprog_write(1, 0, 0);
 246                 usbprog_write(0, 0, 0);
 247         }
 248
 249         LOG_DEBUG_IO("runtest: cur_state %s end_state %s", tap_state_name(
 250                         tap_get_state()), tap_state_name(tap_get_end_state()));
 251
 252         /* finish in end_state */
 253         /*
 254         usbprog_end_state(saved_end_state);
 255         if (tap_get_state() != tap_get_end_state())
 256                 usbprog_state_move();
 257         */
 258 }
 259
 260 static void usbprog_scan(bool ir_scan, enum scan_type type, uint8_t *buffer, int scan_size)
 261 {
 262         tap_state_t saved_end_state = tap_get_end_state();
 263
 264         if (ir_scan)
 265                 usbprog_end_state(TAP_IRSHIFT);
 266         else
 267                 usbprog_end_state(TAP_DRSHIFT);
 268
 269         /* Only move if we're not already there */
 270         if (tap_get_state() != tap_get_end_state())
 271                 usbprog_state_move();
 272
 273         usbprog_end_state(saved_end_state);
 274
 275         usbprog_jtag_tms_send(usbprog_jtag_handle);
 276
 277         void (*f)(struct usbprog_jtag *usbprog_jtag, char *buffer_local, int size);
 278         switch (type) {
 279                 case SCAN_OUT:
 280                         f = &usbprog_jtag_write_tdi;
 281                         break;
 282                 case SCAN_IN:
 283                         f = &usbprog_jtag_read_tdo;
 284                         break;
 285                 case SCAN_IO:
 286                         f = &usbprog_jtag_write_and_read;
 287                         break;
 288                 default:
 289                         LOG_ERROR("unknown scan type: %i", type);
 290                         exit(-1);
 291         }
 292         f(usbprog_jtag_handle, (char *)buffer, scan_size);
 293
 294         /* The adapter does the transition to PAUSE internally */
 295         if (ir_scan)
 296                 tap_set_state(TAP_IRPAUSE);
 297         else
 298                 tap_set_state(TAP_DRPAUSE);
 299
 300         if (tap_get_state() != tap_get_end_state())
 301                 usbprog_state_move();
 302 }
 303
 304 /*************** jtag wrapper functions *********************/
 305
 306 static void usbprog_write(int tck, int tms, int tdi)
 307 {
 308         unsigned char output_value = 0x00;
 309
 310         if (tms)
 311                 output_value |= (1 << TMS_BIT);
 312         if (tdi)
 313                 output_value |= (1 << TDI_BIT);
 314         if (tck)
 315                 output_value |= (1 << TCK_BIT);
 316
 317         usbprog_jtag_write_slice(usbprog_jtag_handle, output_value);
 318 }
 319
 320 /* (1) assert or (0) deassert reset lines */
 321 static void usbprog_reset(int trst, int srst)
 322 {
 323         LOG_DEBUG("trst: %i, srst: %i", trst, srst);
 324
 325         if (trst)
 326                 usbprog_jtag_set_bit(usbprog_jtag_handle, 5, 0);
 327         else
 328                 usbprog_jtag_set_bit(usbprog_jtag_handle, 5, 1);
 329
 330         if (srst)
 331                 usbprog_jtag_set_bit(usbprog_jtag_handle, 4, 0);
 332         else
 333                 usbprog_jtag_set_bit(usbprog_jtag_handle, 4, 1);
 334 }
 335
 336 /*************** jtag lowlevel functions ********************/
 337
 338 struct usb_bus *busses;
 339
 340 struct usbprog_jtag *usbprog_jtag_open(void)
 341 {
 342         const uint16_t vids[] = { VID, 0 };
 343         const uint16_t pids[] = { PID, 0 };
 344         struct libusb_device_handle *dev;
 345
 346         if (jtag_libusb_open(vids, pids, &dev, NULL) != ERROR_OK)
 347                 return NULL;
 348
 349         struct usbprog_jtag *tmp = malloc(sizeof(struct usbprog_jtag));
 350         tmp->usb_handle = dev;
 351
 352         libusb_set_configuration(dev, 1);
 353         libusb_claim_interface(dev, 0);
 354         libusb_set_interface_alt_setting(dev, 0, 0);
 355
 356         return tmp;
 357 }
 358
 359 #if 0
 360 static void usbprog_jtag_close(struct usbprog_jtag *usbprog_jtag)
 361 {
 362         libusb_close(usbprog_jtag->usb_handle);
 363         free(usbprog_jtag);
 364 }
 365 #endif
 366
 367 static unsigned char usbprog_jtag_message(struct usbprog_jtag *usbprog_jtag, char *msg, int msglen)
 368 {
 369         int transferred;
 370
 371         int res = jtag_libusb_bulk_write(usbprog_jtag->usb_handle, 3, msg, msglen, 100, &transferred);
 372         if ((msg[0] == 2) || (msg[0] == 1) || (msg[0] == 4) || (msg[0] == 0) ||
 373             (msg[0] == 6) || (msg[0] == 0x0A) || (msg[0] == 9))
 374                 return 1;
 375         if (res == ERROR_OK && transferred == msglen) {
 376                 /* LOG_INFO("HALLLLOOO %i",(int)msg[0]); */
 377                 res = jtag_libusb_bulk_read(usbprog_jtag->usb_handle, 0x82, msg, 2, 100, &transferred);
 378                 if (res == ERROR_OK && transferred > 0)
 379                         return (unsigned char)msg[1];
 380                 else
 381                         return -1;
 382         } else
 383                 return -1;
 384         return 0;
 385 }
 386
 387 static void usbprog_jtag_init(struct usbprog_jtag *usbprog_jtag)
 388 {
 389         usbprog_jtag_set_direction(usbprog_jtag, 0xFE);
 390 }
 391
 392 static void usbprog_jtag_write_and_read(struct usbprog_jtag *usbprog_jtag, char *buffer, int size)
 393 {
 394         char tmp[64];   /* fastest packet size for usb controller */
 395         int send_bits, bufindex = 0, fillindex = 0, i, loops;
 396
 397         char swap;
 398         /* 61 byte can be transferred (488 bit) */
 399
 400         while (size > 0) {
 401                 if (size > 488) {
 402                         send_bits = 488;
 403                         size = size - 488;
 404                         loops = 61;
 405                 } else {
 406                         send_bits = size;
 407                         loops = size / 8;
 408                         loops++;
 409                         size = 0;
 410                 }
 411                 tmp[0] = WRITE_AND_READ;
 412                 tmp[1] = (char)(send_bits >> 8);        /* high */
 413                 tmp[2] = (char)(send_bits);                     /* low */
 414
 415                 for (i = 0; i < loops; i++) {
 416                         tmp[3 + i] = buffer[bufindex];
 417                         bufindex++;
 418                 }
 419
 420                 int transferred;
 421                 int res = jtag_libusb_bulk_write(usbprog_jtag->usb_handle, 3, tmp, 64, 1000, &transferred);
 422                 if (res == ERROR_OK && transferred == 64) {
 423                         /* LOG_INFO("HALLLLOOO2 %i",(int)tmp[0]); */
 424                         usleep(1);
 425                         int timeout = 0;
 426                         while (jtag_libusb_bulk_read(usbprog_jtag->usb_handle, 0x82, tmp, 64, 1000, &transferred) != ERROR_OK) {
 427                                 timeout++;
 428                                 if (timeout > 10)
 429                                         break;
 430                         }
 431
 432                         for (i = 0; i < loops; i++) {
 433                                 swap = tmp[3 + i];
 434                                 buffer[fillindex++] = swap;
 435                         }
 436                 }
 437         }
 438 }
 439
 440 static void usbprog_jtag_read_tdo(struct usbprog_jtag *usbprog_jtag, char *buffer, int size)
 441 {
 442         char tmp[64];   /* fastest packet size for usb controller */
 443         int send_bits, fillindex = 0, i, loops;
 444
 445         char swap;
 446         /* 61 byte can be transferred (488 bit) */
 447
 448         while (size > 0) {
 449                 if (size > 488) {
 450                         send_bits = 488;
 451                         size = size - 488;
 452                         loops = 61;
 453                 } else {
 454                         send_bits = size;
 455                         loops = size / 8;
 456                         loops++;
 457                         size = 0;
 458                 }
 459                 tmp[0] = WRITE_AND_READ;
 460                 tmp[1] = (char)(send_bits >> 8);        /* high */
 461                 tmp[2] = (char)(send_bits);                     /* low */
 462
 463                 int transferred;
 464                 jtag_libusb_bulk_write(usbprog_jtag->usb_handle, 3, tmp, 3, 1000, &transferred);
 465
 466                 /* LOG_INFO("HALLLLOOO3 %i",(int)tmp[0]); */
 467                 int timeout = 0;
 468                 usleep(1);
 469                 while (jtag_libusb_bulk_read(usbprog_jtag->usb_handle, 0x82, tmp, 64, 10, &transferred) != ERROR_OK) {
 470                         timeout++;
 471                         if (timeout > 10)
 472                                 break;
 473                 }
 474
 475                 for (i = 0; i < loops; i++) {
 476                         swap = tmp[3 + i];
 477                         buffer[fillindex++] = swap;
 478                 }
 479         }
 480 }
 481
 482 static void usbprog_jtag_write_tdi(struct usbprog_jtag *usbprog_jtag, char *buffer, int size)
 483 {
 484         char tmp[64];   /* fastest packet size for usb controller */
 485         int send_bits, bufindex = 0, i, loops;
 486
 487         /* 61 byte can be transferred (488 bit) */
 488         while (size > 0) {
 489                 if (size > 488) {
 490                         send_bits = 488;
 491                         size = size - 488;
 492                         loops = 61;
 493                 } else {
 494                         send_bits = size;
 495                         loops = size/8;
 496                         /* if (loops == 0) */
 497                         loops++;
 498                         size = 0;
 499                 }
 500                 tmp[0] = WRITE_TDI;
 501                 tmp[1] = (char)(send_bits >> 8);        /* high */
 502                 tmp[2] = (char)(send_bits);                     /* low */
 503
 504                 for (i = 0; i < loops; i++) {
 505                         tmp[3 + i] = buffer[bufindex];
 506                         bufindex++;
 507                 }
 508                 int transferred;
 509                 jtag_libusb_bulk_write(usbprog_jtag->usb_handle, 3, tmp, 64, 1000, &transferred);
 510         }
 511 }
 512
 513 static void usbprog_jtag_write_tms(struct usbprog_jtag *usbprog_jtag, char tms_scan)
 514 {
 515         usbprog_jtag_tms_collect(tms_scan);
 516 }
 517
 518 static void usbprog_jtag_set_direction(struct usbprog_jtag *usbprog_jtag, unsigned char direction)
 519 {
 520         char tmp[2];
 521         tmp[0] = PORT_DIRECTION;
 522         tmp[1] = (char)direction;
 523         usbprog_jtag_message(usbprog_jtag, tmp, 2);
 524 }
 525
 526 static void usbprog_jtag_write_slice(struct usbprog_jtag *usbprog_jtag, unsigned char value)
 527 {
 528         char tmp[2];
 529         tmp[0] = PORT_SET;
 530         tmp[1] = (char)value;
 531         usbprog_jtag_message(usbprog_jtag, tmp, 2);
 532 }
 533
 534 #if 0
 535 static unsigned char usbprog_jtag_get_port(struct usbprog_jtag *usbprog_jtag)
 536 {
 537         char tmp[2];
 538         tmp[0] = PORT_GET;
 539         tmp[1] = 0x00;
 540         return usbprog_jtag_message(usbprog_jtag, tmp, 2);
 541 }
 542 #endif
 543
 544 static void usbprog_jtag_set_bit(struct usbprog_jtag *usbprog_jtag, int bit, int value)
 545 {
 546         char tmp[3];
 547         tmp[0] = PORT_SETBIT;
 548         tmp[1] = (char)bit;
 549         if (value == 1)
 550                 tmp[2] = 0x01;
 551         else
 552                 tmp[2] = 0x00;
 553         usbprog_jtag_message(usbprog_jtag, tmp, 3);
 554 }
 555
 556 #if 0
 557 static int usbprog_jtag_get_bit(struct usbprog_jtag *usbprog_jtag, int bit)
 558 {
 559         char tmp[2];
 560         tmp[0] = PORT_GETBIT;
 561         tmp[1] = (char)bit;
 562
 563         if (usbprog_jtag_message(usbprog_jtag, tmp, 2) > 0)
 564                 return 1;
 565         else
 566                 return 0;
 567 }
 568 #endif
 569
 570 static void usbprog_jtag_tms_collect(char tms_scan)
 571 {
 572         tms_chain[tms_chain_index] = tms_scan;
 573         tms_chain_index++;
 574 }
 575
 576 static void usbprog_jtag_tms_send(struct usbprog_jtag *usbprog_jtag)
 577 {
 578         /* LOG_INFO("TMS SEND"); */
 579         if (tms_chain_index > 0) {
 580                 char tmp[tms_chain_index + 2];
 581                 tmp[0] = WRITE_TMS_CHAIN;
 582                 tmp[1] = (char)(tms_chain_index);
 583                 for (int i = 0; i < tms_chain_index + 1; i++)
 584                         tmp[2 + i] = tms_chain[i];
 585                 int transferred;
 586                 jtag_libusb_bulk_write(usbprog_jtag->usb_handle, 3, tmp, tms_chain_index + 2, 1000, &transferred);
 587                 tms_chain_index = 0;
 588         }
 589 }
 590
 591 static struct jtag_interface usbprog_interface = {
 592         .execute_queue = usbprog_execute_queue,
 593 };
 594
 595 struct adapter_driver usbprog_adapter_driver = {
 596         .name = "usbprog",
 597         .transports = jtag_only,
 598
 599         .init = usbprog_init,
 600         .quit = usbprog_quit,
 601
 602         .jtag_ops = &usbprog_interface,
 603 };