597bfcb64f252a90de8e38cdcb48047ce1e8813e
[fw/openocd] / src / target / openrisc / jsp_server.c
1 /***************************************************************************
2  *   Copyright (C) 2014 by Franck Jullien                                  *
3  *   franck.jullien@gmail.com                                              *
4  *                                                                         *
5  *   Based on ./src/server/telnet_server.c                                 *
6  *                                                                         *
7  *   This program is free software; you can redistribute it and/or modify  *
8  *   it under the terms of the GNU General Public License as published by  *
9  *   the Free Software Foundation; either version 2 of the License, or     *
10  *   (at your option) any later version.                                   *
11  *                                                                         *
12  *   This program is distributed in the hope that it will be useful,       *
13  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
14  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
15  *   GNU General Public License for more details.                          *
16  *                                                                         *
17  *   You should have received a copy of the GNU General Public License     *
18  *   along with this program; if not, write to the                         *
19  *   Free Software Foundation, Inc.,                                       *
20  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.           *
21  ***************************************************************************/
22
23 #ifdef HAVE_CONFIG_H
24 #include "config.h"
25 #endif
26
27 #include <server/telnet_server.h>
28
29 #include "or1k_tap.h"
30 #include "or1k_du.h"
31 #include "jsp_server.h"
32
33 static char *jsp_port;
34
35 /**A skim of the relevant RFCs suggests that if my application simply sent the
36  * characters IAC DONT LINEMODE (\377\376\042) as soon as the client connects,
37  * the client should be forced into character mode. However it doesn't make any difference.
38  */
39
40 static char *negotiate =
41         "\xFF\xFB\x03"                  /* IAC WILL Suppress Go Ahead */
42         "\xFF\xFB\x01"                  /* IAC WILL Echo */
43         "\xFF\xFD\x03"                  /* IAC DO Suppress Go Ahead */
44         "\xFF\xFE\x01";                 /* IAC DON'T Echo */
45
46 /* The only way we can detect that the socket is closed is the first time
47  * we write to it, we will fail. Subsequent write operations will
48  * succeed. Shudder!
49  */
50 static int telnet_write(struct connection *connection, const void *data, int len)
51 {
52         struct telnet_connection *t_con = connection->priv;
53         if (t_con->closed)
54                 return ERROR_SERVER_REMOTE_CLOSED;
55
56         if (connection_write(connection, data, len) == len)
57                 return ERROR_OK;
58         t_con->closed = 1;
59         return ERROR_SERVER_REMOTE_CLOSED;
60 }
61
62 int jsp_poll_read(void *priv)
63 {
64         struct jsp_service *jsp_service = (struct jsp_service *)priv;
65         unsigned char out_buffer[10];
66         unsigned char in_buffer[10];
67         int out_len = 0;
68         int in_len;
69
70         if (!jsp_service->connection)
71                 return ERROR_FAIL;
72
73         memset(out_buffer, 0, 10);
74
75         or1k_adv_jtag_jsp_xfer(jsp_service->jtag_info, &out_len, out_buffer, &in_len, in_buffer);
76         if (in_len)
77                 telnet_write(jsp_service->connection, in_buffer, in_len);
78
79         return ERROR_OK;
80 }
81
82 static int jsp_new_connection(struct connection *connection)
83 {
84         struct telnet_connection *telnet_connection = malloc(sizeof(struct telnet_connection));
85         struct jsp_service *jsp_service = connection->service->priv;
86
87         connection->priv = telnet_connection;
88
89         /* initialize telnet connection information */
90         telnet_connection->closed = 0;
91         telnet_connection->line_size = 0;
92         telnet_connection->line_cursor = 0;
93         telnet_connection->option_size = 0;
94         telnet_connection->state = TELNET_STATE_DATA;
95
96         /* negotiate telnet options */
97         telnet_write(connection, negotiate, strlen(negotiate));
98
99         /* print connection banner */
100         if (jsp_service->banner) {
101                 telnet_write(connection, jsp_service->banner, strlen(jsp_service->banner));
102                 telnet_write(connection, "\r\n", 2);
103         }
104
105         jsp_service->connection = connection;
106
107         int retval = target_register_timer_callback(&jsp_poll_read, 1, 1, jsp_service);
108         if (ERROR_OK != retval)
109                 return retval;
110
111         return ERROR_OK;
112 }
113
114 static int jsp_input(struct connection *connection)
115 {
116         int bytes_read;
117         unsigned char buffer[TELNET_BUFFER_SIZE];
118         unsigned char *buf_p;
119         struct telnet_connection *t_con = connection->priv;
120         struct jsp_service *jsp_service = connection->service->priv;
121
122         bytes_read = connection_read(connection, buffer, TELNET_BUFFER_SIZE);
123
124         if (bytes_read == 0)
125                 return ERROR_SERVER_REMOTE_CLOSED;
126         else if (bytes_read == -1) {
127                 LOG_ERROR("error during read: %s", strerror(errno));
128                 return ERROR_SERVER_REMOTE_CLOSED;
129         }
130
131         buf_p = buffer;
132         while (bytes_read) {
133                 switch (t_con->state) {
134                         case TELNET_STATE_DATA:
135                                 if (*buf_p == 0xff)
136                                         t_con->state = TELNET_STATE_IAC;
137                                 else {
138                                         int out_len = 1;
139                                         int in_len;
140                                         unsigned char in_buffer[10];
141                                         or1k_adv_jtag_jsp_xfer(jsp_service->jtag_info,
142                                                                &out_len, buf_p, &in_len,
143                                                                in_buffer);
144                                         if (in_len)
145                                                 telnet_write(connection,
146                                                              in_buffer, in_len);
147                                 }
148                                 break;
149                         case TELNET_STATE_IAC:
150                                 switch (*buf_p) {
151                                 case 0xfe:
152                                         t_con->state = TELNET_STATE_DONT;
153                                         break;
154                                 case 0xfd:
155                                         t_con->state = TELNET_STATE_DO;
156                                         break;
157                                 case 0xfc:
158                                         t_con->state = TELNET_STATE_WONT;
159                                         break;
160                                 case 0xfb:
161                                         t_con->state = TELNET_STATE_WILL;
162                                         break;
163                                 }
164                                 break;
165                         case TELNET_STATE_SB:
166                                 break;
167                         case TELNET_STATE_SE:
168                                 break;
169                         case TELNET_STATE_WILL:
170                         case TELNET_STATE_WONT:
171                         case TELNET_STATE_DO:
172                         case TELNET_STATE_DONT:
173                                 t_con->state = TELNET_STATE_DATA;
174                                 break;
175                         default:
176                                 LOG_ERROR("unknown telnet state");
177                                 exit(-1);
178                 }
179
180                 bytes_read--;
181                 buf_p++;
182         }
183
184         return ERROR_OK;
185 }
186
187 static int jsp_connection_closed(struct connection *connection)
188 {
189         struct telnet_connection *t_con = connection->priv;
190         struct jsp_service *jsp_service = connection->service->priv;
191
192         if (t_con->prompt) {
193                 free(t_con->prompt);
194                 t_con->prompt = NULL;
195         }
196
197         int retval = target_unregister_timer_callback(&jsp_poll_read, jsp_service);
198         if (ERROR_OK != retval)
199                 return retval;
200
201         if (connection->priv) {
202                 free(connection->priv);
203                 connection->priv = NULL;
204         } else
205                 LOG_ERROR("BUG: connection->priv == NULL");
206
207         return ERROR_OK;
208 }
209
210 int jsp_init(struct or1k_jtag *jtag_info, char *banner)
211 {
212         struct jsp_service *jsp_service = malloc(sizeof(struct jsp_service));
213         jsp_service->banner = banner;
214         jsp_service->jtag_info = jtag_info;
215
216         return add_service("jsp",
217                 jsp_port,
218                 1,
219                 jsp_new_connection,
220                 jsp_input,
221                 jsp_connection_closed,
222                 jsp_service);
223 }
224
225 COMMAND_HANDLER(handle_jsp_port_command)
226 {
227         return CALL_COMMAND_HANDLER(server_pipe_command, &jsp_port);
228 }
229
230 static const struct command_registration jsp_command_handlers[] = {
231         {
232                 .name = "jsp_port",
233                 .handler = handle_jsp_port_command,
234                 .mode = COMMAND_ANY,
235                 .help = "Specify port on which to listen "
236                         "for incoming JSP telnet connections.",
237                 .usage = "[port_num]",
238         },
239         COMMAND_REGISTRATION_DONE
240 };
241
242 int jsp_register_commands(struct command_context *cmd_ctx)
243 {
244         jsp_port = strdup("7777");
245         return register_commands(cmd_ctx, NULL, jsp_command_handlers);
246 }
247