1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-25 23:21:54 +03:00
samba-mirror/source3/profile/profile.c

406 lines
16 KiB
C
Raw Normal View History

/*
Unix SMB/CIFS implementation.
store smbd profiling information in shared memory
Copyright (C) Andrew Tridgell 1999
Copyright (C) James Peach 2006
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 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, see <http://www.gnu.org/licenses/>.
*/
#include "includes.h"
2010-05-09 00:03:23 +04:00
#include "librpc/gen_ndr/messaging.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;
#endif
struct profile_header *profile_h;
struct profile_stats *profile_p;
bool do_profile_flag = False;
bool do_profile_times = False;
/****************************************************************************
Set a profiling level.
****************************************************************************/
void set_profile_level(int level, struct server_id src)
{
#ifdef WITH_PROFILE
switch (level) {
case 0: /* turn off profiling */
do_profile_flag = False;
do_profile_times = False;
DEBUG(1,("INFO: Profiling turned OFF from pid %d\n",
(int)procid_to_pid(&src)));
break;
case 1: /* turn on counter profiling only */
do_profile_flag = True;
do_profile_times = False;
DEBUG(1,("INFO: Profiling counts turned ON from pid %d\n",
(int)procid_to_pid(&src)));
break;
case 2: /* turn on complete profiling */
do_profile_flag = True;
do_profile_times = True;
DEBUG(1,("INFO: Full profiling turned ON from pid %d\n",
(int)procid_to_pid(&src)));
break;
case 3: /* reset profile values */
memset((char *)profile_p, 0, sizeof(*profile_p));
DEBUG(1,("INFO: Profiling values cleared from pid %d\n",
(int)procid_to_pid(&src)));
break;
}
#else /* WITH_PROFILE */
DEBUG(1,("INFO: Profiling support unavailable in this build.\n"));
#endif /* WITH_PROFILE */
}
#ifdef WITH_PROFILE
/****************************************************************************
receive a set profile level message
****************************************************************************/
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;
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
****************************************************************************/
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
DEBUG(1,("INFO: Received REQ_PROFILELEVEL message from PID %u\n",
(unsigned int)procid_to_pid(&src)));
messaging_send_buf(msg_ctx, src, MSG_PROFILELEVEL,
(uint8 *)&level, sizeof(level));
}
/*******************************************************************
open the profiling shared memory area
******************************************************************/
bool profile_setup(struct messaging_context *msg_ctx, bool rdonly)
{
struct shmid_ds shm_ds;
read_only = rdonly;
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 (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_h == -1) {
DEBUG(0,("Can't attach to IPC area. Error was %s\n",
strerror(errno)));
return False;
}
/* 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;
}
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;
}
if (shm_ds.shm_segsz != sizeof(*profile_h)) {
DEBUG(0,("WARNING: profile size is %d (expected %d). Deleting\n",
(int)shm_ds.shm_segsz, (int)sizeof(*profile_h)));
if (shmctl(shm_id, IPC_RMID, &shm_ds) == 0) {
goto again;
} else {
return False;
}
}
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"));
}
profile_p = &profile_h->stats;
if (msg_ctx != NULL) {
messaging_register(msg_ctx, NULL, MSG_PROFILE,
profile_message);
messaging_register(msg_ctx, NULL, MSG_REQ_PROFILELEVEL,
reqprofile_message);
}
return True;
}
const char * profile_value_name(enum profile_stats_values val)
{
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_createfile", /* PR_VALUE_SYSCALL_CREATEFILE */
"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_recvfile", /* PR_VALUE_SYSCALL_RECVFILE */
"syscall_rename", /* PR_VALUE_SYSCALL_RENAME */
"syscall_rename_at", /* PR_VALUE_SYSCALL_RENAME_AT */
"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_posix_fallocate", /* PR_VALUE_SYSCALL_POSIX_FALLOCATE */
"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 */
"syscall_get_sd", /* PR_VALUE_SYSCALL_GET_SD */
"syscall_set_sd", /* PR_VALUE_SYSCALL_SET_SD */
"syscall_brl_lock", /* PR_VALUE_SYSCALL_BRL_LOCK */
"syscall_brl_unlock", /* PR_VALUE_SYSCALL_BRL_UNLOCK */
"syscall_brl_cancel", /* PR_VALUE_SYSCALL_BRL_CANCEL */
"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 */
"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 */
"smb2_negprot", /* PR_VALUE_SMB2_NEGPROT */
"smb2_sesssetup", /* PR_VALUE_SMB2_SESSETUP */
"smb2_logoff", /* PR_VALUE_SMB2_LOGOFF */
"smb2_tcon", /* PR_VALUE_SMB2_TCON */
"smb2_tdis", /* PR_VALUE_SMB2_TDIS */
"smb2_create", /* PR_VALUE_SMB2_CREATE */
"smb2_close", /* PR_VALUE_SMB2_CLOSE */
"smb2_flush", /* PR_VALUE_SMB2_FLUSH */
"smb2_read", /* PR_VALUE_SMB2_READ */
"smb2_write", /* PR_VALUE_SMB2_WRITE */
"smb2_lock", /* PR_VALUE_SMB2_LOCK */
"smb2_ioctl", /* PR_VALUE_SMB2_IOCTL */
"smb2_cancel", /* PR_VALUE_SMB2_CANCEL */
"smb2_keepalive", /* PR_VALUE_SMB2_KEEPALIVE */
"smb2_find", /* PR_VALUE_SMB2_FIND */
"smb2_notify", /* PR_VALUE_SMB2_NOTIFY */
"smb2_getinfo", /* PR_VALUE_SMB2_GETINFO */
"smb2_setinfo" /* PR_VALUE_SMB2_SETINFO */
"smb2_break", /* PR_VALUE_SMB2_BREAK */
"" /* PR_VALUE_MAX */
};
SMB_ASSERT(val >= 0);
SMB_ASSERT(val < PR_VALUE_MAX);
return valnames[val];
}
#endif /* WITH_PROFILE */