bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 1 | /* |
| 2 | * tftp.c - a simple, read-only tftp server for qemu |
ths | 5fafdf2 | 2007-09-16 21:08:06 +0000 | [diff] [blame] | 3 | * |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 4 | * Copyright (c) 2004 Magnus Damm <damm@opensource.se> |
ths | 5fafdf2 | 2007-09-16 21:08:06 +0000 | [diff] [blame] | 5 | * |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 6 | * Permission is hereby granted, free of charge, to any person obtaining a copy |
| 7 | * of this software and associated documentation files (the "Software"), to deal |
| 8 | * in the Software without restriction, including without limitation the rights |
| 9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
| 10 | * copies of the Software, and to permit persons to whom the Software is |
| 11 | * furnished to do so, subject to the following conditions: |
| 12 | * |
| 13 | * The above copyright notice and this permission notice shall be included in |
| 14 | * all copies or substantial portions of the Software. |
| 15 | * |
| 16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| 17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| 18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
| 19 | * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| 20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| 21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
| 22 | * THE SOFTWARE. |
| 23 | */ |
| 24 | |
| 25 | #include <slirp.h> |
Jan Kiszka | 0d62c4c | 2009-06-24 14:42:29 +0200 | [diff] [blame] | 26 | #include "qemu-common.h" |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 27 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 28 | static inline int tftp_session_in_use(struct tftp_session *spt) |
| 29 | { |
| 30 | return (spt->slirp != NULL); |
| 31 | } |
ths | 3b46e62 | 2007-09-17 08:09:54 +0000 | [diff] [blame] | 32 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 33 | static inline void tftp_session_update(struct tftp_session *spt) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 34 | { |
bellard | a3504c8 | 2004-08-25 20:55:44 +0000 | [diff] [blame] | 35 | spt->timestamp = curtime; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 36 | } |
| 37 | |
| 38 | static void tftp_session_terminate(struct tftp_session *spt) |
| 39 | { |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 40 | if (spt->fd >= 0) { |
| 41 | close(spt->fd); |
| 42 | spt->fd = -1; |
| 43 | } |
Anthony Liguori | 7267c09 | 2011-08-20 22:09:37 -0500 | [diff] [blame] | 44 | g_free(spt->filename); |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 45 | spt->slirp = NULL; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 46 | } |
| 47 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 48 | static int tftp_session_allocate(Slirp *slirp, struct tftp_t *tp) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 49 | { |
| 50 | struct tftp_session *spt; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 51 | int k; |
| 52 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 53 | for (k = 0; k < TFTP_SESSIONS_MAX; k++) { |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 54 | spt = &slirp->tftp_sessions[k]; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 55 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 56 | if (!tftp_session_in_use(spt)) |
bellard | a3504c8 | 2004-08-25 20:55:44 +0000 | [diff] [blame] | 57 | goto found; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 58 | |
| 59 | /* sessions time out after 5 inactive seconds */ |
Jan Kiszka | 9367964 | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 60 | if ((int)(curtime - spt->timestamp) > 5000) { |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 61 | tftp_session_terminate(spt); |
bellard | a3504c8 | 2004-08-25 20:55:44 +0000 | [diff] [blame] | 62 | goto found; |
Jan Kiszka | 9367964 | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 63 | } |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 64 | } |
| 65 | |
| 66 | return -1; |
| 67 | |
| 68 | found: |
| 69 | memset(spt, 0, sizeof(*spt)); |
| 70 | memcpy(&spt->client_ip, &tp->ip.ip_src, sizeof(spt->client_ip)); |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 71 | spt->fd = -1; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 72 | spt->client_port = tp->udp.uh_sport; |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 73 | spt->slirp = slirp; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 74 | |
| 75 | tftp_session_update(spt); |
| 76 | |
| 77 | return k; |
| 78 | } |
| 79 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 80 | static int tftp_session_find(Slirp *slirp, struct tftp_t *tp) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 81 | { |
| 82 | struct tftp_session *spt; |
| 83 | int k; |
| 84 | |
| 85 | for (k = 0; k < TFTP_SESSIONS_MAX; k++) { |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 86 | spt = &slirp->tftp_sessions[k]; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 87 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 88 | if (tftp_session_in_use(spt)) { |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 89 | if (!memcmp(&spt->client_ip, &tp->ip.ip_src, sizeof(spt->client_ip))) { |
| 90 | if (spt->client_port == tp->udp.uh_sport) { |
| 91 | return k; |
| 92 | } |
| 93 | } |
| 94 | } |
| 95 | } |
| 96 | |
| 97 | return -1; |
| 98 | } |
| 99 | |
Hervé Poussineau | 4aa401f | 2012-09-13 12:39:36 +0200 | [diff] [blame] | 100 | static int tftp_read_data(struct tftp_session *spt, uint32_t block_nr, |
Stefan Weil | b6dce92 | 2010-07-22 22:15:23 +0200 | [diff] [blame] | 101 | uint8_t *buf, int len) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 102 | { |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 103 | int bytes_read = 0; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 104 | |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 105 | if (spt->fd < 0) { |
| 106 | spt->fd = open(spt->filename, O_RDONLY | O_BINARY); |
| 107 | } |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 108 | |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 109 | if (spt->fd < 0) { |
| 110 | return -1; |
| 111 | } |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 112 | |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 113 | if (len) { |
| 114 | lseek(spt->fd, block_nr * 512, SEEK_SET); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 115 | |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 116 | bytes_read = read(spt->fd, buf, len); |
| 117 | } |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 118 | |
Hervé Poussineau | 78be056 | 2012-09-10 20:52:25 +0200 | [diff] [blame] | 119 | return bytes_read; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 120 | } |
| 121 | |
ths | 5fafdf2 | 2007-09-16 21:08:06 +0000 | [diff] [blame] | 122 | static int tftp_send_oack(struct tftp_session *spt, |
Hervé Poussineau | 95b1ad7 | 2012-09-13 07:55:01 +0200 | [diff] [blame] | 123 | const char *keys[], uint32_t values[], int nb, |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 124 | struct tftp_t *recv_tp) |
| 125 | { |
| 126 | struct sockaddr_in saddr, daddr; |
| 127 | struct mbuf *m; |
| 128 | struct tftp_t *tp; |
Hervé Poussineau | 95b1ad7 | 2012-09-13 07:55:01 +0200 | [diff] [blame] | 129 | int i, n = 0; |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 130 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 131 | m = m_get(spt->slirp); |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 132 | |
| 133 | if (!m) |
| 134 | return -1; |
| 135 | |
| 136 | memset(m->m_data, 0, m->m_size); |
| 137 | |
blueswir1 | 9634d90 | 2007-10-26 19:01:16 +0000 | [diff] [blame] | 138 | m->m_data += IF_MAXLINKHDR; |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 139 | tp = (void *)m->m_data; |
| 140 | m->m_data += sizeof(struct udpiphdr); |
ths | 3b46e62 | 2007-09-17 08:09:54 +0000 | [diff] [blame] | 141 | |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 142 | tp->tp_op = htons(TFTP_OACK); |
Hervé Poussineau | 95b1ad7 | 2012-09-13 07:55:01 +0200 | [diff] [blame] | 143 | for (i = 0; i < nb; i++) { |
| 144 | n += snprintf(tp->x.tp_buf + n, sizeof(tp->x.tp_buf) - n, "%s", |
| 145 | keys[i]) + 1; |
| 146 | n += snprintf(tp->x.tp_buf + n, sizeof(tp->x.tp_buf) - n, "%u", |
| 147 | values[i]) + 1; |
| 148 | } |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 149 | |
| 150 | saddr.sin_addr = recv_tp->ip.ip_dst; |
| 151 | saddr.sin_port = recv_tp->udp.uh_dport; |
ths | 3b46e62 | 2007-09-17 08:09:54 +0000 | [diff] [blame] | 152 | |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 153 | daddr.sin_addr = spt->client_ip; |
| 154 | daddr.sin_port = spt->client_port; |
| 155 | |
ths | 5fafdf2 | 2007-09-16 21:08:06 +0000 | [diff] [blame] | 156 | m->m_len = sizeof(struct tftp_t) - 514 + n - |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 157 | sizeof(struct ip) - sizeof(struct udphdr); |
| 158 | udp_output2(NULL, m, &saddr, &daddr, IPTOS_LOWDELAY); |
| 159 | |
| 160 | return 0; |
| 161 | } |
| 162 | |
Jan Kiszka | ef2d54d | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 163 | static void tftp_send_error(struct tftp_session *spt, |
Stefan Weil | b6dce92 | 2010-07-22 22:15:23 +0200 | [diff] [blame] | 164 | uint16_t errorcode, const char *msg, |
Jan Kiszka | ef2d54d | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 165 | struct tftp_t *recv_tp) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 166 | { |
| 167 | struct sockaddr_in saddr, daddr; |
| 168 | struct mbuf *m; |
| 169 | struct tftp_t *tp; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 170 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 171 | m = m_get(spt->slirp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 172 | |
| 173 | if (!m) { |
Jan Kiszka | ef2d54d | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 174 | goto out; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 175 | } |
| 176 | |
| 177 | memset(m->m_data, 0, m->m_size); |
| 178 | |
blueswir1 | 9634d90 | 2007-10-26 19:01:16 +0000 | [diff] [blame] | 179 | m->m_data += IF_MAXLINKHDR; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 180 | tp = (void *)m->m_data; |
| 181 | m->m_data += sizeof(struct udpiphdr); |
ths | 3b46e62 | 2007-09-17 08:09:54 +0000 | [diff] [blame] | 182 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 183 | tp->tp_op = htons(TFTP_ERROR); |
| 184 | tp->x.tp_error.tp_error_code = htons(errorcode); |
blueswir1 | b55266b | 2008-09-20 08:07:15 +0000 | [diff] [blame] | 185 | pstrcpy((char *)tp->x.tp_error.tp_msg, sizeof(tp->x.tp_error.tp_msg), msg); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 186 | |
| 187 | saddr.sin_addr = recv_tp->ip.ip_dst; |
| 188 | saddr.sin_port = recv_tp->udp.uh_dport; |
| 189 | |
| 190 | daddr.sin_addr = spt->client_ip; |
| 191 | daddr.sin_port = spt->client_port; |
| 192 | |
ths | 5fafdf2 | 2007-09-16 21:08:06 +0000 | [diff] [blame] | 193 | m->m_len = sizeof(struct tftp_t) - 514 + 3 + strlen(msg) - |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 194 | sizeof(struct ip) - sizeof(struct udphdr); |
| 195 | |
| 196 | udp_output2(NULL, m, &saddr, &daddr, IPTOS_LOWDELAY); |
| 197 | |
Jan Kiszka | ef2d54d | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 198 | out: |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 199 | tftp_session_terminate(spt); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 200 | } |
| 201 | |
Jan Kiszka | eb7faf0 | 2012-09-13 12:44:27 +0200 | [diff] [blame] | 202 | static void tftp_send_next_block(struct tftp_session *spt, |
| 203 | struct tftp_t *recv_tp) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 204 | { |
| 205 | struct sockaddr_in saddr, daddr; |
| 206 | struct mbuf *m; |
| 207 | struct tftp_t *tp; |
| 208 | int nobytes; |
| 209 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 210 | m = m_get(spt->slirp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 211 | |
| 212 | if (!m) { |
Jan Kiszka | eb7faf0 | 2012-09-13 12:44:27 +0200 | [diff] [blame] | 213 | return; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 214 | } |
| 215 | |
| 216 | memset(m->m_data, 0, m->m_size); |
| 217 | |
blueswir1 | 9634d90 | 2007-10-26 19:01:16 +0000 | [diff] [blame] | 218 | m->m_data += IF_MAXLINKHDR; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 219 | tp = (void *)m->m_data; |
| 220 | m->m_data += sizeof(struct udpiphdr); |
ths | 3b46e62 | 2007-09-17 08:09:54 +0000 | [diff] [blame] | 221 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 222 | tp->tp_op = htons(TFTP_DATA); |
Hervé Poussineau | 4aa401f | 2012-09-13 12:39:36 +0200 | [diff] [blame] | 223 | tp->x.tp_data.tp_block_nr = htons((spt->block_nr + 1) & 0xffff); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 224 | |
| 225 | saddr.sin_addr = recv_tp->ip.ip_dst; |
| 226 | saddr.sin_port = recv_tp->udp.uh_dport; |
| 227 | |
| 228 | daddr.sin_addr = spt->client_ip; |
| 229 | daddr.sin_port = spt->client_port; |
| 230 | |
Hervé Poussineau | 4aa401f | 2012-09-13 12:39:36 +0200 | [diff] [blame] | 231 | nobytes = tftp_read_data(spt, spt->block_nr, tp->x.tp_data.tp_buf, 512); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 232 | |
| 233 | if (nobytes < 0) { |
| 234 | m_free(m); |
| 235 | |
| 236 | /* send "file not found" error back */ |
| 237 | |
| 238 | tftp_send_error(spt, 1, "File not found", tp); |
| 239 | |
Jan Kiszka | eb7faf0 | 2012-09-13 12:44:27 +0200 | [diff] [blame] | 240 | return; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 241 | } |
| 242 | |
ths | 5fafdf2 | 2007-09-16 21:08:06 +0000 | [diff] [blame] | 243 | m->m_len = sizeof(struct tftp_t) - (512 - nobytes) - |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 244 | sizeof(struct ip) - sizeof(struct udphdr); |
| 245 | |
| 246 | udp_output2(NULL, m, &saddr, &daddr, IPTOS_LOWDELAY); |
| 247 | |
| 248 | if (nobytes == 512) { |
| 249 | tftp_session_update(spt); |
| 250 | } |
| 251 | else { |
| 252 | tftp_session_terminate(spt); |
| 253 | } |
| 254 | |
Hervé Poussineau | 4aa401f | 2012-09-13 12:39:36 +0200 | [diff] [blame] | 255 | spt->block_nr++; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 256 | } |
| 257 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 258 | static void tftp_handle_rrq(Slirp *slirp, struct tftp_t *tp, int pktlen) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 259 | { |
| 260 | struct tftp_session *spt; |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 261 | int s, k; |
Jan Kiszka | 9367964 | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 262 | size_t prefix_len; |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 263 | char *req_fname; |
Hervé Poussineau | 95b1ad7 | 2012-09-13 07:55:01 +0200 | [diff] [blame] | 264 | const char *option_name[2]; |
| 265 | uint32_t option_value[2]; |
| 266 | int nb_options = 0; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 267 | |
Thomas Horsten | bfe4e17 | 2010-01-07 17:01:28 +0000 | [diff] [blame] | 268 | /* check if a session already exists and if so terminate it */ |
| 269 | s = tftp_session_find(slirp, tp); |
| 270 | if (s >= 0) { |
| 271 | tftp_session_terminate(&slirp->tftp_sessions[s]); |
| 272 | } |
| 273 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 274 | s = tftp_session_allocate(slirp, tp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 275 | |
| 276 | if (s < 0) { |
| 277 | return; |
| 278 | } |
| 279 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 280 | spt = &slirp->tftp_sessions[s]; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 281 | |
Jan Kiszka | f8e3cbd | 2009-06-24 14:42:29 +0200 | [diff] [blame] | 282 | /* unspecifed prefix means service disabled */ |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 283 | if (!slirp->tftp_prefix) { |
Jan Kiszka | f8e3cbd | 2009-06-24 14:42:29 +0200 | [diff] [blame] | 284 | tftp_send_error(spt, 2, "Access violation", tp); |
| 285 | return; |
| 286 | } |
| 287 | |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 288 | /* skip header fields */ |
| 289 | k = 0; |
Stefan Weil | 89d2d3af | 2011-02-23 19:40:14 +0100 | [diff] [blame] | 290 | pktlen -= offsetof(struct tftp_t, x.tp_buf); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 291 | |
Jan Kiszka | 9367964 | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 292 | /* prepend tftp_prefix */ |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 293 | prefix_len = strlen(slirp->tftp_prefix); |
Anthony Liguori | 7267c09 | 2011-08-20 22:09:37 -0500 | [diff] [blame] | 294 | spt->filename = g_malloc(prefix_len + TFTP_FILENAME_MAX + 2); |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 295 | memcpy(spt->filename, slirp->tftp_prefix, prefix_len); |
Jan Kiszka | 74efd61 | 2009-06-29 08:47:30 +0200 | [diff] [blame] | 296 | spt->filename[prefix_len] = '/'; |
Jan Kiszka | 9367964 | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 297 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 298 | /* get name */ |
Jan Kiszka | 74efd61 | 2009-06-29 08:47:30 +0200 | [diff] [blame] | 299 | req_fname = spt->filename + prefix_len + 1; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 300 | |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 301 | while (1) { |
| 302 | if (k >= TFTP_FILENAME_MAX || k >= pktlen) { |
| 303 | tftp_send_error(spt, 2, "Access violation", tp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 304 | return; |
| 305 | } |
Stefan Weil | 89d2d3af | 2011-02-23 19:40:14 +0100 | [diff] [blame] | 306 | req_fname[k] = tp->x.tp_buf[k]; |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 307 | if (req_fname[k++] == '\0') { |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 308 | break; |
| 309 | } |
| 310 | } |
ths | 3b46e62 | 2007-09-17 08:09:54 +0000 | [diff] [blame] | 311 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 312 | /* check mode */ |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 313 | if ((pktlen - k) < 6) { |
| 314 | tftp_send_error(spt, 2, "Access violation", tp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 315 | return; |
| 316 | } |
ths | 3b46e62 | 2007-09-17 08:09:54 +0000 | [diff] [blame] | 317 | |
Stefan Weil | 89d2d3af | 2011-02-23 19:40:14 +0100 | [diff] [blame] | 318 | if (strcasecmp(&tp->x.tp_buf[k], "octet") != 0) { |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 319 | tftp_send_error(spt, 4, "Unsupported transfer mode", tp); |
| 320 | return; |
| 321 | } |
| 322 | |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 323 | k += 6; /* skipping octet */ |
| 324 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 325 | /* do sanity checks on the filename */ |
Jan Kiszka | 74efd61 | 2009-06-29 08:47:30 +0200 | [diff] [blame] | 326 | if (!strncmp(req_fname, "../", 3) || |
| 327 | req_fname[strlen(req_fname) - 1] == '/' || |
Jan Kiszka | 9367964 | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 328 | strstr(req_fname, "/../")) { |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 329 | tftp_send_error(spt, 2, "Access violation", tp); |
| 330 | return; |
| 331 | } |
| 332 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 333 | /* check if the file exists */ |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 334 | if (tftp_read_data(spt, 0, NULL, 0) < 0) { |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 335 | tftp_send_error(spt, 1, "File not found", tp); |
| 336 | return; |
| 337 | } |
| 338 | |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 339 | if (tp->x.tp_buf[pktlen - 1] != 0) { |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 340 | tftp_send_error(spt, 2, "Access violation", tp); |
| 341 | return; |
| 342 | } |
| 343 | |
Hervé Poussineau | 95b1ad7 | 2012-09-13 07:55:01 +0200 | [diff] [blame] | 344 | while (k < pktlen && nb_options < ARRAY_SIZE(option_name)) { |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 345 | const char *key, *value; |
| 346 | |
Stefan Weil | 89d2d3af | 2011-02-23 19:40:14 +0100 | [diff] [blame] | 347 | key = &tp->x.tp_buf[k]; |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 348 | k += strlen(key) + 1; |
| 349 | |
Jan Kiszka | 20c24bf | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 350 | if (k >= pktlen) { |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 351 | tftp_send_error(spt, 2, "Access violation", tp); |
| 352 | return; |
| 353 | } |
| 354 | |
Stefan Weil | 89d2d3af | 2011-02-23 19:40:14 +0100 | [diff] [blame] | 355 | value = &tp->x.tp_buf[k]; |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 356 | k += strlen(value) + 1; |
| 357 | |
Sergei Gavrikov | facf1a6 | 2011-01-12 15:57:18 +0200 | [diff] [blame] | 358 | if (strcasecmp(key, "tsize") == 0) { |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 359 | int tsize = atoi(value); |
| 360 | struct stat stat_p; |
| 361 | |
Jan Kiszka | f8e3cbd | 2009-06-24 14:42:29 +0200 | [diff] [blame] | 362 | if (tsize == 0) { |
Jan Kiszka | 9367964 | 2009-06-24 14:42:30 +0200 | [diff] [blame] | 363 | if (stat(spt->filename, &stat_p) == 0) |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 364 | tsize = stat_p.st_size; |
| 365 | else { |
| 366 | tftp_send_error(spt, 1, "File not found", tp); |
| 367 | return; |
| 368 | } |
| 369 | } |
| 370 | |
Hervé Poussineau | 95b1ad7 | 2012-09-13 07:55:01 +0200 | [diff] [blame] | 371 | option_name[nb_options] = "tsize"; |
| 372 | option_value[nb_options] = tsize; |
| 373 | nb_options++; |
| 374 | } else if (strcasecmp(key, "blksize") == 0) { |
| 375 | int blksize = atoi(value); |
| 376 | |
| 377 | /* If blksize option is bigger than what we will |
| 378 | * emit, accept the option with our packet size. |
| 379 | * Otherwise, simply do as we didn't see the option. |
| 380 | */ |
| 381 | if (blksize >= 512) { |
| 382 | option_name[nb_options] = "blksize"; |
| 383 | option_value[nb_options] = 512; |
| 384 | nb_options++; |
| 385 | } |
ths | 1f697db | 2007-02-20 00:07:50 +0000 | [diff] [blame] | 386 | } |
| 387 | } |
| 388 | |
Hervé Poussineau | 95b1ad7 | 2012-09-13 07:55:01 +0200 | [diff] [blame] | 389 | if (nb_options > 0) { |
| 390 | assert(nb_options <= ARRAY_SIZE(option_name)); |
| 391 | tftp_send_oack(spt, option_name, option_value, nb_options, tp); |
| 392 | return; |
| 393 | } |
| 394 | |
Hervé Poussineau | 4aa401f | 2012-09-13 12:39:36 +0200 | [diff] [blame] | 395 | spt->block_nr = 0; |
| 396 | tftp_send_next_block(spt, tp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 397 | } |
| 398 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 399 | static void tftp_handle_ack(Slirp *slirp, struct tftp_t *tp, int pktlen) |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 400 | { |
| 401 | int s; |
| 402 | |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 403 | s = tftp_session_find(slirp, tp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 404 | |
| 405 | if (s < 0) { |
| 406 | return; |
| 407 | } |
| 408 | |
Jan Kiszka | eb7faf0 | 2012-09-13 12:44:27 +0200 | [diff] [blame] | 409 | tftp_send_next_block(&slirp->tftp_sessions[s], tp); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 410 | } |
| 411 | |
Thomas Horsten | bfe4e17 | 2010-01-07 17:01:28 +0000 | [diff] [blame] | 412 | static void tftp_handle_error(Slirp *slirp, struct tftp_t *tp, int pktlen) |
| 413 | { |
| 414 | int s; |
| 415 | |
| 416 | s = tftp_session_find(slirp, tp); |
| 417 | |
| 418 | if (s < 0) { |
| 419 | return; |
| 420 | } |
| 421 | |
| 422 | tftp_session_terminate(&slirp->tftp_sessions[s]); |
| 423 | } |
| 424 | |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 425 | void tftp_input(struct mbuf *m) |
| 426 | { |
| 427 | struct tftp_t *tp = (struct tftp_t *)m->m_data; |
| 428 | |
| 429 | switch(ntohs(tp->tp_op)) { |
| 430 | case TFTP_RRQ: |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 431 | tftp_handle_rrq(m->slirp, tp, m->m_len); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 432 | break; |
| 433 | |
| 434 | case TFTP_ACK: |
Jan Kiszka | 460fec6 | 2009-06-24 14:42:31 +0200 | [diff] [blame] | 435 | tftp_handle_ack(m->slirp, tp, m->m_len); |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 436 | break; |
Thomas Horsten | bfe4e17 | 2010-01-07 17:01:28 +0000 | [diff] [blame] | 437 | |
| 438 | case TFTP_ERROR: |
| 439 | tftp_handle_error(m->slirp, tp, m->m_len); |
| 440 | break; |
bellard | c7f7464 | 2004-08-24 21:57:12 +0000 | [diff] [blame] | 441 | } |
| 442 | } |