mirror of
				https://github.com/wangyu-/udp2raw.git
				synced 2025-11-01 02:35:37 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			119 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Bash
		
	
	
	
	
	
			
		
		
	
	
			119 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Bash
		
	
	
	
	
	
| #!/bin/sh /etc/rc.common
 | |
| 
 | |
| START=88
 | |
| STOP=15
 | |
| USE_PROCD=1
 | |
| 
 | |
| NAME=udp2raw
 | |
| 
 | |
| _log() {
 | |
| 	logger -p "daemon.$1" -t "$NAME" "$2"
 | |
| }
 | |
| 
 | |
| has_valid_server() {
 | |
| 	local server
 | |
| 	for server in $@; do
 | |
| 		[ "$(uci_get $NAME $server)" = "servers" ] && return 0
 | |
| 	done
 | |
| 	return 1
 | |
| }
 | |
| 
 | |
| create_config() {
 | |
| 	local config_file="$1"
 | |
| 	echo "# auto-generated config file from /etc/config/udp2raw" > $config_file
 | |
| 
 | |
| 	echo "-c" >> $config_file
 | |
| 	echo "-l ${listen_addr:=127.0.0.1}:${listen_port:=2080}" >> $config_file && _log "info" "listening on: ${listen_addr}:${listen_port}"
 | |
| 	echo "-r ${server_addr}:${server_port:=8080}" >> $config_file
 | |
| 
 | |
| 	[ -n "$raw_mode" ] && echo "--raw-mode ${raw_mode}" >> $config_file && _log "info" "raw-mode: ${raw_mode}"
 | |
| 	[ -n "$key" ] && echo "--key ${key}" >> $config_file
 | |
| 	[ -n "$cipher_mode" ] && echo "--cipher-mode ${cipher_mode}" >> $config_file
 | |
| 	[ -n "$auth_mode" ] && echo "--auth-mode ${auth_mode}" >> $config_file
 | |
| 
 | |
| 	[ $auto_rule -eq 1 ] && echo "--auto-rule" >> $config_file
 | |
| 	[ $auto_rule -eq 1 -a $keep_rule -eq 1 ] && echo "--keep-rule" >> $config_file
 | |
| 
 | |
| 	[ -n "$seq_mode" ] && echo "--seq-mode ${seq_mode}" >> $config_file
 | |
| 	[ -n "$lower_level" ] && echo "--lower-level ${lower_level}" >> $config_file
 | |
| 	[ -n "$source_ip" ] && echo "--source-ip ${source_ip}" >> $config_file
 | |
| 	[ -n "$source_port" ] && echo "--source-port ${source_port}" >> $config_file
 | |
| 
 | |
| 	[ -n "$log_level" ] && echo "--log-level ${log_level}" >> $config_file
 | |
| 	echo "--disable-color" >> $config_file
 | |
| 
 | |
| 	echo "--wait-lock" >> $config_file
 | |
| }
 | |
| 
 | |
| validate_config_section() {
 | |
| 	uci_validate_section "$NAME" general "$1" \
 | |
| 		'server:uciname' \
 | |
| 		'daemon_user:string:root'
 | |
| }
 | |
| 
 | |
| validate_server_section() {
 | |
| 	uci_validate_section "$NAME" servers "$1" \
 | |
| 		'server_addr:host' \
 | |
| 		'server_port:port:8080' \
 | |
| 		'listen_addr:ipaddr:127.0.0.1' \
 | |
| 		'listen_port:port:2080' \
 | |
| 		'raw_mode:or("faketcp", "udp", "icmp"):faketcp' \
 | |
| 		'key:string' \
 | |
| 		'cipher_mode:or("aes128cbc", "xor", "none"):aes128cbc' \
 | |
| 		'auth_mode:or("md5", "crc32", "simple", "none"):md5' \
 | |
| 		'auto_rule:bool:1' \
 | |
| 		'keep_rule:bool:0' \
 | |
| 		'seq_mode:range(0,4)' \
 | |
| 		'lower_level:string' \
 | |
| 		'source_ip:ipaddr' \
 | |
| 		'source_port:port' \
 | |
| 		'log_level:range(0,6)'
 | |
| }
 | |
| 
 | |
| start_instance() {
 | |
| 	local server="$1"
 | |
| 
 | |
| 	if [ -z "$server" -o "$server" == "nil" ]; then
 | |
| 		return 0
 | |
| 	elif ! validate_server_section "$server"; then
 | |
| 		_log "err" "Server config validation failed."
 | |
| 		return 1
 | |
| 	fi
 | |
| 
 | |
| 	/sbin/validate_data "ipaddr" "$server_addr" >/dev/null 2>&1
 | |
| 	[ $? -ne 0 ] && server_addr=$(nslookup "$server_addr" | \
 | |
| 		sed -n 's/^Address[[:space:]]*[0-9]*:[[:space:]]*\(\([0-9]\{1,3\}\.\)\{3\}[0-9]\{1,3\}\)$/\1/p')
 | |
| 	if [ -z "$server_addr" ]; then
 | |
| 		_log "err" "Server address validation failed."
 | |
| 		return 1
 | |
| 	fi
 | |
| 
 | |
| 	[ -d /var/etc ] || mkdir -p /var/etc
 | |
| 	local config_file="/var/etc/${NAME}.${server}.conf"
 | |
| 	create_config "$config_file" || return 1
 | |
| 
 | |
| 	procd_open_instance
 | |
| 	procd_set_param command /usr/bin/udp2raw
 | |
| 	procd_append_param command --conf-file "$config_file"
 | |
| 	procd_set_param respawn
 | |
| 	procd_set_param file "$config_file"
 | |
| 	[ -n "$daemon_user" ] && procd_set_param user "$daemon_user" && _log "info" "running from ${daemon_user} user"
 | |
| 	procd_set_param pidfile "/var/run/${NAME}.${server}.pid"
 | |
| 	procd_close_instance
 | |
| }
 | |
| 
 | |
| service_triggers() {
 | |
| 	procd_add_reload_trigger "$NAME"
 | |
| }
 | |
| 
 | |
| start_service() {
 | |
| 	if ! validate_config_section "general" ; then
 | |
| 		_log "err" "Config validate failed."
 | |
| 		return 1
 | |
| 	fi
 | |
| 	has_valid_server $server || return 1
 | |
| 	for srv in $server; do
 | |
| 		start_instance $srv
 | |
| 	done
 | |
| }
 |