开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
loadave.c
资源名称：net-snmp-5.2.2.rar　[点击查看]
上传用户：wxp200602
上传日期：2007-10-30
资源大小：4028k
文件大小：9k
源码类别：

SNMP编程

开发平台：

Unix_Linux

loadave.c：源码内容
  1. #include <net-snmp/net-snmp-config.h>
  3. #if HAVE_STDLIB_H
  4. #include <stdlib.h>
  5. #endif
  6. #if HAVE_UNISTD_H
  7. #include <unistd.h>
  8. #endif
  9. #if HAVE_FCNTL_H
  10. #include <fcntl.h>
  11. #endif
  12. #include <signal.h>
  13. #if HAVE_MACHINE_PARAM_H
  14. #include <machine/param.h>
  15. #endif
  16. #if HAVE_SYS_PARAM_H
  17. #include <sys/param.h>
  18. #endif
  19. #if HAVE_SYS_VMMETER_H
  20. #if !(defined(bsdi2) || defined(netbsd1))
  21. #include <sys/vmmeter.h>
  22. #endif
  23. #endif
  24. #if HAVE_SYS_CONF_H
  25. #include <sys/conf.h>
  26. #endif
  27. #if HAVE_ASM_PAGE_H
  28. #include <asm/page.h>
  29. #endif
  30. #if HAVE_SYS_SWAP_H
  31. #include <sys/swap.h>
  32. #endif
  33. #if HAVE_SYS_FS_H
  34. #include <sys/fs.h>
  35. #else
  36. #if HAVE_UFS_FS_H
  37. #include <ufs/fs.h>
  38. #else
  39. #ifdef HAVE_SYS_STAT_H
  40. #include <sys/stat.h>
  41. #endif
  42. #ifdef HAVE_SYS_VNODE_H
  43. #include <sys/vnode.h>
  44. #endif
  45. #ifdef HAVE_UFS_UFS_QUOTA_H
  46. #include <ufs/ufs/quota.h>
  47. #endif
  48. #ifdef HAVE_UFS_UFS_INODE_H
  49. #include <ufs/ufs/inode.h>
  50. #endif
  51. #if HAVE_UFS_FFS_FS_H
  52. #include <ufs/ffs/fs.h>
  53. #endif
  54. #endif
  55. #endif
  56. #if HAVE_MTAB_H
  57. #include <mtab.h>
  58. #endif
  59. #include <errno.h>
  60. #if HAVE_FSTAB_H
  61. #include <fstab.h>
  62. #endif
  63. #if HAVE_SYS_STATFS_H
  64. #include <sys/statfs.h>
  65. #endif
  66. #if HAVE_SYS_STATVFS_H
  67. #include <sys/statvfs.h>
  68. #endif
  69. #if HAVE_SYS_VFS_H
  70. #include <sys/vfs.h>
  71. #endif
  72. #if (!defined(HAVE_STATVFS)) && defined(HAVE_STATFS)
  73. #if HAVE_SYS_MOUNT_H
  74. #include <sys/mount.h>
  75. #endif
  76. #if HAVE_SYS_SYSCTL_H
  77. #include <sys/sysctl.h>
  78. #endif
  79. #define statvfs statfs
  80. #endif
  81. #if HAVE_VM_VM_H
  82. #include <vm/vm.h>
  83. #endif
  84. #if HAVE_VM_SWAP_PAGER_H
  85. #include <vm/swap_pager.h>
  86. #endif
  87. #if HAVE_SYS_FIXPOINT_H
  88. #include <sys/fixpoint.h>
  89. #endif
  90. #if HAVE_SYS_LOADAVG_H
  91. #include <sys/loadavg.h>
  92. #endif
  93. #if HAVE_MALLOC_H
  94. #include <malloc.h>
  95. #endif
  96. #if HAVE_STRING_H
  97. #include <string.h>
  98. #endif
  99. #if TIME_WITH_SYS_TIME
  100. # ifdef WIN32
  101. #  include <sys/timeb.h>
  102. # else
  103. #  include <sys/time.h>
  104. # endif
  105. # include <time.h>
  106. #else
  107. # if HAVE_SYS_TIME_H
  108. #  include <sys/time.h>
  109. # else
  110. #  include <time.h>
  111. # endif
  112. #endif
  113. #if HAVE_WINSOCK_H
  114. #include <winsock.h>
  115. #endif
  116. #ifdef dynix
  117. #include <sys/mc_vmparam.h>
  118. #endif
  119. #if defined(hpux10) || defined(hpux11)
  120. #include <sys/pstat.h>
  121. #endif
  122. #if defined(aix4) || defined(aix5)
  123. #include <libperfstat.h>
  124. #endif
  126. #include <net-snmp/net-snmp-includes.h>
  127. #include <net-snmp/agent/net-snmp-agent-includes.h>
  128. #include <net-snmp/agent/auto_nlist.h>
  130. #include "struct.h"
  131. #include "loadave.h"
  132. #include "util_funcs.h"
  133. #include "kernel.h"
  135. double          maxload[3];
  137. void
  138. init_loadave(void)
  139. {
  141.     /*
  142.      * define the structure we're going to ask the agent to register our
  143.      * information at 
  144.      */
  145.     struct variable2 extensible_loadave_variables[] = {
  146.         {MIBINDEX, ASN_INTEGER, RONLY, var_extensible_loadave, 1,
  147.          {MIBINDEX}},
  148.         {ERRORNAME, ASN_OCTET_STR, RONLY, var_extensible_loadave, 1,
  149.          {ERRORNAME}},
  150.         {LOADAVE, ASN_OCTET_STR, RONLY, var_extensible_loadave, 1,
  151.          {LOADAVE}},
  152.         {LOADMAXVAL, ASN_OCTET_STR, RONLY, var_extensible_loadave, 1,
  153.          {LOADMAXVAL}},
  154.         {LOADAVEINT, ASN_INTEGER, RONLY, var_extensible_loadave, 1,
  155.          {LOADAVEINT}},
  156. #ifdef OPAQUE_SPECIAL_TYPES
  157.         {LOADAVEFLOAT, ASN_OPAQUE_FLOAT, RONLY, var_extensible_loadave, 1,
  158.          {LOADAVEFLOAT}},
  159. #endif
  160.         {ERRORFLAG, ASN_INTEGER, RONLY, var_extensible_loadave, 1,
  161.          {ERRORFLAG}},
  162.         {ERRORMSG, ASN_OCTET_STR, RONLY, var_extensible_loadave, 1,
  163.          {ERRORMSG}}
  164.     };
  166.     /*
  167.      * Define the OID pointer to the top of the mib tree that we're
  168.      * registering underneath 
  169.      */
  170.     oid             loadave_variables_oid[] =
  171.         { UCDAVIS_MIB, LOADAVEMIBNUM, 1 };
  173.     /*
  174.      * register ourselves with the agent to handle our mib tree 
  175.      */
  176.     REGISTER_MIB("ucd-snmp/loadave", extensible_loadave_variables,
  177.                  variable2, loadave_variables_oid);
  179.     snmpd_register_config_handler("load", loadave_parse_config,
  180.                                   loadave_free_config,
  181.                                   "max1 [max5] [max15]");
  182. }
  184. void
  185. loadave_parse_config(const char *token, char *cptr)
  186. {
  187.     int             i;
  189.     for (i = 0; i <= 2; i++) {
  190.         if (cptr != NULL)
  191.             maxload[i] = atof(cptr);
  192.         else
  193.             maxload[i] = maxload[i - 1];
  194.         cptr = skip_not_white(cptr);
  195.         cptr = skip_white(cptr);
  196.     }
  197. }
  199. void
  200. loadave_free_config(void)
  201. {
  202.     int             i;
  204.     for (i = 0; i <= 2; i++)
  205.         maxload[i] = DEFMAXLOADAVE;
  206. }
  208. /*
  209.  * try to get load average
  210.  * Inputs: pointer to array of doubles, number of elements in array
  211.  * Returns: 0=array has values, -1=error occurred.
  212.  */
  213. int
  214. try_getloadavg(double *r_ave, size_t s_ave)
  215. {
  216.     double         *pave = r_ave;
  217. #ifndef HAVE_GETLOADAVG
  218. #ifdef HAVE_SYS_FIXPOINT_H
  219.     fix             favenrun[3];
  220. #endif
  221. #if (defined(ultrix) || defined(sun) || defined(__alpha) || defined(dynix))
  222.     int             i;
  223. #if (defined(sun) || defined(__alpha) || defined(dynix))
  224.     long            favenrun[3];
  225.     if (s_ave > 3)              /* bounds check */
  226.         return (-1);
  227. #define FIX_TO_DBL(_IN) (((double) _IN)/((double) FSCALE))
  228. #endif
  229. #endif
  230. #if defined(aix4) || defined(aix5)
  231.     int             favenrun[3];
  232.     perfstat_cpu_total_t cs;
  233. #endif
  234. #if defined(hpux10) || defined(hpux11)
  235.   struct pst_dynamic pst_buf;
  236. #endif
  237. #endif /* HAVE_GETLOADAVG */
  239. #ifdef HAVE_GETLOADAVG
  240.     if (getloadavg(pave, s_ave) == -1)
  241.         return (-1);
  242. #elif defined(linux)
  243.     {
  244.         FILE           *in = fopen("/proc/loadavg", "r");
  245.         if (!in) {
  246.             snmp_log(LOG_ERR, "snmpd: cannot open /proc/loadavgn");
  247.             return (-1);
  248.         }
  249.         fscanf(in, "%lf %lf %lf", pave, (pave + 1), (pave + 2));
  250.         fclose(in);
  251.     }
  252. #elif (defined(ultrix) || defined(sun) || defined(__alpha) || defined(dynix))
  253.     if (auto_nlist(LOADAVE_SYMBOL, (char *) favenrun, sizeof(favenrun)) ==
  254.         0)
  255.         return (-1);
  256.     for (i = 0; i < s_ave; i++)
  257.         *(pave + i) = FIX_TO_DBL(favenrun[i]);
  258. #elif defined(hpux10) || defined(hpux11)
  259.     if (pstat_getdynamic(&pst_buf, sizeof(struct pst_dynamic), 1, 0) < 0)
  260.         return(-1);
  261.     r_ave[0] = pst_buf.psd_avg_1_min;
  262.     r_ave[1] = pst_buf.psd_avg_5_min;
  263.     r_ave[2] = pst_buf.psd_avg_15_min;
  264. #elif !defined(cygwin)
  265. #ifdef CAN_USE_NLIST
  266. #if defined(aix4) || defined(aix5)
  267.     if(perfstat_cpu_total((perfstat_id_t *)NULL, &cs, sizeof(perfstat_cpu_total_t), 1) > 0) {
  268.         r_ave[0] = cs.loadavg[0] / 65536.0;
  269.         r_ave[1] = cs.loadavg[1] / 65536.0;
  270.         r_ave[2] = cs.loadavg[2] / 65536.0;
  271.     }
  272.     return 0;
  273. #else
  274.     if (auto_nlist(LOADAVE_SYMBOL, (char *) pave, sizeof(double) * s_ave)
  275.         == 0)
  276. #endif
  277. #endif
  278.         return (-1);
  279. #endif
  280.     /*
  281.      * XXX
  282.      *   To calculate this, we need to compare
  283.      *   successive values of the kernel array
  284.      *   '_cp_times', and calculate the resulting
  285.      *   percentage changes.
  286.      *     This calculation needs to be performed
  287.      *   regularly - perhaps as a background process.
  288.      *
  289.      *   See the source to 'top' for full details.
  290.      *
  291.      * The linux SNMP HostRes implementation
  292.      *   uses 'avenrun[0]*100' as an approximation.
  293.      *   This is less than accurate, but has the
  294.      *   advantage of being simple to implement!
  295.      *
  296.      * I'm also assuming a single processor
  297.      */
  298.     return 0;
  299. }
  301. u_char         *
  302. var_extensible_loadave(struct variable * vp,
  303.                        oid * name,
  304.                        size_t * length,
  305.                        int exact,
  306.                        size_t * var_len, WriteMethod ** write_method)
  307. {
  309.     static long     long_ret;
  310.     static float    float_ret;
  311.     static char     errmsg[300];
  312.     double          avenrun[3];
  313.     if (header_simple_table
  314.         (vp, name, length, exact, var_len, write_method, 3))
  315.         return (NULL);
  316.     switch (vp->magic) {
  317.     case MIBINDEX:
  318.         long_ret = name[*length - 1];
  319.         return ((u_char *) (&long_ret));
  320.     case ERRORNAME:
  321.         sprintf(errmsg, "Load-%d", ((name[*length - 1] == 1) ? 1 :
  322.                                     ((name[*length - 1] == 2) ? 5 : 15)));
  323.         *var_len = strlen(errmsg);
  324.         return ((u_char *) (errmsg));
  325.     }
  326.     if (try_getloadavg(&avenrun[0], sizeof(avenrun) / sizeof(avenrun[0]))
  327.         == -1)
  328.         return (0);
  329.     switch (vp->magic) {
  330.     case LOADAVE:
  332.         sprintf(errmsg, "%.2f", avenrun[name[*length - 1] - 1]);
  333.         *var_len = strlen(errmsg);
  334.         return ((u_char *) (errmsg));
  335.     case LOADMAXVAL:
  336.         sprintf(errmsg, "%.2f", maxload[name[*length - 1] - 1]);
  337.         *var_len = strlen(errmsg);
  338.         return ((u_char *) (errmsg));
  339.     case LOADAVEINT:
  340.         long_ret = (u_long) (avenrun[name[*length - 1] - 1] * 100);
  341.         return ((u_char *) (&long_ret));
  342. #ifdef OPAQUE_SPECIAL_TYPES
  343.     case LOADAVEFLOAT:
  344.         float_ret = (float) avenrun[name[*length - 1] - 1];
  345.         *var_len = sizeof(float_ret);
  346.         return ((u_char *) (&float_ret));
  347. #endif
  348.     case ERRORFLAG:
  349.         long_ret = (maxload[name[*length - 1] - 1] != 0 &&
  350.                     avenrun[name[*length - 1] - 1] >=
  351.                     maxload[name[*length - 1] - 1]) ? 1 : 0;
  352.         return ((u_char *) (&long_ret));
  353.     case ERRORMSG:
  354.         if (maxload[name[*length - 1] - 1] != 0 &&
  355.             avenrun[name[*length - 1] - 1] >=
  356.             maxload[name[*length - 1] - 1]) {
  357.             sprintf(errmsg, "%d min Load Average too high (= %.2f)",
  358.                     (name[*length - 1] ==
  359.                      1) ? 1 : ((name[*length - 1] == 2) ? 5 : 15),
  360.                     avenrun[name[*length - 1] - 1]);
  361.         } else {
  362.             errmsg[0] = 0;
  363.         }
  364.         *var_len = strlen(errmsg);
  365.         return ((u_char *) errmsg);
  366.     }
  367.     return NULL;
  368. }