]>
git.ipfire.org Git - thirdparty/squid.git/blob - src/icmp/IcmpSquid.cc
2 * Copyright (C) 1996-2016 The Squid Software Foundation and contributors
4 * Squid software is distributed under GPLv2+ license and includes
5 * contributions from numerous individuals and organizations.
6 * Please see the COPYING and CONTRIBUTORS files for details.
9 /* DEBUG: section 37 ICMP Routines */
13 #include "comm/Loops.h"
16 #include "icmp/IcmpConfig.h"
17 #include "icmp/IcmpSquid.h"
18 #include "icmp/net_db.h"
20 #include "SquidConfig.h"
22 #include "SquidTime.h"
26 // Instance global to be available in main() and elsewhere.
39 IcmpSquid::IcmpSquid() : Icmp()
44 IcmpSquid::~IcmpSquid()
52 IcmpSquid::SendEcho(Ip::Address
&to
, int opcode
, const char *payload
, int len
)
54 static pingerEchoData pecho
;
57 /** \li Does nothing if the pinger socket is not available. */
59 debugs(37, 2, HERE
<< " Socket Closed. Aborted send to " << pecho
.to
<< ", opcode " << opcode
<< ", len " << pecho
.psize
);
63 /** \li If no payload is given or is set as NULL it will ignore payload and len */
67 /** \li Otherwise if len is 0, uses strlen() to detect length of payload.
68 \bug This will result in part of the payload being truncated if it contains a NULL character.
69 \bug Or it may result in a buffer over-run if the payload is not nul-terminated properly.
71 else if (payload
&& len
== 0)
72 len
= strlen(payload
);
75 \bug If length specified or auto-detected is greater than the possible payload squid will die with an assert.
76 \todo This should perhapse be reduced to a truncated payload? or no payload. A WARNING is due anyway.
78 assert(len
<= PINGER_PAYLOAD_SZ
);
82 pecho
.opcode
= (unsigned char) opcode
;
87 memcpy(pecho
.payload
, payload
, len
);
89 slen
= sizeof(pingerEchoData
) - PINGER_PAYLOAD_SZ
+ pecho
.psize
;
91 debugs(37, 2, HERE
<< "to " << pecho
.to
<< ", opcode " << opcode
<< ", len " << pecho
.psize
);
93 x
= comm_udp_send(icmp_sock
, (char *)&pecho
, slen
, 0);
97 debugs(37, DBG_IMPORTANT
, MYNAME
<< "send: " << xstrerr(xerrno
));
99 /** \li If the send results in ECONNREFUSED or EPIPE errors from helper, will cleanly shutdown the module. */
100 /** \todo This should try restarting the helper a few times?? before giving up? */
101 if (xerrno
== ECONNREFUSED
|| xerrno
== EPIPE
) {
105 /** All other send errors are ignored. */
106 } else if (x
!= slen
) {
107 debugs(37, DBG_IMPORTANT
, HERE
<< "Wrote " << x
<< " of " << slen
<< " bytes");
111 // static Callback to wrap the squid-side ICMP handler.
112 // the IcmpSquid::Recv cannot be declared both static and virtual.
114 icmpSquidRecv(int unused1
, void *unused2
)
123 static int fail_count
= 0;
124 pingerReplyData preply
;
125 static Ip::Address F
;
127 Comm::SetSelect(icmp_sock
, COMM_SELECT_READ
, icmpSquidRecv
, NULL
, 0);
128 memset(&preply
, '\0', sizeof(pingerReplyData
));
129 n
= comm_udp_recv(icmp_sock
,
131 sizeof(pingerReplyData
),
134 if (n
< 0 && EAGAIN
!= errno
) {
136 debugs(37, DBG_IMPORTANT
, MYNAME
<< "recv: " << xstrerr(xerrno
));
138 if (xerrno
== ECONNREFUSED
)
141 if (xerrno
== ECONNRESET
)
144 if (++fail_count
== 10)
152 /** If its a test probe from the pinger. Do nothing. */
161 switch (preply
.opcode
) {
164 debugs(37,4, HERE
<< " ICMP_ECHO of " << preply
.from
<< " gave: hops=" << preply
.hops
<<", rtt=" << preply
.rtt
);
168 debugs(37,4, HERE
<< " DomainPing of " << preply
.from
<< " gave: hops=" << preply
.hops
<<", rtt=" << preply
.rtt
);
169 netdbHandlePingReply(F
, preply
.hops
, preply
.rtt
);
173 debugs(37, DBG_IMPORTANT
, HERE
<< "Bad opcode: " << preply
.opcode
<< " from " << F
);
178 #endif /* USE_ICMP */
181 IcmpSquid::DomainPing(Ip::Address
&to
, const char *domain
)
184 debugs(37, 4, HERE
<< "'" << domain
<< "' (" << to
<< ")");
185 SendEcho(to
, S_ICMP_DOM
, domain
, 0);
190 IcmpSquid::Open(void)
196 Ip::Address localhost
;
198 /* User configured disabled. */
199 if (!IcmpCfg
.enable
) {
204 args
[0] = "(pinger)";
206 localhost
.setLocalhost();
209 * Do NOT use IPC_DGRAM (=IPC_UNIX_DGRAM) here because you can't
210 * send() more than 4096 bytes on a socketpair() socket (at
213 pid
= ipcCreate(IPC_UDP_SOCKET
,
214 IcmpCfg
.program
.c_str(),
229 fd_note(icmp_sock
, "pinger");
231 Comm::SetSelect(icmp_sock
, COMM_SELECT_READ
, icmpSquidRecv
, NULL
, 0);
233 commUnsetFdTimeout(icmp_sock
);
235 debugs(37, DBG_IMPORTANT
, HERE
<< "Pinger socket opened on FD " << icmp_sock
);
237 /* Tests the pinger immediately using localhost */
239 SendEcho(localhost
, S_ICMP_ECHO
, "ip6-localhost");
240 if (localhost
.setIPv4())
241 SendEcho(localhost
, S_ICMP_ECHO
, "localhost");
245 debugs(37, 4, HERE
<< "Pinger handle: 0x" << std::hex
<< hIpc
<< std::dec
<< ", PID: " << pid
);
247 #endif /* _SQUID_WINDOWS_ */
251 #endif /* USE_ICMP */
255 IcmpSquid::Close(void)
262 debugs(37, DBG_IMPORTANT
, HERE
<< "Closing Pinger socket on FD " << icmp_sock
);
266 send(icmp_sock
, (const void *) "$shutdown\n", 10, 0);
270 comm_close(icmp_sock
);
275 if (WaitForSingleObject(hIpc
, 12000) != WAIT_OBJECT_0
) {
277 debugs(37, DBG_CRITICAL
, HERE
<< "WARNING: (pinger," << pid
<< ") didn't exit in 12 seconds");