Merge branch 'upstream' of git://ftp.linux-mips.org/pub/scm/upstream-linus
[linux-2.6] / net / ipv4 / ipvs / ip_vs_lc.c
1 /*
2  * IPVS:        Least-Connection Scheduling module
3  *
4  * Authors:     Wensong Zhang <wensong@linuxvirtualserver.org>
5  *
6  *              This program is free software; you can redistribute it and/or
7  *              modify it under the terms of the GNU General Public License
8  *              as published by the Free Software Foundation; either version
9  *              2 of the License, or (at your option) any later version.
10  *
11  * Changes:
12  *     Wensong Zhang            :     added the ip_vs_lc_update_svc
13  *     Wensong Zhang            :     added any dest with weight=0 is quiesced
14  *
15  */
16
17 #include <linux/module.h>
18 #include <linux/kernel.h>
19
20 #include <net/ip_vs.h>
21
22
23 static int ip_vs_lc_init_svc(struct ip_vs_service *svc)
24 {
25         return 0;
26 }
27
28
29 static int ip_vs_lc_done_svc(struct ip_vs_service *svc)
30 {
31         return 0;
32 }
33
34
35 static int ip_vs_lc_update_svc(struct ip_vs_service *svc)
36 {
37         return 0;
38 }
39
40
41 static inline unsigned int
42 ip_vs_lc_dest_overhead(struct ip_vs_dest *dest)
43 {
44         /*
45          * We think the overhead of processing active connections is 256
46          * times higher than that of inactive connections in average. (This
47          * 256 times might not be accurate, we will change it later) We
48          * use the following formula to estimate the overhead now:
49          *                dest->activeconns*256 + dest->inactconns
50          */
51         return (atomic_read(&dest->activeconns) << 8) +
52                 atomic_read(&dest->inactconns);
53 }
54
55
56 /*
57  *      Least Connection scheduling
58  */
59 static struct ip_vs_dest *
60 ip_vs_lc_schedule(struct ip_vs_service *svc, const struct sk_buff *skb)
61 {
62         struct ip_vs_dest *dest, *least = NULL;
63         unsigned int loh = 0, doh;
64
65         IP_VS_DBG(6, "ip_vs_lc_schedule(): Scheduling...\n");
66
67         /*
68          * Simply select the server with the least number of
69          *        (activeconns<<5) + inactconns
70          * Except whose weight is equal to zero.
71          * If the weight is equal to zero, it means that the server is
72          * quiesced, the existing connections to the server still get
73          * served, but no new connection is assigned to the server.
74          */
75
76         list_for_each_entry(dest, &svc->destinations, n_list) {
77                 if ((dest->flags & IP_VS_DEST_F_OVERLOAD) ||
78                     atomic_read(&dest->weight) == 0)
79                         continue;
80                 doh = ip_vs_lc_dest_overhead(dest);
81                 if (!least || doh < loh) {
82                         least = dest;
83                         loh = doh;
84                 }
85         }
86
87         if (least)
88         IP_VS_DBG(6, "LC: server %u.%u.%u.%u:%u activeconns %d inactconns %d\n",
89                   NIPQUAD(least->addr), ntohs(least->port),
90                   atomic_read(&least->activeconns),
91                   atomic_read(&least->inactconns));
92
93         return least;
94 }
95
96
97 static struct ip_vs_scheduler ip_vs_lc_scheduler = {
98         .name =                 "lc",
99         .refcnt =               ATOMIC_INIT(0),
100         .module =               THIS_MODULE,
101         .init_service =         ip_vs_lc_init_svc,
102         .done_service =         ip_vs_lc_done_svc,
103         .update_service =       ip_vs_lc_update_svc,
104         .schedule =             ip_vs_lc_schedule,
105 };
106
107
108 static int __init ip_vs_lc_init(void)
109 {
110         INIT_LIST_HEAD(&ip_vs_lc_scheduler.n_list);
111         return register_ip_vs_scheduler(&ip_vs_lc_scheduler) ;
112 }
113
114 static void __exit ip_vs_lc_cleanup(void)
115 {
116         unregister_ip_vs_scheduler(&ip_vs_lc_scheduler);
117 }
118
119 module_init(ip_vs_lc_init);
120 module_exit(ip_vs_lc_cleanup);
121 MODULE_LICENSE("GPL");