spi.c 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * (C) Copyright 2002
  4. * Gerald Van Baren, Custom IDEAS, vanbaren@cideas.com
  5. */
  6. /*
  7. * SPI Read/Write Utilities
  8. */
  9. #include <common.h>
  10. #include <command.h>
  11. #include <dm.h>
  12. #include <errno.h>
  13. #include <spi.h>
  14. /*-----------------------------------------------------------------------
  15. * Definitions
  16. */
  17. #ifndef MAX_SPI_BYTES
  18. #define MAX_SPI_BYTES 64 /* Maximum number of bytes we can handle */
  19. #endif
  20. /*
  21. * Values from last command.
  22. */
  23. static unsigned int bus;
  24. static unsigned int cs;
  25. static unsigned int mode;
  26. static int bitlen;
  27. static unsigned int speed = 10000000;
  28. static uchar dout[MAX_SPI_BYTES];
  29. static uchar din[MAX_SPI_BYTES];
  30. typedef struct
  31. {
  32. /* data */
  33. char* cmd_str;
  34. unsigned char cmd_code;
  35. }ate_cmd_t;
  36. static ate_cmd_t g_ate_cmd_table[] = {
  37. {"rd.b",0xb1},
  38. {"rd.w",0xb3},
  39. {"rd.l",0xb2},
  40. {"wr.b",0xa1},
  41. {"wr.w",0xa3},
  42. {"wr.l",0xa2},
  43. };
  44. static inline int ate_is_write_cmd(unsigned char cmd){
  45. if(cmd == 0xa1 || cmd == 0xa2 || cmd == 0xa3 ) {
  46. return 1;
  47. }
  48. else {
  49. return 0;
  50. }
  51. }
  52. static int ate_get_reg_width(unsigned char cmd)
  53. {
  54. int len = 0;
  55. switch (cmd)
  56. {
  57. case 0xa1:
  58. case 0xb1:
  59. len = 1;
  60. break;
  61. case 0xa2:
  62. case 0xb2:
  63. len = 4;
  64. break;
  65. case 0xa3:
  66. case 0xb3:
  67. len = 2;
  68. break;
  69. default:
  70. break;
  71. }
  72. return len;
  73. }
  74. static inline int ate_get_write_rsp(unsigned char *rsp, unsigned char *err_code)
  75. {
  76. /* 0xc1 + err_code*/
  77. if(rsp == NULL || err_code == NULL)
  78. return -1;
  79. *err_code = *(rsp+1);
  80. return 0;
  81. }
  82. static int ate_get_read_value(unsigned char *rsp, unsigned int reg_len, unsigned int *val)
  83. {
  84. unsigned i;
  85. if(rsp == NULL || val == NULL)
  86. return -1;
  87. /*0xc1 + 1/2/4B */
  88. *val = 0;
  89. for(i = 0; i < reg_len; i++)
  90. {
  91. *val |= *(rsp+1+i) << (i*8);
  92. }
  93. return 0;
  94. }
  95. static int ate_build_cmd_frame(unsigned char cmd,unsigned long long reg_addr, unsigned int reg_value)
  96. {
  97. int spi_frame_index = 0,i,len;
  98. memset(dout,0,sizeof(dout));
  99. memset(din,0,sizeof(din));
  100. /*fill cmd code*/
  101. dout[spi_frame_index++] = cmd;
  102. //printf("spi_frame_index: %d\n",spi_frame_index);
  103. /*fill reg_addr,exp 0xffea000000 -> 0x00 0x00 0x00 0xea 0xff */
  104. for(i = 0; i < 5; i++){
  105. dout[spi_frame_index++] = (unsigned char)((reg_addr >> i*8 ) &0xFF);
  106. }
  107. //printf("spi_frame_index: %d\n",spi_frame_index);
  108. /*check whether send reg_value and reg_value length in bytes */
  109. len = ate_get_reg_width(cmd);
  110. /* fill reg-value */
  111. if(ate_is_write_cmd(cmd)) {
  112. for(i = 0; i < len; i++) {
  113. dout[spi_frame_index++] = (unsigned char)((reg_value >> i*8 ) &0xFF);
  114. }
  115. }
  116. //printf("spi_frame_index: %d\n",spi_frame_index);
  117. /*insert 2-byte or 3byte dummy data */
  118. if(ate_is_write_cmd(cmd)) {
  119. spi_frame_index +=2;
  120. }else {
  121. spi_frame_index +=2;
  122. }
  123. //printf("spi_frame_index: %d %d\n",len,spi_frame_index);
  124. /*pading clk for rsp */
  125. if(ate_is_write_cmd(cmd)) {
  126. /*0xc1 + err_code(1B)*/
  127. spi_frame_index +=2;
  128. }else {
  129. /*0xc1 + 1B/2B/4B data */
  130. spi_frame_index +=len + 1;
  131. }
  132. //printf("spi_frame_index: %d\n",spi_frame_index);
  133. #if 0
  134. for(i= 0; i < spi_frame_index; i++){
  135. if(i == 0)
  136. printf("spi_tx_frame:");
  137. printf("%02x ",dout[i]);
  138. if(i == spi_frame_index-1)
  139. printf("\n");
  140. }
  141. #endif
  142. return spi_frame_index;
  143. }
  144. static int ate_get_cmd_code(char*pcmd, uchar* cmd)
  145. {
  146. unsigned int table_len = sizeof(g_ate_cmd_table)/sizeof(g_ate_cmd_table[0]);
  147. unsigned int i;
  148. for(i = 0 ; i < table_len; i++){
  149. if(!strcmp(pcmd,g_ate_cmd_table[i].cmd_str)){
  150. *cmd = g_ate_cmd_table[i].cmd_code;
  151. return 0;
  152. }
  153. }
  154. return -1;
  155. }
  156. static void ate_spi_config(unsigned int bus, unsigned int cs, unsigned mode, unsigned int speed)
  157. {
  158. bus = bus;
  159. cs = cs;
  160. mode = mode;
  161. speed = speed;
  162. return;
  163. }
  164. static int do_spi_xfer(int bus, int cs ,int bprint_rx)
  165. {
  166. struct spi_slave *slave;
  167. int ret = 0;
  168. int j;
  169. #ifdef CONFIG_DM_SPI
  170. char name[30], *str;
  171. struct udevice *dev;
  172. snprintf(name, sizeof(name), "generic_%d:%d", bus, cs);
  173. str = strdup(name);
  174. if (!str)
  175. return -ENOMEM;
  176. ret = spi_get_bus_and_cs(bus, cs, speed, mode, "spi_generic_drv",
  177. str, &dev, &slave);
  178. if (ret)
  179. return ret;
  180. #else
  181. slave = spi_setup_slave(bus, cs, speed, mode);
  182. if (!slave) {
  183. printf("Invalid device %d:%d\n", bus, cs);
  184. return -EINVAL;
  185. }
  186. #endif
  187. ret = spi_claim_bus(slave);
  188. if (ret)
  189. goto done;
  190. printf("out:\n");
  191. if(bprint_rx)
  192. {
  193. for (j = 0; j < ((bitlen + 7) / 8); j++)
  194. printf("%02X", dout[j]);
  195. printf("\n");
  196. }
  197. ret = spi_xfer(slave, bitlen, dout, din,
  198. SPI_XFER_BEGIN | SPI_XFER_END);
  199. #ifndef CONFIG_DM_SPI
  200. /* We don't get an error code in this case */
  201. if (ret)
  202. ret = -EIO;
  203. #endif
  204. if (ret) {
  205. printf("Error %d during SPI transaction\n", ret);
  206. } else {
  207. printf("in:\n");
  208. if(bprint_rx)
  209. {
  210. for (j = 0; j < ((bitlen + 7) / 8); j++)
  211. printf("%02X", din[j]);
  212. printf("\n");
  213. }
  214. }
  215. done:
  216. spi_release_bus(slave);
  217. #ifndef CONFIG_DM_SPI
  218. spi_free_slave(slave);
  219. #endif
  220. return ret;
  221. }
  222. /*
  223. * SPI read/write
  224. *
  225. * Syntax:
  226. * spi {dev} {num_bits} {dout}
  227. * {dev} is the device number for controlling chip select (see TBD)
  228. * {num_bits} is the number of bits to send & receive (base 10)
  229. * {dout} is a hexadecimal string of data to send
  230. * The command prints out the hexadecimal string received via SPI.
  231. */
  232. int do_spi (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  233. {
  234. char *cp = 0;
  235. uchar tmp;
  236. int j;
  237. /*
  238. * We use the last specified parameters, unless new ones are
  239. * entered.
  240. */
  241. if ((flag & CMD_FLAG_REPEAT) == 0)
  242. {
  243. if (argc >= 2) {
  244. mode = CONFIG_DEFAULT_SPI_MODE;
  245. bus = simple_strtoul(argv[1], &cp, 10);
  246. if (*cp == ':') {
  247. cs = simple_strtoul(cp+1, &cp, 10);
  248. } else {
  249. cs = bus;
  250. bus = CONFIG_DEFAULT_SPI_BUS;
  251. }
  252. if (*cp == '.')
  253. mode = simple_strtoul(cp+1, NULL, 10);
  254. }
  255. if (argc >= 3)
  256. bitlen = simple_strtoul(argv[2], NULL, 10);
  257. if (argc >= 4) {
  258. cp = argv[3];
  259. for(j = 0; *cp; j++, cp++) {
  260. tmp = *cp - '0';
  261. if(tmp > 9)
  262. tmp -= ('A' - '0') - 10;
  263. if(tmp > 15)
  264. tmp -= ('a' - 'A');
  265. if(tmp > 15) {
  266. printf("Hex conversion error on %c\n", *cp);
  267. return 1;
  268. }
  269. if((j % 2) == 0)
  270. dout[j / 2] = (tmp << 4);
  271. else
  272. dout[j / 2] |= tmp;
  273. }
  274. }
  275. }
  276. if ((bitlen < 0) || (bitlen > (MAX_SPI_BYTES * 8))) {
  277. printf("Invalid bitlen %d\n", bitlen);
  278. return 1;
  279. }
  280. if (do_spi_xfer(bus, cs,1))
  281. return 1;
  282. return 0;
  283. }
  284. static char ate_cmd_help [] = {
  285. "ate_spi speed wr{.b .w .l} addr data \n"
  286. "ate_spi speed rd{.b .w .l} addr\n"
  287. "{speed} is baud rate of spi, uinit in hz\n"
  288. "{ rd{.b .w .l}| wr{.b .w .l} } is the reading or writing reg operation\n"
  289. "{addr} is chip reg addr\n"
  290. "{data} is the data while opertion is writing\n"
  291. };
  292. /*
  293. * SPI read/write
  294. *
  295. * Syntax:
  296. * ate_spi {dev} {rd|wr} {b|w|l} {addr} {data}
  297. * {dev} is the device number for controlling chip select (see TBD)
  298. * {rd|wr} is the reading or writing reg operation
  299. * {b|w|l} is the width of reg value
  300. * {addr} is chip reg addr
  301. * {data} is the data while opertion is writing
  302. * The command prints out the hexadecimal string received via SPI.
  303. */
  304. int do_ate_spi (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  305. {
  306. int ret,frame_len;
  307. unsigned char cmd = 0;
  308. unsigned long long reg_addr = 0;
  309. unsigned int reg_value = 0;
  310. /*
  311. * We use the last specified parameters, unless new ones are
  312. * entered.
  313. */
  314. if ((flag & CMD_FLAG_REPEAT) == 0)
  315. {
  316. /* parser argv */
  317. if(argc < 4){
  318. printf("cmd err \n");
  319. printf("%s",ate_cmd_help);
  320. }
  321. /*get speed */
  322. speed = simple_strtoul(argv[1],NULL,10);
  323. if(speed < 1000000) {
  324. printf("speed need bigger than 1000000 hz\n");
  325. return -1;
  326. }
  327. /*
  328. * cs = cs0
  329. * mode = mode0
  330. * speed = 1000000(default)
  331. * bus = 0
  332. */
  333. ate_spi_config(0,0,0,10000000);
  334. /*get cmd code*/
  335. ret = ate_get_cmd_code(argv[2],&cmd);
  336. if(ret){
  337. printf("unknow cmd code \n");
  338. printf("%s",ate_cmd_help);
  339. return ret;
  340. }
  341. /* get reg_addr */
  342. reg_addr = simple_strtoull(argv[3],NULL,16);
  343. if(ate_is_write_cmd(cmd)){
  344. reg_value = simple_strtoul(argv[4],NULL,16);
  345. }
  346. }
  347. /*build cmd frame */
  348. frame_len = ate_build_cmd_frame(cmd,reg_addr,reg_value);
  349. if(frame_len < 0)
  350. return -1;
  351. bitlen = frame_len<<3;
  352. /*send cmd frame and receive result */
  353. if (do_spi_xfer(bus, cs,0))
  354. return 1;
  355. /* get rsp */
  356. if(ate_is_write_cmd(cmd)){
  357. unsigned char err_code = 0xde;
  358. ret = ate_get_write_rsp(&din[frame_len-2],&err_code);
  359. printf("0x%02x\n",err_code);
  360. //printf(" frame_len %d ret = %d rsp = %02x\n",frame_len,ret,err_code);
  361. }else {
  362. unsigned int val = 0;
  363. unsigned int reg_len = 0;
  364. reg_len = ate_get_reg_width(cmd);
  365. ate_get_read_value(&din[frame_len-reg_len-1],reg_len,&val);
  366. if(reg_len == 1){
  367. printf("0x%02x \n",val);
  368. }
  369. if(reg_len == 2){
  370. printf("0x%04x \n",val);
  371. }
  372. if(reg_len == 4){
  373. printf("0x%08x \n",val);
  374. }
  375. }
  376. return 0;
  377. }
  378. /***************************************************/
  379. int spi_tst_cfg (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  380. {
  381. char *cp = 0;
  382. if(argc != 3){
  383. return -1;
  384. }
  385. speed = simple_strtoul(argv[1],&cp,10);
  386. mode = simple_strtoul(argv[2],&cp,10);
  387. return 0;
  388. }
  389. int spi_tst_xfer (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  390. {
  391. char *cp = 0;
  392. uchar tmp;
  393. int j;
  394. /*
  395. * We use the last specified parameters, unless new ones are
  396. * entered.
  397. */
  398. if ((flag & CMD_FLAG_REPEAT) == 0)
  399. {
  400. bus = simple_strtoul(argv[1], &cp, 10);
  401. cs = simple_strtoul(argv[2], &cp, 10);
  402. bitlen = simple_strtoul(argv[3], NULL, 10);
  403. cp = argv[4];
  404. for(j = 0; *cp; j++, cp++) {
  405. tmp = *cp - '0';
  406. if(tmp > 9)
  407. tmp -= ('A' - '0') - 10;
  408. if(tmp > 15)
  409. tmp -= ('a' - 'A');
  410. if(tmp > 15) {
  411. printf("Hex conversion error on %c\n", *cp);
  412. return 1;
  413. }
  414. if((j % 2) == 0)
  415. dout[j / 2] = (tmp << 4);
  416. else
  417. dout[j / 2] |= tmp;
  418. }
  419. }
  420. if ((bitlen < 0) || (bitlen > (MAX_SPI_BYTES * 8))) {
  421. printf("Invalid bitlen %d\n", bitlen);
  422. return -1;
  423. }
  424. if (do_spi_xfer(bus, cs,1))
  425. return -1;
  426. return 0;
  427. }
  428. U_BOOT_CMD(
  429. sspi, 5, 1, do_spi,
  430. "SPI utility command",
  431. "[<bus>:]<cs>[.<mode>] <bit_len> <dout> - Send and receive bits\n"
  432. "<bus> - Identifies the SPI bus\n"
  433. "<cs> - Identifies the chip select\n"
  434. "<mode> - Identifies the SPI mode to use\n"
  435. "<bit_len> - Number of bits to send (base 10)\n"
  436. "<dout> - Hexadecimal string that gets sent"
  437. );
  438. U_BOOT_CMD(
  439. spi_tst_xfer, 5, 1, spi_tst_xfer,
  440. "do spi xfer op",
  441. "spi_tst_xfer bus(n) cs(n) bits_len(nbyte*8) data(e.g:aabbccddee...)\n"
  442. );
  443. U_BOOT_CMD(
  444. spi_tst_cfg, 3, 1, spi_tst_cfg,
  445. "config spi speed and mode",
  446. "spi_tst_cfg speed(hz) mode(0/1/2/3)\n"
  447. );
  448. U_BOOT_CMD(
  449. ate_spi, 5, 1, do_ate_spi,
  450. "ate_spi test command",
  451. ate_cmd_help
  452. );