1
0
mirror of https://github.com/samba-team/samba.git synced 2025-02-23 09:57:40 +03:00

packet.h API: The callback is now responsible to talloc_free() "buf"

This commit is contained in:
Volker Lendecke 2008-12-20 10:44:29 +01:00
parent e000045f15
commit aed8c7bfcf
3 changed files with 50 additions and 37 deletions

View File

@ -44,12 +44,14 @@ NTSTATUS packet_fd_read_sync(struct packet_context *ctx);
* Handle an incoming packet: * Handle an incoming packet:
* Return False if none is available * Return False if none is available
* Otherwise return True and store the callback result in *status * Otherwise return True and store the callback result in *status
* Callback must either talloc_move or talloc_free buf
*/ */
bool packet_handler(struct packet_context *ctx, bool packet_handler(struct packet_context *ctx,
bool (*full_req)(const DATA_BLOB *data, bool (*full_req)(const uint8_t *buf,
size_t available,
size_t *length, size_t *length,
void *private_data), void *private_data),
NTSTATUS (*callback)(const DATA_BLOB *data, NTSTATUS (*callback)(uint8_t *buf, size_t length,
void *private_data), void *private_data),
void *private_data, void *private_data,
NTSTATUS *status); NTSTATUS *status);

View File

@ -155,17 +155,17 @@ static NTSTATUS ctdbd_connect(TALLOC_CTX *mem_ctx,
* Do we have a complete ctdb packet in the queue? * Do we have a complete ctdb packet in the queue?
*/ */
static bool ctdb_req_complete(const DATA_BLOB *data, static bool ctdb_req_complete(const uint8_t *buf, size_t available,
size_t *length, size_t *length,
void *private_data) void *private_data)
{ {
uint32 msglen; uint32 msglen;
if (data->length < sizeof(msglen)) { if (available < sizeof(msglen)) {
return False; return False;
} }
msglen = *((uint32 *)data->data); msglen = *((uint32 *)buf);
DEBUG(10, ("msglen = %d\n", msglen)); DEBUG(10, ("msglen = %d\n", msglen));
@ -176,12 +176,12 @@ static bool ctdb_req_complete(const DATA_BLOB *data,
cluster_fatal("ctdbd protocol error\n"); cluster_fatal("ctdbd protocol error\n");
} }
if (data->length >= msglen) { if (available < msglen) {
*length = msglen; return false;
return True;
} }
return False; *length = msglen;
return true;
} }
/* /*
@ -220,16 +220,13 @@ struct req_pull_state {
* Pull a ctdb request out of the incoming packet queue * Pull a ctdb request out of the incoming packet queue
*/ */
static NTSTATUS ctdb_req_pull(const DATA_BLOB *data, static NTSTATUS ctdb_req_pull(uint8_t *buf, size_t length,
void *private_data) void *private_data)
{ {
struct req_pull_state *state = (struct req_pull_state *)private_data; struct req_pull_state *state = (struct req_pull_state *)private_data;
state->req = data_blob_talloc(state->mem_ctx, data->data, state->req.data = talloc_move(state->mem_ctx, &buf);
data->length); state->req.length = length;
if (state->req.data == NULL) {
return NT_STATUS_NO_MEMORY;
}
return NT_STATUS_OK; return NT_STATUS_OK;
} }
@ -497,7 +494,7 @@ NTSTATUS ctdbd_messaging_connection(TALLOC_CTX *mem_ctx,
/* /*
* Packet handler to receive and handle a ctdb message * Packet handler to receive and handle a ctdb message
*/ */
static NTSTATUS ctdb_handle_message(const DATA_BLOB *data, static NTSTATUS ctdb_handle_message(uint8_t *buf, size_t length,
void *private_data) void *private_data)
{ {
struct ctdbd_connection *conn = talloc_get_type_abort( struct ctdbd_connection *conn = talloc_get_type_abort(
@ -505,11 +502,12 @@ static NTSTATUS ctdb_handle_message(const DATA_BLOB *data,
struct ctdb_req_message *msg; struct ctdb_req_message *msg;
struct messaging_rec *msg_rec; struct messaging_rec *msg_rec;
msg = (struct ctdb_req_message *)data->data; msg = (struct ctdb_req_message *)buf;
if (msg->hdr.operation != CTDB_REQ_MESSAGE) { if (msg->hdr.operation != CTDB_REQ_MESSAGE) {
DEBUG(0, ("Received async msg of type %u, discarding\n", DEBUG(0, ("Received async msg of type %u, discarding\n",
msg->hdr.operation)); msg->hdr.operation));
TALLOC_FREE(buf);
return NT_STATUS_INVALID_PARAMETER; return NT_STATUS_INVALID_PARAMETER;
} }
@ -519,6 +517,7 @@ static NTSTATUS ctdb_handle_message(const DATA_BLOB *data,
DEBUG(10, ("received CTDB_SRVID_RELEASE_IP\n")); DEBUG(10, ("received CTDB_SRVID_RELEASE_IP\n"));
conn->release_ip_handler((const char *)msg->data, conn->release_ip_handler((const char *)msg->data,
conn->release_ip_priv); conn->release_ip_priv);
TALLOC_FREE(buf);
return NT_STATUS_OK; return NT_STATUS_OK;
} }
@ -540,6 +539,8 @@ static NTSTATUS ctdb_handle_message(const DATA_BLOB *data,
*/ */
message_send_all(conn->msg_ctx, MSG_SMB_UNLOCK, NULL, 0, NULL); message_send_all(conn->msg_ctx, MSG_SMB_UNLOCK, NULL, 0, NULL);
TALLOC_FREE(buf);
return NT_STATUS_OK; return NT_STATUS_OK;
} }
@ -548,17 +549,20 @@ static NTSTATUS ctdb_handle_message(const DATA_BLOB *data,
if (msg->srvid != sys_getpid() && msg->srvid != MSG_SRVID_SAMBA) { if (msg->srvid != sys_getpid() && msg->srvid != MSG_SRVID_SAMBA) {
DEBUG(0,("Got unexpected message with srvid=%llu\n", DEBUG(0,("Got unexpected message with srvid=%llu\n",
(unsigned long long)msg->srvid)); (unsigned long long)msg->srvid));
TALLOC_FREE(buf);
return NT_STATUS_OK; return NT_STATUS_OK;
} }
if (!(msg_rec = ctdb_pull_messaging_rec(NULL, data->length, msg))) { if (!(msg_rec = ctdb_pull_messaging_rec(NULL, length, msg))) {
DEBUG(10, ("ctdb_pull_messaging_rec failed\n")); DEBUG(10, ("ctdb_pull_messaging_rec failed\n"));
TALLOC_FREE(buf);
return NT_STATUS_NO_MEMORY; return NT_STATUS_NO_MEMORY;
} }
messaging_dispatch_rec(conn->msg_ctx, msg_rec); messaging_dispatch_rec(conn->msg_ctx, msg_rec);
TALLOC_FREE(msg_rec); TALLOC_FREE(msg_rec);
TALLOC_FREE(buf);
return NT_STATUS_OK; return NT_STATUS_OK;
} }
@ -1025,7 +1029,7 @@ struct ctdbd_traverse_state {
* Handle a traverse record coming in on the ctdbd connection * Handle a traverse record coming in on the ctdbd connection
*/ */
static NTSTATUS ctdb_traverse_handler(const DATA_BLOB *blob, static NTSTATUS ctdb_traverse_handler(uint8_t *buf, size_t length,
void *private_data) void *private_data)
{ {
struct ctdbd_traverse_state *state = struct ctdbd_traverse_state *state =
@ -1035,11 +1039,11 @@ static NTSTATUS ctdb_traverse_handler(const DATA_BLOB *blob,
struct ctdb_rec_data *d; struct ctdb_rec_data *d;
TDB_DATA key, data; TDB_DATA key, data;
m = (struct ctdb_req_message *)blob->data; m = (struct ctdb_req_message *)buf;
if (blob->length < sizeof(*m) || m->hdr.length != blob->length) { if (length < sizeof(*m) || m->hdr.length != length) {
DEBUG(0, ("Got invalid message of length %d\n", DEBUG(0, ("Got invalid message of length %d\n", (int)length));
(int)blob->length)); TALLOC_FREE(buf);
return NT_STATUS_UNEXPECTED_IO_ERROR; return NT_STATUS_UNEXPECTED_IO_ERROR;
} }
@ -1047,6 +1051,7 @@ static NTSTATUS ctdb_traverse_handler(const DATA_BLOB *blob,
if (m->datalen < sizeof(uint32_t) || m->datalen != d->length) { if (m->datalen < sizeof(uint32_t) || m->datalen != d->length) {
DEBUG(0, ("Got invalid traverse data of length %d\n", DEBUG(0, ("Got invalid traverse data of length %d\n",
(int)m->datalen)); (int)m->datalen));
TALLOC_FREE(buf);
return NT_STATUS_UNEXPECTED_IO_ERROR; return NT_STATUS_UNEXPECTED_IO_ERROR;
} }
@ -1063,6 +1068,7 @@ static NTSTATUS ctdb_traverse_handler(const DATA_BLOB *blob,
if (data.dsize < sizeof(struct ctdb_ltdb_header)) { if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
DEBUG(0, ("Got invalid ltdb header length %d\n", DEBUG(0, ("Got invalid ltdb header length %d\n",
(int)data.dsize)); (int)data.dsize));
TALLOC_FREE(buf);
return NT_STATUS_UNEXPECTED_IO_ERROR; return NT_STATUS_UNEXPECTED_IO_ERROR;
} }
data.dsize -= sizeof(struct ctdb_ltdb_header); data.dsize -= sizeof(struct ctdb_ltdb_header);
@ -1072,6 +1078,7 @@ static NTSTATUS ctdb_traverse_handler(const DATA_BLOB *blob,
state->fn(key, data, state->private_data); state->fn(key, data, state->private_data);
} }
TALLOC_FREE(buf);
return NT_STATUS_OK; return NT_STATUS_OK;
} }

View File

@ -120,33 +120,37 @@ NTSTATUS packet_fd_read_sync(struct packet_context *ctx)
} }
bool packet_handler(struct packet_context *ctx, bool packet_handler(struct packet_context *ctx,
bool (*full_req)(const DATA_BLOB *data, bool (*full_req)(const uint8_t *buf,
size_t available,
size_t *length, size_t *length,
void *private_data), void *priv),
NTSTATUS (*callback)(const DATA_BLOB *data, NTSTATUS (*callback)(uint8_t *buf, size_t length,
void *private_data), void *priv),
void *private_data, void *priv, NTSTATUS *status)
NTSTATUS *status)
{ {
size_t length; size_t length;
DATA_BLOB data; uint8_t *buf;
if (!full_req(&ctx->in, &length, private_data)) { if (!full_req(ctx->in.data, ctx->in.length, &length, priv)) {
return False; return False;
} }
SMB_ASSERT(length <= ctx->in.length); if (length > ctx->in.length) {
*status = NT_STATUS_INTERNAL_ERROR;
return true;
}
data = data_blob(ctx->in.data, length); buf = (uint8_t *)TALLOC_MEMDUP(ctx, ctx->in.data, length);
if (buf == NULL) {
*status = NT_STATUS_NO_MEMORY;
return true;
}
memmove(ctx->in.data, ctx->in.data + length, memmove(ctx->in.data, ctx->in.data + length,
ctx->in.length - length); ctx->in.length - length);
ctx->in.length -= length; ctx->in.length -= length;
*status = callback(&data, private_data); *status = callback(buf, length, priv);
data_blob_free(&data);
return True; return True;
} }