[tools][lkboot] dns lookup the host address on the command line

This commit is contained in:
Travis Geiselbrecht
2014-09-16 17:21:38 -07:00
parent 76aba22b46
commit 108cb3b5bd
3 changed files with 54 additions and 2 deletions

View File

@@ -41,7 +41,7 @@ void usage(void) {
" lkboot <hostname> boot <binary>\n"
" lkboot <hostname> getsysparam <name>\n"
" lkboot <hostname> reboot\n"
" lkboot <hostname> :<commandname> [ <arg>* ]\n"
" lkboot <hostname> :<commandname> [ <arg>* ]\n"
"\n"
);
exit(1);

View File

@@ -29,7 +29,35 @@
#include "network.h"
in_addr_t lookup_hostname(const char *hostname) {
return inet_addr(hostname);
int err;
struct addrinfo *info, *temp;
in_addr_t addr = 0;
struct addrinfo hints;
memset(&hints, 0, sizeof(hints));
hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_STREAM;
err = getaddrinfo(hostname, NULL, &hints, &info);
if (err < 0) {
printf("getaddrinfo() returns %d, error '%s'\n", err, gai_strerror(err));
return 0;
}
for (temp = info; temp; temp = temp->ai_next) {
// printf("flags 0x%x, family %d, socktype %d, protocol %d, addrlen %d\n",
// temp->ai_flags, temp->ai_family, temp->ai_socktype, temp->ai_protocol, temp->ai_addrlen);
if (temp->ai_family == AF_INET && temp->ai_protocol == IPPROTO_TCP) {
struct sockaddr_in *sa = (struct sockaddr_in *)temp->ai_addr;
// printf("port %d, addr 0x%x\n", sa->sin_port, sa->sin_addr.s_addr);
addr = sa->sin_addr.s_addr;
}
}
freeaddrinfo(info);
return addr;
}
static int inet_listen(in_addr_t addr, int type, unsigned port, int shared) {

View File

@@ -1,8 +1,32 @@
/*
* Copyright (c) 2014 Brian Swetland
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files
* (the "Software"), to deal in the Software without restriction,
* including without limitation the rights to use, copy, modify, merge,
* publish, distribute, sublicense, and/or sell copies of the Software,
* and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
#pragma once
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
int udp_listen(in_addr_t addr, unsigned port, int shared);
int udp_connect(in_addr_t addr, unsigned port);