mirror of
https://github.com/54shady/kernel_drivers_examples.git
synced 2025-08-11 23:32:00 +00:00
rk3399 ramdisk(busybox)
This commit is contained in:
143
README.md
143
README.md
@ -276,3 +276,146 @@ done
|
||||
结果中能看到如下信息
|
||||
|
||||
Linux version 4.4.16 (gcc version 4.9.4 (Linaro GCC 4.9-2017.01) )
|
||||
|
||||
### 编译ramdisk(busybox-1.22.0)
|
||||
|
||||
修改busybox源码中的xconnect.c解决ping bad addr bug
|
||||
|
||||
[参考文章](http://hankjin.blog.163.com/blog/static/337319372009327101324432/)
|
||||
|
||||
设置交叉编译工具和安装路径等(参考rk3399_busybox_defconfig)
|
||||
|
||||
Build BusyBox as a static binary (no shared libs)
|
||||
|
||||
可以在.config中直接设置
|
||||
|
||||
CONFIG_PREFIX="/home/zeroway/rk3399/bsp/ramdiskfs"
|
||||
CONFIG_CROSS_COMPILER_PREFIX="/home/zeroway/rk3399/tool/gcc-linaro-4.9.4-2017.01-i686_aarch64-linux-gnu/bin/aarch64-linux-gnu-"
|
||||
|
||||
编译安装
|
||||
|
||||
make
|
||||
make install
|
||||
|
||||
经过上一步已经将busybox安装到ramdiskfs文件夹中,有如下内容(linuxrc是指向busybox的链接)
|
||||
|
||||
bin linuxrc sbin usr
|
||||
|
||||
继续在ramdiskfs目录中创建必要的目录结构
|
||||
|
||||
mkdir var tmp sys root proc opt mnt lib home etc dev
|
||||
|
||||
再在etc下创建一个init.d的目录用于保存一些初始化的脚本
|
||||
|
||||
mkdir etc/init.d
|
||||
touch etc/init.d/rcS
|
||||
|
||||
rcS脚本如下
|
||||
|
||||
mount -a
|
||||
mkdir /dev/pts
|
||||
mount -t devpts devpts /dev/pts
|
||||
echo /sbin/mdev > /proc/sys/kernel/hotplug
|
||||
/sbin/mdev -s
|
||||
|
||||
给rcS脚本添加可执行权限
|
||||
|
||||
chmod +x etc/init.d/rcS
|
||||
|
||||
创建sysconfig
|
||||
|
||||
mkdir etc/sysconfig
|
||||
echo "rk3399" > etc/sysconfig/HOSTNAME
|
||||
|
||||
添加/etc/fstab内容如下
|
||||
|
||||
proc /proc proc defaults 0 0
|
||||
tmpfs /tmp tmpfs defaults 0 0
|
||||
sysfs /sys sysfs defaults 0 0
|
||||
tmpfs /dev tmpfs defaults 0 0
|
||||
|
||||
添加必要的设备节点
|
||||
|
||||
sudo mknod dev/console c 5 1
|
||||
sudo mknod dev/null c 1 3
|
||||
|
||||
添加/etc/inittab内容如下
|
||||
|
||||
::sysinit:/etc/init.d/rcS
|
||||
console::askfirst:-/bin/sh
|
||||
::ctrlaltdel:/sbin/reboot
|
||||
::shutdown:/bin/umount -a -r
|
||||
|
||||
添加/etc/profile内容如下
|
||||
|
||||
ulimit -S -c 0 > /dev/null 2>&1
|
||||
USER="`id -un`"
|
||||
LOGNAME=$USER
|
||||
PS1='[\u@\h \W]\# '
|
||||
PATH=$PATH:/usr/local/bin
|
||||
HOSTNAME=`/bin/hostname`
|
||||
export USER LOGNAME PS1 PATH
|
||||
|
||||
添加/etc/passwd内容如下
|
||||
|
||||
root:x:0:0:root:/:/bin/sh
|
||||
ftp::14:50:FTP User:/var/ftp:
|
||||
bin:*:1:1:bin:/bin:
|
||||
daemon:*:2:2:daemon:/sbin:
|
||||
nobody:*:99:99:Nobody:/:
|
||||
|
||||
添加/etc/group内容如下
|
||||
|
||||
root:*:0:
|
||||
daemon:*:1:
|
||||
bin:*:2:
|
||||
sys:*:3:
|
||||
adm:*:4:
|
||||
tty:*:5:
|
||||
disk:*:6:
|
||||
lp:*:7:lp
|
||||
mail:*:8:
|
||||
news:*:9:
|
||||
uucp:*:10:
|
||||
proxy:*:13:
|
||||
kmem:*:15:
|
||||
dialout:*:20:
|
||||
fax:*:21:
|
||||
voice:*:22:
|
||||
cdrom:*:24:
|
||||
floppy:*:25:
|
||||
tape:*:26:
|
||||
sudo:*:27:
|
||||
audio:*:29:
|
||||
ppp:x:99:
|
||||
|
||||
修改[cmdline](./cmdline_linux)参数如下(主要修改init=/linuxrc)
|
||||
|
||||
CMDLINE: androidboot.baseband=N/A androidboot.selinux=disabled androidboot.hardware=rk30board androidboot.console=ttyFIQ0 init=/linuxrc mtdparts=rk29xxnand:0x00002000@0x00002000(uboot),0x00002000@0x00004000(trust),0x00002000@0x00006000(misc),0x00008000@0x00008000(resource),0x0000A000@0x00010000(kernel),0x00010000@0x0001A000(boot),0x00010000@0x0002A000(recovery),0x00038000@0x0003A000(backup),0x00040000@0x00072000(cache),0x00200000@0x000B2000(system),0x00008000@0x002B2000(metadata),0x00002000@0x002BA000(baseparamer),-@0x002BC000(userdata)
|
||||
|
||||
|
||||
制作一个大小位8M的ext2格式镜像文件,并挂载到/mnt/initrd目录下
|
||||
拷贝ramdiskfs里所有文件到/mnt/initrd下
|
||||
|
||||
dd if=/dev/zero of=initrd.img bs=1k count=8192
|
||||
sudo mkfs.ext2 -F initrd.img
|
||||
sudo mkdir /mnt/initrd
|
||||
sudo mount -t ext2 -o loop initrd.img /mnt/initrd
|
||||
sudo cp -rfvd ramdiskfs/* /mnt/initrd
|
||||
sudo umount /mnt/initrd
|
||||
gzip --best -c initrd.img > ramdisk.img
|
||||
sudo chmod 777 ramdisk.img
|
||||
|
||||
使用RK提供的打包工具将ramdisk打包成boot
|
||||
|
||||
rkst/mkkrnlimg ramdisk.img boot.img
|
||||
|
||||
烧写ramdisk镜像
|
||||
|
||||
rkflashtool w boot < boot.img
|
||||
|
||||
上网设置(配置网卡,设置域名服务和默认网关)
|
||||
|
||||
ifconfig eth0 192.168.1.234 up
|
||||
echo "nameserver 192.168.1.1" > /etc/resolv.conf
|
||||
route add default gw 192.168.1.1
|
||||
|
15
cmdline_linux
Normal file
15
cmdline_linux
Normal file
@ -0,0 +1,15 @@
|
||||
FIRMWARE_VER: 6.0.1
|
||||
MACHINE_MODEL: RK3399
|
||||
MACHINE_ID: 007
|
||||
MANUFACTURER: RK3399
|
||||
MAGIC: 0x5041524B
|
||||
ATAG: 0x00200800
|
||||
MACHINE: 3399
|
||||
CHECK_MASK: 0x80
|
||||
PWR_HLD: 0,0,A,0,1
|
||||
#KERNEL_IMG: 0x00280000
|
||||
#FDT_NAME: rk-kernel.dtb
|
||||
#RECOVER_KEY: 1,1,0,20,0
|
||||
#in section; per section 512(0x200) bytes
|
||||
CMDLINE: androidboot.baseband=N/A androidboot.selinux=disabled androidboot.hardware=rk30board androidboot.console=ttyFIQ0 init=/linuxrc mtdparts=rk29xxnand:0x00002000@0x00002000(uboot),0x00002000@0x00004000(trust),0x00002000@0x00006000(misc),0x00008000@0x00008000(resource),0x0000A000@0x00010000(kernel),0x00010000@0x0001A000(boot),0x00010000@0x0002A000(recovery),0x00038000@0x0003A000(backup),0x00040000@0x00072000(cache),0x00200000@0x000B2000(system),0x00008000@0x002B2000(metadata),0x00002000@0x002BA000(baseparamer),-@0x002BC000(userdata)
|
||||
|
5
ramdiskfs/etc/fstab
Executable file
5
ramdiskfs/etc/fstab
Executable file
@ -0,0 +1,5 @@
|
||||
#device mount-point type options dump fsck order
|
||||
proc /proc proc defaults 0 0
|
||||
tmpfs /tmp tmpfs defaults 0 0
|
||||
sysfs /sys sysfs defaults 0 0
|
||||
tmpfs /dev tmpfs defaults 0 0
|
22
ramdiskfs/etc/group
Executable file
22
ramdiskfs/etc/group
Executable file
@ -0,0 +1,22 @@
|
||||
root:*:0:
|
||||
daemon:*:1:
|
||||
bin:*:2:
|
||||
sys:*:3:
|
||||
adm:*:4:
|
||||
tty:*:5:
|
||||
disk:*:6:
|
||||
lp:*:7:lp
|
||||
mail:*:8:
|
||||
news:*:9:
|
||||
uucp:*:10:
|
||||
proxy:*:13:
|
||||
kmem:*:15:
|
||||
dialout:*:20:
|
||||
fax:*:21:
|
||||
voice:*:22:
|
||||
cdrom:*:24:
|
||||
floppy:*:25:
|
||||
tape:*:26:
|
||||
sudo:*:27:
|
||||
audio:*:29:
|
||||
ppp:x:99:
|
7
ramdiskfs/etc/init.d/rcS
Executable file
7
ramdiskfs/etc/init.d/rcS
Executable file
@ -0,0 +1,7 @@
|
||||
#! /bin/sh
|
||||
|
||||
mount -a
|
||||
mkdir /dev/pts
|
||||
mount -t devpts devpts /dev/pts
|
||||
echo /sbin/mdev > /proc/sys/kernel/hotplug
|
||||
/sbin/mdev -s
|
4
ramdiskfs/etc/inittab
Executable file
4
ramdiskfs/etc/inittab
Executable file
@ -0,0 +1,4 @@
|
||||
::sysinit:/etc/init.d/rcS
|
||||
console::askfirst:-/bin/sh
|
||||
::ctrlaltdel:/sbin/reboot
|
||||
::shutdown:/bin/umount -a -r
|
5
ramdiskfs/etc/passwd
Executable file
5
ramdiskfs/etc/passwd
Executable file
@ -0,0 +1,5 @@
|
||||
root:x:0:0:root:/:/bin/sh
|
||||
ftp::14:50:FTP User:/var/ftp:
|
||||
bin:*:1:1:bin:/bin:
|
||||
daemon:*:2:2:daemon:/sbin:
|
||||
nobody:*:99:99:Nobody:/:
|
14
ramdiskfs/etc/profile
Executable file
14
ramdiskfs/etc/profile
Executable file
@ -0,0 +1,14 @@
|
||||
# Ash profile
|
||||
# vim: syntax=sh
|
||||
|
||||
# No core files by default
|
||||
ulimit -S -c 0 > /dev/null 2>&1
|
||||
|
||||
USER="`id -un`"
|
||||
LOGNAME=$USER
|
||||
PS1='[\u@\h \W]\# '
|
||||
PATH=$PATH:/usr/local/bin
|
||||
|
||||
HOSTNAME=`/bin/hostname`
|
||||
|
||||
export USER LOGNAME PS1 PATH
|
1
ramdiskfs/etc/sysconfig/HOSTNAME
Normal file
1
ramdiskfs/etc/sysconfig/HOSTNAME
Normal file
@ -0,0 +1 @@
|
||||
rk3399
|
1038
rk3399_busybox_defconfig
Normal file
1038
rk3399_busybox_defconfig
Normal file
File diff suppressed because it is too large
Load Diff
628
xconnect.c
Normal file
628
xconnect.c
Normal file
@ -0,0 +1,628 @@
|
||||
/* vi: set sw=4 ts=4: */
|
||||
/*
|
||||
* Utility routines.
|
||||
*
|
||||
* Connect to host at port using address resolution from getaddrinfo
|
||||
*
|
||||
* Licensed under GPLv2, see file LICENSE in this source tree.
|
||||
*/
|
||||
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h> /* netinet/in.h needs it */
|
||||
#include <netinet/in.h>
|
||||
#include <net/if.h>
|
||||
#include <sys/un.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <netdb.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#define oops(msg) { perror(msg); exit(1);}
|
||||
#include<ctype.h>
|
||||
|
||||
#include "libbb.h"
|
||||
|
||||
void fillip(char* buffer, const char* ip);
|
||||
char* name2ip(const char* name);
|
||||
/**
|
||||
* * head len: 12
|
||||
* * query: ? + 4
|
||||
* * total: strlen(query.buffer) + 16
|
||||
* */
|
||||
typedef struct{
|
||||
u_short txid;
|
||||
u_short flag;
|
||||
u_short question;
|
||||
u_short answer;
|
||||
u_short authority;
|
||||
u_short additional;
|
||||
char buffer[256];
|
||||
struct {
|
||||
u_short type;
|
||||
u_short cls;
|
||||
}query;
|
||||
}dns_req;
|
||||
/** length of resp: 12 */
|
||||
typedef struct{
|
||||
u_short txid;
|
||||
u_short flag;
|
||||
u_short question;
|
||||
u_short answer;
|
||||
u_short authority;
|
||||
u_short addtional;
|
||||
char buffer[1024];
|
||||
struct{
|
||||
u_short name;
|
||||
u_short type;
|
||||
u_short cls;
|
||||
u_short live_l;
|
||||
u_short live_h;
|
||||
u_short len;
|
||||
struct in_addr addr;
|
||||
}resp;
|
||||
}dns_res;
|
||||
|
||||
int hgetaddrinfo(const char *node, const char *service,
|
||||
const struct addrinfo *hints,
|
||||
struct addrinfo **res){
|
||||
/* int rc = getaddrinfo(node,service,hints,res);
|
||||
* if(rc == 0)//todo: != -> ==
|
||||
* return rc;
|
||||
* */
|
||||
char *ip = name2ip(node); /// 如果 ping www.baidu.com
|
||||
return getaddrinfo(ip,service,hints,res); /// 那么此时 node 指向字符串 "www.baidu.com"
|
||||
}
|
||||
|
||||
char *get_dns_name()
|
||||
{
|
||||
int fd;
|
||||
static char buf[1024],*pstr,*pdns;
|
||||
|
||||
fd = open("/etc/resolv.conf",O_RDWR);
|
||||
read(fd,buf,1020);
|
||||
pstr = strstr(buf,"nameserver");
|
||||
pstr += strlen("nameserver");
|
||||
while(!isdigit(*pstr)){
|
||||
pstr ++;
|
||||
}
|
||||
pdns = pstr;
|
||||
|
||||
while(isdigit(*pstr) || (*pstr == '.') ){
|
||||
pstr ++;
|
||||
}
|
||||
*pstr = '\0';
|
||||
close(fd);
|
||||
return pdns;
|
||||
}
|
||||
|
||||
|
||||
char * name2ip(const char *node){
|
||||
struct sockaddr_in dns;
|
||||
dns_req req;
|
||||
dns_res res;
|
||||
int sockid, len, index;
|
||||
get_dns_name();
|
||||
char dnsip[32];
|
||||
|
||||
strcpy(dnsip,get_dns_name());
|
||||
|
||||
sockid = socket(PF_INET, SOCK_DGRAM, 0);
|
||||
if(sockid == -1)
|
||||
oops("socket");
|
||||
memset((void*)&dns, 0, sizeof(dns));
|
||||
dns.sin_family = AF_INET;
|
||||
dns.sin_port = htons(53);
|
||||
dns.sin_addr.s_addr = inet_addr(dnsip);
|
||||
|
||||
memset((void*)&req, 0, sizeof(req));
|
||||
req.txid = htons(0x4419);
|
||||
req.flag = htons(0x0100);
|
||||
req.question = htons(1);
|
||||
fillip(req.buffer,node);
|
||||
req.query.type=htons(1);
|
||||
req.query.cls=htons(1);
|
||||
memcpy(req.buffer+strlen(req.buffer)+1, (void*)(&req.query), sizeof(req.query));
|
||||
|
||||
sendto(sockid, (void*)&req, strlen(req.buffer)+17, 0,
|
||||
(struct sockaddr*)&dns, sizeof(dns));
|
||||
recvfrom(sockid, (void*)&res, sizeof(res), 0,
|
||||
(struct sockaddr*)&dns, &len);
|
||||
index = strlen(res.buffer)+5;
|
||||
while(1){
|
||||
memcpy((void*)&(res.resp), res.buffer+index, 12);
|
||||
if(ntohs(res.resp.type)==1){
|
||||
memcpy((void*)&(res.resp.addr), res.buffer+index+12, 4);
|
||||
break;
|
||||
}
|
||||
index += ntohs(res.resp.len) + 12;
|
||||
}
|
||||
|
||||
return inet_ntoa(res.resp.addr);
|
||||
}
|
||||
|
||||
void fillip(char* buffer, const char* ip){
|
||||
int i,j=0;
|
||||
for(i = 0; ip[i] != 0; i ++){
|
||||
if(ip[i] != '.'){
|
||||
buffer[i+1] = ip[i];
|
||||
}
|
||||
else{
|
||||
buffer[j] = i - j;
|
||||
j = i + 1;
|
||||
}
|
||||
}
|
||||
buffer[j] = i - j;
|
||||
}
|
||||
|
||||
void FAST_FUNC setsockopt_reuseaddr(int fd)
|
||||
{
|
||||
setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &const_int_1, sizeof(const_int_1));
|
||||
}
|
||||
int FAST_FUNC setsockopt_broadcast(int fd)
|
||||
{
|
||||
return setsockopt(fd, SOL_SOCKET, SO_BROADCAST, &const_int_1, sizeof(const_int_1));
|
||||
}
|
||||
|
||||
#ifdef SO_BINDTODEVICE
|
||||
int FAST_FUNC setsockopt_bindtodevice(int fd, const char *iface)
|
||||
{
|
||||
int r;
|
||||
struct ifreq ifr;
|
||||
strncpy_IFNAMSIZ(ifr.ifr_name, iface);
|
||||
/* NB: passing (iface, strlen(iface) + 1) does not work!
|
||||
* (maybe it works on _some_ kernels, but not on 2.6.26)
|
||||
* Actually, ifr_name is at offset 0, and in practice
|
||||
* just giving char[IFNAMSIZ] instead of struct ifreq works too.
|
||||
* But just in case it's not true on some obscure arch... */
|
||||
r = setsockopt(fd, SOL_SOCKET, SO_BINDTODEVICE, &ifr, sizeof(ifr));
|
||||
if (r)
|
||||
bb_perror_msg("can't bind to interface %s", iface);
|
||||
return r;
|
||||
}
|
||||
#else
|
||||
int FAST_FUNC setsockopt_bindtodevice(int fd UNUSED_PARAM,
|
||||
const char *iface UNUSED_PARAM)
|
||||
{
|
||||
bb_error_msg("SO_BINDTODEVICE is not supported on this system");
|
||||
return -1;
|
||||
}
|
||||
#endif
|
||||
|
||||
static len_and_sockaddr* get_lsa(int fd, int (*get_name)(int fd, struct sockaddr *addr, socklen_t *addrlen))
|
||||
{
|
||||
len_and_sockaddr lsa;
|
||||
len_and_sockaddr *lsa_ptr;
|
||||
|
||||
lsa.len = LSA_SIZEOF_SA;
|
||||
if (get_name(fd, &lsa.u.sa, &lsa.len) != 0)
|
||||
return NULL;
|
||||
|
||||
lsa_ptr = xzalloc(LSA_LEN_SIZE + lsa.len);
|
||||
if (lsa.len > LSA_SIZEOF_SA) { /* rarely (if ever) happens */
|
||||
lsa_ptr->len = lsa.len;
|
||||
get_name(fd, &lsa_ptr->u.sa, &lsa_ptr->len);
|
||||
} else {
|
||||
memcpy(lsa_ptr, &lsa, LSA_LEN_SIZE + lsa.len);
|
||||
}
|
||||
return lsa_ptr;
|
||||
}
|
||||
|
||||
len_and_sockaddr* FAST_FUNC get_sock_lsa(int fd)
|
||||
{
|
||||
return get_lsa(fd, getsockname);
|
||||
}
|
||||
|
||||
len_and_sockaddr* FAST_FUNC get_peer_lsa(int fd)
|
||||
{
|
||||
return get_lsa(fd, getpeername);
|
||||
}
|
||||
|
||||
void FAST_FUNC xconnect(int s, const struct sockaddr *s_addr, socklen_t addrlen)
|
||||
{
|
||||
if (connect(s, s_addr, addrlen) < 0) {
|
||||
if (ENABLE_FEATURE_CLEAN_UP)
|
||||
close(s);
|
||||
if (s_addr->sa_family == AF_INET)
|
||||
bb_perror_msg_and_die("%s (%s)",
|
||||
"can't connect to remote host",
|
||||
inet_ntoa(((struct sockaddr_in *)s_addr)->sin_addr));
|
||||
bb_perror_msg_and_die("can't connect to remote host");
|
||||
}
|
||||
}
|
||||
|
||||
/* Return port number for a service.
|
||||
* If "port" is a number use it as the port.
|
||||
* If "port" is a name it is looked up in /etc/services,
|
||||
* if it isnt found return default_port
|
||||
*/
|
||||
unsigned FAST_FUNC bb_lookup_port(const char *port, const char *protocol, unsigned default_port)
|
||||
{
|
||||
unsigned port_nr = default_port;
|
||||
if (port) {
|
||||
int old_errno;
|
||||
|
||||
/* Since this is a lib function, we're not allowed to reset errno to 0.
|
||||
* Doing so could break an app that is deferring checking of errno. */
|
||||
old_errno = errno;
|
||||
port_nr = bb_strtou(port, NULL, 10);
|
||||
if (errno || port_nr > 65535) {
|
||||
struct servent *tserv = getservbyname(port, protocol);
|
||||
port_nr = default_port;
|
||||
if (tserv)
|
||||
port_nr = ntohs(tserv->s_port);
|
||||
}
|
||||
errno = old_errno;
|
||||
}
|
||||
return (uint16_t)port_nr;
|
||||
}
|
||||
|
||||
|
||||
/* "New" networking API */
|
||||
|
||||
|
||||
int FAST_FUNC get_nport(const struct sockaddr *sa)
|
||||
{
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
if (sa->sa_family == AF_INET6) {
|
||||
return ((struct sockaddr_in6*)sa)->sin6_port;
|
||||
}
|
||||
#endif
|
||||
if (sa->sa_family == AF_INET) {
|
||||
return ((struct sockaddr_in*)sa)->sin_port;
|
||||
}
|
||||
/* What? UNIX socket? IPX?? :) */
|
||||
return -1;
|
||||
}
|
||||
|
||||
void FAST_FUNC set_nport(struct sockaddr *sa, unsigned port)
|
||||
{
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
if (sa->sa_family == AF_INET6) {
|
||||
struct sockaddr_in6 *sin6 = (void*) sa;
|
||||
sin6->sin6_port = port;
|
||||
return;
|
||||
}
|
||||
#endif
|
||||
if (sa->sa_family == AF_INET) {
|
||||
struct sockaddr_in *sin = (void*) sa;
|
||||
sin->sin_port = port;
|
||||
return;
|
||||
}
|
||||
/* What? UNIX socket? IPX?? :) */
|
||||
}
|
||||
|
||||
/* We hijack this constant to mean something else */
|
||||
/* It doesn't hurt because we will remove this bit anyway */
|
||||
#define DIE_ON_ERROR AI_CANONNAME
|
||||
|
||||
|
||||
/* host: "1.2.3.4[:port]", "www.google.com[:port]"
|
||||
* port: if neither of above specifies port # */
|
||||
static len_and_sockaddr* str2sockaddr(
|
||||
const char *host, int port,
|
||||
IF_FEATURE_IPV6(sa_family_t af,)
|
||||
int ai_flags)
|
||||
{
|
||||
IF_NOT_FEATURE_IPV6(sa_family_t af = AF_INET;)
|
||||
int rc;
|
||||
len_and_sockaddr *r;
|
||||
struct addrinfo *result = NULL;
|
||||
struct addrinfo *used_res;
|
||||
const char *org_host = host; /* only for error msg */
|
||||
const char *cp;
|
||||
struct addrinfo hint;
|
||||
|
||||
if (ENABLE_FEATURE_UNIX_LOCAL && strncmp(host, "local:", 6) == 0) {
|
||||
struct sockaddr_un *sun;
|
||||
|
||||
r = xzalloc(LSA_LEN_SIZE + sizeof(struct sockaddr_un));
|
||||
r->len = sizeof(struct sockaddr_un);
|
||||
r->u.sa.sa_family = AF_UNIX;
|
||||
sun = (struct sockaddr_un *)&r->u.sa;
|
||||
safe_strncpy(sun->sun_path, host + 6, sizeof(sun->sun_path));
|
||||
return r;
|
||||
}
|
||||
|
||||
r = NULL;
|
||||
|
||||
/* Ugly parsing of host:addr */
|
||||
if (ENABLE_FEATURE_IPV6 && host[0] == '[') {
|
||||
/* Even uglier parsing of [xx]:nn */
|
||||
host++;
|
||||
cp = strchr(host, ']');
|
||||
if (!cp || (cp[1] != ':' && cp[1] != '\0')) {
|
||||
/* Malformed: must be [xx]:nn or [xx] */
|
||||
bb_error_msg("bad address '%s'", org_host);
|
||||
if (ai_flags & DIE_ON_ERROR)
|
||||
xfunc_die();
|
||||
return NULL;
|
||||
}
|
||||
} else {
|
||||
cp = strrchr(host, ':');
|
||||
if (ENABLE_FEATURE_IPV6 && cp && strchr(host, ':') != cp) {
|
||||
/* There is more than one ':' (e.g. "::1") */
|
||||
cp = NULL; /* it's not a port spec */
|
||||
}
|
||||
}
|
||||
if (cp) { /* points to ":" or "]:" */
|
||||
int sz = cp - host + 1;
|
||||
|
||||
host = safe_strncpy(alloca(sz), host, sz);
|
||||
if (ENABLE_FEATURE_IPV6 && *cp != ':') {
|
||||
cp++; /* skip ']' */
|
||||
if (*cp == '\0') /* [xx] without port */
|
||||
goto skip;
|
||||
}
|
||||
cp++; /* skip ':' */
|
||||
port = bb_strtou(cp, NULL, 10);
|
||||
if (errno || (unsigned)port > 0xffff) {
|
||||
bb_error_msg("bad port spec '%s'", org_host);
|
||||
if (ai_flags & DIE_ON_ERROR)
|
||||
xfunc_die();
|
||||
return NULL;
|
||||
}
|
||||
skip: ;
|
||||
}
|
||||
|
||||
/* Next two if blocks allow to skip getaddrinfo()
|
||||
* in case host name is a numeric IP(v6) address.
|
||||
* getaddrinfo() initializes DNS resolution machinery,
|
||||
* scans network config and such - tens of syscalls.
|
||||
*/
|
||||
/* If we were not asked specifically for IPv6,
|
||||
* check whether this is a numeric IPv4 */
|
||||
IF_FEATURE_IPV6(if(af != AF_INET6)) {
|
||||
struct in_addr in4;
|
||||
if (inet_aton(host, &in4) != 0) {
|
||||
r = xzalloc(LSA_LEN_SIZE + sizeof(struct sockaddr_in));
|
||||
r->len = sizeof(struct sockaddr_in);
|
||||
r->u.sa.sa_family = AF_INET;
|
||||
r->u.sin.sin_addr = in4;
|
||||
goto set_port;
|
||||
}
|
||||
}
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
/* If we were not asked specifically for IPv4,
|
||||
* check whether this is a numeric IPv6 */
|
||||
if (af != AF_INET) {
|
||||
struct in6_addr in6;
|
||||
if (inet_pton(AF_INET6, host, &in6) > 0) {
|
||||
r = xzalloc(LSA_LEN_SIZE + sizeof(struct sockaddr_in6));
|
||||
r->len = sizeof(struct sockaddr_in6);
|
||||
r->u.sa.sa_family = AF_INET6;
|
||||
r->u.sin6.sin6_addr = in6;
|
||||
goto set_port;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
memset(&hint, 0 , sizeof(hint));
|
||||
hint.ai_family = af;
|
||||
/* Need SOCK_STREAM, or else we get each address thrice (or more)
|
||||
* for each possible socket type (tcp,udp,raw...): */
|
||||
hint.ai_socktype = SOCK_STREAM;
|
||||
hint.ai_flags = ai_flags & ~DIE_ON_ERROR;
|
||||
rc = hgetaddrinfo(host, NULL, &hint, &result);
|
||||
//rc = getaddrinfo(host, NULL, &hint, &result);
|
||||
if (rc || !result) {
|
||||
bb_error_msg("bad address '%s'", org_host);
|
||||
if (ai_flags & DIE_ON_ERROR)
|
||||
xfunc_die();
|
||||
goto ret;
|
||||
}
|
||||
used_res = result;
|
||||
#if ENABLE_FEATURE_PREFER_IPV4_ADDRESS
|
||||
while (1) {
|
||||
if (used_res->ai_family == AF_INET)
|
||||
break;
|
||||
used_res = used_res->ai_next;
|
||||
if (!used_res) {
|
||||
used_res = result;
|
||||
break;
|
||||
}
|
||||
}
|
||||
#endif
|
||||
r = xmalloc(LSA_LEN_SIZE + used_res->ai_addrlen);
|
||||
r->len = used_res->ai_addrlen;
|
||||
memcpy(&r->u.sa, used_res->ai_addr, used_res->ai_addrlen);
|
||||
|
||||
set_port:
|
||||
set_nport(&r->u.sa, htons(port));
|
||||
ret:
|
||||
if (result)
|
||||
freeaddrinfo(result);
|
||||
return r;
|
||||
}
|
||||
#if !ENABLE_FEATURE_IPV6
|
||||
#define str2sockaddr(host, port, af, ai_flags) str2sockaddr(host, port, ai_flags)
|
||||
#endif
|
||||
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
len_and_sockaddr* FAST_FUNC host_and_af2sockaddr(const char *host, int port, sa_family_t af)
|
||||
{
|
||||
return str2sockaddr(host, port, af, 0);
|
||||
}
|
||||
|
||||
len_and_sockaddr* FAST_FUNC xhost_and_af2sockaddr(const char *host, int port, sa_family_t af)
|
||||
{
|
||||
return str2sockaddr(host, port, af, DIE_ON_ERROR);
|
||||
}
|
||||
#endif
|
||||
|
||||
len_and_sockaddr* FAST_FUNC host2sockaddr(const char *host, int port)
|
||||
{
|
||||
return str2sockaddr(host, port, AF_UNSPEC, 0);
|
||||
}
|
||||
|
||||
len_and_sockaddr* FAST_FUNC xhost2sockaddr(const char *host, int port)
|
||||
{
|
||||
return str2sockaddr(host, port, AF_UNSPEC, DIE_ON_ERROR);
|
||||
}
|
||||
|
||||
len_and_sockaddr* FAST_FUNC xdotted2sockaddr(const char *host, int port)
|
||||
{
|
||||
return str2sockaddr(host, port, AF_UNSPEC, AI_NUMERICHOST | DIE_ON_ERROR);
|
||||
}
|
||||
|
||||
int FAST_FUNC xsocket_type(len_and_sockaddr **lsap, int family, int sock_type)
|
||||
{
|
||||
len_and_sockaddr *lsa;
|
||||
int fd;
|
||||
int len;
|
||||
|
||||
if (family == AF_UNSPEC) {
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
fd = socket(AF_INET6, sock_type, 0);
|
||||
if (fd >= 0) {
|
||||
family = AF_INET6;
|
||||
goto done;
|
||||
}
|
||||
#endif
|
||||
family = AF_INET;
|
||||
}
|
||||
|
||||
fd = xsocket(family, sock_type, 0);
|
||||
|
||||
len = sizeof(struct sockaddr_in);
|
||||
if (family == AF_UNIX)
|
||||
len = sizeof(struct sockaddr_un);
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
if (family == AF_INET6) {
|
||||
done:
|
||||
len = sizeof(struct sockaddr_in6);
|
||||
}
|
||||
#endif
|
||||
lsa = xzalloc(LSA_LEN_SIZE + len);
|
||||
lsa->len = len;
|
||||
lsa->u.sa.sa_family = family;
|
||||
*lsap = lsa;
|
||||
return fd;
|
||||
}
|
||||
|
||||
int FAST_FUNC xsocket_stream(len_and_sockaddr **lsap)
|
||||
{
|
||||
return xsocket_type(lsap, AF_UNSPEC, SOCK_STREAM);
|
||||
}
|
||||
|
||||
static int create_and_bind_or_die(const char *bindaddr, int port, int sock_type)
|
||||
{
|
||||
int fd;
|
||||
len_and_sockaddr *lsa;
|
||||
|
||||
if (bindaddr && bindaddr[0]) {
|
||||
lsa = xdotted2sockaddr(bindaddr, port);
|
||||
/* user specified bind addr dictates family */
|
||||
fd = xsocket(lsa->u.sa.sa_family, sock_type, 0);
|
||||
} else {
|
||||
fd = xsocket_type(&lsa, AF_UNSPEC, sock_type);
|
||||
set_nport(&lsa->u.sa, htons(port));
|
||||
}
|
||||
setsockopt_reuseaddr(fd);
|
||||
xbind(fd, &lsa->u.sa, lsa->len);
|
||||
free(lsa);
|
||||
return fd;
|
||||
}
|
||||
|
||||
int FAST_FUNC create_and_bind_stream_or_die(const char *bindaddr, int port)
|
||||
{
|
||||
return create_and_bind_or_die(bindaddr, port, SOCK_STREAM);
|
||||
}
|
||||
|
||||
int FAST_FUNC create_and_bind_dgram_or_die(const char *bindaddr, int port)
|
||||
{
|
||||
return create_and_bind_or_die(bindaddr, port, SOCK_DGRAM);
|
||||
}
|
||||
|
||||
|
||||
int FAST_FUNC create_and_connect_stream_or_die(const char *peer, int port)
|
||||
{
|
||||
int fd;
|
||||
len_and_sockaddr *lsa;
|
||||
|
||||
lsa = xhost2sockaddr(peer, port);
|
||||
fd = xsocket(lsa->u.sa.sa_family, SOCK_STREAM, 0);
|
||||
setsockopt_reuseaddr(fd);
|
||||
xconnect(fd, &lsa->u.sa, lsa->len);
|
||||
free(lsa);
|
||||
return fd;
|
||||
}
|
||||
|
||||
int FAST_FUNC xconnect_stream(const len_and_sockaddr *lsa)
|
||||
{
|
||||
int fd = xsocket(lsa->u.sa.sa_family, SOCK_STREAM, 0);
|
||||
xconnect(fd, &lsa->u.sa, lsa->len);
|
||||
return fd;
|
||||
}
|
||||
|
||||
/* We hijack this constant to mean something else */
|
||||
/* It doesn't hurt because we will add this bit anyway */
|
||||
#define IGNORE_PORT NI_NUMERICSERV
|
||||
static char* FAST_FUNC sockaddr2str(const struct sockaddr *sa, int flags)
|
||||
{
|
||||
char host[128];
|
||||
char serv[16];
|
||||
int rc;
|
||||
socklen_t salen;
|
||||
|
||||
if (ENABLE_FEATURE_UNIX_LOCAL && sa->sa_family == AF_UNIX) {
|
||||
struct sockaddr_un *sun = (struct sockaddr_un *)sa;
|
||||
return xasprintf("local:%.*s",
|
||||
(int) sizeof(sun->sun_path),
|
||||
sun->sun_path);
|
||||
}
|
||||
|
||||
salen = LSA_SIZEOF_SA;
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
if (sa->sa_family == AF_INET)
|
||||
salen = sizeof(struct sockaddr_in);
|
||||
if (sa->sa_family == AF_INET6)
|
||||
salen = sizeof(struct sockaddr_in6);
|
||||
#endif
|
||||
rc = getnameinfo(sa, salen,
|
||||
host, sizeof(host),
|
||||
/* can do ((flags & IGNORE_PORT) ? NULL : serv) but why bother? */
|
||||
serv, sizeof(serv),
|
||||
/* do not resolve port# into service _name_ */
|
||||
flags | NI_NUMERICSERV
|
||||
);
|
||||
if (rc)
|
||||
return NULL;
|
||||
if (flags & IGNORE_PORT)
|
||||
return xstrdup(host);
|
||||
#if ENABLE_FEATURE_IPV6
|
||||
if (sa->sa_family == AF_INET6) {
|
||||
if (strchr(host, ':')) /* heh, it's not a resolved hostname */
|
||||
return xasprintf("[%s]:%s", host, serv);
|
||||
/*return xasprintf("%s:%s", host, serv);*/
|
||||
/* - fall through instead */
|
||||
}
|
||||
#endif
|
||||
/* For now we don't support anything else, so it has to be INET */
|
||||
/*if (sa->sa_family == AF_INET)*/
|
||||
return xasprintf("%s:%s", host, serv);
|
||||
/*return xstrdup(host);*/
|
||||
}
|
||||
|
||||
char* FAST_FUNC xmalloc_sockaddr2host(const struct sockaddr *sa)
|
||||
{
|
||||
return sockaddr2str(sa, 0);
|
||||
}
|
||||
|
||||
char* FAST_FUNC xmalloc_sockaddr2host_noport(const struct sockaddr *sa)
|
||||
{
|
||||
return sockaddr2str(sa, IGNORE_PORT);
|
||||
}
|
||||
|
||||
char* FAST_FUNC xmalloc_sockaddr2hostonly_noport(const struct sockaddr *sa)
|
||||
{
|
||||
return sockaddr2str(sa, NI_NAMEREQD | IGNORE_PORT);
|
||||
}
|
||||
char* FAST_FUNC xmalloc_sockaddr2dotted(const struct sockaddr *sa)
|
||||
{
|
||||
return sockaddr2str(sa, NI_NUMERICHOST);
|
||||
}
|
||||
|
||||
char* FAST_FUNC xmalloc_sockaddr2dotted_noport(const struct sockaddr *sa)
|
||||
{
|
||||
return sockaddr2str(sa, NI_NUMERICHOST | IGNORE_PORT);
|
||||
}
|
Reference in New Issue
Block a user