utils.c
上传用户:jlfgdled
上传日期:2013-04-10
资源大小:33168k
文件大小:2k
源码类别:

Linux/Unix编程

开发平台:

Unix_Linux

  1. /*
  2.  * Generic address resultion entity
  3.  *
  4.  * Authors:
  5.  * net_random Alan Cox
  6.  * net_ratelimit Andy Kleen
  7.  *
  8.  * Created by Alexey Kuznetsov <kuznet@ms2.inr.ac.ru>
  9.  *
  10.  * This program is free software; you can redistribute it and/or
  11.  *      modify it under the terms of the GNU General Public License
  12.  *      as published by the Free Software Foundation; either version
  13.  *      2 of the License, or (at your option) any later version.
  14.  */
  15. #include <asm/uaccess.h>
  16. #include <asm/system.h>
  17. #include <linux/types.h>
  18. #include <linux/kernel.h>
  19. #include <linux/sched.h>
  20. #include <linux/string.h>
  21. #include <linux/mm.h>
  22. static unsigned long net_rand_seed = 152L;
  23. unsigned long net_random(void)
  24. {
  25. net_rand_seed=net_rand_seed*69069L+1;
  26.         return net_rand_seed^jiffies;
  27. }
  28. void net_srandom(unsigned long entropy)
  29. {
  30. net_rand_seed ^= entropy;
  31. net_random();
  32. }
  33. int net_msg_cost = 5*HZ;
  34. int net_msg_burst = 10*5*HZ;
  35. /* 
  36.  * This enforces a rate limit: not more than one kernel message
  37.  * every 5secs to make a denial-of-service attack impossible.
  38.  *
  39.  * All warning printk()s should be guarded by this function. 
  40.  */ 
  41. int net_ratelimit(void)
  42. {
  43. static spinlock_t ratelimit_lock = SPIN_LOCK_UNLOCKED;
  44. static unsigned long toks = 10*5*HZ;
  45. static unsigned long last_msg; 
  46. static int missed;
  47. unsigned long flags;
  48. unsigned long now = jiffies;
  49. spin_lock_irqsave(&ratelimit_lock, flags);
  50. toks += now - last_msg;
  51. last_msg = now;
  52. if (toks > net_msg_burst)
  53. toks = net_msg_burst;
  54. if (toks >= net_msg_cost) {
  55. int lost = missed;
  56. missed = 0;
  57. toks -= net_msg_cost;
  58. spin_unlock_irqrestore(&ratelimit_lock, flags);
  59. if (lost)
  60. printk(KERN_WARNING "NET: %d messages suppressed.n", lost);
  61. return 1;
  62. }
  63. missed++;
  64. spin_unlock_irqrestore(&ratelimit_lock, flags);
  65. return 0;
  66. }