static int check_name_real (const grub_uint8_t *name_at, const grub_uint8_t *head, const grub_uint8_t *tail, const char *check_with, int *length, char *set) { const char *readable_ptr = check_with; const grub_uint8_t *ptr; char *optr = set; int bytes_processed = 0; if (length) *length = 0; for (ptr = name_at; ptr < tail && bytes_processed < tail - head + 2; ) { /* End marker. */ if (!*ptr) { if (length && *length) (*length)--; if (optr && optr != set) optr--; if (optr) *optr = 0; return !readable_ptr || (*readable_ptr == 0); } if (*ptr & 0xc0) { bytes_processed += 2; if (ptr + 1 >= tail) return 0; ptr = head + (((ptr[0] & 0x3f) << 8) | ptr[1]); continue; } if (readable_ptr && grub_memcmp (ptr + 1, readable_ptr, *ptr) != 0) return 0; if (grub_memchr (ptr + 1, 0, *ptr) || grub_memchr (ptr + 1, '.', *ptr)) return 0; if (readable_ptr) readable_ptr += *ptr; if (readable_ptr && *readable_ptr != '.' && *readable_ptr != 0) return 0; bytes_processed += *ptr + 1; if (length) *length += *ptr + 1; if (optr) { grub_memcpy (optr, ptr + 1, *ptr); optr += *ptr; } if (optr) *optr++ = '.'; if (readable_ptr && *readable_ptr) readable_ptr++; ptr += *ptr + 1; } return 0; }
static grub_err_t http_receive (grub_net_tcp_socket_t sock __attribute__ ((unused)), struct grub_net_buff *nb, void *f) { grub_file_t file = f; http_data_t data = file->data; grub_err_t err; while (1) { char *ptr = (char *) nb->data; if ((!data->headers_recv || data->in_chunk_len) && data->current_line) { int have_line = 1; char *t; ptr = grub_memchr (nb->data, '\n', nb->tail - nb->data); if (ptr) ptr++; else { have_line = 0; ptr = (char *) nb->tail; } t = grub_realloc (data->current_line, data->current_line_len + (ptr - (char *) nb->data)); if (!t) { grub_netbuff_free (nb); grub_net_tcp_close (data->sock, GRUB_NET_TCP_ABORT); return grub_errno; } data->current_line = t; grub_memcpy (data->current_line + data->current_line_len, nb->data, ptr - (char *) nb->data); data->current_line_len += ptr - (char *) nb->data; if (!have_line) { grub_netbuff_free (nb); return GRUB_ERR_NONE; } err = parse_line (file, data, data->current_line, data->current_line_len); grub_free (data->current_line); data->current_line = 0; data->current_line_len = 0; if (err) { grub_net_tcp_close (data->sock, GRUB_NET_TCP_ABORT); grub_netbuff_free (nb); return err; } } while (ptr < (char *) nb->tail && (!data->headers_recv || data->in_chunk_len)) { char *ptr2; ptr2 = grub_memchr (ptr, '\n', (char *) nb->tail - ptr); if (!ptr2) { data->current_line = grub_malloc ((char *) nb->tail - ptr); if (!data->current_line) { grub_netbuff_free (nb); grub_net_tcp_close (data->sock, GRUB_NET_TCP_ABORT); return grub_errno; } data->current_line_len = (char *) nb->tail - ptr; grub_memcpy (data->current_line, ptr, data->current_line_len); grub_netbuff_free (nb); return GRUB_ERR_NONE; } err = parse_line (file, data, ptr, ptr2 - ptr); if (err) { grub_net_tcp_close (data->sock, GRUB_NET_TCP_ABORT); grub_netbuff_free (nb); return err; } ptr = ptr2 + 1; } if (((char *) nb->tail - ptr) <= 0) { grub_netbuff_free (nb); return GRUB_ERR_NONE; } err = grub_netbuff_pull (nb, ptr - (char *) nb->data); if (err) { grub_net_tcp_close (data->sock, GRUB_NET_TCP_ABORT); grub_netbuff_free (nb); return err; } if (!(data->chunked && (grub_ssize_t) data->chunk_rem < nb->tail - nb->data)) { grub_net_put_packet (&file->device->net->packs, nb); if (file->device->net->packs.count >= 20) file->device->net->stall = 1; if (file->device->net->packs.count >= 100) grub_net_tcp_stall (data->sock); if (data->chunked) data->chunk_rem -= nb->tail - nb->data; return GRUB_ERR_NONE; } if (data->chunk_rem) { struct grub_net_buff *nb2; nb2 = grub_netbuff_alloc (data->chunk_rem); if (!nb2) return grub_errno; grub_netbuff_put (nb2, data->chunk_rem); grub_memcpy (nb2->data, nb->data, data->chunk_rem); if (file->device->net->packs.count >= 20) { file->device->net->stall = 1; grub_net_tcp_stall (data->sock); } grub_net_put_packet (&file->device->net->packs, nb2); grub_netbuff_pull (nb, data->chunk_rem); } data->in_chunk_len = 1; } }