author | Tero Marttila <terom@fixme.fi> |
Fri, 06 Jun 2008 18:35:46 +0300 | |
changeset 11 | 082bfaf38cf0 |
parent 9 | fb6632e6c1bb |
child 13 | ee426f453cf5 |
permissions | -rw-r--r-- |
9
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
1 |
#include <string.h> |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
2 |
#include <stdio.h> |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
3 |
#include <ctype.h> |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
4 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
5 |
#include "remote_pool.h" |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
6 |
#include "common.h" |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
7 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
8 |
void remote_pool_init (struct remote_pool *pool_info) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
9 |
memset(pool_info, 0, sizeof(*pool_info)); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
10 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
11 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
12 |
int remote_pool_add (struct remote_pool *pool_info, const char *hostname, const char *portname) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
13 |
int i; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
14 |
struct remote_node *node_info = NULL; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
15 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
16 |
// find a free remote_node entry and store its pointer in node_info |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
17 |
for (i = 0; i < REMOTE_POOL_MAX; i++) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
18 |
if (!pool_info->nodes[i].valid) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
19 |
node_info = &pool_info->nodes[i]; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
20 |
break; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
21 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
22 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
23 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
24 |
if (!node_info) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
25 |
error("remote_pool_add: pool full, consider increasing REMOTE_POOL_MAX"); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
26 |
return -1; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
27 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
28 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
29 |
return remote_node_init(node_info, hostname, portname); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
30 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
31 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
32 |
int remote_pool_load (struct remote_pool *pool_info, const char *filename) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
33 |
// open the file |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
34 |
FILE *fh = fopen(filename, "r"); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
35 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
36 |
if (!fh) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
37 |
perr("remote_pool_load: fopen(%s)", filename); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
38 |
return -1; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
39 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
40 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
41 |
// read it in line-by-line |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
42 |
char line_buf[POOL_FILE_LINE_LENGTH], *hostname, *portname, *c; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
43 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
44 |
// non-ferror error indicator |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
45 |
int error_flag = 0; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
46 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
47 |
while (1) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
48 |
// read in a line |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
49 |
if (!fgets(line_buf, POOL_FILE_LINE_LENGTH, fh)) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
50 |
perr("remote_pool_load: fgets"); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
51 |
break; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
52 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
53 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
54 |
// strip comments |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
55 |
c = strchr(line_buf, '#'); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
56 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
57 |
if (c) |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
58 |
*c = '\0'; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
59 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
60 |
// skip empty lines |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
61 |
for (c = line_buf; *c != '\0'; c++) |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
62 |
if (!isspace(*c)) |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
63 |
break; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
64 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
65 |
if (*c == '\0') |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
66 |
continue; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
67 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
68 |
// parse |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
69 |
if (!parse_hostport(line_buf, &hostname, &portname)) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
70 |
error_flag = 1; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
71 |
break; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
72 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
73 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
74 |
// add it to the pool |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
75 |
if (remote_pool_add(pool_info, hostname, portname)) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
76 |
error_flag = 1; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
77 |
break; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
78 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
79 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
80 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
81 |
// pick up ferrors if otherwise fine |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
82 |
error_flag = error_flag || ferror(fh); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
83 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
84 |
// close the fd |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
85 |
fclose(fh); |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
86 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
87 |
// error return? |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
88 |
if (error_flag) |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
89 |
return -1; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
90 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
91 |
return 0; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
92 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
93 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
94 |
struct remote_node *remote_pool_get (struct remote_pool *pool_info) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
95 |
int i; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
96 |
struct remote_node *node_info = NULL; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
97 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
98 |
// find a the appropriate remote_node entry and store its pointer in node_info |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
99 |
for (i = 0; i < REMOTE_POOL_MAX; i++) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
100 |
if (pool_info->nodes[i].valid && (node_info == NULL || pool_info->nodes[i].current_load < node_info->current_load)) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
101 |
node_info = &pool_info->nodes[i]; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
102 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
103 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
104 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
105 |
// either NULL or the right remote_info |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
106 |
return node_info; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
107 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
108 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
109 |
int remote_pool_size (struct remote_pool *pool_info) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
110 |
int i, size = 0; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
111 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
112 |
for (i = 0; i < REMOTE_POOL_MAX; i++) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
113 |
if (pool_info->nodes[i].valid) { |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
114 |
size++; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
115 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
116 |
} |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
117 |
|
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
118 |
return size; |
fb6632e6c1bb
two new modules, remote_node and remote_pool
Tero Marttila <terom@fixme.fi>
parents:
diff
changeset
|
119 |
} |