1
0
mirror of https://github.com/samba-team/samba.git synced 2025-07-06 08:59:08 +03:00

r23112: Trim down the message.c API slightly: The messages_pending_for_pid is now

replaced by MSG_FLAG_LOWPRIORITY or'ed into the msg_type. To enable this,
changed the msg_type definitions to hexadecimal.

This way we could theoretically add the MSG_FLAG_NODUPLICATES again, but I
would rather not do this, because that one is racy and can't be guaranteed at
all.
(This used to be commit 3f5eb8a960)
This commit is contained in:
Volker Lendecke
2007-05-24 11:09:37 +00:00
committed by Gerald (Jerry) Carter
parent c3c7e1b423
commit ff0ac5b0e7
4 changed files with 106 additions and 116 deletions

View File

@ -984,7 +984,7 @@ cac: SHOWFLAGS $(LIBMSRPC)
## Perl IDL Compiler ## Perl IDL Compiler
IDL_FILES = unixinfo.idl lsa.idl dfs.idl echo.idl winreg.idl initshutdown.idl \ IDL_FILES = unixinfo.idl lsa.idl dfs.idl echo.idl winreg.idl initshutdown.idl \
srvsvc.idl svcctl.idl eventlog.idl wkssvc.idl netlogon.idl notify.idl \ srvsvc.idl svcctl.idl eventlog.idl wkssvc.idl netlogon.idl notify.idl \
epmapper.idl epmapper.idl messaging.idl
idl: idl:
@IDL_FILES="$(IDL_FILES)" CPP="$(CPP)" PERL="$(PERL)" \ @IDL_FILES="$(IDL_FILES)" CPP="$(CPP)" PERL="$(PERL)" \

View File

@ -22,79 +22,93 @@
#ifndef _MESSAGES_H_ #ifndef _MESSAGES_H_
#define _MESSAGES_H_ #define _MESSAGES_H_
#define MSG_TYPE_MASK 0xFFFF
/* general messages */ /* general messages */
#define MSG_DEBUG 1 #define MSG_DEBUG 0x0001
#define MSG_PING 2 #define MSG_PING 0x0002
#define MSG_PONG 3 #define MSG_PONG 0x0003
#define MSG_PROFILE 4 #define MSG_PROFILE 0x0004
#define MSG_REQ_DEBUGLEVEL 5 #define MSG_REQ_DEBUGLEVEL 0x0005
#define MSG_DEBUGLEVEL 6 #define MSG_DEBUGLEVEL 0x0006
#define MSG_REQ_PROFILELEVEL 7 #define MSG_REQ_PROFILELEVEL 0x0007
#define MSG_PROFILELEVEL 8 #define MSG_PROFILELEVEL 0x0008
#define MSG_REQ_POOL_USAGE 9 #define MSG_REQ_POOL_USAGE 0x0009
#define MSG_POOL_USAGE 10 #define MSG_POOL_USAGE 0x000A
/* If dmalloc is included, set a steady-state mark */ /* If dmalloc is included, set a steady-state mark */
#define MSG_REQ_DMALLOC_MARK 11 #define MSG_REQ_DMALLOC_MARK 0x000B
/* If dmalloc is included, dump to the dmalloc log a description of /* If dmalloc is included, dump to the dmalloc log a description of
* what has changed since the last MARK */ * what has changed since the last MARK */
#define MSG_REQ_DMALLOC_LOG_CHANGED 12 #define MSG_REQ_DMALLOC_LOG_CHANGED 0x000C
#define MSG_SHUTDOWN 13 #define MSG_SHUTDOWN 0x000D
/* nmbd messages */ /* nmbd messages */
#define MSG_FORCE_ELECTION 1001 #define MSG_FORCE_ELECTION 0x0101
#define MSG_WINS_NEW_ENTRY 1002 #define MSG_WINS_NEW_ENTRY 0x0102
#define MSG_SEND_PACKET 1003 #define MSG_SEND_PACKET 0x0103
/* printing messages */ /* printing messages */
/* #define MSG_PRINTER_NOTIFY 2001*/ /* Obsolete */ /* #define MSG_PRINTER_NOTIFY 2001*/ /* Obsolete */
#define MSG_PRINTER_NOTIFY2 2002 #define MSG_PRINTER_NOTIFY2 0x0202
#define MSG_PRINTER_DRVUPGRADE 2101 #define MSG_PRINTER_DRVUPGRADE 0x0203
#define MSG_PRINTERDATA_INIT_RESET 2102 #define MSG_PRINTERDATA_INIT_RESET 0x0204
#define MSG_PRINTER_UPDATE 2103 #define MSG_PRINTER_UPDATE 0x0205
#define MSG_PRINTER_MOD 2104 #define MSG_PRINTER_MOD 0x0206
/* smbd messages */ /* smbd messages */
#define MSG_SMB_CONF_UPDATED 3001 #define MSG_SMB_CONF_UPDATED 0x0301
#define MSG_SMB_FORCE_TDIS 3002 #define MSG_SMB_FORCE_TDIS 0x0302
#define MSG_SMB_SAM_SYNC 3003 #define MSG_SMB_SAM_SYNC 0x0303
#define MSG_SMB_SAM_REPL 3004 #define MSG_SMB_SAM_REPL 0x0304
#define MSG_SMB_UNLOCK 3005 #define MSG_SMB_UNLOCK 0x0305
#define MSG_SMB_BREAK_REQUEST 3006 #define MSG_SMB_BREAK_REQUEST 0x0306
#define MSG_SMB_BREAK_RESPONSE 3007 #define MSG_SMB_BREAK_RESPONSE 0x0307
#define MSG_SMB_ASYNC_LEVEL2_BREAK 3008 #define MSG_SMB_ASYNC_LEVEL2_BREAK 0x0308
#define MSG_SMB_OPEN_RETRY 3009 #define MSG_SMB_OPEN_RETRY 0x0309
#define MSG_SMB_KERNEL_BREAK 3010 #define MSG_SMB_KERNEL_BREAK 0x030A
#define MSG_SMB_FILE_RENAME 3011 #define MSG_SMB_FILE_RENAME 0x030B
#define MSG_SMB_INJECT_FAULT 3012 #define MSG_SMB_INJECT_FAULT 0x030C
#define MSG_SMB_BLOCKING_LOCK_CANCEL 3013 #define MSG_SMB_BLOCKING_LOCK_CANCEL 0x030D
#define MSG_SMB_NOTIFY 3014 #define MSG_SMB_NOTIFY 0x030E
#define MSG_SMB_STAT_CACHE_DELETE 3015 #define MSG_SMB_STAT_CACHE_DELETE 0x030F
/* /*
* Samba4 compatibility * Samba4 compatibility
*/ */
#define MSG_PVFS_NOTIFY 3016 #define MSG_PVFS_NOTIFY 0x0310
/* winbind messages */ /* winbind messages */
#define MSG_WINBIND_FINISHED 4001 #define MSG_WINBIND_FINISHED 0x0401
#define MSG_WINBIND_FORGET_STATE 4002 #define MSG_WINBIND_FORGET_STATE 0x0402
#define MSG_WINBIND_ONLINE 4003 #define MSG_WINBIND_ONLINE 0x0403
#define MSG_WINBIND_OFFLINE 4004 #define MSG_WINBIND_OFFLINE 0x0404
#define MSG_WINBIND_ONLINESTATUS 4005 #define MSG_WINBIND_ONLINESTATUS 0x0405
#define MSG_WINBIND_TRY_TO_GO_ONLINE 4006 #define MSG_WINBIND_TRY_TO_GO_ONLINE 0x0406
#define MSG_WINBIND_FAILED_TO_GO_ONLINE 4007 #define MSG_WINBIND_FAILED_TO_GO_ONLINE 0x0407
/*
* Special flags passed to message_send. Allocated from the top, lets see when
* it collides with the message types in the lower 16 bits :-)
*/
/*
* Under high load, this message can be dropped. Use for notify-style
* messages that are not critical for correct operation.
*/
#define MSG_FLAG_LOWPRIORITY 0x80000000
/* Flags to classify messages - used in message_send_all() */ /* Flags to classify messages - used in message_send_all() */
/* Sender will filter by flag. */ /* Sender will filter by flag. */
#define FLAG_MSG_GENERAL 0x0001 #define FLAG_MSG_GENERAL 0x0001
#define FLAG_MSG_SMBD 0x0002 #define FLAG_MSG_SMBD 0x0002
#define FLAG_MSG_NMBD 0x0004 #define FLAG_MSG_NMBD 0x0004
#define FLAG_MSG_PRINT_NOTIFY 0x0008 #define FLAG_MSG_PRINT_NOTIFY 0x0008
#define FLAG_MSG_PRINT_GENERAL 0x0010 #define FLAG_MSG_PRINT_GENERAL 0x0010
struct server_id { struct server_id {
pid_t pid; pid_t pid;
@ -103,8 +117,6 @@ struct server_id {
struct messaging_context; struct messaging_context;
struct data_blob; struct data_blob;
unsigned int messages_pending_for_pid(struct messaging_context *msg_ctx,
struct server_id pid);
void message_dispatch(struct messaging_context *msg_ctx); void message_dispatch(struct messaging_context *msg_ctx);
BOOL message_send_all(struct messaging_context *msg_ctx, BOOL message_send_all(struct messaging_context *msg_ctx,
int msg_type, int msg_type,

View File

@ -70,6 +70,8 @@ struct messaging_context {
struct server_id id; struct server_id id;
struct event_context *event_ctx; struct event_context *event_ctx;
struct messaging_callback *callbacks; struct messaging_callback *callbacks;
}; };
/**************************************************************************** /****************************************************************************
@ -103,7 +105,7 @@ static void ping_message(struct messaging_context *msg_ctx,
Initialise the messaging functions. Initialise the messaging functions.
****************************************************************************/ ****************************************************************************/
static BOOL message_init(struct messaging_context *msg_ctx) static BOOL message_tdb_init(struct messaging_context *msg_ctx)
{ {
sec_init(); sec_init();
@ -121,14 +123,6 @@ static BOOL message_init(struct messaging_context *msg_ctx)
CatchSignal(SIGUSR1, SIGNAL_CAST sig_usr1); CatchSignal(SIGUSR1, SIGNAL_CAST sig_usr1);
messaging_register(msg_ctx, NULL, MSG_PING, ping_message);
/* Register some debugging related messages */
register_msg_pool_usage(msg_ctx);
register_dmalloc_msgs(msg_ctx);
debug_register_msgs(msg_ctx);
return True; return True;
} }
@ -300,9 +294,9 @@ static NTSTATUS message_notify(struct server_id procid)
Send a message to a particular pid. Send a message to a particular pid.
****************************************************************************/ ****************************************************************************/
static NTSTATUS messaging_tdb_send(TDB_CONTEXT *msg_tdb, static NTSTATUS messaging_tdb_send(struct messaging_context *msg_ctx,
struct server_id pid, int msg_type, struct server_id pid, int msg_type,
const void *buf, size_t len) const DATA_BLOB *data)
{ {
struct messaging_array *msg_array; struct messaging_array *msg_array;
struct messaging_rec *rec; struct messaging_rec *rec;
@ -311,8 +305,8 @@ static NTSTATUS messaging_tdb_send(TDB_CONTEXT *msg_tdb,
TDB_DATA key = message_key_pid(pid); TDB_DATA key = message_key_pid(pid);
/* NULL pointer means implicit length zero. */ /* NULL pointer means implicit length zero. */
if (!buf) { if (!data->data) {
SMB_ASSERT(len == 0); SMB_ASSERT(data->length == 0);
} }
/* /*
@ -326,42 +320,45 @@ static NTSTATUS messaging_tdb_send(TDB_CONTEXT *msg_tdb,
return NT_STATUS_NO_MEMORY; return NT_STATUS_NO_MEMORY;
} }
if (tdb_chainlock(msg_tdb, key) == -1) { if (tdb_chainlock(msg_ctx->tdb, key) == -1) {
TALLOC_FREE(mem_ctx);
return NT_STATUS_LOCK_NOT_GRANTED; return NT_STATUS_LOCK_NOT_GRANTED;
} }
status = messaging_tdb_fetch(msg_tdb, key, mem_ctx, &msg_array); status = messaging_tdb_fetch(msg_ctx->tdb, key, mem_ctx, &msg_array);
if (!NT_STATUS_IS_OK(status)) { if (!NT_STATUS_IS_OK(status)) {
tdb_chainunlock(msg_tdb, key); goto done;
TALLOC_FREE(mem_ctx); }
return status;
if ((msg_type & MSG_FLAG_LOWPRIORITY)
&& (msg_array->num_messages > 1000)) {
DEBUG(5, ("Dropping message for PID %s\n",
procid_str_static(&pid)));
status = NT_STATUS_INSUFFICIENT_RESOURCES;
goto done;
} }
if (!(rec = TALLOC_REALLOC_ARRAY(mem_ctx, msg_array->messages, if (!(rec = TALLOC_REALLOC_ARRAY(mem_ctx, msg_array->messages,
struct messaging_rec, struct messaging_rec,
msg_array->num_messages+1))) { msg_array->num_messages+1))) {
tdb_chainunlock(msg_tdb, key); status = NT_STATUS_NO_MEMORY;
TALLOC_FREE(mem_ctx); goto done;
return NT_STATUS_NO_MEMORY;
} }
rec[msg_array->num_messages].msg_version = MESSAGE_VERSION; rec[msg_array->num_messages].msg_version = MESSAGE_VERSION;
rec[msg_array->num_messages].msg_type = msg_type; rec[msg_array->num_messages].msg_type = msg_type & MSG_TYPE_MASK;
rec[msg_array->num_messages].dest = pid; rec[msg_array->num_messages].dest = pid;
rec[msg_array->num_messages].src = procid_self(); rec[msg_array->num_messages].src = procid_self();
rec[msg_array->num_messages].buf = data_blob_const(buf, len); rec[msg_array->num_messages].buf = *data;
msg_array->messages = rec; msg_array->messages = rec;
msg_array->num_messages += 1; msg_array->num_messages += 1;
status = messaging_tdb_store(msg_tdb, key, msg_array); status = messaging_tdb_store(msg_ctx->tdb, key, msg_array);
tdb_chainunlock(msg_tdb, key);
TALLOC_FREE(mem_ctx);
if (!NT_STATUS_IS_OK(status)) { if (!NT_STATUS_IS_OK(status)) {
return status; goto done;
} }
status = message_notify(pid); status = message_notify(pid);
@ -369,34 +366,15 @@ static NTSTATUS messaging_tdb_send(TDB_CONTEXT *msg_tdb,
if (NT_STATUS_EQUAL(status, NT_STATUS_INVALID_HANDLE)) { if (NT_STATUS_EQUAL(status, NT_STATUS_INVALID_HANDLE)) {
DEBUG(2, ("pid %s doesn't exist - deleting messages record\n", DEBUG(2, ("pid %s doesn't exist - deleting messages record\n",
procid_str_static(&pid))); procid_str_static(&pid)));
tdb_delete(msg_tdb, message_key_pid(pid)); tdb_delete(msg_ctx->tdb, message_key_pid(pid));
} }
done:
tdb_chainunlock(msg_ctx->tdb, key);
TALLOC_FREE(mem_ctx);
return status; return status;
} }
/****************************************************************************
Count the messages pending for a particular pid. Expensive....
****************************************************************************/
unsigned int messages_pending_for_pid(struct messaging_context *msg_ctx,
struct server_id pid)
{
struct messaging_array *msg_array;
unsigned int result;
if (!NT_STATUS_IS_OK(messaging_tdb_fetch(msg_ctx->tdb,
message_key_pid(pid), NULL,
&msg_array))) {
DEBUG(10, ("messaging_tdb_fetch failed\n"));
return 0;
}
result = msg_array->num_messages;
TALLOC_FREE(msg_array);
return result;
}
/**************************************************************************** /****************************************************************************
Retrieve all messages for the current process. Retrieve all messages for the current process.
****************************************************************************/ ****************************************************************************/
@ -619,11 +597,19 @@ struct messaging_context *messaging_init(TALLOC_CTX *mem_ctx,
ctx->id = server_id; ctx->id = server_id;
ctx->event_ctx = ev; ctx->event_ctx = ev;
if (!message_init(ctx)) { if (!message_tdb_init(ctx)) {
DEBUG(0, ("message_init failed: %s\n", strerror(errno))); DEBUG(0, ("message_init failed: %s\n", strerror(errno)));
TALLOC_FREE(ctx); TALLOC_FREE(ctx);
} }
messaging_register(ctx, NULL, MSG_PING, ping_message);
/* Register some debugging related messages */
register_msg_pool_usage(ctx);
register_dmalloc_msgs(ctx);
debug_register_msgs(ctx);
return ctx; return ctx;
} }
@ -688,11 +674,10 @@ void messaging_deregister(struct messaging_context *ctx, uint32_t msg_type,
Send a message to a particular server Send a message to a particular server
*/ */
NTSTATUS messaging_send(struct messaging_context *msg_ctx, NTSTATUS messaging_send(struct messaging_context *msg_ctx,
struct server_id server, struct server_id server, uint32_t msg_type,
uint32_t msg_type, const DATA_BLOB *data) const DATA_BLOB *data)
{ {
return messaging_tdb_send(msg_ctx->tdb, server, msg_type, return messaging_tdb_send(msg_ctx, server, msg_type, data);
data->data, data->length);
} }
NTSTATUS messaging_send_buf(struct messaging_context *msg_ctx, NTSTATUS messaging_send_buf(struct messaging_context *msg_ctx,

View File

@ -183,16 +183,9 @@ static void print_notify_send_messages_to_printer(struct messaging_context *msg_
} }
for (i = 0; i < num_pids; i++) { for (i = 0; i < num_pids; i++) {
unsigned int q_len = messages_pending_for_pid(
msg_ctx, pid_to_procid(pid_list[i]));
if (q_len > 1000) {
DEBUG(5, ("print_notify_send_messages_to_printer: discarding notify to printer %s as queue length = %u\n",
printer, q_len ));
continue;
}
messaging_send_buf(msg_ctx, messaging_send_buf(msg_ctx,
pid_to_procid(pid_list[i]), pid_to_procid(pid_list[i]),
MSG_PRINTER_NOTIFY2, MSG_PRINTER_NOTIFY2 | MSG_FLAG_LOWPRIORITY,
(uint8 *)buf, offset); (uint8 *)buf, offset);
if ((timeout != 0) && timeval_expired(&end_time)) { if ((timeout != 0) && timeval_expired(&end_time)) {