/* 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 . */ #include "includes.h" #include "system/shmem.h" #include "system/filesys.h" #include "messages.h" #include "smbprofile.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_fallocate", /* PR_VALUE_SYSCALL_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 */ "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 */