smbd/notify.c

ソースコードを見る。

データ構造
struct	notify_change_request
struct	notify_mid_map
関数
static void	notify_fsp (files_struct *fsp, uint32 action, const char *name)
static BOOL	notify_change_record_identical (struct notify_change *c1, struct notify_change *c2)
static BOOL	notify_marshall_changes (int num_changes, uint32 max_offset, struct notify_change *changes, prs_struct *ps)
static void	change_notify_reply_packet (const char *request_buf, NTSTATUS error_code)
void	change_notify_reply (const char *request_buf, uint32 max_param, struct notify_change_buf *notify_buf)
static void	notify_callback (void *private_data, const struct notify_event *e)
NTSTATUS	change_notify_create (struct files_struct *fsp, uint32 filter, BOOL recursive)
NTSTATUS	change_notify_add_request (const char *inbuf, uint32 max_param, uint32 filter, BOOL recursive, struct files_struct *fsp)
static void	change_notify_remove_request (struct notify_change_request *remove_req)
void	remove_pending_change_notify_requests_by_mid (uint16 mid)
void	remove_pending_change_notify_requests_by_fid (files_struct *fsp, NTSTATUS status)
void	notify_fname (connection_struct *conn, uint32 action, uint32 filter, const char *path)
char *	notify_filter_string (TALLOC_CTX *mem_ctx, uint32 filter)
sys_notify_context *	sys_notify_context_create (connection_struct *conn, TALLOC_CTX *mem_ctx, struct event_context *ev)
NTSTATUS	sys_notify_watch (struct sys_notify_context *ctx, struct notify_entry *e, void(*callback)(struct sys_notify_context *ctx, void *private_data, struct notify_event *ev), void *private_data, void *handle)
変数
int	max_send
static struct notify_mid_map *	notify_changes_by_mid

---

関数

static void notify_fsp	(	files_struct *	fsp,
		uint32	action,
		const char *	name
	)			[static]

notify.c の 373 行で定義されています。

参照先 change・change_notify_remove_request()・change_notify_reply()・notify_change_buf::changes・notify_change_request::max_param・files_struct::notify・notify_change_buf::num_changes・notify_change_request::request_buf・notify_change_buf::requests・string_replace()・talloc_strdup().

参照元 notify_callback().

{
        struct notify_change *change, *changes;
        pstring name2;

        if (fsp->notify == NULL) {
                /*
                 * Nobody is waiting, don't queue
                 */
                return;
        }

        pstrcpy(name2, name);
        string_replace(name2, '/', '\\');

        /*
         * Someone has triggered a notify previously, queue the change for
         * later.
         */

        if ((fsp->notify->num_changes > 1000) || (name == NULL)) {
                /*
                 * The real number depends on the client buf, just provide a
                 * guard against a DoS here.
                 */
                TALLOC_FREE(fsp->notify->changes);
                fsp->notify->num_changes = -1;
                return;
        }

        if (fsp->notify->num_changes == -1) {
                return;
        }

        if (!(changes = TALLOC_REALLOC_ARRAY(
                      fsp->notify, fsp->notify->changes,
                      struct notify_change, fsp->notify->num_changes+1))) {
                DEBUG(0, ("talloc_realloc failed\n"));
                return;
        }

        fsp->notify->changes = changes;

        change = &(fsp->notify->changes[fsp->notify->num_changes]);

        if (!(change->name = talloc_strdup(changes, name2))) {
                DEBUG(0, ("talloc_strdup failed\n"));
                return;
        }

        change->action = action;
        fsp->notify->num_changes += 1;

        if (fsp->notify->requests == NULL) {
                /*
                 * Nobody is waiting, so don't send anything. The ot
                 */
                return;
        }

        if (action == NOTIFY_ACTION_OLD_NAME) {
                /*
                 * We have to send the two rename events in one reply. So hold
                 * the first part back.
                 */
                return;
        }

        /*
         * Someone is waiting for the change, trigger the reply immediately.
         *
         * TODO: do we have to walk the lists of requests pending?
         */

        change_notify_reply(fsp->notify->requests->request_buf,
                            fsp->notify->requests->max_param,
                            fsp->notify);

        change_notify_remove_request(fsp->notify->requests);
}

static BOOL notify_change_record_identical	(	struct notify_change *	c1,
		struct notify_change *	c2
	)			[static]

notify.c の 53 行で定義されています。

参照先 notify_change::action・notify_change::name.

参照元 notify_marshall_changes().

{
        /* Note this is deliberately case sensitive. */
        if (c1->action == c2->action &&
                        strcmp(c1->name, c2->name) == 0) {
                return True;
        }
        return False;
}

static BOOL notify_marshall_changes	(	int	num_changes,
		uint32	max_offset,
		struct notify_change *	changes,
		prs_struct *	ps
	)			[static]

notify.c の 64 行で定義されています。

参照先 UNISTR::buffer・c・CH_UNIX・CH_UTF16LE・convert_string_allocate()・notify_change_record_identical().

参照元 change_notify_reply().

{
        int i;
        UNISTR uni_name;

        uni_name.buffer = NULL;

        for (i=0; i<num_changes; i++) {
                struct notify_change *c;
                size_t namelen;
                uint32 u32_tmp; /* Temp arg to prs_uint32 to avoid
                                 * signed/unsigned issues */

                /* Coalesce any identical records. */
                while (i+1 < num_changes &&
                        notify_change_record_identical(&changes[i],
                                                &changes[i+1])) {
                        i++;
                }

                c = &changes[i];

                namelen = convert_string_allocate(
                        NULL, CH_UNIX, CH_UTF16LE, c->name, strlen(c->name)+1,
                        &uni_name.buffer, True);
                if ((namelen == -1) || (uni_name.buffer == NULL)) {
                        goto fail;
                }

                namelen -= 2;   /* Dump NULL termination */

                /*
                 * Offset to next entry, only if there is one
                 */

                u32_tmp = (i == num_changes-1) ? 0 : namelen + 12;
                if (!prs_uint32("offset", ps, 1, &u32_tmp)) goto fail;

                u32_tmp = c->action;
                if (!prs_uint32("action", ps, 1, &u32_tmp)) goto fail;

                u32_tmp = namelen;
                if (!prs_uint32("namelen", ps, 1, &u32_tmp)) goto fail;

                if (!prs_unistr("name", ps, 1, &uni_name)) goto fail;

                /*
                 * Not NULL terminated, decrease by the 2 UCS2 \0 chars
                 */
                prs_set_offset(ps, prs_offset(ps)-2);

                SAFE_FREE(uni_name.buffer);

                if (prs_offset(ps) > max_offset) {
                        /* Too much data for client. */
                        return False;
                }
        }

        return True;

 fail:
        SAFE_FREE(uni_name.buffer);
        return False;
}

static void change_notify_reply_packet	(	const char *	request_buf,
		NTSTATUS	error_code
	)			[static]

notify.c の 137 行で定義されています。

参照先 construct_reply_common()・exit_server_cleanly()・send_smb()・set_message()・show_msg()・smbd_server_fd().

参照元 change_notify_reply()・remove_pending_change_notify_requests_by_fid().

{
        char outbuf[smb_size+38];

        memset(outbuf, '\0', sizeof(outbuf));
        construct_reply_common(request_buf, outbuf);

        ERROR_NT(error_code);

        /*
         * Seems NT needs a transact command with an error code
         * in it. This is a longer packet than a simple error.
         */
        set_message(outbuf,18,0,False);

        show_msg(outbuf);
        if (!send_smb(smbd_server_fd(),outbuf))
                exit_server_cleanly("change_notify_reply_packet: send_smb "
                                    "failed.");
}

void change_notify_reply	(	const char *	request_buf,
		uint32	max_param,
		struct notify_change_buf *	notify_buf
	)

notify.c の 159 行で定義されています。

参照先 change_notify_reply_packet()・notify_change_buf::changes・construct_reply_common()・exit_server()・max_send・notify_marshall_changes()・notify_change_buf::num_changes・prs_init()・prs_mem_free()・send_nt_replies().

参照元 call_nt_transact_notify_change()・notify_fsp().

{
        char *outbuf = NULL;
        prs_struct ps;
        size_t buflen;

        if (notify_buf->num_changes == -1) {
                change_notify_reply_packet(request_buf, NT_STATUS_OK);
                notify_buf->num_changes = 0;
                return;
        }

        prs_init(&ps, 0, NULL, MARSHALL);

        if (!notify_marshall_changes(notify_buf->num_changes, max_param,
                                        notify_buf->changes, &ps)) {
                /*
                 * We exceed what the client is willing to accept. Send
                 * nothing.
                 */
                change_notify_reply_packet(request_buf, NT_STATUS_OK);
                goto done;
        }

        buflen = smb_size+38+prs_offset(&ps) + 4 /* padding */;

        if (buflen > max_send) {
                /*
                 * We exceed what the client is willing to accept. Send
                 * nothing.
                 */
                change_notify_reply_packet(request_buf, NT_STATUS_OK);
                goto done;
        }

        if (!(outbuf = SMB_MALLOC_ARRAY(char, buflen))) {
                change_notify_reply_packet(request_buf, NT_STATUS_NO_MEMORY);
                goto done;
        }

        construct_reply_common(request_buf, outbuf);

        if (send_nt_replies(outbuf, buflen, NT_STATUS_OK, prs_data_p(&ps),
                            prs_offset(&ps), NULL, 0) == -1) {
                exit_server("change_notify_reply_packet: send_smb failed.");
        }

 done:
        SAFE_FREE(outbuf);
        prs_mem_free(&ps);

        TALLOC_FREE(notify_buf->changes);
        notify_buf->num_changes = 0;
}

static void notify_callback	(	void *	private_data,
		const struct notify_event *	e
	)			[static]

notify.c の 215 行で定義されています。

参照先 notify_event::action・notify_change_request::fsp・files_struct::fsp_name・notify_fsp()・notify_event::path.

参照元 change_notify_create().

{
        files_struct *fsp = (files_struct *)private_data;
        DEBUG(10, ("notify_callback called for %s\n", fsp->fsp_name));
        notify_fsp(fsp, e->action, e->path);
}

NTSTATUS change_notify_create	(	struct files_struct *	fsp,
		uint32	filter,
		BOOL	recursive
	)

notify.c の 222 行で定義されています。

参照先 asprintf()・files_struct::conn・connection_struct::connectpath・notify_entry::filter・files_struct::fsp_name・fullpath()・files_struct::notify・notify_add()・notify_callback()・connection_struct::notify_ctx・notify_entry::path・status・notify_entry::subdir_filter.

参照元 call_nt_transact_notify_change().

{
        char *fullpath;
        struct notify_entry e;
        NTSTATUS status;

        SMB_ASSERT(fsp->notify == NULL);

        if (!(fsp->notify = TALLOC_ZERO_P(NULL, struct notify_change_buf))) {
                DEBUG(0, ("talloc failed\n"));
                return NT_STATUS_NO_MEMORY;
        }

        if (asprintf(&fullpath, "%s/%s", fsp->conn->connectpath,
                     fsp->fsp_name) == -1) {
                DEBUG(0, ("asprintf failed\n"));
                return NT_STATUS_NO_MEMORY;
        }

        e.path = fullpath;
        e.filter = filter;
        e.subdir_filter = 0;
        if (recursive) {
                e.subdir_filter = filter;
        }

        status = notify_add(fsp->conn->notify_ctx, &e, notify_callback, fsp);
        SAFE_FREE(fullpath);

        return status;
}

NTSTATUS change_notify_add_request	(	const char *	inbuf,
		uint32	max_param,
		uint32	filter,
		BOOL	recursive,
		struct files_struct *	fsp
	)

notify.c の 255 行で定義されています。

参照先 notify_change_request::backend_data・notify_change_request::filter・notify_change_request::fsp・notify_change_request::max_param・notify_mid_map::mid・notify_change_request::mid_map・files_struct::notify・notify_changes_by_mid・notify_mid_map::req・notify_change_request::request_buf・notify_change_buf::requests・srv_defer_sign_response().

参照元 call_nt_transact_notify_change().

{
        struct notify_change_request *request = NULL;
        struct notify_mid_map *map = NULL;

        if (!(request = SMB_MALLOC_P(struct notify_change_request))
            || !(map = SMB_MALLOC_P(struct notify_mid_map))) {
                SAFE_FREE(request);
                return NT_STATUS_NO_MEMORY;
        }

        request->mid_map = map;
        map->req = request;

        memcpy(request->request_buf, inbuf, sizeof(request->request_buf));
        request->max_param = max_param;
        request->filter = filter;
        request->fsp = fsp;
        request->backend_data = NULL;

        DLIST_ADD_END(fsp->notify->requests, request,
                      struct notify_change_request *);

        map->mid = SVAL(inbuf, smb_mid);
        DLIST_ADD(notify_changes_by_mid, map);

        /* Push the MID of this packet on the signing queue. */
        srv_defer_sign_response(SVAL(inbuf,smb_mid));

        return NT_STATUS_OK;
}

static void change_notify_remove_request

(

struct notify_change_request *

remove_req

)

[static]

notify.c の 289 行で定義されています。

参照先 notify_change_request::backend_data・notify_change_request::fsp・notify_change_request::mid_map・notify_change_request::next・files_struct::notify・notify_changes_by_mid・notify_change_buf::requests・smb_panic().

参照元 notify_fsp()・remove_pending_change_notify_requests_by_fid().

{
        files_struct *fsp;
        struct notify_change_request *req;

        /*
         * Paranoia checks, the fsp referenced must must have the request in
         * its list of pending requests
         */

        fsp = remove_req->fsp;
        SMB_ASSERT(fsp->notify != NULL);

        for (req = fsp->notify->requests; req; req = req->next) {
                if (req == remove_req) {
                        break;
                }
        }

        if (req == NULL) {
                smb_panic("notify_req not found in fsp's requests\n");
        }

        DLIST_REMOVE(fsp->notify->requests, req);
        DLIST_REMOVE(notify_changes_by_mid, req->mid_map);
        SAFE_FREE(req->mid_map);
        TALLOC_FREE(req->backend_data);
        SAFE_FREE(req);
}

void remove_pending_change_notify_requests_by_mid

(

uint16

mid

)

notify.c の 323 行で定義されています。

参照先 notify_mid_map::mid・notify_mid_map::next・notify_changes_by_mid.

参照元 reply_ntcancel().

{
        struct notify_mid_map *map;

        for (map = notify_changes_by_mid; map; map = map->next) {
                if (map->mid == mid) {
                        break;
                }
        }

        if (map == NULL) {
                return;
        }

        change_notify_reply_packet(map->req->request_buf, NT_STATUS_CANCELLED);
        change_notify_remove_request(map->req);
}

void remove_pending_change_notify_requests_by_fid	(	files_struct *	fsp,
		NTSTATUS	status
	)

notify.c の 345 行で定義されています。

参照先 change_notify_remove_request()・change_notify_reply_packet()・files_struct::notify・notify_change_request::request_buf・notify_change_buf::requests・status.

{
        if (fsp->notify == NULL) {
                return;
        }

        while (fsp->notify->requests != NULL) {
                change_notify_reply_packet(
                        fsp->notify->requests->request_buf, status);
                change_notify_remove_request(fsp->notify->requests);
        }
}

void notify_fname	(	connection_struct *	conn,
		uint32	action,
		uint32	filter,
		const char *	path
	)

notify.c の 359 行で定義されています。

参照先 asprintf()・connection_struct::connectpath・fullpath()・connection_struct::notify_ctx・notify_trigger().

参照元 mkdir_internal()・notify_rename()・open_file()・rmdir_internals()・set_ea_dos_attribute()・smb_set_file_time()・unlink_internals()・vfs_set_filelen().

{
        char *fullpath;

        if (asprintf(&fullpath, "%s/%s", conn->connectpath, path) == -1) {
                DEBUG(0, ("asprintf failed\n"));
                return;
        }

        notify_trigger(conn->notify_ctx, action, filter, fullpath);
        SAFE_FREE(fullpath);
}

char* notify_filter_string	(	TALLOC_CTX *	mem_ctx,
		uint32	filter
	)

notify.c の 454 行で定義されています。

参照先 result・talloc_asprintf_append()・talloc_strdup().

参照元 call_nt_transact_notify_change().

{
        char *result = NULL;

        result = talloc_strdup(mem_ctx, "");

        if (filter & FILE_NOTIFY_CHANGE_FILE_NAME)
                result = talloc_asprintf_append(result, "FILE_NAME|");
        if (filter & FILE_NOTIFY_CHANGE_DIR_NAME)
                result = talloc_asprintf_append(result, "DIR_NAME|");
        if (filter & FILE_NOTIFY_CHANGE_ATTRIBUTES)
                result = talloc_asprintf_append(result, "ATTRIBUTES|");
        if (filter & FILE_NOTIFY_CHANGE_SIZE)
                result = talloc_asprintf_append(result, "SIZE|");
        if (filter & FILE_NOTIFY_CHANGE_LAST_WRITE)
                result = talloc_asprintf_append(result, "LAST_WRITE|");
        if (filter & FILE_NOTIFY_CHANGE_LAST_ACCESS)
                result = talloc_asprintf_append(result, "LAST_ACCESS|");
        if (filter & FILE_NOTIFY_CHANGE_CREATION)
                result = talloc_asprintf_append(result, "CREATION|");
        if (filter & FILE_NOTIFY_CHANGE_EA)
                result = talloc_asprintf_append(result, "EA|");
        if (filter & FILE_NOTIFY_CHANGE_SECURITY)
                result = talloc_asprintf_append(result, "SECURITY|");
        if (filter & FILE_NOTIFY_CHANGE_STREAM_NAME)
                result = talloc_asprintf_append(result, "STREAM_NAME|");
        if (filter & FILE_NOTIFY_CHANGE_STREAM_SIZE)
                result = talloc_asprintf_append(result, "STREAM_SIZE|");
        if (filter & FILE_NOTIFY_CHANGE_STREAM_WRITE)
                result = talloc_asprintf_append(result, "STREAM_WRITE|");

        if (result == NULL) return NULL;
        if (*result == '\0') return result;

        result[strlen(result)-1] = '\0';
        return result;
}

struct sys_notify_context* sys_notify_context_create	(	connection_struct *	conn,
		TALLOC_CTX *	mem_ctx,
		struct event_context *	ev
	)

notify.c の 492 行で定義されています。

参照先 sys_notify_context::conn・ctx・sys_notify_context::ev.

参照元 notify_init().

{
        struct sys_notify_context *ctx;

        if (!(ctx = TALLOC_P(mem_ctx, struct sys_notify_context))) {
                DEBUG(0, ("talloc failed\n"));
                return NULL;
        }

        ctx->ev = ev;
        ctx->conn = conn;
        ctx->private_data = NULL;
        return ctx;
}

NTSTATUS sys_notify_watch	(	struct sys_notify_context *	ctx,
		struct notify_entry *	e,
		void(*)(struct sys_notify_context *ctx, void *private_data, struct notify_event *ev)	callback,
		void *	private_data,
		void *	handle
	)

notify.c の 509 行で定義されています。

参照元 notify_add().

{
        return SMB_VFS_NOTIFY_WATCH(ctx->conn, ctx, e, callback, private_data,
                                    handle);
}

---

変数

int max_send

process.c の 38 行で定義されています。

struct notify_mid_map* notify_changes_by_mid [static]

notify.c の 40 行で定義されています。

参照元 change_notify_add_request()・change_notify_remove_request()・remove_pending_change_notify_requests_by_mid().

---