mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-23 21:01:08 +00:00
275 lines
6.0 KiB
C++
275 lines
6.0 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include <stdio.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
|
|
#if defined(AIX) || defined(__linux)
|
|
#include <sys/select.h> // for fd_set
|
|
#endif
|
|
|
|
#if defined(__linux)
|
|
// Didn't find gettdtablehi() or gettdtablesize() on linux. Using FD_SETSIZE
|
|
#define getdtablehi() FD_SETSIZE
|
|
#else
|
|
#define getdtablehi() getdtablesize()
|
|
|
|
// If you find a system doesn't have getdtablesize try #define getdtablesize
|
|
// to FD_SETSIZE. And if you encounter a system that doesn't even have
|
|
// FD_SETSIZE, just grab your ankles and use 255.
|
|
#endif
|
|
|
|
|
|
#include "nspr.h"
|
|
#include "nsCRT.h"
|
|
#include "unix_dns.h"
|
|
|
|
struct sockaddr_un unix_addr;
|
|
|
|
int async_dns_lookup(char* hostName)
|
|
{
|
|
fprintf(stderr, "start async_dns_lookup\n");
|
|
int socket_fd = socket(PF_UNIX, SOCK_STREAM, 0);
|
|
if (socket_fd == -1) {
|
|
fprintf(stderr, "socket returned error.\n");
|
|
return -1;
|
|
}
|
|
|
|
unix_addr.sun_family = AF_UNIX;
|
|
strcpy(unix_addr.sun_path, DNS_SOCK_NAME);
|
|
|
|
int err = connect(socket_fd,(struct sockaddr*)&unix_addr, sizeof(unix_addr));
|
|
if (err == -1) {
|
|
fprintf(stderr, "connect failed (errno = %d).\n",errno);
|
|
close(socket_fd);
|
|
return -1;
|
|
}
|
|
|
|
char buf[256];
|
|
strcpy(buf, "lookup: ");
|
|
strcpy(&buf[8], hostName);
|
|
|
|
err = send(socket_fd, buf, strlen(buf)+1, 0);
|
|
if (err < 0)
|
|
fprintf(stderr, "send(%s) returned error (errno=%d).\n",buf, errno);
|
|
|
|
// receive 4 byte ID
|
|
err = recv(socket_fd, buf, 256, 0);
|
|
if (err < 0)
|
|
fprintf(stderr, "recv() returned error (errno=%d).\n", errno);
|
|
else
|
|
{
|
|
// printf("recv() returned %d bytes.");
|
|
int id = *(int *)buf;
|
|
fprintf(stderr, "id: %d\n", id);
|
|
}
|
|
|
|
return socket_fd;
|
|
}
|
|
|
|
static char *
|
|
string_trim(char *s)
|
|
{
|
|
char *s2;
|
|
if (!s) return 0;
|
|
s2 = s + strlen(s) - 1;
|
|
while (s2 > s && (*s2 == '\n' || *s2 == '\r' || *s2 == ' ' || *s2 == '\t'))
|
|
*s2-- = 0;
|
|
while (*s == ' ' || *s == '\t' || *s == '\n' || *s == '\r')
|
|
s++;
|
|
return s;
|
|
}
|
|
|
|
hostent *
|
|
bytesToHostent(char *buf)
|
|
{
|
|
int i;
|
|
// int size = 0;
|
|
int len, aliasCount, addressCount;
|
|
int addrtype, addrlength;
|
|
char* p = buf;
|
|
char s[1024];
|
|
|
|
len = *(int *)p; // length of name
|
|
p += sizeof(int); // advance past name length
|
|
|
|
memcpy(s, p, len); s[len] = 0;
|
|
fprintf(stderr, "hostname: %s\n", s);
|
|
|
|
p += len; // advance past name
|
|
aliasCount = *(int *)p; // number of aliases
|
|
p += sizeof(int); // advance past alias count
|
|
|
|
for (i=0; i<aliasCount; i++) {
|
|
len = *(int *)p; // length of alias name
|
|
p += sizeof(int); // advance past alias name length
|
|
|
|
memcpy(s, p, len); s[len] = 0;
|
|
fprintf(stderr, "alias: %s\n", s);
|
|
|
|
p += len; // advance past alias name
|
|
}
|
|
|
|
addrtype = *(int *)p;
|
|
|
|
fprintf(stderr, "addrtype: %d\n", addrtype);
|
|
|
|
p += sizeof(int);
|
|
addrlength = *(int *)p;
|
|
|
|
fprintf(stderr, "addrlength: %d\n", addrlength);
|
|
|
|
p += sizeof(int);
|
|
addressCount = *(int *)p;
|
|
p += sizeof(int);
|
|
|
|
for (i=0; i<addressCount; i++) {
|
|
len = *(int *)p;
|
|
p += sizeof(int);
|
|
|
|
fprintf(stderr, "addr len: %d\n", len);
|
|
fprintf(stderr, "addr : %x\n", *(int *)p);
|
|
|
|
p += len;
|
|
}
|
|
|
|
// size = p - buf;
|
|
// size += 1 + aliasCount;
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main(int argc, char* argv[])
|
|
{
|
|
PRStatus status;
|
|
|
|
// launch daemon
|
|
printf("### launch daemon...\n");
|
|
|
|
PRProcessAttr *attributes = PR_NewProcessAttr();
|
|
if (attributes == nullptr) {
|
|
printf("PR_NewProcessAttr() failed.\n");
|
|
return -1;
|
|
}
|
|
|
|
PRProcess *daemon = PR_CreateProcess("nsDnsAsyncLookup", nullptr, nullptr, attributes);
|
|
if (daemon == nullptr) {
|
|
printf("PR_CreateProcess failed.\n");
|
|
} else {
|
|
// status = PR_DetachProcess(daemon);
|
|
//if (status != 0)
|
|
// printf("PR_DetachProcess returned %d\n", status);
|
|
//daemon = nullptr;
|
|
}
|
|
|
|
PR_DestroyProcessAttr(attributes);
|
|
|
|
// create socket and connect to daemon
|
|
int socket_fd = 0;
|
|
|
|
|
|
bool notDone = true;
|
|
char buf[1024];
|
|
|
|
while(notDone) {
|
|
int status = 0;
|
|
fd_set fdset;
|
|
|
|
FD_ZERO(&fdset);
|
|
FD_SET(fileno(stdin), &fdset);
|
|
if (socket_fd > 0)
|
|
FD_SET(socket_fd, &fdset);
|
|
|
|
status = select(getdtablehi(), &fdset, 0, 0, 0);
|
|
if (status <= 0)
|
|
{
|
|
fprintf(stderr, "%s: select() returned %d\n", argv[0], status);
|
|
exit(-1);
|
|
}
|
|
|
|
// which fd is set?
|
|
|
|
if (FD_ISSET(fileno(stdin), &fdset))
|
|
{
|
|
char *line = fgets(buf, sizeof(buf)-1, stdin);
|
|
line = string_trim(line);
|
|
|
|
if(!strcmp(line, "quit") || !strcmp(line, "exit"))
|
|
{
|
|
fprintf(stderr, "bye now.\n");
|
|
notDone = false;
|
|
}
|
|
else if (!strncmp(line, "abort ", 6))
|
|
{
|
|
// abort id
|
|
}
|
|
else if (strchr(line, ' ') || strchr(line, '\t'))
|
|
{
|
|
fprintf(stderr, "%s: unrecognized command %s.\n", argv[0], line);
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr, "%s: looking up %s...\n", argv[0], line);
|
|
// initiate dns lookup
|
|
socket_fd = async_dns_lookup(line);
|
|
}
|
|
}
|
|
|
|
if (socket_fd && FD_ISSET(socket_fd, &fdset))
|
|
{
|
|
// read from socket, parse results
|
|
int size = read(socket_fd, buf, 1024);
|
|
if (size > 0)
|
|
{
|
|
// parse buffer into hostent
|
|
char *p = buf;
|
|
fprintf(stderr, "bytes read: %d\n", size);
|
|
fprintf(stderr, "response code: %d\n", *(int *)p);
|
|
p += sizeof(int);
|
|
|
|
for (int i=0; i < size; i++) {
|
|
if (!(i%8))
|
|
fprintf(stderr, "\n");
|
|
fprintf(stderr, "%2.2x ",(unsigned char)buf[i]);
|
|
}
|
|
fprintf(stderr, "\n");
|
|
hostent *h;
|
|
h = bytesToHostent(p);
|
|
}
|
|
close(socket_fd);
|
|
socket_fd = 0;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
buffer
|
|
|
|
int nameLen;
|
|
|
|
if (nameLen > 0)
|
|
char [nameLen+1] name
|
|
|
|
int aliasCount
|
|
for each alias
|
|
int aliasNameLen
|
|
char [aliasNameLen+1] aliasName
|
|
|
|
int h_addrtype
|
|
int h_length
|
|
int addrCount
|
|
for each addr
|
|
char[h_length] addr
|
|
|
|
|
|
|
|
*/
|