check_dhcp.c 34 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253
  1. /******************************************************************************
  2. *
  3. * CHECK_DHCP.C
  4. *
  5. * Program: DHCP plugin for Nagios
  6. * License: GPL
  7. * Copyright (c) 2001-2004 Ethan Galstad (nagios@nagios.org)
  8. * Copyright (c) 2001-2006 Nagios Plugin Development Team
  9. *
  10. * License Information:
  11. *
  12. * This program is free software; you can redistribute it and/or modify
  13. * it under the terms of the GNU General Public License as published by
  14. * the Free Software Foundation; either version 2 of the License, or
  15. * (at your option) any later version.
  16. *
  17. * This program is distributed in the hope that it will be useful,
  18. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  20. * GNU General Public License for more details.
  21. *
  22. * You should have received a copy of the GNU General Public License
  23. * along with this program; if not, write to the Free Software
  24. * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  25. *
  26. * $Id$
  27. *
  28. *****************************************************************************/
  29. const char *progname = "check_dhcp";
  30. const char *revision = "$Revision$";
  31. const char *copyright = "2001-2006";
  32. const char *email = "nagiosplug-devel@lists.sourceforge.net";
  33. #include "common.h"
  34. #include "netutils.h"
  35. #include "utils.h"
  36. #include <stdio.h>
  37. #include <stdlib.h>
  38. #include <string.h>
  39. #include <errno.h>
  40. #include <unistd.h>
  41. #include <sys/time.h>
  42. #include <sys/ioctl.h>
  43. #include <fcntl.h>
  44. #include <getopt.h>
  45. #include <sys/socket.h>
  46. #include <sys/types.h>
  47. #include <netdb.h>
  48. #include <netinet/in.h>
  49. #include <net/if.h>
  50. #include <arpa/inet.h>
  51. #if defined( __linux__ )
  52. #include <linux/if_ether.h>
  53. #include <features.h>
  54. #elif defined (__bsd__)
  55. #include <netinet/if_ether.h>
  56. #include <sys/sysctl.h>
  57. #include <net/if_dl.h>
  58. #elif defined(__sun__) || defined(__solaris__) || defined(__hpux__)
  59. #define INSAP 22
  60. #define OUTSAP 24
  61. #include <signal.h>
  62. #include <ctype.h>
  63. #include <sys/stropts.h>
  64. #include <sys/poll.h>
  65. #include <sys/dlpi.h>
  66. #define bcopy(source, destination, length) memcpy(destination, source, length)
  67. #define AREA_SZ 5000 /* buffer length in bytes */
  68. static u_long ctl_area[AREA_SZ];
  69. static u_long dat_area[AREA_SZ];
  70. static struct strbuf ctl = {AREA_SZ, 0, (char *)ctl_area};
  71. static struct strbuf dat = {AREA_SZ, 0, (char *)dat_area};
  72. #define GOT_CTRL 1
  73. #define GOT_DATA 2
  74. #define GOT_BOTH 3
  75. #define GOT_INTR 4
  76. #define GOT_ERR 128
  77. #define u_int8_t uint8_t
  78. #define u_int16_t uint16_t
  79. #define u_int32_t uint32_t
  80. static int get_msg(int);
  81. static int check_ctrl(int);
  82. static int put_ctrl(int, int, int);
  83. static int put_both(int, int, int, int);
  84. static int dl_open(const char *, int, int *);
  85. static int dl_bind(int, int, u_char *);
  86. long mac_addr_dlpi( const char *, int, u_char *);
  87. #endif
  88. /**** Common definitions ****/
  89. #define OK 0
  90. #define ERROR -1
  91. #define FALSE 0
  92. #define TRUE 1
  93. /**** DHCP definitions ****/
  94. #define MAX_DHCP_CHADDR_LENGTH 16
  95. #define MAX_DHCP_SNAME_LENGTH 64
  96. #define MAX_DHCP_FILE_LENGTH 128
  97. #define MAX_DHCP_OPTIONS_LENGTH 312
  98. typedef struct dhcp_packet_struct{
  99. u_int8_t op; /* packet type */
  100. u_int8_t htype; /* type of hardware address for this machine (Ethernet, etc) */
  101. u_int8_t hlen; /* length of hardware address (of this machine) */
  102. u_int8_t hops; /* hops */
  103. u_int32_t xid; /* random transaction id number - chosen by this machine */
  104. u_int16_t secs; /* seconds used in timing */
  105. u_int16_t flags; /* flags */
  106. struct in_addr ciaddr; /* IP address of this machine (if we already have one) */
  107. struct in_addr yiaddr; /* IP address of this machine (offered by the DHCP server) */
  108. struct in_addr siaddr; /* IP address of DHCP server */
  109. struct in_addr giaddr; /* IP address of DHCP relay */
  110. unsigned char chaddr [MAX_DHCP_CHADDR_LENGTH]; /* hardware address of this machine */
  111. char sname [MAX_DHCP_SNAME_LENGTH]; /* name of DHCP server */
  112. char file [MAX_DHCP_FILE_LENGTH]; /* boot file name (used for diskless booting?) */
  113. char options[MAX_DHCP_OPTIONS_LENGTH]; /* options */
  114. }dhcp_packet;
  115. typedef struct dhcp_offer_struct{
  116. struct in_addr server_address; /* address of DHCP server that sent this offer */
  117. struct in_addr offered_address; /* the IP address that was offered to us */
  118. u_int32_t lease_time; /* lease time in seconds */
  119. u_int32_t renewal_time; /* renewal time in seconds */
  120. u_int32_t rebinding_time; /* rebinding time in seconds */
  121. struct dhcp_offer_struct *next;
  122. }dhcp_offer;
  123. typedef struct requested_server_struct{
  124. struct in_addr server_address;
  125. struct requested_server_struct *next;
  126. }requested_server;
  127. #define BOOTREQUEST 1
  128. #define BOOTREPLY 2
  129. #define DHCPDISCOVER 1
  130. #define DHCPOFFER 2
  131. #define DHCPREQUEST 3
  132. #define DHCPDECLINE 4
  133. #define DHCPACK 5
  134. #define DHCPNACK 6
  135. #define DHCPRELEASE 7
  136. #define DHCP_OPTION_MESSAGE_TYPE 53
  137. #define DHCP_OPTION_HOST_NAME 12
  138. #define DHCP_OPTION_BROADCAST_ADDRESS 28
  139. #define DHCP_OPTION_REQUESTED_ADDRESS 50
  140. #define DHCP_OPTION_LEASE_TIME 51
  141. #define DHCP_OPTION_RENEWAL_TIME 58
  142. #define DHCP_OPTION_REBINDING_TIME 59
  143. #define DHCP_INFINITE_TIME 0xFFFFFFFF
  144. #define DHCP_BROADCAST_FLAG 32768
  145. #define DHCP_SERVER_PORT 67
  146. #define DHCP_CLIENT_PORT 68
  147. #define ETHERNET_HARDWARE_ADDRESS 1 /* used in htype field of dhcp packet */
  148. #define ETHERNET_HARDWARE_ADDRESS_LENGTH 6 /* length of Ethernet hardware addresses */
  149. unsigned char client_hardware_address[MAX_DHCP_CHADDR_LENGTH]="";
  150. char network_interface_name[IFNAMSIZ]="eth0";
  151. u_int32_t packet_xid=0;
  152. u_int32_t dhcp_lease_time=0;
  153. u_int32_t dhcp_renewal_time=0;
  154. u_int32_t dhcp_rebinding_time=0;
  155. int dhcpoffer_timeout=2;
  156. dhcp_offer *dhcp_offer_list=NULL;
  157. requested_server *requested_server_list=NULL;
  158. int valid_responses=0; /* number of valid DHCPOFFERs we received */
  159. int requested_servers=0;
  160. int requested_responses=0;
  161. int request_specific_address=FALSE;
  162. int received_requested_address=FALSE;
  163. int verbose=0;
  164. struct in_addr requested_address;
  165. int process_arguments(int, char **);
  166. int call_getopt(int, char **);
  167. int validate_arguments(void);
  168. void print_usage(void);
  169. void print_help(void);
  170. int get_hardware_address(int,char *);
  171. int send_dhcp_discover(int);
  172. int get_dhcp_offer(int);
  173. int get_results(void);
  174. int add_dhcp_offer(struct in_addr,dhcp_packet *);
  175. int free_dhcp_offer_list(void);
  176. int free_requested_server_list(void);
  177. int create_dhcp_socket(void);
  178. int close_dhcp_socket(int);
  179. int send_dhcp_packet(void *,int,int,struct sockaddr_in *);
  180. int receive_dhcp_packet(void *,int,int,int,struct sockaddr_in *);
  181. int main(int argc, char **argv){
  182. int dhcp_socket;
  183. int result = STATE_UNKNOWN;
  184. setlocale (LC_ALL, "");
  185. bindtextdomain (PACKAGE, LOCALEDIR);
  186. textdomain (PACKAGE);
  187. if(process_arguments(argc,argv)!=OK){
  188. usage4 (_("Could not parse arguments"));
  189. }
  190. /* create socket for DHCP communications */
  191. dhcp_socket=create_dhcp_socket();
  192. /* get hardware address of client machine */
  193. get_hardware_address(dhcp_socket,network_interface_name);
  194. /* send DHCPDISCOVER packet */
  195. send_dhcp_discover(dhcp_socket);
  196. /* wait for a DHCPOFFER packet */
  197. get_dhcp_offer(dhcp_socket);
  198. /* close socket we created */
  199. close_dhcp_socket(dhcp_socket);
  200. /* determine state/plugin output to return */
  201. result=get_results();
  202. /* free allocated memory */
  203. free_dhcp_offer_list();
  204. free_requested_server_list();
  205. return result;
  206. }
  207. /* determines hardware address on client machine */
  208. int get_hardware_address(int sock,char *interface_name){
  209. int i;
  210. #if defined(__linux__)
  211. struct ifreq ifr;
  212. strncpy((char *)&ifr.ifr_name,interface_name,sizeof(ifr.ifr_name));
  213. /* try and grab hardware address of requested interface */
  214. if(ioctl(sock,SIOCGIFHWADDR,&ifr)<0){
  215. printf(_("Error: Could not get hardware address of interface '%s'\n"),interface_name);
  216. exit(STATE_UNKNOWN);
  217. }
  218. memcpy(&client_hardware_address[0],&ifr.ifr_hwaddr.sa_data,6);
  219. #elif defined(__bsd__)
  220. /* King 2004 see ACKNOWLEDGEMENTS */
  221. int mib[6], len;
  222. char *buf;
  223. unsigned char *ptr;
  224. struct if_msghdr *ifm;
  225. struct sockaddr_dl *sdl;
  226. mib[0] = CTL_NET;
  227. mib[1] = AF_ROUTE;
  228. mib[2] = 0;
  229. mib[3] = AF_LINK;
  230. mib[4] = NET_RT_IFLIST;
  231. if ((mib[5] = if_nametoindex(interface_name)) == 0) {
  232. printf(_("Error: if_nametoindex error - %s.\n"), strerror(errno));
  233. exit(STATE_UNKNOWN);
  234. }
  235. if (sysctl(mib, 6, NULL, &len, NULL, 0) < 0) {
  236. printf(_("Error: Couldn't get hardware address from %s. sysctl 1 error - %s.\n"), interface_name, strerror(errno));
  237. exit(STATE_UNKNOWN);
  238. }
  239. if ((buf = malloc(len)) == NULL) {
  240. printf(_("Error: Couldn't get hardware address from interface %s. malloc error - %s.\n"), interface_name, strerror(errno));
  241. exit(4);
  242. }
  243. if (sysctl(mib, 6, buf, &len, NULL, 0) < 0) {
  244. printf(_("Error: Couldn't get hardware address from %s. sysctl 2 error - %s.\n"), interface_name, strerror(errno));
  245. exit(STATE_UNKNOWN);
  246. }
  247. ifm = (struct if_msghdr *)buf;
  248. sdl = (struct sockaddr_dl *)(ifm + 1);
  249. ptr = (unsigned char *)LLADDR(sdl);
  250. memcpy(&client_hardware_address[0], ptr, 6) ;
  251. /* King 2004 */
  252. #elif defined(__sun__) || defined(__solaris__)
  253. /* Kompf 2000-2003 see ACKNOWLEDGEMENTS */
  254. long stat;
  255. char dev[20] = "/dev/";
  256. char *p;
  257. int unit;
  258. for (p = interface_name; *p && isalpha(*p); p++)
  259. /* no-op */ ;
  260. if ( p != '\0' ) {
  261. unit = atoi(p) ;
  262. *p = '\0' ;
  263. strncat(dev, interface_name, 6) ;
  264. } else {
  265. printf(_("Error: can't find unit number in interface_name (%s) - expecting TypeNumber eg lnc0.\n"), interface_name);
  266. exit(STATE_UNKNOWN);
  267. }
  268. stat = mac_addr_dlpi(dev, unit, client_hardware_address);
  269. if (stat != 0) {
  270. printf(_("Error: can't read MAC address from DLPI streams interface for device %s unit %d.\n"), dev, unit);
  271. exit(STATE_UNKNOWN);
  272. }
  273. #elif defined(__hpux__)
  274. long stat;
  275. char dev[20] = "/dev/dlpi" ;
  276. int unit = 0;
  277. stat = mac_addr_dlpi(dev, unit, client_hardware_address);
  278. if (stat != 0) {
  279. printf(_("Error: can't read MAC address from DLPI streams interface for device %s unit %d.\n"), dev, unit);
  280. exit(STATE_UNKNOWN);
  281. }
  282. /* Kompf 2000-2003 */
  283. #else
  284. printf(_("Error: can't get MAC address for this architecture.\n"));
  285. exit(STATE_UNKNOWN);
  286. #endif
  287. if (verbose) {
  288. printf(_("Hardware address: "));
  289. for (i=0; i<6; ++i)
  290. printf("%2.2x", client_hardware_address[i]);
  291. printf( "\n");
  292. }
  293. return OK;
  294. }
  295. /* sends a DHCPDISCOVER broadcast message in an attempt to find DHCP servers */
  296. int send_dhcp_discover(int sock){
  297. dhcp_packet discover_packet;
  298. struct sockaddr_in sockaddr_broadcast;
  299. /* clear the packet data structure */
  300. bzero(&discover_packet,sizeof(discover_packet));
  301. /* boot request flag (backward compatible with BOOTP servers) */
  302. discover_packet.op=BOOTREQUEST;
  303. /* hardware address type */
  304. discover_packet.htype=ETHERNET_HARDWARE_ADDRESS;
  305. /* length of our hardware address */
  306. discover_packet.hlen=ETHERNET_HARDWARE_ADDRESS_LENGTH;
  307. discover_packet.hops=0;
  308. /* transaction id is supposed to be random */
  309. srand(time(NULL));
  310. packet_xid=random();
  311. discover_packet.xid=htonl(packet_xid);
  312. /**** WHAT THE HECK IS UP WITH THIS?!? IF I DON'T MAKE THIS CALL, ONLY ONE SERVER RESPONSE IS PROCESSED!!!! ****/
  313. /* downright bizzarre... */
  314. ntohl(discover_packet.xid);
  315. /*discover_packet.secs=htons(65535);*/
  316. discover_packet.secs=0xFF;
  317. /* tell server it should broadcast its response */
  318. discover_packet.flags=htons(DHCP_BROADCAST_FLAG);
  319. /* our hardware address */
  320. memcpy(discover_packet.chaddr,client_hardware_address,ETHERNET_HARDWARE_ADDRESS_LENGTH);
  321. /* first four bytes of options field is magic cookie (as per RFC 2132) */
  322. discover_packet.options[0]='\x63';
  323. discover_packet.options[1]='\x82';
  324. discover_packet.options[2]='\x53';
  325. discover_packet.options[3]='\x63';
  326. /* DHCP message type is embedded in options field */
  327. discover_packet.options[4]=DHCP_OPTION_MESSAGE_TYPE; /* DHCP message type option identifier */
  328. discover_packet.options[5]='\x01'; /* DHCP message option length in bytes */
  329. discover_packet.options[6]=DHCPDISCOVER;
  330. /* the IP address we're requesting */
  331. if(request_specific_address==TRUE){
  332. discover_packet.options[7]=DHCP_OPTION_REQUESTED_ADDRESS;
  333. discover_packet.options[8]='\x04';
  334. memcpy(&discover_packet.options[9],&requested_address,sizeof(requested_address));
  335. }
  336. /* send the DHCPDISCOVER packet to broadcast address */
  337. sockaddr_broadcast.sin_family=AF_INET;
  338. sockaddr_broadcast.sin_port=htons(DHCP_SERVER_PORT);
  339. sockaddr_broadcast.sin_addr.s_addr=INADDR_BROADCAST;
  340. bzero(&sockaddr_broadcast.sin_zero,sizeof(sockaddr_broadcast.sin_zero));
  341. if (verbose) {
  342. printf(_("DHCPDISCOVER to %s port %d\n"),inet_ntoa(sockaddr_broadcast.sin_addr),ntohs(sockaddr_broadcast.sin_port));
  343. printf("DHCPDISCOVER XID: %lu (0x%X)\n",ntohl(discover_packet.xid),ntohl(discover_packet.xid));
  344. printf("DHCDISCOVER ciaddr: %s\n",inet_ntoa(discover_packet.ciaddr));
  345. printf("DHCDISCOVER yiaddr: %s\n",inet_ntoa(discover_packet.yiaddr));
  346. printf("DHCDISCOVER siaddr: %s\n",inet_ntoa(discover_packet.siaddr));
  347. printf("DHCDISCOVER giaddr: %s\n",inet_ntoa(discover_packet.giaddr));
  348. }
  349. /* send the DHCPDISCOVER packet out */
  350. send_dhcp_packet(&discover_packet,sizeof(discover_packet),sock,&sockaddr_broadcast);
  351. if (verbose)
  352. printf("\n\n");
  353. return OK;
  354. }
  355. /* waits for a DHCPOFFER message from one or more DHCP servers */
  356. int get_dhcp_offer(int sock){
  357. dhcp_packet offer_packet;
  358. struct sockaddr_in source;
  359. int result=OK;
  360. int timeout=1;
  361. int responses=0;
  362. int x;
  363. time_t start_time;
  364. time_t current_time;
  365. time(&start_time);
  366. /* receive as many responses as we can */
  367. for(responses=0,valid_responses=0;;){
  368. time(&current_time);
  369. if((current_time-start_time)>=dhcpoffer_timeout)
  370. break;
  371. if (verbose)
  372. printf("\n\n");
  373. bzero(&source,sizeof(source));
  374. bzero(&offer_packet,sizeof(offer_packet));
  375. result=OK;
  376. result=receive_dhcp_packet(&offer_packet,sizeof(offer_packet),sock,dhcpoffer_timeout,&source);
  377. if(result!=OK){
  378. if (verbose)
  379. printf(_("Result=ERROR\n"));
  380. continue;
  381. }
  382. else{
  383. if (verbose)
  384. printf(_("Result=OK\n"));
  385. responses++;
  386. }
  387. if (verbose) {
  388. printf(_("DHCPOFFER from IP address %s\n"),inet_ntoa(source.sin_addr));
  389. printf("DHCPOFFER XID: %lu (0x%X)\n",ntohl(offer_packet.xid),ntohl(offer_packet.xid));
  390. }
  391. /* check packet xid to see if its the same as the one we used in the discover packet */
  392. if(ntohl(offer_packet.xid)!=packet_xid){
  393. if (verbose)
  394. printf(_("DHCPOFFER XID (%lu) did not match DHCPDISCOVER XID (%lu) - ignoring packet\n"),ntohl(offer_packet.xid),packet_xid);
  395. continue;
  396. }
  397. /* check hardware address */
  398. result=OK;
  399. if (verbose)
  400. printf("DHCPOFFER chaddr: ");
  401. for(x=0;x<ETHERNET_HARDWARE_ADDRESS_LENGTH;x++){
  402. if (verbose)
  403. printf("%02X",(unsigned char)offer_packet.chaddr[x]);
  404. if(offer_packet.chaddr[x]!=client_hardware_address[x])
  405. result=ERROR;
  406. }
  407. if (verbose)
  408. printf("\n");
  409. if(result==ERROR){
  410. if (verbose)
  411. printf(_("DHCPOFFER hardware address did not match our own - ignoring packet\n"));
  412. continue;
  413. }
  414. if (verbose) {
  415. printf("DHCPOFFER ciaddr: %s\n",inet_ntoa(offer_packet.ciaddr));
  416. printf("DHCPOFFER yiaddr: %s\n",inet_ntoa(offer_packet.yiaddr));
  417. printf("DHCPOFFER siaddr: %s\n",inet_ntoa(offer_packet.siaddr));
  418. printf("DHCPOFFER giaddr: %s\n",inet_ntoa(offer_packet.giaddr));
  419. }
  420. add_dhcp_offer(source.sin_addr,&offer_packet);
  421. valid_responses++;
  422. }
  423. if (verbose) {
  424. printf(_("Total responses seen on the wire: %d\n"),responses);
  425. printf(_("Valid responses for this machine: %d\n"),valid_responses);
  426. }
  427. return OK;
  428. }
  429. /* sends a DHCP packet */
  430. int send_dhcp_packet(void *buffer, int buffer_size, int sock, struct sockaddr_in *dest){
  431. struct sockaddr_in myname;
  432. int result;
  433. result=sendto(sock,(char *)buffer,buffer_size,0,(struct sockaddr *)dest,sizeof(*dest));
  434. if (verbose)
  435. printf(_("send_dhcp_packet result: %d\n"),result);
  436. if(result<0)
  437. return ERROR;
  438. return OK;
  439. }
  440. /* receives a DHCP packet */
  441. int receive_dhcp_packet(void *buffer, int buffer_size, int sock, int timeout, struct sockaddr_in *address){
  442. struct timeval tv;
  443. fd_set readfds;
  444. int recv_result;
  445. socklen_t address_size;
  446. struct sockaddr_in source_address;
  447. /* wait for data to arrive (up time timeout) */
  448. tv.tv_sec=timeout;
  449. tv.tv_usec=0;
  450. FD_ZERO(&readfds);
  451. FD_SET(sock,&readfds);
  452. select(sock+1,&readfds,NULL,NULL,&tv);
  453. /* make sure some data has arrived */
  454. if(!FD_ISSET(sock,&readfds)){
  455. if (verbose)
  456. printf(_("No (more) data received\n"));
  457. return ERROR;
  458. }
  459. else{
  460. /* why do we need to peek first? i don't know, its a hack. without it, the source address of the first packet received was
  461. not being interpreted correctly. sigh... */
  462. bzero(&source_address,sizeof(source_address));
  463. address_size=sizeof(source_address);
  464. recv_result=recvfrom(sock,(char *)buffer,buffer_size,MSG_PEEK,(struct sockaddr *)&source_address,&address_size);
  465. if (verbose)
  466. printf("recv_result_1: %d\n",recv_result);
  467. recv_result=recvfrom(sock,(char *)buffer,buffer_size,0,(struct sockaddr *)&source_address,&address_size);
  468. if (verbose)
  469. printf("recv_result_2: %d\n",recv_result);
  470. if(recv_result==-1){
  471. if (verbose) {
  472. printf(_("recvfrom() failed, "));
  473. printf("errno: (%d) -> %s\n",errno,strerror(errno));
  474. }
  475. return ERROR;
  476. }
  477. else{
  478. if (verbose) {
  479. printf(_("receive_dhcp_packet() result: %d\n"),recv_result);
  480. printf(_("receive_dhcp_packet() source: %s\n"),inet_ntoa(source_address.sin_addr));
  481. }
  482. memcpy(address,&source_address,sizeof(source_address));
  483. return OK;
  484. }
  485. }
  486. return OK;
  487. }
  488. /* creates a socket for DHCP communication */
  489. int create_dhcp_socket(void){
  490. struct sockaddr_in myname;
  491. struct ifreq interface;
  492. int sock;
  493. int flag=1;
  494. /* Set up the address we're going to bind to. */
  495. bzero(&myname,sizeof(myname));
  496. myname.sin_family=AF_INET;
  497. myname.sin_port=htons(DHCP_CLIENT_PORT);
  498. myname.sin_addr.s_addr=INADDR_ANY; /* listen on any address */
  499. bzero(&myname.sin_zero,sizeof(myname.sin_zero));
  500. /* create a socket for DHCP communications */
  501. sock=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
  502. if(sock<0){
  503. printf(_("Error: Could not create socket!\n"));
  504. exit(STATE_UNKNOWN);
  505. }
  506. if (verbose)
  507. printf("DHCP socket: %d\n",sock);
  508. /* set the reuse address flag so we don't get errors when restarting */
  509. flag=1;
  510. if(setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,(char *)&flag,sizeof(flag))<0){
  511. printf(_("Error: Could not set reuse address option on DHCP socket!\n"));
  512. exit(STATE_UNKNOWN);
  513. }
  514. /* set the broadcast option - we need this to listen to DHCP broadcast messages */
  515. if(setsockopt(sock,SOL_SOCKET,SO_BROADCAST,(char *)&flag,sizeof flag)<0){
  516. printf(_("Error: Could not set broadcast option on DHCP socket!\n"));
  517. exit(STATE_UNKNOWN);
  518. }
  519. /* bind socket to interface */
  520. #if defined(__linux__)
  521. strncpy(interface.ifr_ifrn.ifrn_name,network_interface_name,IFNAMSIZ);
  522. if(setsockopt(sock,SOL_SOCKET,SO_BINDTODEVICE,(char *)&interface,sizeof(interface))<0){
  523. printf(_("Error: Could not bind socket to interface %s. Check your privileges...\n"),network_interface_name);
  524. exit(STATE_UNKNOWN);
  525. }
  526. #else
  527. strncpy(interface.ifr_name,network_interface_name,IFNAMSIZ);
  528. #endif
  529. /* bind the socket */
  530. if(bind(sock,(struct sockaddr *)&myname,sizeof(myname))<0){
  531. printf(_("Error: Could not bind to DHCP socket (port %d)! Check your privileges...\n"),DHCP_CLIENT_PORT);
  532. exit(STATE_UNKNOWN);
  533. }
  534. return sock;
  535. }
  536. /* closes DHCP socket */
  537. int close_dhcp_socket(int sock){
  538. close(sock);
  539. return OK;
  540. }
  541. /* adds a requested server address to list in memory */
  542. int add_requested_server(struct in_addr server_address){
  543. requested_server *new_server;
  544. new_server=(requested_server *)malloc(sizeof(requested_server));
  545. if(new_server==NULL)
  546. return ERROR;
  547. new_server->server_address=server_address;
  548. new_server->next=requested_server_list;
  549. requested_server_list=new_server;
  550. requested_servers++;
  551. if (verbose)
  552. printf(_("Requested server address: %s\n"),inet_ntoa(new_server->server_address));
  553. return OK;
  554. }
  555. /* adds a DHCP OFFER to list in memory */
  556. int add_dhcp_offer(struct in_addr source,dhcp_packet *offer_packet){
  557. dhcp_offer *new_offer;
  558. int x;
  559. int y;
  560. unsigned option_type;
  561. unsigned option_length;
  562. if(offer_packet==NULL)
  563. return ERROR;
  564. /* process all DHCP options present in the packet */
  565. for(x=4;x<MAX_DHCP_OPTIONS_LENGTH;){
  566. /* end of options (0 is really just a pad, but bail out anyway) */
  567. if((int)offer_packet->options[x]==-1 || (int)offer_packet->options[x]==0)
  568. break;
  569. /* get option type */
  570. option_type=offer_packet->options[x++];
  571. /* get option length */
  572. option_length=offer_packet->options[x++];
  573. if (verbose)
  574. printf("Option: %d (0x%02X)\n",option_type,option_length);
  575. /* get option data */
  576. if(option_type==DHCP_OPTION_LEASE_TIME) {
  577. memcpy(&dhcp_lease_time, &offer_packet->options[x],
  578. sizeof(dhcp_lease_time));
  579. dhcp_lease_time = ntohl(dhcp_lease_time);
  580. }
  581. if(option_type==DHCP_OPTION_RENEWAL_TIME) {
  582. memcpy(&dhcp_renewal_time, &offer_packet->options[x],
  583. sizeof(dhcp_renewal_time));
  584. dhcp_renewal_time = ntohl(dhcp_renewal_time);
  585. }
  586. if(option_type==DHCP_OPTION_REBINDING_TIME) {
  587. memcpy(&dhcp_rebinding_time, &offer_packet->options[x],
  588. sizeof(dhcp_rebinding_time));
  589. dhcp_rebinding_time = ntohl(dhcp_rebinding_time);
  590. }
  591. /* skip option data we're ignoring */
  592. else
  593. for(y=0;y<option_length;y++,x++);
  594. }
  595. if (verbose) {
  596. if(dhcp_lease_time==DHCP_INFINITE_TIME)
  597. printf(_("Lease Time: Infinite\n"));
  598. else
  599. printf(_("Lease Time: %lu seconds\n"),(unsigned long)dhcp_lease_time);
  600. if(dhcp_renewal_time==DHCP_INFINITE_TIME)
  601. printf(_("Renewal Time: Infinite\n"));
  602. else
  603. printf(_("Renewal Time: %lu seconds\n"),(unsigned long)dhcp_renewal_time);
  604. if(dhcp_rebinding_time==DHCP_INFINITE_TIME)
  605. printf(_("Rebinding Time: Infinite\n"));
  606. printf(_("Rebinding Time: %lu seconds\n"),(unsigned long)dhcp_rebinding_time);
  607. }
  608. new_offer=(dhcp_offer *)malloc(sizeof(dhcp_offer));
  609. if(new_offer==NULL)
  610. return ERROR;
  611. new_offer->server_address=source;
  612. new_offer->offered_address=offer_packet->yiaddr;
  613. new_offer->lease_time=dhcp_lease_time;
  614. new_offer->renewal_time=dhcp_renewal_time;
  615. new_offer->rebinding_time=dhcp_rebinding_time;
  616. if (verbose) {
  617. printf(_("Added offer from server @ %s"),inet_ntoa(new_offer->server_address));
  618. printf(_(" of IP address %s\n"),inet_ntoa(new_offer->offered_address));
  619. }
  620. /* add new offer to head of list */
  621. new_offer->next=dhcp_offer_list;
  622. dhcp_offer_list=new_offer;
  623. return OK;
  624. }
  625. /* frees memory allocated to DHCP OFFER list */
  626. int free_dhcp_offer_list(void){
  627. dhcp_offer *this_offer;
  628. dhcp_offer *next_offer;
  629. for(this_offer=dhcp_offer_list;this_offer!=NULL;this_offer=next_offer){
  630. next_offer=this_offer->next;
  631. free(this_offer);
  632. }
  633. return OK;
  634. }
  635. /* frees memory allocated to requested server list */
  636. int free_requested_server_list(void){
  637. requested_server *this_server;
  638. requested_server *next_server;
  639. for(this_server=requested_server_list;this_server!=NULL;this_server=next_server){
  640. next_server=this_server->next;
  641. free(this_server);
  642. }
  643. return OK;
  644. }
  645. /* gets state and plugin output to return */
  646. int get_results(void){
  647. dhcp_offer *temp_offer;
  648. requested_server *temp_server;
  649. int result;
  650. u_int32_t max_lease_time=0;
  651. received_requested_address=FALSE;
  652. /* checks responses from requested servers */
  653. requested_responses=0;
  654. if(requested_servers>0){
  655. for(temp_server=requested_server_list;temp_server!=NULL;temp_server=temp_server->next){
  656. for(temp_offer=dhcp_offer_list;temp_offer!=NULL;temp_offer=temp_offer->next){
  657. /* get max lease time we were offered */
  658. if(temp_offer->lease_time>max_lease_time || temp_offer->lease_time==DHCP_INFINITE_TIME)
  659. max_lease_time=temp_offer->lease_time;
  660. /* see if we got the address we requested */
  661. if(!memcmp(&requested_address,&temp_offer->offered_address,sizeof(requested_address)))
  662. received_requested_address=TRUE;
  663. /* see if the servers we wanted a response from talked to us or not */
  664. if(!memcmp(&temp_offer->server_address,&temp_server->server_address,sizeof(temp_server->server_address))){
  665. if (verbose) {
  666. printf(_("DHCP Server Match: Offerer=%s"),inet_ntoa(temp_offer->server_address));
  667. printf(_(" Requested=%s\n"),inet_ntoa(temp_server->server_address));
  668. }
  669. requested_responses++;
  670. }
  671. }
  672. }
  673. }
  674. /* else check and see if we got our requested address from any server */
  675. else{
  676. for(temp_offer=dhcp_offer_list;temp_offer!=NULL;temp_offer=temp_offer->next){
  677. /* get max lease time we were offered */
  678. if(temp_offer->lease_time>max_lease_time || temp_offer->lease_time==DHCP_INFINITE_TIME)
  679. max_lease_time=temp_offer->lease_time;
  680. /* see if we got the address we requested */
  681. if(!memcmp(&requested_address,&temp_offer->offered_address,sizeof(requested_address)))
  682. received_requested_address=TRUE;
  683. }
  684. }
  685. result=STATE_OK;
  686. if(valid_responses==0)
  687. result=STATE_CRITICAL;
  688. else if(requested_servers>0 && requested_responses==0)
  689. result=STATE_CRITICAL;
  690. else if(requested_responses<requested_servers)
  691. result=STATE_WARNING;
  692. else if(request_specific_address==TRUE && received_requested_address==FALSE)
  693. result=STATE_WARNING;
  694. if(result==0) /* garrett honeycutt 2005 */
  695. printf("OK: ");
  696. else if(result==1)
  697. printf("WARNING: ");
  698. else if(result==2)
  699. printf("CRITICAL: ");
  700. else if(result==3)
  701. printf("UNKNOWN: ");
  702. /* we didn't receive any DHCPOFFERs */
  703. if(dhcp_offer_list==NULL){
  704. printf(_("No DHCPOFFERs were received.\n"));
  705. return result;
  706. }
  707. printf(_("Received %d DHCPOFFER(s)"),valid_responses);
  708. if(requested_servers>0)
  709. printf(_(", %s%d of %d requested servers responded"),((requested_responses<requested_servers) && requested_responses>0)?"only ":"",requested_responses,requested_servers);
  710. if(request_specific_address==TRUE)
  711. printf(_(", requested address (%s) was %soffered"),inet_ntoa(requested_address),(received_requested_address==TRUE)?"":_("not "));
  712. printf(_(", max lease time = "));
  713. if(max_lease_time==DHCP_INFINITE_TIME)
  714. printf(_("Infinity"));
  715. else
  716. printf("%lu sec",(unsigned long)max_lease_time);
  717. printf(".\n");
  718. return result;
  719. }
  720. /* process command-line arguments */
  721. int process_arguments(int argc, char **argv){
  722. int c;
  723. if(argc<1)
  724. return ERROR;
  725. c=0;
  726. while((c+=(call_getopt(argc-c,&argv[c])))<argc){
  727. /*
  728. if(is_option(argv[c]))
  729. continue;
  730. */
  731. }
  732. return validate_arguments();
  733. }
  734. int call_getopt(int argc, char **argv){
  735. int c=0;
  736. int i=0;
  737. struct in_addr ipaddress;
  738. int option_index = 0;
  739. static struct option long_options[] =
  740. {
  741. {"serverip", required_argument,0,'s'},
  742. {"requestedip", required_argument,0,'r'},
  743. {"timeout", required_argument,0,'t'},
  744. {"interface", required_argument,0,'i'},
  745. {"verbose", no_argument, 0,'v'},
  746. {"version", no_argument, 0,'V'},
  747. {"help", no_argument, 0,'h'},
  748. {0,0,0,0}
  749. };
  750. while(1){
  751. c=getopt_long(argc,argv,"+hVvt:s:r:t:i:",long_options,&option_index);
  752. i++;
  753. if(c==-1||c==EOF||c==1)
  754. break;
  755. switch(c){
  756. case 'w':
  757. case 'r':
  758. case 't':
  759. case 'i':
  760. i++;
  761. break;
  762. default:
  763. break;
  764. }
  765. switch(c){
  766. case 's': /* DHCP server address */
  767. if(inet_aton(optarg,&ipaddress))
  768. add_requested_server(ipaddress);
  769. /*
  770. else
  771. usage("Invalid server IP address\n");
  772. */
  773. break;
  774. case 'r': /* address we are requested from DHCP servers */
  775. if(inet_aton(optarg,&ipaddress)){
  776. requested_address=ipaddress;
  777. request_specific_address=TRUE;
  778. }
  779. /*
  780. else
  781. usage("Invalid requested IP address\n");
  782. */
  783. break;
  784. case 't': /* timeout */
  785. /*
  786. if(is_intnonneg(optarg))
  787. */
  788. if(atoi(optarg)>0)
  789. dhcpoffer_timeout=atoi(optarg);
  790. /*
  791. else
  792. usage("Time interval must be a nonnegative integer\n");
  793. */
  794. break;
  795. case 'i': /* interface name */
  796. strncpy(network_interface_name,optarg,sizeof(network_interface_name)-1);
  797. network_interface_name[sizeof(network_interface_name)-1]='\x0';
  798. break;
  799. case 'V': /* version */
  800. print_revision(progname,revision);
  801. exit(STATE_OK);
  802. case 'h': /* help */
  803. print_help();
  804. exit(STATE_OK);
  805. case 'v': /* verbose */
  806. verbose=1;
  807. break;
  808. case '?': /* help */
  809. usage2 (_("Unknown argument"), optarg);
  810. break;
  811. default:
  812. break;
  813. }
  814. }
  815. return i;
  816. }
  817. int validate_arguments(void){
  818. return OK;
  819. }
  820. #if defined(__sun__) || defined(__solaris__) || defined(__hpux__)
  821. /* Kompf 2000-2003 see ACKNOWLEDGEMENTS */
  822. /* get a message from a stream; return type of message */
  823. static int get_msg(int fd)
  824. {
  825. int flags = 0;
  826. int res, ret;
  827. ctl_area[0] = 0;
  828. dat_area[0] = 0;
  829. ret = 0;
  830. res = getmsg(fd, &ctl, &dat, &flags);
  831. if(res < 0) {
  832. if(errno == EINTR) {
  833. return(GOT_INTR);
  834. } else {
  835. printf("%s\n", "get_msg FAILED.");
  836. return(GOT_ERR);
  837. }
  838. }
  839. if(ctl.len > 0) {
  840. ret |= GOT_CTRL;
  841. }
  842. if(dat.len > 0) {
  843. ret |= GOT_DATA;
  844. }
  845. return(ret);
  846. }
  847. /* verify that dl_primitive in ctl_area = prim */
  848. static int check_ctrl(int prim)
  849. {
  850. dl_error_ack_t *err_ack = (dl_error_ack_t *)ctl_area;
  851. if(err_ack->dl_primitive != prim) {
  852. printf(_("Error: DLPI stream API failed to get MAC in check_ctrl: %s.\n"), strerror(errno));
  853. exit(STATE_UNKNOWN);
  854. }
  855. return 0;
  856. }
  857. /* put a control message on a stream */
  858. static int put_ctrl(int fd, int len, int pri)
  859. {
  860. ctl.len = len;
  861. if(putmsg(fd, &ctl, 0, pri) < 0) {
  862. printf(_("Error: DLPI stream API failed to get MAC in put_ctrl/putmsg(): %s.\n"), strerror(errno));
  863. exit(STATE_UNKNOWN);
  864. }
  865. return 0;
  866. }
  867. /* put a control + data message on a stream */
  868. static int put_both(int fd, int clen, int dlen, int pri)
  869. {
  870. ctl.len = clen;
  871. dat.len = dlen;
  872. if(putmsg(fd, &ctl, &dat, pri) < 0) {
  873. printf(_("Error: DLPI stream API failed to get MAC in put_both/putmsg().\n"), strerror(errno));
  874. exit(STATE_UNKNOWN);
  875. }
  876. return 0;
  877. }
  878. /* open file descriptor and attach */
  879. static int dl_open(const char *dev, int unit, int *fd)
  880. {
  881. dl_attach_req_t *attach_req = (dl_attach_req_t *)ctl_area;
  882. if((*fd = open(dev, O_RDWR)) == -1) {
  883. printf(_("Error: DLPI stream API failed to get MAC in dl_attach_req/open(%s..): %s.\n"), dev, strerror(errno));
  884. exit(STATE_UNKNOWN);
  885. }
  886. attach_req->dl_primitive = DL_ATTACH_REQ;
  887. attach_req->dl_ppa = unit;
  888. put_ctrl(*fd, sizeof(dl_attach_req_t), 0);
  889. get_msg(*fd);
  890. return check_ctrl(DL_OK_ACK);
  891. }
  892. /* send DL_BIND_REQ */
  893. static int dl_bind(int fd, int sap, u_char *addr)
  894. {
  895. dl_bind_req_t *bind_req = (dl_bind_req_t *)ctl_area;
  896. dl_bind_ack_t *bind_ack = (dl_bind_ack_t *)ctl_area;
  897. bind_req->dl_primitive = DL_BIND_REQ;
  898. bind_req->dl_sap = sap;
  899. bind_req->dl_max_conind = 1;
  900. bind_req->dl_service_mode = DL_CLDLS;
  901. bind_req->dl_conn_mgmt = 0;
  902. bind_req->dl_xidtest_flg = 0;
  903. put_ctrl(fd, sizeof(dl_bind_req_t), 0);
  904. get_msg(fd);
  905. if (GOT_ERR == check_ctrl(DL_BIND_ACK)) {
  906. printf(_("Error: DLPI stream API failed to get MAC in dl_bind/check_ctrl(): %s.\n"), strerror(errno));
  907. exit(STATE_UNKNOWN);
  908. }
  909. bcopy((u_char *)bind_ack + bind_ack->dl_addr_offset, addr,
  910. bind_ack->dl_addr_length);
  911. return 0;
  912. }
  913. /***********************************************************************
  914. * interface:
  915. * function mac_addr_dlpi - get the mac address of the interface with
  916. * type dev (eg lnc, hme) and unit (0, 1 ..)
  917. *
  918. * parameter: addr: an array of six bytes, has to be allocated by the caller
  919. *
  920. * return: 0 if OK, -1 if the address could not be determined
  921. *
  922. *
  923. ***********************************************************************/
  924. long mac_addr_dlpi( const char *dev, int unit, u_char *addr) {
  925. int fd;
  926. u_char mac_addr[25];
  927. if (GOT_ERR != dl_open(dev, unit, &fd)) {
  928. if (GOT_ERR != dl_bind(fd, INSAP, mac_addr)) {
  929. bcopy( mac_addr, addr, 6);
  930. return 0;
  931. }
  932. }
  933. close(fd);
  934. return -1;
  935. }
  936. /* Kompf 2000-2003 */
  937. #endif
  938. /* print usage help */
  939. void print_help(void){
  940. print_revision(progname,revision);
  941. printf("Copyright (c) 2001-2004 Ethan Galstad (nagios@nagios.org)\n");
  942. printf (COPYRIGHT, copyright, email);
  943. printf(_("This plugin tests the availability of DHCP servers on a network.\n\n"));
  944. print_usage();
  945. printf (_(UT_HELP_VRSN));
  946. printf (_(UT_VERBOSE));
  947. printf(_("\
  948. -s, --serverip=IPADDRESS\n\
  949. IP address of DHCP server that we must hear from\n\
  950. -r, --requestedip=IPADDRESS\n\
  951. IP address that should be offered by at least one DHCP server\n\
  952. -t, --timeout=INTEGER\n\
  953. Seconds to wait for DHCPOFFER before timeout occurs\n\
  954. -i, --interface=STRING\n\
  955. Interface to to use for listening (i.e. eth0)\n"));
  956. }
  957. void print_usage(void)
  958. {
  959. printf("\
  960. Usage: %s [-s serverip] [-r requestedip] [-t timeout] [-i interface] [-v]\n",progname);
  961. }