Shift buffer to gain space if necessary.
This commit is contained in:
parent
e66a64be26
commit
e7c81cd9b3
@ -142,6 +142,14 @@ typedef struct guac_instruction {
|
|||||||
*/
|
*/
|
||||||
guac_instruction* guac_instruction_alloc();
|
guac_instruction* guac_instruction_alloc();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Resets the parse state and contents of the given instruction, such that the
|
||||||
|
* memory of that instruction can be reused for another parse cycle.
|
||||||
|
*
|
||||||
|
* @param instruction The instruction to reset.
|
||||||
|
*/
|
||||||
|
void guac_instruction_reset(guac_instruction* instruction);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Appends data from the given buffer to the given instruction. The data will
|
* Appends data from the given buffer to the given instruction. The data will
|
||||||
* be appended, if possible, to this instruction as a reference and thus the
|
* be appended, if possible, to this instruction as a reference and thus the
|
||||||
|
@ -55,14 +55,16 @@ guac_instruction* guac_instruction_alloc() {
|
|||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Initialize state */
|
guac_instruction_reset(instruction);
|
||||||
|
return instruction;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
void guac_instruction_reset(guac_instruction* instruction) {
|
||||||
instruction->opcode = NULL;
|
instruction->opcode = NULL;
|
||||||
instruction->argc = 0;
|
instruction->argc = 0;
|
||||||
instruction->state = GUAC_INSTRUCTION_PARSE_LENGTH;
|
instruction->state = GUAC_INSTRUCTION_PARSE_LENGTH;
|
||||||
instruction->__elementc = 0;
|
instruction->__elementc = 0;
|
||||||
|
|
||||||
return instruction;
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int guac_instruction_append(guac_instruction* instr,
|
int guac_instruction_append(guac_instruction* instr,
|
||||||
@ -178,6 +180,7 @@ guac_instruction* guac_instruction_read(guac_socket* socket,
|
|||||||
|
|
||||||
char* unparsed_end = socket->__instructionbuf_unparsed_end;
|
char* unparsed_end = socket->__instructionbuf_unparsed_end;
|
||||||
char* unparsed_start = socket->__instructionbuf_unparsed_start;
|
char* unparsed_start = socket->__instructionbuf_unparsed_start;
|
||||||
|
char* instr_start = socket->__instructionbuf_unparsed_start;
|
||||||
char* buffer_end = socket->__instructionbuf
|
char* buffer_end = socket->__instructionbuf
|
||||||
+ sizeof(socket->__instructionbuf);
|
+ sizeof(socket->__instructionbuf);
|
||||||
|
|
||||||
@ -197,9 +200,23 @@ guac_instruction* guac_instruction_read(guac_socket* socket,
|
|||||||
|
|
||||||
/* If no space left to read, fail */
|
/* If no space left to read, fail */
|
||||||
if (unparsed_end == buffer_end) {
|
if (unparsed_end == buffer_end) {
|
||||||
guac_error = GUAC_STATUS_NO_MEMORY;
|
|
||||||
guac_error_message = "Instruction too long";
|
/* Shift backward if possible */
|
||||||
return NULL;
|
if (instr_start != socket->__instructionbuf) {
|
||||||
|
memmove(socket->__instructionbuf, instr_start,
|
||||||
|
unparsed_end - instr_start);
|
||||||
|
unparsed_end -= instr_start - socket->__instructionbuf;
|
||||||
|
unparsed_start = instr_start = socket->__instructionbuf;
|
||||||
|
guac_instruction_reset(instruction);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Otherwise, no memory to read */
|
||||||
|
else {
|
||||||
|
guac_error = GUAC_STATUS_NO_MEMORY;
|
||||||
|
guac_error_message = "Instruction too long";
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* No instruction yet? Get more data ... */
|
/* No instruction yet? Get more data ... */
|
||||||
|
Loading…
Reference in New Issue
Block a user