-/*
+/*
Unix SMB/CIFS implementation.
store smbd profiling information in shared memory
Copyright (C) Andrew Tridgell 1999
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 2 of the License, or
+ the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
-
+
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
-
+
You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "includes.h"
-
-#ifdef WITH_PROFILE
-#define IPC_PERMS ((S_IRUSR | S_IWUSR) | S_IRGRP | S_IROTH)
-#endif /* WITH_PROFILE */
-
-#ifdef WITH_PROFILE
-static int shm_id;
-static BOOL read_only;
-#if defined(HAVE_CLOCK_GETTIME)
-clockid_t __profile_clock;
-BOOL have_profiling_clock = False;
-#endif
+#include "system/shmem.h"
+#include "system/filesys.h"
+#include "system/time.h"
+#include "messages.h"
+#include "smbprofile.h"
+#include "lib/tdb_wrap/tdb_wrap.h"
+#include <tevent.h>
+#include "../lib/crypto/crypto.h"
+
+#ifdef HAVE_SYS_RESOURCE_H
+#include <sys/resource.h>
#endif
-struct profile_header *profile_h;
struct profile_stats *profile_p;
-
-BOOL do_profile_flag = False;
-BOOL do_profile_times = False;
+struct smbprofile_global_state smbprofile_state;
/****************************************************************************
Set a profiling level.
****************************************************************************/
-void set_profile_level(int level, struct process_id src)
+void set_profile_level(int level, const struct server_id *src)
{
-#ifdef WITH_PROFILE
+ SMB_ASSERT(smbprofile_state.internal.db != NULL);
+
switch (level) {
case 0: /* turn off profiling */
- do_profile_flag = False;
- do_profile_times = False;
+ smbprofile_state.config.do_count = false;
+ smbprofile_state.config.do_times = false;
DEBUG(1,("INFO: Profiling turned OFF from pid %d\n",
- (int)procid_to_pid(&src)));
+ (int)procid_to_pid(src)));
break;
case 1: /* turn on counter profiling only */
- do_profile_flag = True;
- do_profile_times = False;
+ smbprofile_state.config.do_count = true;
+ smbprofile_state.config.do_times = false;
DEBUG(1,("INFO: Profiling counts turned ON from pid %d\n",
- (int)procid_to_pid(&src)));
+ (int)procid_to_pid(src)));
break;
case 2: /* turn on complete profiling */
-
-#if defined(HAVE_CLOCK_GETTIME)
- if (!have_profiling_clock) {
- do_profile_flag = True;
- do_profile_times = False;
- DEBUG(1,("INFO: Profiling counts turned ON from "
- "pid %d\n", (int)procid_to_pid(&src)));
- DEBUGADD(1,("INFO: Profiling times disabled "
- "due to lack of a suitable clock\n"));
- break;
- }
-#endif
-
- do_profile_flag = True;
- do_profile_times = True;
+ smbprofile_state.config.do_count = true;
+ smbprofile_state.config.do_times = true;
DEBUG(1,("INFO: Full profiling turned ON from pid %d\n",
- (int)procid_to_pid(&src)));
+ (int)procid_to_pid(src)));
break;
case 3: /* reset profile values */
- memset((char *)profile_p, 0, sizeof(*profile_p));
+ ZERO_STRUCT(profile_p->values);
+ tdb_wipe_all(smbprofile_state.internal.db->tdb);
DEBUG(1,("INFO: Profiling values cleared from pid %d\n",
- (int)procid_to_pid(&src)));
+ (int)procid_to_pid(src)));
break;
}
-#else /* WITH_PROFILE */
- DEBUG(1,("INFO: Profiling support unavailable in this build.\n"));
-#endif /* WITH_PROFILE */
}
/****************************************************************************
receive a set profile level message
****************************************************************************/
-void profile_message(int msg_type, struct process_id src, void *buf, size_t len, void *private_data)
+static void profile_message(struct messaging_context *msg_ctx,
+ void *private_data,
+ uint32_t msg_type,
+ struct server_id src,
+ DATA_BLOB *data)
{
int level;
- memcpy(&level, buf, sizeof(int));
- set_profile_level(level, src);
+ if (data->length != sizeof(level)) {
+ DEBUG(0, ("got invalid profile message\n"));
+ return;
+ }
+
+ memcpy(&level, data->data, sizeof(level));
+ set_profile_level(level, &src);
}
/****************************************************************************
receive a request profile level message
****************************************************************************/
-void reqprofile_message(int msg_type, struct process_id src,
- void *buf, size_t len, void *private_data)
+static void reqprofile_message(struct messaging_context *msg_ctx,
+ void *private_data,
+ uint32_t msg_type,
+ struct server_id src,
+ DATA_BLOB *data)
{
int level;
-#ifdef WITH_PROFILE
- level = 1 + (do_profile_flag?2:0) + (do_profile_times?4:0);
-#else
- level = 0;
-#endif
+ level = 1;
+ if (smbprofile_state.config.do_count) {
+ level += 2;
+ }
+ if (smbprofile_state.config.do_times) {
+ level += 4;
+ }
+
DEBUG(1,("INFO: Received REQ_PROFILELEVEL message from PID %u\n",
(unsigned int)procid_to_pid(&src)));
- message_send_pid(src, MSG_PROFILELEVEL, &level, sizeof(int), True);
+ messaging_send_buf(msg_ctx, src, MSG_PROFILELEVEL,
+ (uint8_t *)&level, sizeof(level));
}
/*******************************************************************
open the profiling shared memory area
******************************************************************/
-#ifdef WITH_PROFILE
-
-#ifdef HAVE_CLOCK_GETTIME
-
-/* Find a clock. Just because the definition for a particular clock ID is
- * present doesn't mean the system actually supports it.
- */
-static void init_clock_gettime(void)
+bool profile_setup(struct messaging_context *msg_ctx, bool rdonly)
{
- struct timespec ts;
+ unsigned char tmp[16] = {};
+ MD5_CTX md5;
+ char *db_name;
- have_profiling_clock = False;
+ if (smbprofile_state.internal.db != NULL) {
+ return true;
+ }
-#ifdef HAVE_CLOCK_PROCESS_CPUTIME_ID
- /* CLOCK_PROCESS_CPUTIME_ID is sufficiently fast that the
- * always profiling times is plausible. Unfortunately on Linux
- * it is only accurate if we can guarantee we will not be scheduled
- * scheduled onto a different CPU between samples. Until there is
- * some way to set processor affinity, we can only use this on
- * uniprocessors.
- */
- if (!this_is_smp()) {
- if (clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &ts) == 0) {
- DEBUG(10, ("Using CLOCK_PROCESS_CPUTIME_ID "
- "for profile_clock\n"));
- __profile_clock = CLOCK_PROCESS_CPUTIME_ID;
- have_profiling_clock = True;
- }
+ db_name = cache_path("smbprofile.tdb");
+ if (db_name == NULL) {
+ return false;
}
-#endif
-#ifdef HAVE_CLOCK_MONOTONIC
- if (!have_profiling_clock &&
- clock_gettime(CLOCK_MONOTONIC, &ts) == 0) {
- DEBUG(10, ("Using CLOCK_MONOTONIC for profile_clock\n"));
- __profile_clock = CLOCK_MONOTONIC;
- have_profiling_clock = True;
+ smbprofile_state.internal.db = tdb_wrap_open(
+ NULL, db_name, 0,
+ rdonly ? 0 : TDB_CLEAR_IF_FIRST|TDB_MUTEX_LOCKING,
+ O_CREAT | (rdonly ? O_RDONLY : O_RDWR), 0644);
+ if (smbprofile_state.internal.db == NULL) {
+ return false;
}
-#endif
-#ifdef HAVE_CLOCK_REALTIME
- /* POSIX says that CLOCK_REALTIME should be defined everywhere
- * where we have clock_gettime...
- */
- if (!have_profiling_clock &&
- clock_gettime(CLOCK_REALTIME, &ts) == 0) {
- __profile_clock = CLOCK_REALTIME;
- have_profiling_clock = True;
+ if (msg_ctx != NULL) {
+ messaging_register(msg_ctx, NULL, MSG_PROFILE,
+ profile_message);
+ messaging_register(msg_ctx, NULL, MSG_REQ_PROFILELEVEL,
+ reqprofile_message);
+ }
- SMB_WARN(__profile_clock != CLOCK_REALTIME,
- ("forced to use a slow profiling clock"));
+ MD5Init(&md5);
+
+ MD5Update(&md5,
+ (const uint8_t *)&smbprofile_state.stats.global,
+ sizeof(smbprofile_state.stats.global));
+
+#define __UPDATE(str) do { \
+ MD5Update(&md5, (const uint8_t *)str, strlen(str)); \
+} while(0)
+#define SMBPROFILE_STATS_START
+#define SMBPROFILE_STATS_SECTION_START(name, display) do { \
+ __UPDATE(#name "+" #display); \
+} while(0);
+#define SMBPROFILE_STATS_COUNT(name) do { \
+ __UPDATE(#name "+count"); \
+} while(0);
+#define SMBPROFILE_STATS_TIME(name) do { \
+ __UPDATE(#name "+time"); \
+} while(0);
+#define SMBPROFILE_STATS_BASIC(name) do { \
+ __UPDATE(#name "+count"); \
+ __UPDATE(#name "+time"); \
+} while(0);
+#define SMBPROFILE_STATS_BYTES(name) do { \
+ __UPDATE(#name "+count"); \
+ __UPDATE(#name "+time"); \
+ __UPDATE(#name "+idle"); \
+ __UPDATE(#name "+bytes"); \
+} while(0);
+#define SMBPROFILE_STATS_IOBYTES(name) do { \
+ __UPDATE(#name "+count"); \
+ __UPDATE(#name "+time"); \
+ __UPDATE(#name "+idle"); \
+ __UPDATE(#name "+inbytes"); \
+ __UPDATE(#name "+outbytes"); \
+} while(0);
+#define SMBPROFILE_STATS_SECTION_END
+#define SMBPROFILE_STATS_END
+ SMBPROFILE_STATS_ALL_SECTIONS
+#undef __UPDATE
+#undef SMBPROFILE_STATS_START
+#undef SMBPROFILE_STATS_SECTION_START
+#undef SMBPROFILE_STATS_COUNT
+#undef SMBPROFILE_STATS_TIME
+#undef SMBPROFILE_STATS_BASIC
+#undef SMBPROFILE_STATS_BYTES
+#undef SMBPROFILE_STATS_IOBYTES
+#undef SMBPROFILE_STATS_SECTION_END
+#undef SMBPROFILE_STATS_END
+
+ MD5Final(tmp, &md5);
+
+ profile_p = &smbprofile_state.stats.global;
+
+ profile_p->magic = BVAL(tmp, 0);
+ if (profile_p->magic == 0) {
+ profile_p->magic = BVAL(tmp, 8);
}
-#endif
+ return True;
+}
- SMB_WARN(have_profiling_clock == True,
- ("could not find a working clock for profiling"));
- return;
+void smbprofile_dump_setup(struct tevent_context *ev)
+{
+ TALLOC_FREE(smbprofile_state.internal.te);
+ smbprofile_state.internal.ev = ev;
}
-#endif
-BOOL profile_setup(BOOL rdonly)
+static void smbprofile_dump_timer(struct tevent_context *ev,
+ struct tevent_timer *te,
+ struct timeval current_time,
+ void *private_data)
{
- struct shmid_ds shm_ds;
+ smbprofile_dump();
+}
- read_only = rdonly;
+void smbprofile_dump_schedule_timer(void)
+{
+ struct timeval tv;
-#ifdef HAVE_CLOCK_GETTIME
- init_clock_gettime();
-#endif
+ GetTimeOfDay(&tv);
+ tv.tv_sec += 1;
+
+ smbprofile_state.internal.te = tevent_add_timer(
+ smbprofile_state.internal.ev,
+ smbprofile_state.internal.ev,
+ tv,
+ smbprofile_dump_timer,
+ NULL);
+}
+
+static int profile_stats_parser(TDB_DATA key, TDB_DATA value,
+ void *private_data)
+{
+ struct profile_stats *s = private_data;
- again:
- /* try to use an existing key */
- shm_id = shmget(PROF_SHMEM_KEY, 0, 0);
-
- /* if that failed then create one. There is a race condition here
- if we are running from inetd. Bad luck. */
- if (shm_id == -1) {
- if (read_only) return False;
- shm_id = shmget(PROF_SHMEM_KEY, sizeof(*profile_h),
- IPC_CREAT | IPC_EXCL | IPC_PERMS);
+ if (value.dsize != sizeof(struct profile_stats)) {
+ *s = (struct profile_stats) {};
+ return 0;
}
-
- if (shm_id == -1) {
- DEBUG(0,("Can't create or use IPC area. Error was %s\n",
- strerror(errno)));
- return False;
- }
-
-
- profile_h = (struct profile_header *)shmat(shm_id, 0,
- read_only?SHM_RDONLY:0);
- if ((long)profile_p == -1) {
- DEBUG(0,("Can't attach to IPC area. Error was %s\n",
- strerror(errno)));
- return False;
+
+ memcpy(s, value.dptr, value.dsize);
+ if (s->magic != profile_p->magic) {
+ *s = (struct profile_stats) {};
+ return 0;
}
- /* find out who created this memory area */
- if (shmctl(shm_id, IPC_STAT, &shm_ds) != 0) {
- DEBUG(0,("ERROR shmctl : can't IPC_STAT. Error was %s\n",
- strerror(errno)));
- return False;
+ return 0;
+}
+
+void smbprofile_dump(void)
+{
+ pid_t pid = getpid();
+ TDB_DATA key = { .dptr = (uint8_t *)&pid, .dsize = sizeof(pid) };
+ struct profile_stats s = {};
+ int ret;
+#ifdef HAVE_GETRUSAGE
+ struct rusage rself;
+#endif /* HAVE_GETRUSAGE */
+
+ TALLOC_FREE(smbprofile_state.internal.te);
+
+ if (smbprofile_state.internal.db == NULL) {
+ return;
}
- if (shm_ds.shm_perm.cuid != sec_initial_uid() ||
- shm_ds.shm_perm.cgid != sec_initial_gid()) {
- DEBUG(0,("ERROR: we did not create the shmem "
- "(owned by another user, uid %u, gid %u)\n",
- shm_ds.shm_perm.cuid,
- shm_ds.shm_perm.cgid));
- return False;
+#ifdef HAVE_GETRUSAGE
+ ret = getrusage(RUSAGE_SELF, &rself);
+ if (ret != 0) {
+ ZERO_STRUCT(rself);
}
- if (shm_ds.shm_segsz != sizeof(*profile_h)) {
- DEBUG(0,("WARNING: profile size is %d (expected %lu). Deleting\n",
- (int)shm_ds.shm_segsz, sizeof(*profile_h)));
- if (shmctl(shm_id, IPC_RMID, &shm_ds) == 0) {
- goto again;
- } else {
- return False;
- }
+ profile_p->values.cpu_user_stats.time =
+ (rself.ru_utime.tv_sec * 1000000) +
+ rself.ru_utime.tv_usec;
+ profile_p->values.cpu_system_stats.time =
+ (rself.ru_stime.tv_sec * 1000000) +
+ rself.ru_stime.tv_usec;
+#endif /* HAVE_GETRUSAGE */
+
+ ret = tdb_chainlock(smbprofile_state.internal.db->tdb, key);
+ if (ret != 0) {
+ return;
}
- if (!read_only && (shm_ds.shm_nattch == 1)) {
- memset((char *)profile_h, 0, sizeof(*profile_h));
- profile_h->prof_shm_magic = PROF_SHM_MAGIC;
- profile_h->prof_shm_version = PROF_SHM_VERSION;
- DEBUG(3,("Initialised profile area\n"));
+ tdb_parse_record(smbprofile_state.internal.db->tdb,
+ key, profile_stats_parser, &s);
+
+ smbprofile_stats_accumulate(profile_p, &s);
+
+ tdb_store(smbprofile_state.internal.db->tdb, key,
+ (TDB_DATA) {
+ .dptr = (uint8_t *)profile_p,
+ .dsize = sizeof(*profile_p)
+ },
+ 0);
+
+ tdb_chainunlock(smbprofile_state.internal.db->tdb, key);
+ ZERO_STRUCT(profile_p->values);
+
+ return;
+}
+
+void smbprofile_cleanup(pid_t pid, pid_t dst)
+{
+ TDB_DATA key = { .dptr = (uint8_t *)&pid, .dsize = sizeof(pid) };
+ struct profile_stats s = {};
+ struct profile_stats acc = {};
+ int ret;
+
+ if (smbprofile_state.internal.db == NULL) {
+ return;
}
- profile_p = &profile_h->stats;
- message_register(MSG_PROFILE, profile_message, NULL);
- message_register(MSG_REQ_PROFILELEVEL, reqprofile_message, NULL);
- return True;
+ ret = tdb_chainlock(smbprofile_state.internal.db->tdb, key);
+ if (ret != 0) {
+ return;
+ }
+ ret = tdb_parse_record(smbprofile_state.internal.db->tdb,
+ key, profile_stats_parser, &s);
+ if (ret == -1) {
+ tdb_chainunlock(smbprofile_state.internal.db->tdb, key);
+ return;
+ }
+ tdb_delete(smbprofile_state.internal.db->tdb, key);
+ tdb_chainunlock(smbprofile_state.internal.db->tdb, key);
+
+ pid = dst;
+ ret = tdb_chainlock(smbprofile_state.internal.db->tdb, key);
+ if (ret != 0) {
+ return;
+ }
+ tdb_parse_record(smbprofile_state.internal.db->tdb,
+ key, profile_stats_parser, &acc);
+
+ /*
+ * We may have to fix the disconnect count
+ * in case the process died
+ */
+ s.values.disconnect_stats.count = s.values.connect_stats.count;
+
+ smbprofile_stats_accumulate(&acc, &s);
+
+ acc.magic = profile_p->magic;
+ tdb_store(smbprofile_state.internal.db->tdb, key,
+ (TDB_DATA) {
+ .dptr = (uint8_t *)&acc,
+ .dsize = sizeof(acc)
+ },
+ 0);
+
+ tdb_chainunlock(smbprofile_state.internal.db->tdb, key);
}
- const char * profile_value_name(enum profile_stats_values val)
+void smbprofile_stats_accumulate(struct profile_stats *acc,
+ const struct profile_stats *add)
{
- static const char * valnames[PR_VALUE_MAX + 1] =
- {
- "smbd_idle", /* PR_VALUE_SMBD_IDLE */
- "syscall_opendir", /* PR_VALUE_SYSCALL_OPENDIR */
- "syscall_readdir", /* PR_VALUE_SYSCALL_READDIR */
- "syscall_seekdir", /* PR_VALUE_SYSCALL_SEEKDIR */
- "syscall_telldir", /* PR_VALUE_SYSCALL_TELLDIR */
- "syscall_rewinddir", /* PR_VALUE_SYSCALL_REWINDDIR */
- "syscall_mkdir", /* PR_VALUE_SYSCALL_MKDIR */
- "syscall_rmdir", /* PR_VALUE_SYSCALL_RMDIR */
- "syscall_closedir", /* PR_VALUE_SYSCALL_CLOSEDIR */
- "syscall_open", /* PR_VALUE_SYSCALL_OPEN */
- "syscall_close", /* PR_VALUE_SYSCALL_CLOSE */
- "syscall_read", /* PR_VALUE_SYSCALL_READ */
- "syscall_pread", /* PR_VALUE_SYSCALL_PREAD */
- "syscall_write", /* PR_VALUE_SYSCALL_WRITE */
- "syscall_pwrite", /* PR_VALUE_SYSCALL_PWRITE */
- "syscall_lseek", /* PR_VALUE_SYSCALL_LSEEK */
- "syscall_sendfile", /* PR_VALUE_SYSCALL_SENDFILE */
- "syscall_rename", /* PR_VALUE_SYSCALL_RENAME */
- "syscall_fsync", /* PR_VALUE_SYSCALL_FSYNC */
- "syscall_stat", /* PR_VALUE_SYSCALL_STAT */
- "syscall_fstat", /* PR_VALUE_SYSCALL_FSTAT */
- "syscall_lstat", /* PR_VALUE_SYSCALL_LSTAT */
- "syscall_unlink", /* PR_VALUE_SYSCALL_UNLINK */
- "syscall_chmod", /* PR_VALUE_SYSCALL_CHMOD */
- "syscall_fchmod", /* PR_VALUE_SYSCALL_FCHMOD */
- "syscall_chown", /* PR_VALUE_SYSCALL_CHOWN */
- "syscall_fchown", /* PR_VALUE_SYSCALL_FCHOWN */
- "syscall_chdir", /* PR_VALUE_SYSCALL_CHDIR */
- "syscall_getwd", /* PR_VALUE_SYSCALL_GETWD */
- "syscall_ntimes", /* PR_VALUE_SYSCALL_NTIMES */
- "syscall_ftruncate", /* PR_VALUE_SYSCALL_FTRUNCATE */
- "syscall_fcntl_lock", /* PR_VALUE_SYSCALL_FCNTL_LOCK */
- "syscall_kernel_flock", /* PR_VALUE_SYSCALL_KERNEL_FLOCK */
- "syscall_linux_setlease", /* PR_VALUE_SYSCALL_LINUX_SETLEASE */
- "syscall_fcntl_getlock", /* PR_VALUE_SYSCALL_FCNTL_GETLOCK */
- "syscall_readlink", /* PR_VALUE_SYSCALL_READLINK */
- "syscall_symlink", /* PR_VALUE_SYSCALL_SYMLINK */
- "syscall_link", /* PR_VALUE_SYSCALL_LINK */
- "syscall_mknod", /* PR_VALUE_SYSCALL_MKNOD */
- "syscall_realpath", /* PR_VALUE_SYSCALL_REALPATH */
- "syscall_get_quota", /* PR_VALUE_SYSCALL_GET_QUOTA */
- "syscall_set_quota", /* PR_VALUE_SYSCALL_SET_QUOTA */
- "SMBmkdir", /* PR_VALUE_SMBMKDIR */
- "SMBrmdir", /* PR_VALUE_SMBRMDIR */
- "SMBopen", /* PR_VALUE_SMBOPEN */
- "SMBcreate", /* PR_VALUE_SMBCREATE */
- "SMBclose", /* PR_VALUE_SMBCLOSE */
- "SMBflush", /* PR_VALUE_SMBFLUSH */
- "SMBunlink", /* PR_VALUE_SMBUNLINK */
- "SMBmv", /* PR_VALUE_SMBMV */
- "SMBgetatr", /* PR_VALUE_SMBGETATR */
- "SMBsetatr", /* PR_VALUE_SMBSETATR */
- "SMBread", /* PR_VALUE_SMBREAD */
- "SMBwrite", /* PR_VALUE_SMBWRITE */
- "SMBlock", /* PR_VALUE_SMBLOCK */
- "SMBunlock", /* PR_VALUE_SMBUNLOCK */
- "SMBctemp", /* PR_VALUE_SMBCTEMP */
- "SMBmknew", /* PR_VALUE_SMBMKNEW */
- "SMBcheckpath", /* PR_VALUE_SMBCHECKPATH */
- "SMBexit", /* PR_VALUE_SMBEXIT */
- "SMBlseek", /* PR_VALUE_SMBLSEEK */
- "SMBlockread", /* PR_VALUE_SMBLOCKREAD */
- "SMBwriteunlock", /* PR_VALUE_SMBWRITEUNLOCK */
- "SMBreadbraw", /* PR_VALUE_SMBREADBRAW */
- "SMBreadBmpx", /* PR_VALUE_SMBREADBMPX */
- "SMBreadBs", /* PR_VALUE_SMBREADBS */
- "SMBwritebraw", /* PR_VALUE_SMBWRITEBRAW */
- "SMBwriteBmpx", /* PR_VALUE_SMBWRITEBMPX */
- "SMBwriteBs", /* PR_VALUE_SMBWRITEBS */
- "SMBwritec", /* PR_VALUE_SMBWRITEC */
- "SMBsetattrE", /* PR_VALUE_SMBSETATTRE */
- "SMBgetattrE", /* PR_VALUE_SMBGETATTRE */
- "SMBlockingX", /* PR_VALUE_SMBLOCKINGX */
- "SMBtrans", /* PR_VALUE_SMBTRANS */
- "SMBtranss", /* PR_VALUE_SMBTRANSS */
- "SMBioctl", /* PR_VALUE_SMBIOCTL */
- "SMBioctls", /* PR_VALUE_SMBIOCTLS */
- "SMBcopy", /* PR_VALUE_SMBCOPY */
- "SMBmove", /* PR_VALUE_SMBMOVE */
- "SMBecho", /* PR_VALUE_SMBECHO */
- "SMBwriteclose", /* PR_VALUE_SMBWRITECLOSE */
- "SMBopenX", /* PR_VALUE_SMBOPENX */
- "SMBreadX", /* PR_VALUE_SMBREADX */
- "SMBwriteX", /* PR_VALUE_SMBWRITEX */
- "SMBtrans2", /* PR_VALUE_SMBTRANS2 */
- "SMBtranss2", /* PR_VALUE_SMBTRANSS2 */
- "SMBfindclose", /* PR_VALUE_SMBFINDCLOSE */
- "SMBfindnclose", /* PR_VALUE_SMBFINDNCLOSE */
- "SMBtcon", /* PR_VALUE_SMBTCON */
- "SMBtdis", /* PR_VALUE_SMBTDIS */
- "SMBnegprot", /* PR_VALUE_SMBNEGPROT */
- "SMBsesssetupX", /* PR_VALUE_SMBSESSSETUPX */
- "SMBulogoffX", /* PR_VALUE_SMBULOGOFFX */
- "SMBtconX", /* PR_VALUE_SMBTCONX */
- "SMBdskattr", /* PR_VALUE_SMBDSKATTR */
- "SMBsearch", /* PR_VALUE_SMBSEARCH */
- "SMBffirst", /* PR_VALUE_SMBFFIRST */
- "SMBfunique", /* PR_VALUE_SMBFUNIQUE */
- "SMBfclose", /* PR_VALUE_SMBFCLOSE */
- "SMBnttrans", /* PR_VALUE_SMBNTTRANS */
- "SMBnttranss", /* PR_VALUE_SMBNTTRANSS */
- "SMBntcreateX", /* PR_VALUE_SMBNTCREATEX */
- "SMBntcancel", /* PR_VALUE_SMBNTCANCEL */
- "SMBntrename", /* PR_VALUE_SMBNTRENAME */
- "SMBsplopen", /* PR_VALUE_SMBSPLOPEN */
- "SMBsplwr", /* PR_VALUE_SMBSPLWR */
- "SMBsplclose", /* PR_VALUE_SMBSPLCLOSE */
- "SMBsplretq", /* PR_VALUE_SMBSPLRETQ */
- "SMBsends", /* PR_VALUE_SMBSENDS */
- "SMBsendb", /* PR_VALUE_SMBSENDB */
- "SMBfwdname", /* PR_VALUE_SMBFWDNAME */
- "SMBcancelf", /* PR_VALUE_SMBCANCELF */
- "SMBgetmac", /* PR_VALUE_SMBGETMAC */
- "SMBsendstrt", /* PR_VALUE_SMBSENDSTRT */
- "SMBsendend", /* PR_VALUE_SMBSENDEND */
- "SMBsendtxt", /* PR_VALUE_SMBSENDTXT */
- "SMBinvalid", /* PR_VALUE_SMBINVALID */
- "pathworks_setdir", /* PR_VALUE_PATHWORKS_SETDIR */
- "Trans2_open", /* PR_VALUE_TRANS2_OPEN */
- "Trans2_findfirst", /* PR_VALUE_TRANS2_FINDFIRST */
- "Trans2_findnext", /* PR_VALUE_TRANS2_FINDNEXT */
- "Trans2_qfsinfo", /* PR_VALUE_TRANS2_QFSINFO */
- "Trans2_setfsinfo", /* PR_VALUE_TRANS2_SETFSINFO */
- "Trans2_qpathinfo", /* PR_VALUE_TRANS2_QPATHINFO */
- "Trans2_setpathinfo", /* PR_VALUE_TRANS2_SETPATHINFO */
- "Trans2_qfileinfo", /* PR_VALUE_TRANS2_QFILEINFO */
- "Trans2_setfileinfo", /* PR_VALUE_TRANS2_SETFILEINFO */
- "Trans2_fsctl", /* PR_VALUE_TRANS2_FSCTL */
- "Trans2_ioctl", /* PR_VALUE_TRANS2_IOCTL */
- "Trans2_findnotifyfirst", /* PR_VALUE_TRANS2_FINDNOTIFYFIRST */
- "Trans2_findnotifynext", /* PR_VALUE_TRANS2_FINDNOTIFYNEXT */
- "Trans2_mkdir", /* PR_VALUE_TRANS2_MKDIR */
- "Trans2_session_setup", /* PR_VALUE_TRANS2_SESSION_SETUP */
- "Trans2_get_dfs_referral", /* PR_VALUE_TRANS2_GET_DFS_REFERRAL */
- "Trans2_report_dfs_inconsistancy", /* PR_VALUE_TRANS2_REPORT_DFS_INCONSISTANCY */
- "NT_transact_create", /* PR_VALUE_NT_TRANSACT_CREATE */
- "NT_transact_ioctl", /* PR_VALUE_NT_TRANSACT_IOCTL */
- "NT_transact_set_security_desc", /* PR_VALUE_NT_TRANSACT_SET_SECURITY_DESC */
- "NT_transact_notify_change",/* PR_VALUE_NT_TRANSACT_NOTIFY_CHANGE */
- "NT_transact_rename", /* PR_VALUE_NT_TRANSACT_RENAME */
- "NT_transact_query_security_desc", /* PR_VALUE_NT_TRANSACT_QUERY_SECURITY_DESC */
- "NT_transact_get_user_quota",/* PR_VALUE_NT_TRANSACT_GET_USER_QUOTA */
- "NT_transact_set_user_quota",/* PR_VALUE_NT_TRANSACT_SET_USER_QUOTA */
- "get_nt_acl", /* PR_VALUE_GET_NT_ACL */
- "fget_nt_acl", /* PR_VALUE_FGET_NT_ACL */
- "set_nt_acl", /* PR_VALUE_SET_NT_ACL */
- "fset_nt_acl", /* PR_VALUE_FSET_NT_ACL */
- "chmod_acl", /* PR_VALUE_CHMOD_ACL */
- "fchmod_acl", /* PR_VALUE_FCHMOD_ACL */
- "name_release", /* PR_VALUE_NAME_RELEASE */
- "name_refresh", /* PR_VALUE_NAME_REFRESH */
- "name_registration", /* PR_VALUE_NAME_REGISTRATION */
- "node_status", /* PR_VALUE_NODE_STATUS */
- "name_query", /* PR_VALUE_NAME_QUERY */
- "host_announce", /* PR_VALUE_HOST_ANNOUNCE */
- "workgroup_announce", /* PR_VALUE_WORKGROUP_ANNOUNCE */
- "local_master_announce", /* PR_VALUE_LOCAL_MASTER_ANNOUNCE */
- "master_browser_announce", /* PR_VALUE_MASTER_BROWSER_ANNOUNCE */
- "lm_host_announce", /* PR_VALUE_LM_HOST_ANNOUNCE */
- "get_backup_list", /* PR_VALUE_GET_BACKUP_LIST */
- "reset_browser", /* PR_VALUE_RESET_BROWSER */
- "announce_request", /* PR_VALUE_ANNOUNCE_REQUEST */
- "lm_announce_request", /* PR_VALUE_LM_ANNOUNCE_REQUEST */
- "domain_logon", /* PR_VALUE_DOMAIN_LOGON */
- "sync_browse_lists", /* PR_VALUE_SYNC_BROWSE_LISTS */
- "run_elections", /* PR_VALUE_RUN_ELECTIONS */
- "election", /* PR_VALUE_ELECTION */
- "" /* PR_VALUE_MAX */
- };
-
- SMB_ASSERT(val >= 0);
- SMB_ASSERT(val < PR_VALUE_MAX);
- return valnames[val];
+#define SMBPROFILE_STATS_START
+#define SMBPROFILE_STATS_SECTION_START(name, display)
+#define SMBPROFILE_STATS_COUNT(name) do { \
+ acc->values.name##_stats.count += add->values.name##_stats.count; \
+} while(0);
+#define SMBPROFILE_STATS_TIME(name) do { \
+ acc->values.name##_stats.time += add->values.name##_stats.time; \
+} while(0);
+#define SMBPROFILE_STATS_BASIC(name) do { \
+ acc->values.name##_stats.count += add->values.name##_stats.count; \
+ acc->values.name##_stats.time += add->values.name##_stats.time; \
+} while(0);
+#define SMBPROFILE_STATS_BYTES(name) do { \
+ acc->values.name##_stats.count += add->values.name##_stats.count; \
+ acc->values.name##_stats.time += add->values.name##_stats.time; \
+ acc->values.name##_stats.idle += add->values.name##_stats.idle; \
+ acc->values.name##_stats.bytes += add->values.name##_stats.bytes; \
+} while(0);
+#define SMBPROFILE_STATS_IOBYTES(name) do { \
+ acc->values.name##_stats.count += add->values.name##_stats.count; \
+ acc->values.name##_stats.time += add->values.name##_stats.time; \
+ acc->values.name##_stats.idle += add->values.name##_stats.idle; \
+ acc->values.name##_stats.inbytes += add->values.name##_stats.inbytes; \
+ acc->values.name##_stats.outbytes += add->values.name##_stats.outbytes; \
+} while(0);
+#define SMBPROFILE_STATS_SECTION_END
+#define SMBPROFILE_STATS_END
+ SMBPROFILE_STATS_ALL_SECTIONS
+#undef SMBPROFILE_STATS_START
+#undef SMBPROFILE_STATS_SECTION_START
+#undef SMBPROFILE_STATS_COUNT
+#undef SMBPROFILE_STATS_TIME
+#undef SMBPROFILE_STATS_BASIC
+#undef SMBPROFILE_STATS_BYTES
+#undef SMBPROFILE_STATS_IOBYTES
+#undef SMBPROFILE_STATS_SECTION_END
+#undef SMBPROFILE_STATS_END
+}
+
+static int smbprofile_collect_fn(struct tdb_context *tdb,
+ TDB_DATA key, TDB_DATA value,
+ void *private_data)
+{
+ struct profile_stats *acc = (struct profile_stats *)private_data;
+ const struct profile_stats *v;
+
+ if (value.dsize != sizeof(struct profile_stats)) {
+ return 0;
+ }
+
+ v = (const struct profile_stats *)value.dptr;
+
+ if (v->magic != profile_p->magic) {
+ return 0;
+ }
+
+ smbprofile_stats_accumulate(acc, v);
+ return 0;
}
-#endif /* WITH_PROFILE */
+void smbprofile_collect(struct profile_stats *stats)
+{
+ *stats = (struct profile_stats) {};
+
+ if (smbprofile_state.internal.db == NULL) {
+ return;
+ }
+
+ tdb_traverse_read(smbprofile_state.internal.db->tdb,
+ smbprofile_collect_fn, stats);
+}