summaryrefslogtreecommitdiff
path: root/digital/avr/modules/flash/test/test-flash.c
blob: bbfcfd6ab54a6b3b06c07a7acdccbb85572e4045 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/* test-erase.c */
/* avr.flash - AVR Flash SPI use. {{{
 *
 * Copyright (C) 2008 Nélio Laranjeiro
 *
 * APBTeam:
 *        Web: http://apbteam.org/
 *      Email: team AT apbteam DOT org
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * }}} */
#include "common.h"
#include "io.h"
#include "../flash.h"
#include "modules/proto/proto.h"
#include "modules/utils/utils.h"
#include "modules/utils/byte.h"
#include "modules/uart/uart.h"

void
proto_callback (uint8_t cmd, uint8_t size, uint8_t *args)
{
    /* May be unused. */
    uint32_t addr = v8_to_v32 (0, args[0], args[1], args[2]);
    uint8_t buf[FLASH_LOG_BUFFER_SIZE+1];
    uint8_t status;
#define c(cmd, size) (cmd << 8 | size)
    switch (c (cmd, size))
      {
      case c ('z', 0):
	/* Reset */
	utils_reset ();
	break;
      case c ('e', 0):
	/* Erase full */
	flash_erase (FLASH_ERASE_FULL, 0);
	break;
      case c ('e', 3):
	/* Erase 4k:
	 *  - 3b: address. */
	flash_erase (FLASH_ERASE_4K, addr);
	break;
      case c ('s', 0):
	/* print flash status */
	proto_send1b ('s', flash_read_status());
	break;
      case c ('w', 0):
	/* Send the write enable flash command. */
	flash_send_command (FLASH_WREN);
	break;
      case c ('r', 3):
	/* Read one byte:
	 *  - 3b: address. */
	proto_send1b ('r', flash_read (addr));
	break;
      case c ('r', 4):
	/* Read several bytes:
	 *  - 3b: address.
	 *  - 1b: number of bytes. */
	if (args[3] > sizeof (buf))
	  {
	    proto_send0 ('?');
	    return;
	  }
	else
	  {
	    flash_read_array (addr, buf, args[3]);
	    proto_send ('r', args[3], buf);
	  }
	break;
      case c ('w', 4):
	/* Write one byte:
	 *  - 3b: address.
	 *  - 1b: byte. */
	flash_write (addr, args[3]);
	break;
      case c ('i', 0):
	status = flash_init ();
	proto_send1b ('f', status);
	break;
      default:
	if (cmd == 'w' && size > 4)
	  {
	    /* Write several bytes:
	     *  - 3b: address.
	     *  - nb: bytes. */
	    flash_write_array (addr, args + 3, size - 3);
	  }
	else
	  {
	    /* Error */
	    proto_send0 ('?');
	    return;
	  }
	  break;
      }
    /* Acknowledge what has been done */
    proto_send (cmd, size, args);
}

int
main (void)
{
    uart0_init ();
    proto_send0 ('z');

    while (1)
	proto_accept (uart0_getc ());
}