Compare commits

...

86 Commits

Author SHA1 Message Date
Nick Peng
86902d2e34 serve-expired: support config reply ttl 2020-09-06 15:21:37 +08:00
Nick Peng
6f30fe6d05 compile: remove O_CLOEXEC flags 2020-09-05 19:37:50 +08:00
Nick Peng
0b45da29c7 cache: support persist cache when restart smartdns 2020-09-05 12:55:30 +08:00
Nick Peng
2b81fffb7e dns_client: fix ssl race condition issue. 2020-09-02 22:03:24 +08:00
Nick Peng
7af6f475da dns-client: reduce ssl session cache size 2020-08-27 23:34:22 +08:00
Purple Grape
c3b6560b46 add rpm support 2020-08-25 22:02:45 +08:00
Nick Peng
a3d3364a32 dns_client: avoid disconnect with server after query complete 2020-08-24 23:54:06 +08:00
Nick Peng
b8a36ccb8c dns_client: avoid connect failure 2020-08-24 23:18:35 +08:00
Nick Peng
aee19be262 dns-cache: support cache when speed check is disabled 2020-08-23 20:12:50 +08:00
Nick Peng
98429e88f1 speed_check: fix query failure when speed-check is none and dualstack-selection is enable. 2020-08-23 15:28:56 +08:00
Nick Peng
80cb27c795 tlog: update tlog 2020-08-22 08:13:55 +08:00
Kevin Zhou
cb73eadf01 Make the ipset working as expected when the domain passthrough 2020-08-22 07:57:00 +08:00
Nick Peng
69ba3f8789 pending: fix pending server resolve issue 2020-08-16 17:05:51 +08:00
Nick Peng
c380bbe0e3 tlog: update tlog 2020-08-13 01:23:40 +08:00
Nick Peng
da74e877c5 dns_client: repeat same arguments with SSL_write 2020-08-13 00:56:51 +08:00
Nick Peng
a300873b3f dns-client: fix ssl send request issue 2020-08-11 00:09:38 +08:00
Nick Peng
a9829f6155 dns-server: fix race condition issue 2020-08-09 23:54:23 +08:00
Nick Peng
1923271630 compile: fix comile warnings 2020-08-09 18:01:33 +08:00
Nick Peng
c23ec7ea8f client: suppress log. 2020-08-03 01:07:47 +08:00
rhettor
aad751c1f5 Support docker 2020-08-02 18:10:31 +08:00
Nick Peng
138df2fd5d luci: fix no-check-certificate config not working issue 2020-08-02 17:43:38 +08:00
Nick Peng
297ea29639 Client: Fix ssl crash issue. 2020-08-02 01:00:46 +08:00
Nick Peng
bc0d6b89ca tlog: update tlog 2020-07-30 00:03:20 +08:00
Nick Peng
2148efd262 client: retry upstream when network restart 2020-07-12 11:12:55 +08:00
Nick Peng
ce46ac58a7 dns_client: check ssl before set certificate 2020-05-10 22:04:48 +08:00
Nick Peng
c1f9941427 dualstack: update cache when ipv6 not exists 2020-05-04 02:24:55 +08:00
Nick Peng
770ce9e8bc dns-client: fix add pending server issue when AAAA record not exists 2020-05-04 00:03:31 +08:00
Nick Peng
60d0fc856e luci-compate: fix typo 2020-05-03 23:32:49 +08:00
Nick Peng
95472129ee dns-client: increase dns query retry times. 2020-05-03 17:00:50 +08:00
Nick Peng
f79ceeac43 luci: update luci 2020-05-01 00:23:21 +08:00
Nick Peng
54801e1ed6 tls: support certificate verify. 2020-05-01 00:23:19 +08:00
Nick Peng
304e94bc6f fast-ping: force send RST after ping finish. 2020-04-19 09:43:47 +08:00
Nick Peng
3985ea8269 pending-server: retry for 2 minutes when add pending server. 2020-04-17 22:37:49 +08:00
Nick Peng
c2c418ed89 dns-client: fix https memory leak issue. 2020-04-10 23:37:13 +08:00
clay_your
b78c9eb6fe fix missing the FILE field..
fix  Download/default is missing the FILE field..  Stop. error on openwrt
2020-04-06 21:55:37 +08:00
Nick Peng
0aec326d55 QRCode: update QR code. 2020-04-01 22:43:54 +08:00
Nick Peng
29e4d4b48b init-script: fix openwrt init-script issue. 2020-03-22 17:11:33 +08:00
Felix Yan
a6fe329105 Fix hostname option name in conf example
The hostname option is actually `-host-name` as in the code and README. Setting `-hostname` doesn't work.
2020-03-09 22:13:37 +08:00
Nick Peng
c824c0abc5 CodeStyle: update clang-format, limite column to 120 2020-03-02 23:45:47 +08:00
Nick Peng
3ad7cd7f45 install-script: Follow FHS rules 2020-02-27 00:50:43 +08:00
Nick Peng
b31792ad9b luci-app-smartdns: clean javascript code 2020-02-25 21:52:48 +08:00
Nick Peng
5a411c3937 luci-app-smartdns: remove uci track 2020-02-25 21:52:47 +08:00
Nick Peng
0092212f46 Bugfix: fix crash issue when has pending server 2020-02-25 21:50:13 +08:00
Nick Peng
f269f582f3 Bugfix: ping test not working issue 2020-02-24 21:06:50 +08:00
Nick Peng
9bc32f9f01 luci-build: fix luci build script 2020-02-24 20:45:52 +08:00
Nick Peng
f6255da2ad License: update license file 2020-02-24 20:26:23 +08:00
Nick Peng
c5985d3f2d Bugfix: dns_server_query is not call callback function when rule is hit. 2020-02-24 20:16:37 +08:00
Nick Peng
0d2d08586a ReadMe: Update package list information 2020-02-19 23:17:34 +08:00
Nick Peng
c8ab6db39e BugFix: fix install script issue, and change openwrt script default port for seconddns 2020-02-18 19:39:01 +08:00
Nick Peng
2afd4e7d4a luci-app-smartdns: Service redirect status is not displayed when smartdns is disabled 2020-02-15 14:48:56 +08:00
Nick Peng
20051a3698 second DNS: change default port 7053 to 6553 2020-02-15 14:35:51 +08:00
Nick Peng
5f9167cf07 bind-option: support force aaaa soa bind option, secondary DNS support Force-AAAA-SOA options 2020-02-14 13:14:03 +08:00
Nick Peng
3cf2d35c13 dualstack-selection: force IPV4 when not found IPV6 in cache 2020-02-14 13:13:47 +08:00
Nick Peng
8990446411 serve-expired: new feature, support cache serve-expired feature 2020-02-14 13:13:22 +08:00
Felix Yan
8f40eba083 smartdns.conf: correct some typos 2020-02-14 09:41:12 +08:00
Nick Peng
4bd6c025db domain-rules: new feature, support set speed-check-mode 2020-02-05 23:21:59 +08:00
Nick Peng
80174d9da5 luci-app-smartdns: rewrite luci-app in JavaScript 2020-02-01 14:05:05 +08:00
Nick Peng
7d1b850119 luci-app-smartdns: rename luci-app-smartdns to luci-app-smartdns-compat 2020-02-01 11:55:43 +08:00
Nick Peng
744a5409e0 init-script: fix a script typo 2020-01-31 14:16:28 +08:00
Rosen Penev
d965df2c44 run openwrt init script through shellcheck
Signed-off-by: Rosen Penev <rosenp@gmail.com>
2020-01-30 13:12:52 +08:00
Nick Peng
2322a9dbd0 ssl: Stop using deprecated Openssl APIs, by neheb. 2020-01-21 22:48:02 +08:00
Nick Peng
aa7237f770 Codesytle: replace CRLF with LF 2020-01-21 20:00:38 +08:00
Soff
50572acc11 Update ReadMe.md
`bind` -> `bind-tcp`
2020-01-13 23:59:21 +08:00
Nick Peng
511f8c135c BugFix: ssl crash issue. 2020-01-04 00:09:57 +08:00
Nick Peng
198d18f18a BugFix: tcp server process failed 2019-12-28 21:57:56 +08:00
Nick Peng
ba9fbc981f add some SSL exception check 2019-12-27 00:44:49 +08:00
Nick Peng
5a91e90ee3 update Makefile for openwrt 2019-12-21 11:56:04 +08:00
Nick Peng
982002e836 Modify the detection method of smartdns 2019-12-21 11:28:37 +08:00
Nick Peng
aaeae7167b Fix shellcheck issues 2019-12-21 11:02:55 +08:00
Nick Peng
0228a79991 Fix smartdns luci typo 2019-12-19 22:38:53 +08:00
Nick Peng
d6c34658af Add github issue template 2019-12-15 10:32:45 +08:00
Nick Peng
cc93156ee3 Add package build script 2019-12-15 01:28:37 +08:00
Nick Peng
8433729186 Update ReadMe.md 2019-12-15 01:28:37 +08:00
Nick Peng
4464bb982e update makefile for openwrt 2019-12-15 01:28:36 +08:00
Nick Peng
371aab6d4b support override CFLAGS and CXXFLAGS 2019-12-15 01:28:35 +08:00
Nick Peng
01e4140722 Add license 2019-12-15 01:28:32 +08:00
Nick Peng
2b08ffee98 Add some log, fix tlog issue 2019-12-15 01:28:31 +08:00
Nick Peng
9f27bb9b70 Fix compile error 2019-12-15 01:28:30 +08:00
Nick Peng
528d5c6ccc Update tlog 2019-12-15 01:28:30 +08:00
Nick Peng
6bf6315a6d Fix ssl crash issue 2019-12-15 01:28:29 +08:00
Nick Peng
aaca1c013b Fix byte alignment issue 2019-12-15 01:28:27 +08:00
Nick Peng
25c6cab4a8 Fix installation issue on arch-linux 2019-12-15 01:28:26 +08:00
Nick Peng
408c199564 create pid before load config file to avoid systemd complain 2019-12-15 01:28:24 +08:00
Nick Peng
044536aef6 optimize tcp connection lag 2019-12-15 01:28:22 +08:00
Nick Peng
1fb5cbbcd9 Fix ignore flags not working issue 2019-12-15 01:28:18 +08:00
Nick Peng
4e0cbc50b3 Update ReadMe.md 2019-12-15 01:28:17 +08:00
98 changed files with 17806 additions and 13004 deletions

View File

@@ -1,9 +1,10 @@
#http://clang.llvm.org/docs/ClangFormatStyleOptions.html #http://clang.llvm.org/docs/ClangFormatStyleOptions.html
BasedOnStyle: LLVM BasedOnStyle: LLVM
IndentWidth: 4 IndentWidth: 4
TabWidth: 4 TabWidth: 4
UseTab: ForContinuationAndIndentation UseTab: ForContinuationAndIndentation
MaxEmptyLinesToKeep: 1 MaxEmptyLinesToKeep: 1
AllowShortFunctionsOnASingleLine: Empty AllowShortFunctionsOnASingleLine: Empty
BreakBeforeBraces: Linux BreakBeforeBraces: Linux
ColumnLimit: 120

20
.github/ISSUE_TEMPLATE/feature.md vendored Normal file
View File

@@ -0,0 +1,20 @@
---
name: 需求建议
about: 需求建议描述
title: ''
labels: ''
assignees: ''
---
**需求应用场景**
请描述需求应用的场景和方式。
**建议的方案**
实现上述场景建议的方案。
**设备信息**
1. 设备信息CPU厂家
2. 固件信息

29
.github/ISSUE_TEMPLATE/issue.md vendored Normal file
View File

@@ -0,0 +1,29 @@
---
name: 问题报告
about: 问题现象描述
title: ''
labels: ''
assignees: ''
---
**问题现象**
简要描述问题出现的现象
**运行环境**
1. 固件型号
2. 运营商
**重现步骤**
1. 上游DNS配置。
2. 访问的域名。
**信息收集**
1. 将/var/log/smrtdns.log日志作为附件上传。
2. 如进程异常请将coredump功能开启上传coredump信息文件。
在自定义界面,开启设置->自定义设置->生成coredump配置重现问题后提交coredump文件
coredump文件在/tmp目录下

18
Dockerfile Normal file
View File

@@ -0,0 +1,18 @@
FROM debian:buster-slim
RUN apt update && \
apt install -y git make gcc libssl-dev && \
git clone https://github.com/pymumu/smartdns.git --depth 1 && \
cd smartdns && \
sh ./package/build-pkg.sh --platform debian --arch `dpkg --print-architecture` && \
dpkg -i package/*.deb && \
cd / && \
rm -rf smartdns/ && \
apt autoremove -y git make gcc libssl-dev && \
apt clean && \
rm -rf /var/lib/apt/lists/*
EXPOSE 53/udp
VOLUME "/etc/smartdns/"
CMD ["/usr/sbin/smartdns", "-f"]

149
LICENSE
View File

@@ -1,21 +1,23 @@
GNU AFFERO GENERAL PUBLIC LICENSE GNU GENERAL PUBLIC LICENSE
Version 3, 19 November 2007 Version 3, 29 June 2007
Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/> Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
Everyone is permitted to copy and distribute verbatim copies Everyone is permitted to copy and distribute verbatim copies
of this license document, but changing it is not allowed. of this license document, but changing it is not allowed.
Preamble Preamble
The GNU Affero General Public License is a free, copyleft license for The GNU General Public License is a free, copyleft license for
software and other kinds of works, specifically designed to ensure software and other kinds of works.
cooperation with the community in the case of network server software.
The licenses for most software and other practical works are designed The licenses for most software and other practical works are designed
to take away your freedom to share and change the works. By contrast, to take away your freedom to share and change the works. By contrast,
our General Public Licenses are intended to guarantee your freedom to the GNU General Public License is intended to guarantee your freedom to
share and change all versions of a program--to make sure it remains free share and change all versions of a program--to make sure it remains free
software for all its users. software for all its users. We, the Free Software Foundation, use the
GNU General Public License for most of our software; it applies also to
any other work released this way by its authors. You can apply it to
your programs, too.
When we speak of free software, we are referring to freedom, not When we speak of free software, we are referring to freedom, not
price. Our General Public Licenses are designed to make sure that you price. Our General Public Licenses are designed to make sure that you
@@ -24,34 +26,44 @@ them if you wish), that you receive source code or can get it if you
want it, that you can change the software or use pieces of it in new want it, that you can change the software or use pieces of it in new
free programs, and that you know you can do these things. free programs, and that you know you can do these things.
Developers that use our General Public Licenses protect your rights To protect your rights, we need to prevent others from denying you
with two steps: (1) assert copyright on the software, and (2) offer these rights or asking you to surrender the rights. Therefore, you have
you this License which gives you legal permission to copy, distribute certain responsibilities if you distribute copies of the software, or if
and/or modify the software. you modify it: responsibilities to respect the freedom of others.
A secondary benefit of defending all users' freedom is that For example, if you distribute copies of such a program, whether
improvements made in alternate versions of the program, if they gratis or for a fee, you must pass on to the recipients the same
receive widespread use, become available for other developers to freedoms that you received. You must make sure that they, too, receive
incorporate. Many developers of free software are heartened and or can get the source code. And you must show them these terms so they
encouraged by the resulting cooperation. However, in the case of know their rights.
software used on network servers, this result may fail to come about.
The GNU General Public License permits making a modified version and
letting the public access it on a server without ever releasing its
source code to the public.
The GNU Affero General Public License is designed specifically to Developers that use the GNU GPL protect your rights with two steps:
ensure that, in such cases, the modified source code becomes available (1) assert copyright on the software, and (2) offer you this License
to the community. It requires the operator of a network server to giving you legal permission to copy, distribute and/or modify it.
provide the source code of the modified version running there to the
users of that server. Therefore, public use of a modified version, on
a publicly accessible server, gives the public access to the source
code of the modified version.
An older license, called the Affero General Public License and For the developers' and authors' protection, the GPL clearly explains
published by Affero, was designed to accomplish similar goals. This is that there is no warranty for this free software. For both users' and
a different license, not a version of the Affero GPL, but Affero has authors' sake, the GPL requires that modified versions be marked as
released a new version of the Affero GPL which permits relicensing under changed, so that their problems will not be attributed erroneously to
this license. authors of previous versions.
Some devices are designed to deny users access to install or run
modified versions of the software inside them, although the manufacturer
can do so. This is fundamentally incompatible with the aim of
protecting users' freedom to change the software. The systematic
pattern of such abuse occurs in the area of products for individuals to
use, which is precisely where it is most unacceptable. Therefore, we
have designed this version of the GPL to prohibit the practice for those
products. If such problems arise substantially in other domains, we
stand ready to extend this provision to those domains in future versions
of the GPL, as needed to protect the freedom of users.
Finally, every program is threatened constantly by software patents.
States should not allow patents to restrict development and use of
software on general-purpose computers, but in those that do, we wish to
avoid the special danger that patents applied to a free program could
make it effectively proprietary. To prevent this, the GPL assures that
patents cannot be used to render the program non-free.
The precise terms and conditions for copying, distribution and The precise terms and conditions for copying, distribution and
modification follow. modification follow.
@@ -60,7 +72,7 @@ modification follow.
0. Definitions. 0. Definitions.
"This License" refers to version 3 of the GNU Affero General Public License. "This License" refers to version 3 of the GNU General Public License.
"Copyright" also means copyright-like laws that apply to other kinds of "Copyright" also means copyright-like laws that apply to other kinds of
works, such as semiconductor masks. works, such as semiconductor masks.
@@ -537,45 +549,35 @@ to collect a royalty for further conveying from those to whom you convey
the Program, the only way you could satisfy both those terms and this the Program, the only way you could satisfy both those terms and this
License would be to refrain entirely from conveying the Program. License would be to refrain entirely from conveying the Program.
13. Remote Network Interaction; Use with the GNU General Public License. 13. Use with the GNU Affero General Public License.
Notwithstanding any other provision of this License, if you modify the
Program, your modified version must prominently offer all users
interacting with it remotely through a computer network (if your version
supports such interaction) an opportunity to receive the Corresponding
Source of your version by providing access to the Corresponding Source
from a network server at no charge, through some standard or customary
means of facilitating copying of software. This Corresponding Source
shall include the Corresponding Source for any work covered by version 3
of the GNU General Public License that is incorporated pursuant to the
following paragraph.
Notwithstanding any other provision of this License, you have Notwithstanding any other provision of this License, you have
permission to link or combine any covered work with a work licensed permission to link or combine any covered work with a work licensed
under version 3 of the GNU General Public License into a single under version 3 of the GNU Affero General Public License into a single
combined work, and to convey the resulting work. The terms of this combined work, and to convey the resulting work. The terms of this
License will continue to apply to the part which is the covered work, License will continue to apply to the part which is the covered work,
but the work with which it is combined will remain governed by version but the special requirements of the GNU Affero General Public License,
3 of the GNU General Public License. section 13, concerning interaction through a network will apply to the
combination as such.
14. Revised Versions of this License. 14. Revised Versions of this License.
The Free Software Foundation may publish revised and/or new versions of The Free Software Foundation may publish revised and/or new versions of
the GNU Affero General Public License from time to time. Such new versions the GNU General Public License from time to time. Such new versions will
will be similar in spirit to the present version, but may differ in detail to be similar in spirit to the present version, but may differ in detail to
address new problems or concerns. address new problems or concerns.
Each version is given a distinguishing version number. If the Each version is given a distinguishing version number. If the
Program specifies that a certain numbered version of the GNU Affero General Program specifies that a certain numbered version of the GNU General
Public License "or any later version" applies to it, you have the Public License "or any later version" applies to it, you have the
option of following the terms and conditions either of that numbered option of following the terms and conditions either of that numbered
version or of any later version published by the Free Software version or of any later version published by the Free Software
Foundation. If the Program does not specify a version number of the Foundation. If the Program does not specify a version number of the
GNU Affero General Public License, you may choose any version ever published GNU General Public License, you may choose any version ever published
by the Free Software Foundation. by the Free Software Foundation.
If the Program specifies that a proxy can decide which future If the Program specifies that a proxy can decide which future
versions of the GNU Affero General Public License can be used, that proxy's versions of the GNU General Public License can be used, that proxy's
public statement of acceptance of a version permanently authorizes you public statement of acceptance of a version permanently authorizes you
to choose that version for the Program. to choose that version for the Program.
@@ -633,29 +635,40 @@ the "copyright" line and a pointer to where the full notice is found.
Copyright (C) <year> <name of author> Copyright (C) <year> <name of author>
This program is free software: you can redistribute it and/or modify This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published it under the terms of the GNU General Public License as published by
by the Free Software Foundation, either version 3 of the License, or the Free Software Foundation, either version 3 of the License, or
(at your option) any later version. (at your option) any later version.
This program is distributed in the hope that it will be useful, This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details. GNU General Public License for more details.
You should have received a copy of the GNU Affero General Public License You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. along with this program. If not, see <https://www.gnu.org/licenses/>.
Also add information on how to contact you by electronic and paper mail. Also add information on how to contact you by electronic and paper mail.
If your software can interact with users remotely through a computer If the program does terminal interaction, make it output a short
network, you should also make sure that it provides a way for users to notice like this when it starts in an interactive mode:
get its source. For example, if your program is a web application, its
interface could display a "Source" link that leads users to an archive <program> Copyright (C) <year> <name of author>
of the code. There are many ways you could offer source, and different This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
solutions will be better for different programs; see section 13 for the This is free software, and you are welcome to redistribute it
specific requirements. under certain conditions; type `show c' for details.
The hypothetical commands `show w' and `show c' should show the appropriate
parts of the General Public License. Of course, your program's commands
might be different; for a GUI interface, you would use an "about box".
You should also get your employer (if you work as a programmer) or school, You should also get your employer (if you work as a programmer) or school,
if any, to sign a "copyright disclaimer" for the program, if necessary. if any, to sign a "copyright disclaimer" for the program, if necessary.
For more information on this, and how to apply and follow the GNU AGPL, see For more information on this, and how to apply and follow the GNU GPL, see
<http://www.gnu.org/licenses/>. <https://www.gnu.org/licenses/>.
The GNU General Public License does not permit incorporating your program
into proprietary programs. If your program is a subroutine library, you
may consider it more useful to permit linking proprietary applications with
the library. If this is what you want to do, use the GNU Lesser General
Public License instead of this License. But first, please read
<https://www.gnu.org/licenses/why-not-lgpl.html>.

47
Makefile Normal file
View File

@@ -0,0 +1,47 @@
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
PKG_CONFIG := pkg-config
DESTDIR :=
PREFIX := /usr
SBINDIR := $(PREFIX)/sbin
SYSCONFDIR := /etc
RUNSTATEDIR := /var/run
SYSTEMDSYSTEMUNITDIR := $(shell ${PKG_CONFIG} --variable=systemdsystemunitdir systemd)
SMARTDNS_SYSTEMD = systemd/smartdns.service
.PHONY: all clean install SMARTDNS_BIN
all: SMARTDNS_BIN
SMARTDNS_BIN: $(SMARTDNS_SYSTEMD)
$(MAKE) $(MFLAGS) -C src all
$(SMARTDNS_SYSTEMD): systemd/smartdns.service.in
cp $< $@
sed -i 's|@SBINDIR@|$(SBINDIR)|' $@
sed -i 's|@SYSCONFDIR@|$(SYSCONFDIR)|' $@
sed -i 's|@RUNSTATEDIR@|$(RUNSTATEDIR)|' $@
clean:
$(MAKE) $(MFLAGS) -C src clean
$(RM) $(SMARTDNS_SYSTEMD)
install: SMARTDNS_BIN
install -v -m 0640 -D -t $(DESTDIR)$(SYSCONFDIR)/default etc/default/smartdns
install -v -m 0755 -D -t $(DESTDIR)$(SYSCONFDIR)/init.d etc/init.d/smartdns
install -v -m 0640 -D -t $(DESTDIR)$(SYSCONFDIR)/smartdns etc/smartdns/smartdns.conf
install -v -m 0755 -D -t $(DESTDIR)$(SBINDIR) src/smartdns
install -v -m 0644 -D -t $(DESTDIR)$(SYSTEMDSYSTEMUNITDIR) systemd/smartdns.service

1371
ReadMe.md

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

Binary file not shown.

Before

Width:  |  Height:  |  Size: 68 KiB

After

Width:  |  Height:  |  Size: 52 KiB

View File

@@ -1,4 +1,19 @@
#!/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
### BEGIN INIT INFO ### BEGIN INIT INFO
# Provides: smartdns # Provides: smartdns
@@ -19,14 +34,14 @@ test -x $SMARTDNS || exit 5
case $1 in case $1 in
start) start)
$SMARTDNS $SMART_DNS_OPTS $SMARTDNS "$SMART_DNS_OPTS"
while true; do while true; do
if [ -e "$PIDFILE" ]; then if [ -e "$PIDFILE" ]; then
break; break;
fi fi
sleep .5 sleep .5
done done
PID="`cat $PIDFILE 2>/dev/null`" PID="$(cat $PIDFILE 2>/dev/null)"
if [ -z "$PID" ]; then if [ -z "$PID" ]; then
echo "start smartdns server failed." echo "start smartdns server failed."
exit 1 exit 1
@@ -42,25 +57,25 @@ case $1 in
echo "smartdns server is stopped." echo "smartdns server is stopped."
exit 0 exit 0
fi fi
PID="`cat $PIDFILE 2>/dev/null`" PID="$(cat $PIDFILE 2>/dev/null)"
if [ ! -e "/proc/$PID" ] || [ -z "$PID" ]; then if [ ! -e "/proc/$PID" ] || [ -z "$PID" ]; then
echo "smartdns server is stopped" echo "smartdns server is stopped"
exit 0 exit 0
fi fi
kill -TERM $PID kill -TERM "$PID"
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
echo "Stop smartdns server failed." echo "Stop smartdns server failed."
exit 1; exit 1;
fi fi
rm -f $PIDFILE rm -f "$PIDFILE"
echo "Stop smartdns server success." echo "Stop smartdns server success."
;; ;;
restart) restart)
$0 stop && sleep 1 && $0 start "$0" stop && sleep 1 && "$0" start
;; ;;
status) status)
PID="`cat $PIDFILE 2>/dev/null`" PID="$(cat "$PIDFILE" 2>/dev/null)"
if [ ! -e "/proc/$PID" ] || [ -z "$PID" ]; then if [ ! -e "/proc/$PID" ] || [ -z "$PID" ]; then
echo "smartdns server is not running." echo "smartdns server is not running."
exit 1 exit 1

View File

@@ -1,4 +1,4 @@
# dns server name, defaut is host name # dns server name, default is host name
# server-name, # server-name,
# example: # example:
# server-name smartdns # server-name smartdns
@@ -22,6 +22,7 @@
# -no-cache: skip cache. # -no-cache: skip cache.
# -no-rule-soa: Skip address SOA(#) rules. # -no-rule-soa: Skip address SOA(#) rules.
# -no-dualstack-selection: Disable dualstack ip selection. # -no-dualstack-selection: Disable dualstack ip selection.
# -force-aaaa-soa: force AAAA query return SOA.
# example: # example:
# IPV4: # IPV4:
# bind :53 # bind :53
@@ -37,12 +38,30 @@ bind [::]:53
# dns cache size # dns cache size
# cache-size [number] # cache-size [number]
# 0: for no cache # 0: for no cache
cache-size 512 cache-size 4096
# enable persist cache when restart
# cache-persist yes
# cache persist file
# cache-file /tmp/smartdns.cache
# prefetch domain # prefetch domain
# prefetch-domain [yes|no] # prefetch-domain [yes|no]
# prefetch-domain yes # prefetch-domain yes
# cache serve expired
# serve-expired [yes|no]
# serve-expired yes
# cache serve expired TTL
# serve-expired-ttl [num]
# serve-expired-ttl 0
# reply TTL value to use when replying with expired data
# serve-expired-reply-ttl [num]
# serve-expired-reply-ttl 30
# List of hosts that supply bogus NX domain results # List of hosts that supply bogus NX domain results
# bogus-nxdomain [ip/subnet] # bogus-nxdomain [ip/subnet]
@@ -97,12 +116,20 @@ log-level info
# dns audit # dns audit
# audit-enable [yes|no]: enable or disable audit. # audit-enable [yes|no]: enable or disable audit.
# audit-enable yes # audit-enable yes
# audit-SOA [yes|no]: enable or disalbe log soa result. # audit-SOA [yes|no]: enable or disable log soa result.
# audit-size size of each audit file, support k,m,g # audit-size size of each audit file, support k,m,g
# audit-file /var/log/smartdns-audit.log # audit-file /var/log/smartdns-audit.log
# audit-size 128k # audit-size 128k
# audit-num 2 # audit-num 2
# certificate file
# ca-file [file]
# ca-file /etc/ssl/certs/ca-certificates.crt
# certificate path
# ca-path [path]
# ca-path /etc/ss/certs
# remote udp dns server list # remote udp dns server list
# server [IP]:[PORT] [-blacklist-ip] [-whitelist-ip] [-check-edns] [-group [group] ...] [-exclude-default-group] # server [IP]:[PORT] [-blacklist-ip] [-whitelist-ip] [-check-edns] [-group [group] ...] [-exclude-default-group]
# default port is 53 # default port is 53
@@ -121,8 +148,9 @@ log-level info
# remote tls dns server list # remote tls dns server list
# server-tls [IP]:[PORT] [-blacklist-ip] [-whitelist-ip] [-spki-pin [sha256-pin]] [-group [group] ...] [-exclude-default-group] # server-tls [IP]:[PORT] [-blacklist-ip] [-whitelist-ip] [-spki-pin [sha256-pin]] [-group [group] ...] [-exclude-default-group]
# -spki-pin: TLS spki pin to verify. # -spki-pin: TLS spki pin to verify.
# -tls-host-check: cert hostname to verify. # -tls-host-verify: cert hostname to verify.
# -hostname: TLS sni hostname. # -host-name: TLS sni hostname.
# -no-check-certificate: no check certificate.
# Get SPKI with this command: # Get SPKI with this command:
# echo | openssl s_client -connect '[ip]:853' | openssl x509 -pubkey -noout | openssl pkey -pubin -outform der | openssl dgst -sha256 -binary | openssl enc -base64 # echo | openssl s_client -connect '[ip]:853' | openssl x509 -pubkey -noout | openssl pkey -pubin -outform der | openssl dgst -sha256 -binary | openssl enc -base64
# default port is 853 # default port is 853
@@ -132,9 +160,10 @@ log-level info
# remote https dns server list # remote https dns server list
# server-https https://[host]:[port]/path [-blacklist-ip] [-whitelist-ip] [-spki-pin [sha256-pin]] [-group [group] ...] [-exclude-default-group] # server-https https://[host]:[port]/path [-blacklist-ip] [-whitelist-ip] [-spki-pin [sha256-pin]] [-group [group] ...] [-exclude-default-group]
# -spki-pin: TLS spki pin to verify. # -spki-pin: TLS spki pin to verify.
# -tls-host-check: cert hostname to verify. # -tls-host-verify: cert hostname to verify.
# -hostname: TLS sni hostname. # -host-name: TLS sni hostname.
# -http-host: http host. # -http-host: http host.
# -no-check-certificate: no check certificate.
# default port is 443 # default port is 443
# server-https https://cloudflare-dns.com/dns-query # server-https https://cloudflare-dns.com/dns-query
@@ -156,3 +185,12 @@ log-level info
# ipset /domain/[ipset|-] # ipset /domain/[ipset|-]
# ipset /www.example.com/block, set ipset with ipset name of block # ipset /www.example.com/block, set ipset with ipset name of block
# ipset /www.example.com/-, ignore this domain # ipset /www.example.com/-, ignore this domain
# set domain rules
# domain-rules /domain/ [-speed-check-mode [...]]
# rules:
# -speed-check-mode [mode]: speed check mode
# speed-check-mode [ping|tcp:port|none|,]
# -address [address|-]: same as address option
# -nameserver [group|-]: same as nameserver option
# -ipset [ipset|-]: same as ipset option

182
package/build-pkg.sh Normal file
View File

@@ -0,0 +1,182 @@
#!/bin/sh
# Copyright (C) 2018-2019 Nick Peng (pymumu@gmail.com)
CURR_DIR=$(cd $(dirname $0);pwd)
VER="`date +"1.%Y.%m.%d-%H%M"`"
CODE_DIR="$CURR_DIR/.."
IS_BUILD_SMARTDNS=1
OUTPUTDIR=$CURR_DIR
export CC
export STRIP
showhelp()
{
echo "Usage: $0 [OPTION]"
echo "Options:"
echo " --platform [luci|luci-compat|debian|openwrt|optware|linux] build for platform. "
echo " --arch [all|armhf|arm64|x86-64|...] build for architecture, e.g. "
echo " --cross-tool [cross-tool] cross compiler, e.g. mips-openwrt-linux-"
echo ""
echo "Advance Options:"
echo " --static static link smartdns"
echo " --only-package only package, not build source"
echo " --filearch [arch] output file arch, default: equal --arch"
echo " --outputdir [dir] output package to specific directory"
echo " "
echo "Example:"
echo " build luci:"
echo " $0 --platform luci"
echo " build luci:"
echo " $0 --platform luci-compat"
echo " build debian:"
echo " $0 --platform debian --arch x86-64"
echo " build raspbian pi:"
echo " $0 --platform debian --arch armhf"
echo " build optware mips:"
echo " $0 --platform optware --arch mipsbig"
echo " build openwrt mips:"
echo " $0 --platform openwrt --arch mips_24kc"
echo " build generic linux:"
echo " $0 --platform linux --arch x86-64"
}
build_smartdns()
{
if [ "$PLATFORM" != "luci" ]; then
make -C $CODE_DIR clean $MAKE_ARGS
make -C $CODE_DIR all -j8 VER=$VER $MAKE_ARGS
if [ $? -ne 0 ]; then
echo "make smartdns failed"
exit 1
fi
fi
$STRIP -d $CODE_DIR/src/smartdns >/dev/null 2>&1
return 0
}
build()
{
echo "build package for $PLATFORM"
if [ $IS_BUILD_SMARTDNS -eq 1 ]; then
build_smartdns
if [ $? -ne 0 ]; then
return 1
fi
fi
chmod +x $CODE_DIR/package/$PLATFORM/make.sh
$CODE_DIR/package/$PLATFORM/make.sh -o $CURR_DIR --arch $ARCH --ver $VER --filearch $FILEARCH -o $OUTPUTDIR
if [ $? -ne 0 ]; then
echo "build package for $PLATFORM failed"
return 1
fi
echo "build package for $PLATFORM success."
return 0
}
main()
{
OPTS=`getopt -o o:h --long arch:,filearch:,ver:,platform:,cross-tool:,static,only-package,outputdir: \
-n "" -- "$@"`
if [ "$#" -le "1" ]; then
showhelp
exit 1
fi
if [ $? != 0 ] ; then echo "Terminating..." >&2 ; exit 1 ; fi
# Note the quotes around `$TEMP': they are essential!
eval set -- "$OPTS"
while true; do
case "$1" in
--arch)
ARCH="$2"
shift 2;;
--filearch)
FILEARCH="$2"
shift 2;;
--platform)
PLATFORM="$2"
shift 2;;
--cross-tool)
CROSS_TOOL="$2"
shift 2;;
--static)
export STATIC="yes"
shift 1;;
--only-package)
IS_BUILD_SMARTDNS=0
shift 1;;
--outputdir)
OUTPUTDIR="$2"
shift 2;;
--ver)
VER="$2"
shift 2;;
-h | --help )
showhelp
return 0
shift ;;
-- ) shift; break ;;
* ) break ;;
esac
done
if [ -z "$PLATFORM" ]; then
echo "please input platform"
echo "run $0 -h for help."
return 1
fi
if [ "$PLATFORM" = "luci" ]; then
ARCH="all"
fi
if [ -z "$ARCH" ]; then
echo "please input arch."
echo "run $0 -h for help."
return 1
fi
if [ -z "$FILEARCH" ]; then
FILEARCH="$ARCH"
fi
if [ -z "$OUTPUTDIR" ]; then
OUTPUTDIR=$CURR_DIR
fi
if [ ! -z "$CROSS_TOOL" ]; then
CC="${CROSS_TOOL}gcc"
STRIP="${CROSS_TOOL}strip"
fi
if [ -z "$CC" ]; then
CC="gcc"
fi
if [ -z "$STRIP" ]; then
if [ ! -z "`echo $CC | grep '\-gcc'`" ]; then
STRIP="`echo "$CC" | sed 's/-gcc\$/-strip/g'`"
else
STRIP="strip"
fi
fi
if [ ! -e "`which $CC`" ]; then
echo "Cannot find compiler $CC"
return 1
fi
build
}
main $@
exit $?

View File

@@ -3,5 +3,5 @@ Upstream-Name: smartdns
Source: http://github.com/pymumu/smartdns Source: http://github.com/pymumu/smartdns
Files: * Files: *
Copyright: 2018 Nick peng Copyright: 2018-2019 Nick peng
License: proprietary License: proprietary

View File

@@ -1,5 +1,19 @@
#/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
CURR_DIR=$(cd $(dirname $0);pwd) CURR_DIR=$(cd $(dirname $0);pwd)
VER="`date +"1.%Y.%m.%d-%H%M"`" VER="`date +"1.%Y.%m.%d-%H%M"`"
SMARTDNS_DIR=$CURR_DIR/../../ SMARTDNS_DIR=$CURR_DIR/../../
@@ -11,37 +25,41 @@ showhelp()
echo "Options:" echo "Options:"
echo " -o output directory." echo " -o output directory."
echo " --arch archtecture." echo " --arch archtecture."
echo " --ver version." echo " --ver version."
echo " -h show this message." echo " -h show this message."
} }
build() build()
{ {
ROOT=/tmp/smartdns-deiban ROOT=/tmp/smartdns-deiban
rm -fr $ROOT rm -fr $ROOT
mkdir -p $ROOT mkdir -p $ROOT
cd $ROOT/ cd $ROOT/
cp $CURR_DIR/DEBIAN $ROOT/ -af cp $CURR_DIR/DEBIAN $ROOT/ -af
CONTROL=$ROOT/DEBIAN/control CONTROL=$ROOT/DEBIAN/control
mkdir $ROOT/usr/sbin -p mkdir $ROOT/usr/sbin -p
mkdir $ROOT/etc/smartdns/ -p mkdir $ROOT/etc/smartdns/ -p
mkdir $ROOT/etc/default/ -p mkdir $ROOT/etc/default/ -p
mkdir $ROOT/lib/systemd/system/ -p mkdir $ROOT/lib/systemd/system/ -p
sed -i "s/Version:.*/Version: $VER/" $ROOT/DEBIAN/control sed -i "s/Version:.*/Version: $VER/" $ROOT/DEBIAN/control
sed -i "s/Architecture:.*/Architecture: $ARCH/" $ROOT/DEBIAN/control sed -i "s/Architecture:.*/Architecture: $ARCH/" $ROOT/DEBIAN/control
chmod 0755 $ROOT/DEBIAN/prerm chmod 0755 $ROOT/DEBIAN/prerm
cp $SMARTDNS_DIR/etc/smartdns/smartdns.conf $ROOT/etc/smartdns/ cp $SMARTDNS_DIR/etc/smartdns/smartdns.conf $ROOT/etc/smartdns/
cp $SMARTDNS_DIR/etc/default/smartdns $ROOT/etc/default/ cp $SMARTDNS_DIR/etc/default/smartdns $ROOT/etc/default/
cp $SMARTDNS_DIR/systemd/smartdns.service $ROOT/lib/systemd/system/ cp $SMARTDNS_DIR/systemd/smartdns.service $ROOT/lib/systemd/system/
cp $SMARTDNS_DIR/src/smartdns $ROOT/usr/sbin cp $SMARTDNS_DIR/src/smartdns $ROOT/usr/sbin
chmod +x $ROOT/usr/sbin/smartdns if [ $? -ne 0 ]; then
echo "copy smartdns file failed."
return 1
fi
chmod +x $ROOT/usr/sbin/smartdns
dpkg -b $ROOT $OUTPUTDIR/smartdns.$VER.$FILEARCH.deb dpkg -b $ROOT $OUTPUTDIR/smartdns.$VER.$FILEARCH.deb
rm -fr $ROOT/ rm -fr $ROOT/
} }
main() main()
@@ -59,38 +77,38 @@ main()
--arch) --arch)
ARCH="$2" ARCH="$2"
shift 2;; shift 2;;
--filearch) --filearch)
FILEARCH="$2" FILEARCH="$2"
shift 2;; shift 2;;
--ver) --ver)
VER="$2" VER="$2"
shift 2;; shift 2;;
-o ) -o )
OUTPUTDIR="$2" OUTPUTDIR="$2"
shift 2;; shift 2;;
-h | --help ) -h | --help )
showhelp showhelp
return 0 return 0
shift ;; shift ;;
-- ) shift; break ;; -- ) shift; break ;;
* ) break ;; * ) break ;;
esac esac
done done
if [ -z "$ARCH" ]; then if [ -z "$ARCH" ]; then
echo "please input arch." echo "please input arch."
return 1; return 1;
fi fi
if [ -z "$FILEARCH" ]; then if [ -z "$FILEARCH" ]; then
FILEARCH=$ARCH FILEARCH=$ARCH
fi fi
if [ -z "$OUTPUTDIR" ]; then if [ -z "$OUTPUTDIR" ]; then
OUTPUTDIR=$CURR_DIR; OUTPUTDIR=$CURR_DIR;
fi fi
build build
} }
main $@ main $@

View File

@@ -1,7 +1,19 @@
#!/bin/sh #!/bin/sh
# #
# Copyright (C) 2018 Ruilin Peng (Nick) <pymumu@gmail.com> # Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
# #
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
INST_DIR=$(cd $(dirname $0);pwd) INST_DIR=$(cd $(dirname $0);pwd)
ISWSL=1 # 1 means not WSL, 0 means wsl ISWSL=1 # 1 means not WSL, 0 means wsl
@@ -53,7 +65,7 @@ clean_service()
get_systemd_path() get_systemd_path()
{ {
service="`systemctl --no-legend| grep .service | head -n 1 | awk '{print $1}'`" service="`systemctl --no-legend| grep '\.service' | head -n 1 | awk '{print $1}'`"
SERVICE_PATH="`systemctl show $service | grep FragmentPath | awk -F'=' '{print $2}'`" SERVICE_PATH="`systemctl show $service | grep FragmentPath | awk -F'=' '{print $2}'`"
dirname $SERVICE_PATH dirname $SERVICE_PATH
} }
@@ -65,14 +77,18 @@ install_files()
return 1 return 1
fi fi
install -v -m 0755 -t $PREFIX/usr/sbin src/smartdns install -v -m 0755 -t $PREFIX/usr/sbin usr/sbin/smartdns
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
return 1 return 1
fi fi
install -v -m 0640 -t $PREFIX$SMARTDNS_CONF_DIR etc/smartdns/smartdns.conf if [ -e "$PREFIX$SMARTDNS_CONF_DIR/smartdns.conf" ]; then
if [ $? -ne 0 ]; then cp etc/smartdns/smartdns.conf $PREFIX$SMARTDNS_CONF_DIR/smartdns.conf.pkg
return 1 else
install -v -m 0640 -t $PREFIX$SMARTDNS_CONF_DIR etc/smartdns/smartdns.conf
if [ $? -ne 0 ]; then
return 1
fi
fi fi
install -v -m 0640 -t $PREFIX/etc/default etc/default/smartdns install -v -m 0640 -t $PREFIX/etc/default etc/default/smartdns
@@ -82,7 +98,9 @@ install_files()
install -v -m 0755 -t $SMARTDNS_INIT_DIR etc/init.d/smartdns install -v -m 0755 -t $SMARTDNS_INIT_DIR etc/init.d/smartdns
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
return 1 if [ $ISSYSTEMD -ne 0 ]; then
return 1
fi
fi fi
if [ $ISSYSTEMD -eq 0 ]; then if [ $ISSYSTEMD -eq 0 ]; then
@@ -104,7 +122,6 @@ uninstall_smartdns()
if [ -z "$PREFIX" ]; then if [ -z "$PREFIX" ]; then
stop_service 2>/dev/null stop_service 2>/dev/null
fi fi
rm -f $PREFIX$SMARTDNS_CONF_DIR/smartdns.conf
rmdir $PREFIX$SMARTDNS_CONF_DIR 2>/dev/null rmdir $PREFIX$SMARTDNS_CONF_DIR 2>/dev/null
rm -f $PREFIX/usr/sbin/smartdns rm -f $PREFIX/usr/sbin/smartdns
rm -f $PREFIX/etc/default/smartdns rm -f $PREFIX/etc/default/smartdns

103
package/linux/make.sh Normal file
View File

@@ -0,0 +1,103 @@
#!/bin/sh
CURR_DIR=$(cd $(dirname $0);pwd)
VER="`date +"1.%Y.%m.%d-%H%M"`"
SMARTDNS_DIR=$CURR_DIR/../../
SMARTDNS_BIN=$SMARTDNS_DIR/src/smartdns
showhelp()
{
echo "Usage: make [OPTION]"
echo "Options:"
echo " -o output directory."
echo " --arch archtecture."
echo " --ver version."
echo " -h show this message."
}
build()
{
PKG_ROOT=/tmp/smartdns-linux
rm -fr $PKG_ROOT
mkdir -p $PKG_ROOT/smartdns
cd $PKG_ROOT/
# Generic x86_64
mkdir $PKG_ROOT/smartdns/usr/sbin -p
mkdir $PKG_ROOT/smartdns/package -p
mkdir $PKG_ROOT/smartdns/systemd -p
cd $SMARTDNS_DIR
cp package/windows $PKG_ROOT/smartdns/package/ -a
cp etc *.md LICENSE package/linux/install $PKG_ROOT/smartdns/ -a
cp systemd/smartdns.service $PKG_ROOT/smartdns/systemd
cp src/smartdns $PKG_ROOT/smartdns/usr/sbin -a
chmod +x $PKG_ROOT/smartdns/install
if [ $? -ne 0 ]; then
echo "copy smartdns file failed"
rm -fr $PKG_ROOT
exit 1
fi
cd $PKG_ROOT
tar zcf $OUTPUTDIR/smartdns.$VER.$FILEARCH.tar.gz smartdns
if [ $? -ne 0 ]; then
echo "create package failed"
rm -fr $PKG_ROOT
exit 1
fi
cd $CURR_DIR
rm -fr $PKG_ROOT
}
main()
{
OPTS=`getopt -o o:h --long arch:,ver:,filearch: \
-n "" -- "$@"`
if [ $? != 0 ] ; then echo "Terminating..." >&2 ; exit 1 ; fi
# Note the quotes around `$TEMP': they are essential!
eval set -- "$OPTS"
while true; do
case "$1" in
--arch)
ARCH="$2"
shift 2;;
--filearch)
FILEARCH="$2"
shift 2;;
--ver)
VER="$2"
shift 2;;
-o )
OUTPUTDIR="$2"
shift 2;;
-h | --help )
showhelp
return 0
shift ;;
-- ) shift; break ;;
* ) break ;;
esac
done
if [ -z "$ARCH" ]; then
echo "please input arch."
return 1;
fi
if [ -z "$FILEARCH" ]; then
FILEARCH=$ARCH
fi
if [ -z "$OUTPUTDIR" ]; then
OUTPUTDIR=$CURR_DIR;
fi
build
}
main $@
exit $?

View File

@@ -0,0 +1,7 @@
Package: luci-app-smartdns
Version: git-18.201.27126-7bf0367-1
Depends: libc, smartdns
Source: feeds/luci/applications/luci-app-smartdns
Section: luci
Architecture: all
Description: A smartdns server

View File

@@ -0,0 +1,21 @@
#!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
[ "${IPKG_NO_SCRIPT}" = "1" ] && exit 0
[ -x ${IPKG_INSTROOT}/lib/functions.sh ] || exit 0
. ${IPKG_INSTROOT}/lib/functions.sh
default_postinst $0 $@

View File

@@ -0,0 +1,4 @@
[ -n "${IPKG_INSTROOT}" ] || {
(. /etc/uci-defaults/50_luci-smartdns) && rm -f /etc/uci-defaults/50_luci-smartdns
exit 0
}

View File

@@ -0,0 +1,20 @@
#!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
[ -x ${IPKG_INSTROOT}/lib/functions.sh ] || exit 0
. ${IPKG_INSTROOT}/lib/functions.sh
default_prerm $0 $@

View File

@@ -0,0 +1 @@
2.0

View File

@@ -0,0 +1,26 @@
#!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
uci -q batch <<-EOF >/dev/null
delete ucitrack.@smartdns[-1]
add ucitrack smartdns
set ucitrack.@smartdns[-1].init=smartdns
commit ucitrack
EOF
rm -f /tmp/luci-indexcache
exit 0

View File

@@ -1,4 +1,18 @@
-- Copyright 2018 Nick Peng (pymumu@gmail.com) --
-- Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
--
-- smartdns is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- smartdns is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
module("luci.controller.smartdns", package.seeall) module("luci.controller.smartdns", package.seeall)
local smartdns = require "luci.model.smartdns" local smartdns = require "luci.model.smartdns"

View File

@@ -0,0 +1,299 @@
msgid ""
msgstr "Content-Type: text/plain; charset=UTF-8\n"
msgid "SmartDNS"
msgstr "SmartDNS"
msgid "SmartDNS is a local high-performance DNS server"
msgstr "SmartDNS是一个本地高性能DNS服务器"
msgid "SmartDNS Server"
msgstr "SmartDNS 服务器"
msgid "SmartDNS is a local high-performance DNS server, supports finding fastest IP, supports ad filtering, and supports avoiding DNS poisoning."
msgstr "SmartDNS是一个本地高性能DNS服务器支持返回最快IP支持广告过滤。"
msgid "Custom Settings"
msgstr "自定义设置"
msgid "General Settings"
msgstr "基本设置"
msgid "Settings"
msgstr "设置"
msgid "Advanced Settings"
msgstr "高级设置"
msgid "Generate Coredump"
msgstr "生成coredump"
msgid "Generate Coredump file when smartdns crash, coredump file is located at /tmp/smartdns.xxx.core."
msgstr "当smartdns异常时生成coredump文件coredump文件在/tmp/smartdns.xxx.core."
msgid "Server Name"
msgstr "服务器名称"
msgid "Smartdns server name"
msgstr "SmartDNS的服务器名称默认为smartdns留空为主机名"
msgid "SmartDNS is a local dns server to find fastest ip."
msgstr "本地高性能服务器,优化网络访问性能。"
msgid "Enable or disable smartdns server"
msgstr "启用或禁用SmartDNS服务"
msgid "Local Port"
msgstr "本地端口"
msgid "Smartdns local server port"
msgstr "SmartDNS本地服务端口"
msgid "IPV4 53 Port Redirect Failure"
msgstr "IPV4 53端口重定向失败"
msgid "IPV6 53 Port Redirect Failure"
msgstr "IPV6 53端口重定向失败"
msgid "Dnsmasq Forwared To Smartdns Failure"
msgstr "重定向dnsmasq到smartdns失败"
msgid "TCP Server"
msgstr "TCP服务器"
msgid "Enable TCP DNS Server"
msgstr "启用TCP服务器"
msgid "IPV6 Server"
msgstr "IPV6服务器"
msgid "Enable IPV6 DNS Server"
msgstr "启用IPV6服务器"
msgid "Dual-stack IP Selection"
msgstr "双栈IP优选"
msgid "Enable IP selection between IPV4 and IPV6"
msgstr "启用或禁用IPV4IPV6间的IP优选策略。"
msgid "Domain prefetch"
msgstr "域名预加载"
msgid "Enable domain prefetch, accelerate domain response speed."
msgstr "启用域名预加载,加速域名响应速度。"
msgid "Serve expired"
msgstr "过期缓存服务"
msgid "Attempts to serve old responses from cache with a TTL of 0 in the response without waiting for the actual resolution to finish."
msgstr "查询性能优化有请求时尝试回应TTL为0的过期记录以避免查询等待。"
msgid "Redirect"
msgstr "重定向"
msgid "SmartDNS redirect mode"
msgstr "SmartDNS 重定向模式"
msgid "Run as dnsmasq upstream server"
msgstr "作为dnsmasq的上游服务器"
msgid "Redirect 53 port to SmartDNS"
msgstr "重定向53端口到SmartDNS"
msgid "Cache Size"
msgstr "缓存大小"
msgid "DNS domain result cache size"
msgstr "缓存DNS的结果缓存大小配置零则不缓存"
msgid "Domain TTL"
msgstr "域名TTL"
msgid "TTL for all domain result."
msgstr "设置所有域名的TTL值"
msgid "Domain TTL Min"
msgstr "域名TTL最小值"
msgid "Minimum TTL for all domain result."
msgstr "设置所有域名的TTL最小值"
msgid "Domain TTL Max"
msgstr "域名TTL最大值"
msgid "Maximum TTL for all domain result."
msgstr "设置所有域名的TTL最大值"
msgid "smartdns custom settings"
msgstr "smartdns 自定义设置,具体配置参数参考指导"
msgid "Second Server Settings"
msgstr "第二DNS服务器"
msgid "Enable or disable second DNS server."
msgstr "是否启用第二DNS服务器。"
msgid "Skip Speed Check"
msgstr "跳过测速"
msgid "Do not check speed."
msgstr "禁用测速。"
msgid "Server Group"
msgstr "服务器组"
msgid "Query DNS through specific dns server group, such as office, home."
msgstr "使用指定服务器组查询比如office, home。"
msgid "Skip Address Rules"
msgstr "跳过address规则"
msgid "Skip address rules."
msgstr "跳过address规则。"
msgid "Skip Nameserver Rule"
msgstr "跳过Nameserver规则"
msgid "Skip nameserver rules."
msgstr "跳过Nameserver规则。"
msgid "Skip Ipset Rule"
msgstr "跳过ipset规则"
msgid "Skip ipset rules."
msgstr "跳过ipset规则。"
msgid "Skip SOA Address Rule"
msgstr "跳过address SOA(#)规则"
msgid "Skip SOA address rules."
msgstr "跳过address SOA(#)规则。"
msgid "Skip Dualstack Selection"
msgstr "跳过双栈优选"
msgid "Skip Dualstack Selection."
msgstr "跳过双栈优选。"
msgid "Skip Cache"
msgstr "跳过cache"
msgid "Skip Cache."
msgstr "跳过cache。"
msgid "Upstream Servers"
msgstr "上游服务器"
msgid "Upstream Servers, support UDP, TCP protocol. Please configure multiple DNS servers, including multiple foreign DNS servers."
msgstr "上游DNS服务器列表支持UDPTCP协议请配置多个上游DNS服务器包括多个国内外服务器"
msgid "DNS Server Name"
msgstr "DNS服务器名称"
msgid "port"
msgstr "端口"
msgid "DNS Server port"
msgstr "DNS服务器端口"
msgid "DNS Server ip"
msgstr "DNS服务器IP"
msgid "type"
msgstr "类型"
msgid "DNS Server type"
msgstr "协议类型"
msgid "Domain Address"
msgstr "域名地址"
msgid "TLS Hostname Verify"
msgstr "校验TLS主机名"
msgid "Set TLS hostname to verify."
msgstr "设置校验TLS主机名。"
msgid "No check certificate"
msgstr "停用证书校验"
msgid "Do not check certificate."
msgstr "不校验证书的合法性。"
msgid "TLS SNI name"
msgstr "TLS SNI名称"
msgid "HTTP Host"
msgstr "HTTP主机"
msgid "Sets the server name indication for query."
msgstr "设置查询时使用的服务器SNI名称。"
msgid "Set the HTTP host used for the query. Use this parameter when the host of the URL address is an IP address."
msgstr "设置查询时使用的HTTP主机当URL地址的host是IP地址时使用此参数。"
msgid "Server Group"
msgstr "服务器组"
msgid "DNS Server group belongs to, used with nameserver, such as office, home."
msgstr "DNS服务器所属组 配合nameserver使用例如officehome。"
msgid "IP Blacklist Filtering"
msgstr "IP黑名单过滤"
msgid "Anti Answer Forgery"
msgstr "反回答伪造"
msgid "Anti answer forgery, if DNS does not work properly after enabling, please turn off this feature"
msgstr "反回答伪造如果启用后DNS工作不正常请关闭此功能。"
msgid "Filtering IP with blacklist"
msgstr "使用IP黑名单过滤"
msgid "TLS SPKI Pinning"
msgstr "TLS SPKI 指纹"
msgid "Used to verify the validity of the TLS server, The value is Base64 encoded SPKI fingerprint, leaving blank to indicate that the validity of TLS is not verified."
msgstr "用于校验TLS服务器的有效性数值为Base64编码的SPKI指纹, 留空表示不验证TLS的合法性"
msgid "Additional Server Args"
msgstr "额外的服务器参数"
msgid "Additional Args for upstream dns servers"
msgstr "额外的上游DNS服务器参数"
msgid "Upstream DNS Server Configuration"
msgstr "上游DNS服务器配置"
msgid "Set Specific domain ip address."
msgstr "指定特定域名的IP地址"
msgid "Specify an IP address to return for any host in the given domains, Queries in the domains are never forwarded and always replied to with the specified IP address which may be IPv4 or IPv6."
msgstr "配置特定域名返回特定的IP地址域名查询将不到上游服务器请求直接返回配置的IP地址可用于广告屏蔽。"
msgid "IP Blacklist"
msgstr "IP黑名单"
msgid "Set Specific ip blacklist."
msgstr "设置IP黑名单列表"
msgid "Configure IP blacklists that will be filtered from the results of specific DNS server."
msgstr "配置需要从指定域名服务器结果过滤的IP黑名单。"
msgid "Technical Support"
msgstr "技术支持"
msgid "If you like this software, please buy me a cup of coffee."
msgstr "如果本软件对你有帮助,请给作者加个蛋。"
msgid "SmartDNS official website"
msgstr "SmartDNS官方网站"
msgid "open website"
msgstr "打开网站"
msgid "Donate to smartdns"
msgstr "捐助smartdns项目"
msgid "Donate"
msgstr "捐助"

View File

@@ -1,4 +1,18 @@
-- Copyright 2018 Nick Peng (pymumu@gmail.com) --
-- Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
--
-- smartdns is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- smartdns is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
require ("nixio.fs") require ("nixio.fs")
require ("luci.http") require ("luci.http")
@@ -69,6 +83,15 @@ o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0" return Flag.cfgvalue(...) or "0"
end end
---- Domain Serve expired
o = s:taboption("settings", Flag, "serve_expired", translate("Serve expired"),
translate("Attempts to serve old responses from cache with a TTL of 0 in the response without waiting for the actual resolution to finish."))
o.rmempty = false
o.default = o.disabled
o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0"
end
---- Redirect ---- Redirect
o = s:taboption("settings", ListValue, "redirect", translate("Redirect"), translate("SmartDNS redirect mode")) o = s:taboption("settings", ListValue, "redirect", translate("Redirect"), translate("SmartDNS redirect mode"))
o.placeholder = "none" o.placeholder = "none"
@@ -105,8 +128,8 @@ o.rempty = false
---- Port ---- Port
o = s:taboption("seconddns", Value, "seconddns_port", translate("Local Port"), translate("Smartdns local server port")) o = s:taboption("seconddns", Value, "seconddns_port", translate("Local Port"), translate("Smartdns local server port"))
o.placeholder = 7053 o.placeholder = 6553
o.default = 7053 o.default = 6553
o.datatype = "port" o.datatype = "port"
o.rempty = false o.rempty = false
@@ -118,13 +141,6 @@ o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "1" return Flag.cfgvalue(...) or "1"
end end
o = s:taboption("seconddns", Flag, "seconddns_no_speed_check", translate("Skip Speed Check"), translate("Do not check speed."))
o.rmempty = false
o.default = o.disabled
o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0"
end
---- dns server group ---- dns server group
o = s:taboption("seconddns", Value, "seconddns_server_group", translate("Server Group"), translate("Query DNS through specific dns server group, such as office, home.")) o = s:taboption("seconddns", Value, "seconddns_server_group", translate("Server Group"), translate("Query DNS through specific dns server group, such as office, home."))
o.rmempty = true o.rmempty = true
@@ -132,6 +148,13 @@ o.placeholder = "default"
o.datatype = "hostname" o.datatype = "hostname"
o.rempty = true o.rempty = true
o = s:taboption("seconddns", Flag, "seconddns_no_speed_check", translate("Skip Speed Check"), translate("Do not check speed."))
o.rmempty = false
o.default = o.disabled
o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0"
end
---- skip address rules ---- skip address rules
o = s:taboption("seconddns", Flag, "seconddns_no_rule_addr", translate("Skip Address Rules"), translate("Skip address rules.")) o = s:taboption("seconddns", Flag, "seconddns_no_rule_addr", translate("Skip Address Rules"), translate("Skip address rules."))
o.rmempty = false o.rmempty = false
@@ -164,7 +187,7 @@ o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0" return Flag.cfgvalue(...) or "0"
end end
o = s:taboption("seconddns", Flag, "seconddns_no_dualstack_selection", translate("Skip Dualstack Selection"), translate("Skip Sualstack Selection.")) o = s:taboption("seconddns", Flag, "seconddns_no_dualstack_selection", translate("Skip Dualstack Selection"), translate("Skip Dualstack Selection."))
o.rmempty = false o.rmempty = false
o.default = o.disabled o.default = o.disabled
o.cfgvalue = function(...) o.cfgvalue = function(...)
@@ -179,6 +202,14 @@ o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0" return Flag.cfgvalue(...) or "0"
end end
---- Force AAAA SOA
o = s:taboption("seconddns", Flag, "force_aaaa_soa", translate("Force AAAA SOA"), translate("Force AAAA SOA."))
o.rmempty = false
o.default = o.disabled
o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0"
end
----- custom settings ----- custom settings
custom = s:taboption("custom", Value, "Custom Settings", custom = s:taboption("custom", Value, "Custom Settings",
translate(""), translate(""),
@@ -246,13 +277,14 @@ o:value("https", translate("https"))
o.default = "udp" o.default = "udp"
o.rempty = false o.rempty = false
-- Doman addresss s = m:section(TypedSection, "smartdns", translate("Advanced Settings"), translate("Advanced Settings"));
s = m:section(TypedSection, "smartdns", translate("Domain Address"), s.anonymous = true;
translate("Set Specific domain ip address."))
s.anonymous = true
---- address s:tab("domain-address", translate("Domain Address"), translate("Set Specific domain ip address."));
addr = s:option(Value, "address", s:tab("blackip-list", translate("IP Blacklist"), translate("Set Specific ip blacklist."));
-- Doman addresss
addr = s:taboption("domain-address", Value, "address",
translate(""), translate(""),
translate("Specify an IP address to return for any host in the given domains, Queries in the domains are never forwarded and always replied to with the specified IP address which may be IPv4 or IPv6.")) translate("Specify an IP address to return for any host in the given domains, Queries in the domains are never forwarded and always replied to with the specified IP address which may be IPv4 or IPv6."))
@@ -269,12 +301,7 @@ function addr.write(self, section, value)
end end
-- IP Blacklist -- IP Blacklist
s = m:section(TypedSection, "smartdns", translate("IP Blacklist"), addr = s:taboption("blackip-list", Value, "blacklist_ip",
translate("Set Specific ip blacklist."))
s.anonymous = true
---- blacklist
addr = s:option(Value, "blacklist_ip",
translate(""), translate(""),
translate("Configure IP blacklists that will be filtered from the results of specific DNS server.")) translate("Configure IP blacklists that will be filtered from the results of specific DNS server."))
@@ -290,7 +317,7 @@ function addr.write(self, section, value)
nixio.fs.writefile("/etc/smartdns/blacklist-ip.conf", value) nixio.fs.writefile("/etc/smartdns/blacklist-ip.conf", value)
end end
-- Doman addresss -- Technical Support
s = m:section(TypedSection, "smartdns", translate("Technical Support"), s = m:section(TypedSection, "smartdns", translate("Technical Support"),
translate("If you like this software, please buy me a cup of coffee.")) translate("If you like this software, please buy me a cup of coffee."))
s.anonymous = true s.anonymous = true

View File

@@ -1,3 +1,19 @@
--
-- Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
--
-- smartdns is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- smartdns is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
local sid = arg[1] local sid = arg[1]
m = Map("smartdns", "%s - %s" %{translate("SmartDNS Server"), translate("Upstream DNS Server Configuration")}) m = Map("smartdns", "%s - %s" %{translate("SmartDNS Server"), translate("Upstream DNS Server Configuration")})
@@ -39,6 +55,21 @@ o:value("https", translate("https"))
o.default = "udp" o.default = "udp"
o.rempty = false o.rempty = false
---- server group
o = s:option(Value, "server_group", translate("Server Group"), translate("DNS Server group belongs to, used with nameserver, such as office, home."))
o.rmempty = true
o.placeholder = "default"
o.datatype = "hostname"
o.rempty = true
---- blacklist_ip
o = s:option(Flag, "blacklist_ip", translate("IP Blacklist Filtering"), translate("Filtering IP with blacklist"))
o.rmempty = false
o.default = o.disabled
o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0"
end
---- TLS host verify ---- TLS host verify
o = s:option(Value, "tls_host_verify", translate("TLS Hostname Verify"), translate("Set TLS hostname to verify.")) o = s:option(Value, "tls_host_verify", translate("TLS Hostname Verify"), translate("Set TLS hostname to verify."))
o.default = "" o.default = ""
@@ -47,6 +78,16 @@ o.rempty = true
o:depends("type", "tls") o:depends("type", "tls")
o:depends("type", "https") o:depends("type", "https")
---- certificate verify
o = s:option(Flag, "no_check_certificate", translate("No check certificate"), translate("Do not check certificate."))
o.rmempty = false
o.default = o.disabled
o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0"
end
o:depends("type", "tls")
o:depends("type", "https")
---- SNI host name ---- SNI host name
o = s:option(Value, "host_name", translate("TLS SNI name"), translate("Sets the server name indication for query.")) o = s:option(Value, "host_name", translate("TLS SNI name"), translate("Sets the server name indication for query."))
o.default = "" o.default = ""
@@ -62,21 +103,6 @@ o.datatype = "hostname"
o.rempty = true o.rempty = true
o:depends("type", "https") o:depends("type", "https")
---- server group
o = s:option(Value, "server_group", translate("Server Group"), translate("DNS Server group belongs to, used with nameserver, such as office, home."))
o.rmempty = true
o.placeholder = "default"
o.datatype = "hostname"
o.rempty = true
---- blacklist_ip
o = s:option(Flag, "blacklist_ip", translate("IP Blacklist Filtering"), translate("Filtering IP with blacklist"))
o.rmempty = false
o.default = o.disabled
o.cfgvalue = function(...)
return Flag.cfgvalue(...) or "0"
end
---- anti-Answer-Forgery ---- anti-Answer-Forgery
-- o = s:option(Flag, "check_edns", translate("Anti Answer Forgery"), translate("Anti answer forgery, if DNS does not work properly after enabling, please turn off this feature")) -- o = s:option(Flag, "check_edns", translate("Anti Answer Forgery"), translate("Anti answer forgery, if DNS does not work properly after enabling, please turn off this feature"))
-- o.rmempty = false -- o.rmempty = false
@@ -94,7 +120,6 @@ o.rempty = true
o:depends("type", "tls") o:depends("type", "tls")
o:depends("type", "https") o:depends("type", "https")
---- other args ---- other args
o = s:option(Value, "addition_arg", translate("Additional Server Args"), translate("Additional Args for upstream dns servers")) o = s:option(Value, "addition_arg", translate("Additional Server Args"), translate("Additional Args for upstream dns servers"))
o.default = "" o.default = ""

View File

@@ -0,0 +1,31 @@
--
-- Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
--
-- smartdns is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- smartdns is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
require ("nixio.fs")
require ("luci.http")
require ("luci.dispatcher")
require ("nixio.fs")
local uci = require "luci.model.uci".cursor()
module("luci.model.smartdns", package.seeall)
function get_config_option(module, section, option, default)
return uci:get_first(module, section, option) or default
end
return m

135
package/luci-compat/make.sh Normal file
View File

@@ -0,0 +1,135 @@
#!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
CURR_DIR=$(cd $(dirname $0);pwd)
VER="`date +"1.%Y.%m.%d-%H%M"`"
SMARTDNS_DIR=$CURR_DIR/../../
PO2LMO=
showhelp()
{
echo "Usage: make [OPTION]"
echo "Options:"
echo " -o output directory."
echo " --arch archtecture."
echo " --ver version."
echo " -h show this message."
}
build_tool()
{
make -C $ROOT/tool/po2lmo -j
PO2LMO="$ROOT/tool/po2lmo/src/po2lmo"
}
clean_tool()
{
make -C $ROOT/tool/po2lmo clean
}
build()
{
ROOT=/tmp/luci-app-smartdns
rm -fr $ROOT
mkdir -p $ROOT
cp $CURR_DIR/* $ROOT/ -af
cd $ROOT/
build_tool
mkdir $ROOT/root/usr/lib/lua/ -p
cp $ROOT/files/luci $ROOT/root/usr/lib/lua/ -af
#Generate Language
$PO2LMO $ROOT/files/luci/i18n/smartdns.zh-cn.po $ROOT/root/usr/lib/lua/luci/i18n/smartdns.zh-cn.lmo
rm $ROOT/root/usr/lib/lua/luci/i18n/smartdns.zh-cn.po
cp $ROOT/files/etc $ROOT/root/ -af
INST_SIZE="`du -sb $ROOT/root/ | awk '{print $1}'`"
sed -i "s/^Architecture.*/Architecture: all/g" $ROOT/control/control
sed -i "s/Version:.*/Version: $VER/" $ROOT/control/control
if [ ! -z "$INST_SIZE" ]; then
echo "Installed-Size: $INST_SIZE" >> $ROOT/control/control
fi
cd $ROOT/control
chmod +x *
tar zcf ../control.tar.gz ./
cd $ROOT
tar zcf $ROOT/data.tar.gz -C root .
tar zcf $OUTPUTDIR/luci-app-smartdns.$VER.$FILEARCH.ipk ./control.tar.gz ./data.tar.gz ./debian-binary
rm -fr $ROOT/
}
main()
{
OPTS=`getopt -o o:h --long arch:,ver:,filearch: \
-n "" -- "$@"`
if [ $? != 0 ] ; then echo "Terminating..." >&2 ; exit 1 ; fi
# Note the quotes around `$TEMP': they are essential!
eval set -- "$OPTS"
while true; do
case "$1" in
--arch)
ARCH="$2"
shift 2;;
--filearch)
FILEARCH="$2"
shift 2;;
--ver)
VER="$2"
shift 2;;
-o )
OUTPUTDIR="$2"
shift 2;;
-h | --help )
showhelp
return 0
shift ;;
-- ) shift; break ;;
* ) break ;;
esac
done
if [ -z "$ARCH" ]; then
echo "please input arch."
return 1;
fi
if [ -z "$FILEARCH" ]; then
FILEARCH=$ARCH
fi
if [ -z "$OUTPUTDIR" ]; then
OUTPUTDIR=$CURR_DIR;
fi
build
}
main $@
exit $?

View File

@@ -0,0 +1,12 @@
INSTALL = install
PREFIX = /usr/bin
po2lmo: src/po2lmo.o src/template_lmo.o
$(CC) $(LDFLAGS) -o src/po2lmo src/po2lmo.o src/template_lmo.o
install:
$(INSTALL) -m 755 src/po2lmo $(PREFIX)
clean:
$(RM) src/po2lmo src/*.o

View File

@@ -0,0 +1,247 @@
/*
* lmo - Lua Machine Objects - PO to LMO conversion tool
*
* Copyright (C) 2009-2012 Jo-Philipp Wich <xm@subsignal.org>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "template_lmo.h"
static void die(const char *msg)
{
fprintf(stderr, "Error: %s\n", msg);
exit(1);
}
static void usage(const char *name)
{
fprintf(stderr, "Usage: %s input.po output.lmo\n", name);
exit(1);
}
static void print(const void *ptr, size_t size, size_t nmemb, FILE *stream)
{
if( fwrite(ptr, size, nmemb, stream) == 0 )
die("Failed to write stdout");
}
static int extract_string(const char *src, char *dest, int len)
{
int pos = 0;
int esc = 0;
int off = -1;
for( pos = 0; (pos < strlen(src)) && (pos < len); pos++ )
{
if( (off == -1) && (src[pos] == '"') )
{
off = pos + 1;
}
else if( off >= 0 )
{
if( esc == 1 )
{
switch (src[pos])
{
case '"':
case '\\':
off++;
break;
}
dest[pos-off] = src[pos];
esc = 0;
}
else if( src[pos] == '\\' )
{
dest[pos-off] = src[pos];
esc = 1;
}
else if( src[pos] != '"' )
{
dest[pos-off] = src[pos];
}
else
{
dest[pos-off] = '\0';
break;
}
}
}
return (off > -1) ? strlen(dest) : -1;
}
static int cmp_index(const void *a, const void *b)
{
uint32_t x = ((const lmo_entry_t *)a)->key_id;
uint32_t y = ((const lmo_entry_t *)b)->key_id;
if (x < y)
return -1;
else if (x > y)
return 1;
return 0;
}
static void print_uint32(uint32_t x, FILE *out)
{
uint32_t y = htonl(x);
print(&y, sizeof(uint32_t), 1, out);
}
static void print_index(void *array, int n, FILE *out)
{
lmo_entry_t *e;
qsort(array, n, sizeof(*e), cmp_index);
for (e = array; n > 0; n--, e++)
{
print_uint32(e->key_id, out);
print_uint32(e->val_id, out);
print_uint32(e->offset, out);
print_uint32(e->length, out);
}
}
int main(int argc, char *argv[])
{
char line[4096];
char key[4096];
char val[4096];
char tmp[4096];
int state = 0;
int offset = 0;
int length = 0;
int n_entries = 0;
void *array = NULL;
lmo_entry_t *entry = NULL;
uint32_t key_id, val_id;
FILE *in;
FILE *out;
if( (argc != 3) || ((in = fopen(argv[1], "r")) == NULL) || ((out = fopen(argv[2], "w")) == NULL) )
usage(argv[0]);
memset(line, 0, sizeof(key));
memset(key, 0, sizeof(val));
memset(val, 0, sizeof(val));
while( (NULL != fgets(line, sizeof(line), in)) || (state >= 2 && feof(in)) )
{
if( state == 0 && strstr(line, "msgid \"") == line )
{
switch(extract_string(line, key, sizeof(key)))
{
case -1:
die("Syntax error in msgid");
case 0:
state = 1;
break;
default:
state = 2;
}
}
else if( state == 1 || state == 2 )
{
if( strstr(line, "msgstr \"") == line || state == 2 )
{
switch(extract_string(line, val, sizeof(val)))
{
case -1:
state = 4;
break;
default:
state = 3;
}
}
else
{
switch(extract_string(line, tmp, sizeof(tmp)))
{
case -1:
state = 2;
break;
default:
strcat(key, tmp);
}
}
}
else if( state == 3 )
{
switch(extract_string(line, tmp, sizeof(tmp)))
{
case -1:
state = 4;
break;
default:
strcat(val, tmp);
}
}
if( state == 4 )
{
if( strlen(key) > 0 && strlen(val) > 0 )
{
key_id = sfh_hash(key, strlen(key));
val_id = sfh_hash(val, strlen(val));
if( key_id != val_id )
{
n_entries++;
array = realloc(array, n_entries * sizeof(lmo_entry_t));
entry = (lmo_entry_t *)array + n_entries - 1;
if (!array)
die("Out of memory");
entry->key_id = key_id;
entry->val_id = val_id;
entry->offset = offset;
entry->length = strlen(val);
length = strlen(val) + ((4 - (strlen(val) % 4)) % 4);
print(val, length, 1, out);
offset += length;
}
}
state = 0;
memset(key, 0, sizeof(key));
memset(val, 0, sizeof(val));
}
memset(line, 0, sizeof(line));
}
print_index(array, n_entries, out);
if( offset > 0 )
{
print_uint32(offset, out);
fsync(fileno(out));
fclose(out);
}
else
{
fclose(out);
unlink(argv[2]);
}
fclose(in);
return(0);
}

View File

@@ -0,0 +1,328 @@
/*
* lmo - Lua Machine Objects - Base functions
*
* Copyright (C) 2009-2010 Jo-Philipp Wich <xm@subsignal.org>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "template_lmo.h"
/*
* Hash function from http://www.azillionmonkeys.com/qed/hash.html
* Copyright (C) 2004-2008 by Paul Hsieh
*/
uint32_t sfh_hash(const char *data, int len)
{
uint32_t hash = len, tmp;
int rem;
if (len <= 0 || data == NULL) return 0;
rem = len & 3;
len >>= 2;
/* Main loop */
for (;len > 0; len--) {
hash += sfh_get16(data);
tmp = (sfh_get16(data+2) << 11) ^ hash;
hash = (hash << 16) ^ tmp;
data += 2*sizeof(uint16_t);
hash += hash >> 11;
}
/* Handle end cases */
switch (rem) {
case 3: hash += sfh_get16(data);
hash ^= hash << 16;
hash ^= data[sizeof(uint16_t)] << 18;
hash += hash >> 11;
break;
case 2: hash += sfh_get16(data);
hash ^= hash << 11;
hash += hash >> 17;
break;
case 1: hash += *data;
hash ^= hash << 10;
hash += hash >> 1;
}
/* Force "avalanching" of final 127 bits */
hash ^= hash << 3;
hash += hash >> 5;
hash ^= hash << 4;
hash += hash >> 17;
hash ^= hash << 25;
hash += hash >> 6;
return hash;
}
uint32_t lmo_canon_hash(const char *str, int len)
{
char res[4096];
char *ptr, prev;
int off;
if (!str || len >= sizeof(res))
return 0;
for (prev = ' ', ptr = res, off = 0; off < len; prev = *str, off++, str++)
{
if (isspace(*str))
{
if (!isspace(prev))
*ptr++ = ' ';
}
else
{
*ptr++ = *str;
}
}
if ((ptr > res) && isspace(*(ptr-1)))
ptr--;
return sfh_hash(res, ptr - res);
}
lmo_archive_t * lmo_open(const char *file)
{
int in = -1;
uint32_t idx_offset = 0;
struct stat s;
lmo_archive_t *ar = NULL;
if (stat(file, &s) == -1)
goto err;
if ((in = open(file, O_RDONLY)) == -1)
goto err;
if ((ar = (lmo_archive_t *)malloc(sizeof(*ar))) != NULL)
{
memset(ar, 0, sizeof(*ar));
ar->fd = in;
ar->size = s.st_size;
fcntl(ar->fd, F_SETFD, fcntl(ar->fd, F_GETFD) | FD_CLOEXEC);
if ((ar->mmap = mmap(NULL, ar->size, PROT_READ, MAP_SHARED, ar->fd, 0)) == MAP_FAILED)
goto err;
idx_offset = ntohl(*((const uint32_t *)
(ar->mmap + ar->size - sizeof(uint32_t))));
if (idx_offset >= ar->size)
goto err;
ar->index = (lmo_entry_t *)(ar->mmap + idx_offset);
ar->length = (ar->size - idx_offset - sizeof(uint32_t)) / sizeof(lmo_entry_t);
ar->end = ar->mmap + ar->size;
return ar;
}
err:
if (in > -1)
close(in);
if (ar != NULL)
{
if ((ar->mmap != NULL) && (ar->mmap != MAP_FAILED))
munmap(ar->mmap, ar->size);
free(ar);
}
return NULL;
}
void lmo_close(lmo_archive_t *ar)
{
if (ar != NULL)
{
if ((ar->mmap != NULL) && (ar->mmap != MAP_FAILED))
munmap(ar->mmap, ar->size);
close(ar->fd);
free(ar);
ar = NULL;
}
}
lmo_catalog_t *_lmo_catalogs = NULL;
lmo_catalog_t *_lmo_active_catalog = NULL;
int lmo_load_catalog(const char *lang, const char *dir)
{
DIR *dh = NULL;
char pattern[16];
char path[PATH_MAX];
struct dirent *de = NULL;
lmo_archive_t *ar = NULL;
lmo_catalog_t *cat = NULL;
if (!lmo_change_catalog(lang))
return 0;
if (!dir || !(dh = opendir(dir)))
goto err;
if (!(cat = malloc(sizeof(*cat))))
goto err;
memset(cat, 0, sizeof(*cat));
snprintf(cat->lang, sizeof(cat->lang), "%s", lang);
snprintf(pattern, sizeof(pattern), "*.%s.lmo", lang);
while ((de = readdir(dh)) != NULL)
{
if (!fnmatch(pattern, de->d_name, 0))
{
snprintf(path, sizeof(path), "%s/%s", dir, de->d_name);
ar = lmo_open(path);
if (ar)
{
ar->next = cat->archives;
cat->archives = ar;
}
}
}
closedir(dh);
cat->next = _lmo_catalogs;
_lmo_catalogs = cat;
if (!_lmo_active_catalog)
_lmo_active_catalog = cat;
return 0;
err:
if (dh) closedir(dh);
if (cat) free(cat);
return -1;
}
int lmo_change_catalog(const char *lang)
{
lmo_catalog_t *cat;
for (cat = _lmo_catalogs; cat; cat = cat->next)
{
if (!strncmp(cat->lang, lang, sizeof(cat->lang)))
{
_lmo_active_catalog = cat;
return 0;
}
}
return -1;
}
static lmo_entry_t * lmo_find_entry(lmo_archive_t *ar, uint32_t hash)
{
unsigned int m, l, r;
uint32_t k;
l = 0;
r = ar->length - 1;
while (1)
{
m = l + ((r - l) / 2);
if (r < l)
break;
k = ntohl(ar->index[m].key_id);
if (k == hash)
return &ar->index[m];
if (k > hash)
{
if (!m)
break;
r = m - 1;
}
else
{
l = m + 1;
}
}
return NULL;
}
int lmo_translate(const char *key, int keylen, char **out, int *outlen)
{
uint32_t hash;
lmo_entry_t *e;
lmo_archive_t *ar;
if (!key || !_lmo_active_catalog)
return -2;
hash = lmo_canon_hash(key, keylen);
for (ar = _lmo_active_catalog->archives; ar; ar = ar->next)
{
if ((e = lmo_find_entry(ar, hash)) != NULL)
{
*out = ar->mmap + ntohl(e->offset);
*outlen = ntohl(e->length);
return 0;
}
}
return -1;
}
void lmo_close_catalog(const char *lang)
{
lmo_archive_t *ar, *next;
lmo_catalog_t *cat, *prev;
for (prev = NULL, cat = _lmo_catalogs; cat; prev = cat, cat = cat->next)
{
if (!strncmp(cat->lang, lang, sizeof(cat->lang)))
{
if (prev)
prev->next = cat->next;
else
_lmo_catalogs = cat->next;
for (ar = cat->archives; ar; ar = next)
{
next = ar->next;
lmo_close(ar);
}
free(cat);
break;
}
}
}

View File

@@ -0,0 +1,92 @@
/*
* lmo - Lua Machine Objects - General header
*
* Copyright (C) 2009-2012 Jo-Philipp Wich <xm@subsignal.org>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef _TEMPLATE_LMO_H_
#define _TEMPLATE_LMO_H_
#include <stdlib.h>
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <errno.h>
#include <fnmatch.h>
#include <dirent.h>
#include <ctype.h>
#include <limits.h>
#if (defined(__GNUC__) && defined(__i386__))
#define sfh_get16(d) (*((const uint16_t *) (d)))
#else
#define sfh_get16(d) ((((uint32_t)(((const uint8_t *)(d))[1])) << 8)\
+(uint32_t)(((const uint8_t *)(d))[0]) )
#endif
struct lmo_entry {
uint32_t key_id;
uint32_t val_id;
uint32_t offset;
uint32_t length;
} __attribute__((packed));
typedef struct lmo_entry lmo_entry_t;
struct lmo_archive {
int fd;
int length;
uint32_t size;
lmo_entry_t *index;
char *mmap;
char *end;
struct lmo_archive *next;
};
typedef struct lmo_archive lmo_archive_t;
struct lmo_catalog {
char lang[6];
struct lmo_archive *archives;
struct lmo_catalog *next;
};
typedef struct lmo_catalog lmo_catalog_t;
uint32_t sfh_hash(const char *data, int len);
uint32_t lmo_canon_hash(const char *data, int len);
lmo_archive_t * lmo_open(const char *file);
void lmo_close(lmo_archive_t *ar);
extern lmo_catalog_t *_lmo_catalogs;
extern lmo_catalog_t *_lmo_active_catalog;
int lmo_load_catalog(const char *lang, const char *dir);
int lmo_change_catalog(const char *lang);
int lmo_translate(const char *key, int keylen, char **out, int *outlen);
void lmo_close_catalog(const char *lang);
#endif

View File

@@ -4,5 +4,4 @@ Depends: libc, smartdns
Source: feeds/luci/applications/luci-app-smartdns Source: feeds/luci/applications/luci-app-smartdns
Section: luci Section: luci
Architecture: all Architecture: all
Installed-Size: 1040 Description: A smartdns server
Description: A smartdns server module

View File

@@ -1,4 +1,20 @@
#!/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
[ "${IPKG_NO_SCRIPT}" = "1" ] && exit 0 [ "${IPKG_NO_SCRIPT}" = "1" ] && exit 0
[ -x ${IPKG_INSTROOT}/lib/functions.sh ] || exit 0 [ -x ${IPKG_INSTROOT}/lib/functions.sh ] || exit 0
. ${IPKG_INSTROOT}/lib/functions.sh . ${IPKG_INSTROOT}/lib/functions.sh

View File

@@ -1,4 +0,0 @@
[ -n "${IPKG_INSTROOT}" ] || {
(. /etc/uci-defaults/50_luci-smartdns) && rm -f /etc/uci-defaults/50_luci-smartdns
exit 0
}

View File

@@ -1,4 +1,20 @@
#!/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
[ -x ${IPKG_INSTROOT}/lib/functions.sh ] || exit 0 [ -x ${IPKG_INSTROOT}/lib/functions.sh ] || exit 0
. ${IPKG_INSTROOT}/lib/functions.sh . ${IPKG_INSTROOT}/lib/functions.sh
default_prerm $0 $@ default_prerm $0 $@

View File

@@ -1,11 +0,0 @@
#!/bin/sh
uci -q batch <<-EOF >/dev/null
delete ucitrack.@smartdns[-1]
add ucitrack smartdns
set ucitrack.@smartdns[-1].init=smartdns
commit ucitrack
EOF
rm -f /tmp/luci-indexcache
exit 0

View File

@@ -1,282 +1,311 @@
msgid "" msgid ""
msgstr "Content-Type: text/plain; charset=UTF-8\n" msgstr "Content-Type: text/plain; charset=UTF-8\n"
msgid "SmartDNS" msgid "SmartDNS"
msgstr "SmartDNS" msgstr "SmartDNS"
msgid "SmartDNS is a local high-performance DNS server" msgid "SmartDNS is a local high-performance DNS server"
msgstr "SmartDNS是一个本地高性能DNS服务器" msgstr "SmartDNS是一个本地高性能DNS服务器"
msgid "SmartDNS Server" msgid "SmartDNS Server"
msgstr "SmartDNS 服务器" msgstr "SmartDNS 服务器"
msgid "SmartDNS is a local high-performance DNS server, supports finding fastest IP, supports ad filtering, and supports avoiding DNS poisoning." msgid "SmartDNS is a local high-performance DNS server, supports finding fastest IP, supports ad filtering, and supports avoiding DNS poisoning."
msgstr "SmartDNS是一个本地高性能DNS服务器支持返回最快IP支持广告过滤。" msgstr "SmartDNS是一个本地高性能DNS服务器支持返回最快IP支持广告过滤。"
msgid "Custom Settings" msgid "Custom Settings"
msgstr "自定义设置" msgstr "自定义设置"
msgid "Generate Coredump" msgid "General Settings"
msgstr "生成coredump" msgstr "基本设置"
msgid "Generate Coredump file when smartdns crash, coredump file is located at /tmp/smartdns.xxx.core." msgid "Settings"
msgstr "当smartdns异常时生成coredump文件coredump文件在/tmp/smartdns.xxx.core." msgstr "设置"
msgid "Server Name" msgid "Advanced Settings"
msgstr "服务器名称" msgstr "高级设置"
msgid "Smartdns server name" msgid "RUNNING"
msgstr "SmartDNS的服务器名称默认为smartdns留空为主机名" msgstr "运行中"
msgid "SmartDNS is a local dns server to find fastest ip." msgid "NOT RUNNING"
msgstr "本地高性能服务器,优化网络访问性能。" msgstr "未运行"
msgid "Enable or disable smartdns server" msgid "Generate Coredump"
msgstr "启用或禁用SmartDNS服务" msgstr "生成coredump"
msgid "Local Port" msgid "Generate Coredump file when smartdns crash, coredump file is located at /tmp/smartdns.xxx.core."
msgstr "本地端口" msgstr "当smartdns异常时生成coredump文件coredump文件在/tmp/smartdns.xxx.core."
msgid "Smartdns local server port" msgid "Server Name"
msgstr "SmartDNS本地服务端口" msgstr "服务器名称"
msgid "IPV4 53 Port Redirect Failure" msgid "Smartdns server name"
msgstr "IPV4 53端口重定向失败" msgstr "SmartDNS的服务器名称默认为smartdns留空为主机名"
msgid "IPV6 53 Port Redirect Failure" msgid "SmartDNS is a local dns server to find fastest ip."
msgstr "IPV6 53端口重定向失败" msgstr "本地高性能服务器,优化网络访问性能。"
msgid "Dnsmasq Forwared To Smartdns Failure" msgid "Enable or disable smartdns server"
msgstr "重定向dnsmasq到smartdns失败" msgstr "启用或禁用SmartDNS服务"
msgid "TCP Server" msgid "Local Port"
msgstr "TCP服务器" msgstr "本地端口"
msgid "Enable TCP DNS Server" msgid "Smartdns local server port"
msgstr "启用TCP服务器" msgstr "SmartDNS本地服务端口"
msgid "IPV6 Server" msgid "IPV4 53 Port Redirect Failure"
msgstr "IPV6服务器" msgstr "IPV4 53端口重定向失败"
msgid "Enable IPV6 DNS Server" msgid "IPV6 53 Port Redirect Failure"
msgstr "启用IPV6服务器" msgstr "IPV6 53端口重定向失败"
msgid "Dual-stack IP Selection" msgid "Dnsmasq Forwared To Smartdns Failure"
msgstr "双栈IP优选" msgstr "重定向dnsmasq到smartdns失败"
msgid "Enable IP selection between IPV4 and IPV6" msgid "TCP Server"
msgstr "启用或禁用IPV4IPV6间的IP优选策略。" msgstr "TCP服务器"
msgid "Domain prefetch" msgid "Enable TCP DNS Server"
msgstr "域名预加载" msgstr "启用TCP服务器"
msgid "Enable domain prefetch, accelerate domain response speed." msgid "IPV6 Server"
msgstr "启用域名预加载,加速域名响应速度。" msgstr "IPV6服务器"
msgid "Redirect" msgid "Enable IPV6 DNS Server"
msgstr "重定向" msgstr "启用IPV6服务器"
msgid "SmartDNS redirect mode" msgid "Dual-stack IP Selection"
msgstr "SmartDNS 重定向模式" msgstr "双栈IP优选"
msgid "Run as dnsmasq upstream server" msgid "Enable IP selection between IPV4 and IPV6"
msgstr "作为dnsmasq的上游服务器" msgstr "启用或禁用IPV4IPV6间的IP优选策略。"
msgid "Redirect 53 port to SmartDNS" msgid "Domain prefetch"
msgstr "重定向53端口到SmartDNS" msgstr "域名预加载"
msgid "Cache Size" msgid "Enable domain prefetch, accelerate domain response speed."
msgstr "缓存大小" msgstr "启用域名预加载,加速域名响应速度。"
msgid "DNS domain result cache size" msgid "Serve expired"
msgstr "缓存DNS的结果缓存大小配置零则不缓存" msgstr "缓存过期服务"
msgid "Domain TTL" msgid "Attempts to serve old responses from cache with a TTL of 0 in the response without waiting for the actual resolution to finish."
msgstr "域名TTL" msgstr "查询性能优化有请求时尝试回应TTL为0的过期记录以避免查询等待。"
msgid "TTL for all domain result." msgid "Redirect"
msgstr "设置所有域名的TTL值" msgstr "重定向"
msgid "Domain TTL Min" msgid "SmartDNS redirect mode"
msgstr "域名TTL最小值" msgstr "SmartDNS 重定向模式"
msgid "Minimum TTL for all domain result." msgid "Run as dnsmasq upstream server"
msgstr "设置所有域名的TTL最小值" msgstr "作为dnsmasq的上游服务器"
msgid "Domain TTL Max" msgid "Redirect 53 port to SmartDNS"
msgstr "域名TTL最大值" msgstr "重定向53端口到SmartDNS"
msgid "Maximum TTL for all domain result." msgid "Cache Size"
msgstr "设置所有域名的TTL最大值" msgstr "缓存大小"
msgid "smartdns custom settings" msgid "DNS domain result cache size"
msgstr "smartdns 自定义设置,具体配置参数参考指导" msgstr "缓存DNS的结果缓存大小配置零则不缓存"
msgid "Second Server Settings" msgid "Domain TTL"
msgstr "第二DNS服务器" msgstr "域名TTL"
msgid "Enable or disable second DNS server." msgid "TTL for all domain result."
msgstr "是否启用第二DNS服务器。" msgstr "设置所有域名的TTL值"
msgid "Skip Speed Check" msgid "Domain TTL Min"
msgstr "跳过测速" msgstr "域名TTL最小值"
msgid "Do not check speed." msgid "Minimum TTL for all domain result."
msgstr "禁用测速。" msgstr "设置所有域名的TTL最小值"
msgid "Server Group" msgid "Domain TTL Max"
msgstr "服务器组" msgstr "域名TTL最大值"
msgid "Query DNS through specific dns server group, such as office, home." msgid "Maximum TTL for all domain result."
msgstr "使用指定服务器组查询比如office, home。" msgstr "设置所有域名的TTL最大值"
msgid "Skip Address Rules" msgid "smartdns custom settings"
msgstr "跳过address规则" msgstr "smartdns 自定义设置,具体配置参数参考指导"
msgid "Skip address rules." msgid "Second Server Settings"
msgstr "跳过address规则。" msgstr "第二DNS服务器"
msgid "Skip Nameserver Rule" msgid "Enable or disable second DNS server."
msgstr "跳过Nameserver规则" msgstr "是否启用第二DNS服务器。"
msgid "Skip nameserver rules." msgid "Skip Speed Check"
msgstr "跳过Nameserver规则。" msgstr "跳过测速"
msgid "Skip Ipset Rule" msgid "Do not check speed."
msgstr "跳过ipset规则" msgstr "禁用测速。"
msgid "Skip ipset rules." msgid "Server Group"
msgstr "跳过ipset规则。" msgstr "服务器组"
msgid "Skip SOA Address Rule" msgid "Query DNS through specific dns server group, such as office, home."
msgstr "跳过address SOA(#)规则" msgstr "使用指定服务器组查询比如office, home。"
msgid "Skip SOA address rules." msgid "Skip Address Rules"
msgstr "跳过address SOA(#)规则" msgstr "跳过address规则"
msgid "Skip Dualstack Selection" msgid "Skip address rules."
msgstr "跳过双栈优选" msgstr "跳过address规则。"
msgid "Skip Sualstack Selection." msgid "Skip Nameserver Rule"
msgstr "跳过双栈优选。" msgstr "跳过Nameserver规则"
msgid "Skip Cache" msgid "Skip nameserver rules."
msgstr "跳过cache" msgstr "跳过Nameserver规则。"
msgid "Skip Cache." msgid "Skip Ipset Rule"
msgstr "跳过cache。" msgstr "跳过ipset规则"
msgid "Upstream Servers" msgid "Skip ipset rules."
msgstr "上游服务器" msgstr "跳过ipset规则。"
msgid "Upstream Servers, support UDP, TCP protocol. Please configure multiple DNS servers, including multiple foreign DNS servers." msgid "Skip SOA Address Rule"
msgstr "上游DNS服务器列表支持UDPTCP协议请配置多个上游DNS服务器包括多个国内外服务器" msgstr "跳过address SOA(#)规则"
msgid "DNS Server Name" msgid "Skip SOA address rules."
msgstr "DNS服务器名称" msgstr "跳过address SOA(#)规则。"
msgid "port" msgid "Skip Dualstack Selection"
msgstr "端口" msgstr "跳过双栈优选"
msgid "DNS Server port" msgid "Skip Dualstack Selection."
msgstr "DNS服务器端口" msgstr "跳过双栈优选。"
msgid "DNS Server ip" msgid "Skip Cache"
msgstr "DNS服务器IP" msgstr "跳过cache"
msgid "type" msgid "Skip Cache."
msgstr "类型" msgstr "跳过cache。"
msgid "DNS Server type" msgid "Force AAAA SOA"
msgstr "协议类型" msgstr "停用IPV6地址解析"
msgid "Domain Address" msgid "Force AAAA SOA."
msgstr "域名地址" msgstr "停用IPV6地址解析。"
msgid "TLS Hostname Verify" msgid "Upstream Servers"
msgstr "校验TLS主机名" msgstr "上游服务器"
msgid "Set TLS hostname to verify." msgid "Upstream Servers, support UDP, TCP protocol. Please configure multiple DNS servers, including multiple foreign DNS servers."
msgstr "设置校验TLS主机名。" msgstr "上游DNS服务器列表支持UDPTCP协议请配置多个上游DNS服务器包括多个国内外服务器"
msgid "TLS SNI name" msgid "DNS Server Name"
msgstr "TLS SNI名称" msgstr "DNS服务器名称"
msgid "HTTP Host" msgid "port"
msgstr "HTTP主机" msgstr "端口"
msgid "Sets the server name indication for query." msgid "DNS Server port"
msgstr "设置查询时使用的服务器SNI名称。" msgstr "DNS服务器端口"
msgid "Set the HTTP host used for the query. Use this parameter when the host of the URL address is an IP address." msgid "DNS Server ip"
msgstr "设置查询时使用的HTTP主机当URL地址的host是IP地址时使用此参数。" msgstr "DNS服务器IP"
msgid "Server Group" msgid "type"
msgstr "服务器组" msgstr "类型"
msgid "DNS Server group belongs to, used with nameserver, such as office, home." msgid "DNS Server type"
msgsr "DNS服务器所属组 配合nameserver使用例如officehome。" msgstr "协议类型"
msgid "IP Blacklist Filtering" msgid "Domain Address"
msgstr "IP黑名单过滤" msgstr "域名地址"
msgid "Anti Answer Forgery" msgid "TLS Hostname Verify"
msgstr "反回答伪造" msgstr "校验TLS主机名"
msgid "Anti answer forgery, if DNS does not work properly after enabling, please turn off this feature" msgid "Set TLS hostname to verify."
msgstr "反回答伪造如果启用后DNS工作不正常请关闭此功能。" msgstr "设置校验TLS主机名。"
msgid "Filtering IP with blacklist" msgid "No check certificate"
msgstr "使用IP黑名单过滤" msgstr "停用证书校验"
msgid "TLS SPKI Pinning" msgid "Do not check certificate."
msgstr "TLS SPKI 指纹" msgstr "不校验证书的合法性。"
msgid "Used to verify the validity of the TLS server, The value is Base64 encoded SPKI fingerprint, leaving blank to indicate that the validity of TLS is not verified." msgid "TLS SNI name"
msgstr "用于校验TLS服务器的有效性数值为Base64编码的SPKI指纹, 留空表示不验证TLS的合法性" msgstr "TLS SNI名称"
msgid "Additional Server Args" msgid "HTTP Host"
msgstr "额外的服务器参数" msgstr "HTTP主机"
msgid "Additional Args for upstream dns servers" msgid "Sets the server name indication for query."
msgstr "额外的上游DNS服务器参数" msgstr "设置查询时使用的服务器SNI名称。"
msgid "Upstream DNS Server Configuration" msgid "Set the HTTP host used for the query. Use this parameter when the host of the URL address is an IP address."
msgstr "上游DNS服务器配置" msgstr "设置查询时使用的HTTP主机当URL地址的host是IP地址时使用此参数。"
msgid "Set Specific domain ip address." msgid "Server Group"
msgstr "指定特定域名的IP地址" msgstr "服务器组"
msgid "Specify an IP address to return for any host in the given domains, Queries in the domains are never forwarded and always replied to with the specified IP address which may be IPv4 or IPv6." msgid "DNS Server group belongs to, used with nameserver, such as office, home."
msgstr "配置特定域名返回特定的IP地址域名查询将不到上游服务器请求直接返回配置的IP地址可用于广告屏蔽。" msgstr "DNS服务器所属组 配合nameserver使用例如officehome。"
msgid "IP Blacklist" msgid "IP Blacklist Filtering"
msgstr "IP黑名单" msgstr "IP黑名单过滤"
msgid "Set Specific ip blacklist." msgid "Anti Answer Forgery"
msgstr "设置IP黑名单列表" msgstr "反回答伪造"
msgid "Configure IP blacklists that will be filtered from the results of specific DNS server." msgid "Anti answer forgery, if DNS does not work properly after enabling, please turn off this feature"
msgstr "配置需要从指定域名服务器结果过滤的IP黑名单。" msgstr "反回答伪造如果启用后DNS工作不正常请关闭此功能。"
msgid "Technical Support" msgid "Filtering IP with blacklist"
msgstr "技术支持" msgstr "使用IP黑名单过滤"
msgid "If you like this software, please buy me a cup of coffee." msgid "TLS SPKI Pinning"
msgstr "如果本软件对你有帮助,请给作者加个蛋。" msgstr "TLS SPKI 指纹"
msgid "SmartDNS official website" msgid "Used to verify the validity of the TLS server, The value is Base64 encoded SPKI fingerprint, leaving blank to indicate that the validity of TLS is not verified."
msgstr "SmartDNS官方网站" msgstr "用于校验TLS服务器的有效性数值为Base64编码的SPKI指纹, 留空表示不验证TLS的合法性"
msgid "open website" msgid "Additional Server Args"
msgstr "打开网站" msgstr "额外的服务器参数"
msgid "Donate to smartdns" msgid "Additional Args for upstream dns servers"
msgstr "捐助smartdns项目" msgstr "额外的上游DNS服务器参数"
msgid "Donate" msgid "Upstream DNS Server Configuration"
msgstr "捐助" msgstr "上游DNS服务器配置"
msgid "Set Specific domain ip address."
msgstr "指定特定域名的IP地址"
msgid "Specify an IP address to return for any host in the given domains, Queries in the domains are never forwarded and always replied to with the specified IP address which may be IPv4 or IPv6."
msgstr "配置特定域名返回特定的IP地址域名查询将不到上游服务器请求直接返回配置的IP地址可用于广告屏蔽。"
msgid "IP Blacklist"
msgstr "IP黑名单"
msgid "Set Specific ip blacklist."
msgstr "设置IP黑名单列表"
msgid "Configure IP blacklists that will be filtered from the results of specific DNS server."
msgstr "配置需要从指定域名服务器结果过滤的IP黑名单。"
msgid "Technical Support"
msgstr "技术支持"
msgid "If you like this software, please buy me a cup of coffee."
msgstr "如果本软件对你有帮助,请给作者加个蛋。"
msgid "SmartDNS official website"
msgstr "SmartDNS官方网站"
msgid "open website"
msgstr "打开网站"
msgid "Donate to smartdns"
msgstr "捐助smartdns项目"
msgid "Donate"
msgstr "捐助"

View File

@@ -1,17 +0,0 @@
-- Copyright 2018 Nick Peng (pymumu@gmail.com)
require ("nixio.fs")
require ("luci.http")
require ("luci.dispatcher")
require ("nixio.fs")
local uci = require "luci.model.uci".cursor()
module("luci.model.smartdns", package.seeall)
function get_config_option(module, section, option, default)
return uci:get_first(module, section, option) or default
end
return m

View File

@@ -0,0 +1,12 @@
{
"admin/services/smartdns": {
"title": "SmartDNS",
"action": {
"type": "view",
"path": "smartdns/smartdns"
},
"depends": {
"uci": { "smartdns": true }
}
}
}

View File

@@ -0,0 +1,23 @@
{
"luci-app-smartdns": {
"description": "Grant access to LuCI app smartdns",
"read": {
"file": {
"/etc/smartdns/*": [ "read" ],
"/usr/sbin/iptables -t nat -nL PREROUTING": [ "exec" ],
"/usr/sbin/ip6tables -t nat -nL PREROUTING": [ "exec" ],
"/usr/sbin/smartdns": [ "exec" ]
},
"ubus": {
"service": [ "list" ]
},
"uci": [ "smartdns" ]
},
"write": {
"file": {
"/etc/smartdns/*": [ "write" ]
},
"uci": [ "smartdns" ]
}
}
}

View File

@@ -0,0 +1,492 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
'use strict';
'require fs';
'require uci';
'require form';
'require rpc';
var conf = 'smartdns';
var callServiceList = rpc.declare({
object: 'service',
method: 'list',
params: ['name'],
expect: { '': {} }
});
function getPidOfSmartdns() {
return L.resolveDefault(callServiceList(conf), {})
.then(function (res) {
var isrunning = false;
try {
isrunning = res[conf]['instances']['smartdns']['running'];
} catch (e) { }
return isrunning;
});
}
function getIPTablesRedirect() {
return fs.exec('/usr/sbin/iptables', ['-t', 'nat', '-nL', 'PREROUTING']).then(function (res) {
if (res.code === 0) {
return res.stdout.trim();
} else {
return "";
}
});
}
function getIP6TablesRedirect() {
return fs.exec('/usr/sbin/ip6tables', ['-t', 'nat', '-nL', 'PREROUTING']).then(function (res) {
if (res.code === 0) {
return res.stdout.trim();
} else {
return "";
}
});
}
function smartdnsServiceStatus() {
return Promise.all([
getPidOfSmartdns(),
getIPTablesRedirect(),
getIP6TablesRedirect()
]);
}
function smartdnsRenderStatus(res) {
var renderHTML = "";
var isRunning = res[0];
var ipt = res[1];
var ip6t = res[2];
var serverPort = uci.get_first('smartdns', 'smartdns', 'port');
var redirectMode = uci.get_first('smartdns', 'smartdns', 'redirect');
var ipv6Enabled = uci.get_first('smartdns', 'smartdns', 'ipv6_server');
if (isRunning) {
renderHTML += "<span style=\"color:green;font-weight:bold\">SmartDNS - " + _("RUNNING") + "</span>";
} else {
renderHTML += "<span style=\"color:red;font-weight:bold\">SmartDNS - " + _("NOT RUNNING") + "</span>";
return renderHTML;
}
if (redirectMode === "dnsmasq-upstream") {
var matchLine = "127.0.0.1#" + serverPort;
var dnsmasqServer = uci.get_first('dhcp', 'dnsmasq', 'server') || "";
if (dnsmasqServer.indexOf(matchLine) < 0) {
renderHTML += "<br /><span style=\"color:red;font-weight:bold\">" + _("Dnsmasq Forwared To Smartdns Failure") + "</span>";
}
} else if (redirectMode === "redirect") {
var redirectRules = (ipt || '').split(/\n/).filter(function (rule) {
return rule.match(/REDIRECT/) && rule.match(/dpt:53/) && rule.match("ports " + serverPort);
});
if (redirectRules.length <= 0) {
renderHTML += "<br /><span style=\"color:red;font-weight:bold\">" + _("IPV4 53 Port Redirect Failure") + "</span>";
if (ipv6Enabled) {
var redirectRules = (ip6t || '').split(/\n/).filter(function (rule) {
return rule.match(/REDIRECT/) && rule.match(/dpt:53/) && rule.match("ports " + serverPort);
});
if (redirectRules.length <= 0) {
renderHTML += "<br /><span style=\"color:red;font-weight:bold\">" + _("IPV6 53 Port Redirect Failure") + "</span>";
}
}
}
}
return renderHTML;
}
return L.view.extend({
load: function () {
return Promise.all([
uci.load('smartdns'),
uci.load('dhcp')
]);
},
render: function (stats) {
var m, s, o;
m = new form.Map('smartdns', _('SmartDNS'));
m.title = _("SmartDNS Server");
m.description = _("SmartDNS is a local high-performance DNS server, supports finding fastest IP, "
+ "supports ad filtering, and supports avoiding DNS poisoning.");
s = m.section(form.NamedSection, '_status');
s.anonymous = true;
s.render = function (section_id) {
L.Poll.add(function () {
return L.resolveDefault(smartdnsServiceStatus()).then(function (res) {
var view = document.getElementById("service_status");
view.innerHTML = smartdnsRenderStatus(res);
});
});
return E('div', { class: 'cbi-map' },
E('div', { class: 'cbi-section' }, [
E('div', { id: 'service_status' },
_('Collecting data ...'))
])
);
}
// Basic;
s = m.section(form.TypedSection, "smartdns", _("Settings"), _("General Settings"));
s.anonymous = true;
s.tab("settings", _("General Settings"));
s.tab("seconddns", _("Second Server Settings"));
s.tab("custom", _("Custom Settings"));
// Eanble;
o = s.taboption("settings", form.Flag, "enabled", _("Enable"), _("Enable or disable smartdns server"));
o.default = o.disabled;
o.rempty = false;
// server name;
o = s.taboption("settings", form.Value, "server_name", _("Server Name"), _("Smartdns server name"));
o.default = "smartdns";
o.datatype = "hostname";
o.rempty = false;
// Port;
o = s.taboption("settings", form.Value, "port", _("Local Port"), _("Smartdns local server port"));
o.placeholder = 6053;
o.default = 6053;
o.datatype = "port";
o.rempty = false;
// Enable TCP server;
o = s.taboption("settings", form.Flag, "tcp_server", _("TCP Server"), _("Enable TCP DNS Server"));
o.rmempty = false;
o.default = o.enabled;
// Support IPV6;
o = s.taboption("settings", form.Flag, "ipv6_server", _("IPV6 Server"), _("Enable IPV6 DNS Server"));
o.rmempty = false;
o.default = o.enabled;
// Support DualStack ip selection;
o = s.taboption("settings", form.Flag, "dualstack_ip_selection", _("Dual-stack IP Selection"),
_("Enable IP selection between IPV4 and IPV6"));
o.rmempty = false;
o.default = o.disabled;
// Domain prefetch load ;
o = s.taboption("settings", form.Flag, "prefetch_domain", _("Domain prefetch"),
_("Enable domain prefetch, accelerate domain response speed."));
o.rmempty = false;
o.default = o.disabled;
// Domain Serve expired
o = s.taboption("settings", form.Flag, "serve_expired", _("Serve expired"),
_("Attempts to serve old responses from cache with a TTL of 0 in the response without waiting for the actual resolution to finish."));
o.rmempty = false;
o.default = o.disabled;
// Redirect;
o = s.taboption("settings", form.ListValue, "redirect", _("Redirect"), _("SmartDNS redirect mode"));
o.placeholder = "none";
o.value("none", _("none"));
o.value("dnsmasq-upstream", _("Run as dnsmasq upstream server"));
o.value("redirect", _("Redirect 53 port to SmartDNS"));
o.default = "none";
o.rempty = false;
// cache-size;
o = s.taboption("settings", form.Value, "cache_size", _("Cache Size"), _("DNS domain result cache size"));
o.rempty = true;
// rr-ttl;
o = s.taboption("settings", form.Value, "rr_ttl", _("Domain TTL"), _("TTL for all domain result."));
o.rempty = true;
// rr-ttl-min;
o = s.taboption("settings", form.Value, "rr_ttl_min", _("Domain TTL Min"),
_("Minimum TTL for all domain result."));
o.rempty = true;
o.placeholder = "300";
o.default = 300;
o.optional = true;
// second dns server;
// rr-ttl-max;
o = s.taboption("settings", form.Value, "rr_ttl_max", _("Domain TTL Max"),
_("Maximum TTL for all domain result."));
o.rempty = true;
// Eanble;
o = s.taboption("seconddns", form.Flag, "seconddns_enabled", _("Enable"),
_("Enable or disable second DNS server."));
o.default = o.disabled;
o.rempty = false;
// Port;
o = s.taboption("seconddns", form.Value, "seconddns_port", _("Local Port"), _("Smartdns local server port"));
o.placeholder = 6553;
o.default = 6553;
o.datatype = "port";
o.rempty = false;
// Enable TCP server;
o = s.taboption("seconddns", form.Flag, "seconddns_tcp_server", _("TCP Server"), _("Enable TCP DNS Server"));
o.rmempty = false;
o.default = o.enabled;
// dns server group;
o = s.taboption("seconddns", form.Value, "seconddns_server_group", _("Server Group"),
_("Query DNS through specific dns server group, such as office, home."));
o.rmempty = true;
o.placeholder = "default";
o.datatype = "hostname";
o.rempty = true;
o = s.taboption("seconddns", form.Flag, "seconddns_no_speed_check", _("Skip Speed Check"),
_("Do not check speed."));
o.rmempty = false;
o.default = o.disabled;
// skip address rules;
o = s.taboption("seconddns", form.Flag, "seconddns_no_rule_addr", _("Skip Address Rules"),
_("Skip address rules."));
o.rmempty = false;
o.default = o.disabled;
// skip name server rules;
o = s.taboption("seconddns", form.Flag, "seconddns_no_rule_nameserver", _("Skip Nameserver Rule"),
_("Skip nameserver rules."));
o.rmempty = false;
o.default = o.disabled;
// skip ipset rules;
o = s.taboption("seconddns", form.Flag, "seconddns_no_rule_ipset", _("Skip Ipset Rule"),
_("Skip ipset rules."));
o.rmempty = false;
o.default = o.disabled;
// skip soa address rule;
o = s.taboption("seconddns", form.Flag, "seconddns_no_rule_soa", _("Skip SOA Address Rule"),
_("Skip SOA address rules."));
o.rmempty = false;
o.default = o.disabled;
o = s.taboption("seconddns", form.Flag, "seconddns_no_dualstack_selection", _("Skip Dualstack Selection"),
_("Skip Dualstack Selection."));
o.rmempty = false;
o.default = o.disabled;
// skip cache;
o = s.taboption("seconddns", form.Flag, "seconddns_no_cache", _("Skip Cache"), _("Skip Cache."));
o.rmempty = false;
o.default = o.disabled;
// Force AAAA SOA
o = s.taboption("seconddns", form.Flag, "force_aaaa_soa", _("Force AAAA SOA"), _("Force AAAA SOA."));
o.rmempty = false;
o.default = o.disabled;
// custom settings;
o = s.taboption("custom", form.TextValue, "custom_conf",
"", _("smartdns custom settings"));
o.rows = 20;
o.cfgvalue = function (section_id) {
return fs.trimmed('/etc/smartdns/custom.conf');
};
o.write = function (section_id, formvalue) {
return fs.write('/etc/smartdns/custom.conf', formvalue.trim().replace(/\r\n/g, '\n') + '\n');
};
o = s.taboption("custom", form.Flag, "coredump", _("Generate Coredump"),
_("Generate Coredump file when smartdns crash, coredump file is located at /tmp/smartdns.xxx.core."));
o.rmempty = false;
o.default = o.disabled;
// Upstream servers;
s = m.section(form.GridSection, "server", _("Upstream Servers"),
_("Upstream Servers, support UDP, TCP protocol. Please configure multiple DNS servers, "
+ "including multiple foreign DNS servers."));
s.anonymous = true;
s.addremove = true;
s.tab('general', _('General Settings'));
s.tab('advanced', _('Advanced Settings'));
// enable flag;
o = s.taboption("general", form.Flag, "enabled", _("Enable"), _("Enable"));
o.rmempty = false;
o.default = o.enabled;
o.editable = true;
// name;
o = s.taboption("general", form.Value, "name", _("DNS Server Name"), _("DNS Server Name"));
// IP address;
o = s.taboption("general", form.Value, "ip", _("ip"), _("DNS Server ip"));
o.datatype = "or(ipaddr, string)";
o.rmempty = false;
// port;
o = s.taboption("general", form.Value, "port", _("port"), _("DNS Server port"));
o.placeholder = "default";
o.datatype = "port";
o.rempty = true;
o.depends("type", "udp");
o.depends("type", "tcp");
o.depends("type", "tls");
// type;
o = s.taboption("general", form.ListValue, "type", _("type"), _("DNS Server type"));
o.placeholder = "udp";
o.value("udp", _("udp"));
o.value("tcp", _("tcp"));
o.value("tls", _("tls"));
o.value("https", _("https"));
o.default = "udp";
o.rempty = false;
// Advanced Options
// server group
o = s.taboption("advanced", form.Value, "server_group", _("Server Group"), _("DNS Server group belongs to, "
+ "used with nameserver, such as office, home."))
o.rmempty = true
o.placeholder = "default"
o.datatype = "hostname"
o.rempty = true
o.modalonly = true;
// blacklist_ip
o = s.taboption("advanced", form.Flag, "blacklist_ip", _("IP Blacklist Filtering"),
_("Filtering IP with blacklist"))
o.rmempty = false
o.default = o.disabled
o.modalonly = true;
// TLS host verify
o = s.taboption("advanced", form.Value, "tls_host_verify", _("TLS Hostname Verify"),
_("Set TLS hostname to verify."))
o.default = ""
o.datatype = "string"
o.rempty = true
o.modalonly = true;
o.depends("type", "tls")
o.depends("type", "https")
// certificate verify
o = s.taboption("advanced", form.Flag, "no_check_certificate", _("No check certificate"),
_("Do not check certificate."))
o.rmempty = false
o.default = o.disabled
o.modalonly = true;
o.depends("type", "tls")
o.depends("type", "https")
// SNI host name
o = s.taboption("advanced", form.Value, "host_name", _("TLS SNI name"),
_("Sets the server name indication for query."))
o.default = ""
o.datatype = "hostname"
o.rempty = true
o.modalonly = true;
o.depends("type", "tls")
o.depends("type", "https")
// http host
o = s.taboption("advanced", form.Value, "http_host", _("HTTP Host"),
_("Set the HTTP host used for the query. Use this parameter when the host of the URL address is an IP address."))
o.default = ""
o.datatype = "hostname"
o.rempty = true
o.modalonly = true;
o.depends("type", "https")
// SPKI pin
o = s.taboption("advanced", form.Value, "spki_pin", _("TLS SPKI Pinning"),
_("Used to verify the validity of the TLS server, The value is Base64 encoded SPKI fingerprint, "
+ "leaving blank to indicate that the validity of TLS is not verified."))
o.default = ""
o.datatype = "string"
o.rempty = true
o.modalonly = true;
o.depends("type", "tls")
o.depends("type", "https")
// other args
o = s.taboption("advanced", form.Value, "addition_arg", _("Additional Server Args"),
_("Additional Args for upstream dns servers"))
o.default = ""
o.rempty = true
o.modalonly = true;
// Doman addresss;
s = m.section(form.TypedSection, "smartdns", _("Advanced Settings"), _("Advanced Settings"));
s.anonymous = true;
s.tab("domain-address", _("Domain Address"), _("Set Specific domain ip address."));
s.tab("blackip-list", _("IP Blacklist"), _("Set Specific ip blacklist."));
o = s.taboption("domain-address", form.TextValue, "address_conf",
"",
_("Specify an IP address to return for any host in the given domains, Queries in the domains are never "
+ "forwarded and always replied to with the specified IP address which may be IPv4 or IPv6."));
o.rows = 20;
o.cfgvalue = function (section_id) {
return fs.trimmed('/etc/smartdns/address.conf');
};
o.write = function (section_id, formvalue) {
return fs.write('/etc/smartdns/address.conf', formvalue.trim().replace(/\r\n/g, '\n') + '\n');
};
// IP Blacklist;
// blacklist;
o = s.taboption("blackip-list", form.TextValue, "blackip_ip_conf",
"", _("Configure IP blacklists that will be filtered from the results of specific DNS server."));
o.rows = 20;
o.cfgvalue = function (section_id) {
return fs.trimmed('/etc/smartdns/blacklist-ip.conf');
};
o.write = function (section_id, formvalue) {
return fs.write('/etc/smartdns/blacklist-ip.conf', formvalue.trim().replace(/\r\n/g, '\n') + '\n');
};
// Doman addresss;
s = m.section(form.TypedSection, "smartdns", _("Technical Support"),
_("If you like this software, please buy me a cup of coffee."));
s.anonymous = true;
o = s.option(form.Button, "web");
o.title = _("SmartDNS official website");
o.inputtitle = _("open website");
o.inputstyle = "apply";
o.onclick = function () {
window.open("https://pymumu.github.io/smartdns", '_blank');
};
o = s.option(form.Button, "Donate");
o.title = _("Donate to smartdns");
o.inputtitle = _("Donate");
o.inputstyle = "apply";
o.onclick = function () {
window.open("https://pymumu.github.io/smartdns/#donate", '_blank');
};
return m.render();
}
});

View File

@@ -1,5 +1,19 @@
#/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
CURR_DIR=$(cd $(dirname $0);pwd) CURR_DIR=$(cd $(dirname $0);pwd)
VER="`date +"1.%Y.%m.%d-%H%M"`" VER="`date +"1.%Y.%m.%d-%H%M"`"
@@ -12,58 +26,60 @@ showhelp()
echo "Options:" echo "Options:"
echo " -o output directory." echo " -o output directory."
echo " --arch archtecture." echo " --arch archtecture."
echo " --ver version." echo " --ver version."
echo " -h show this message." echo " -h show this message."
} }
build_tool() build_tool()
{ {
make -C $ROOT/tool/po2lmo -j make -C $ROOT/tool/po2lmo -j
PO2LMO="$ROOT/tool/po2lmo/src/po2lmo" PO2LMO="$ROOT/tool/po2lmo/src/po2lmo"
} }
clean_tool() clean_tool()
{ {
make -C $ROOT/tool/po2lmo clean make -C $ROOT/tool/po2lmo clean
} }
build() build()
{ {
ROOT=/tmp/luci-app-smartdns ROOT=/tmp/luci-app-smartdns
rm -fr $ROOT rm -fr $ROOT
mkdir -p $ROOT mkdir -p $ROOT
cp $CURR_DIR/* $ROOT/ -af cp $CURR_DIR/* $ROOT/ -af
cd $ROOT/ cd $ROOT/
build_tool build_tool
mkdir $ROOT/root/usr/lib/lua/ -p
cp $ROOT/files/luci $ROOT/root/usr/lib/lua/ -af mkdir $ROOT/root/usr/lib/lua/luci -p
mkdir $ROOT/root/usr/share/rpcd/acl.d/ -p
#Generate Language cp $ROOT/files/luci/i18n $ROOT/root/usr/lib/lua/luci/ -avf
$PO2LMO $ROOT/files/luci/i18n/smartdns.zh-cn.po $ROOT/root/usr/lib/lua/luci/i18n/smartdns.zh-cn.lmo
rm $ROOT/root/usr/lib/lua/luci/i18n/smartdns.zh-cn.po
cp $ROOT/files/etc $ROOT/root/ -af #Generate Language
INST_SIZE="`du -sb $ROOT/root/ | awk '{print $1}'`" $PO2LMO $ROOT/files/luci/i18n/smartdns.zh-cn.po $ROOT/root/usr/lib/lua/luci/i18n/smartdns.zh-cn.lmo
rm $ROOT/root/usr/lib/lua/luci/i18n/smartdns.zh-cn.po
sed -i "s/^Architecture.*/Architecture: $ARCH/g" $ROOT/control/control
sed -i "s/Version:.*/Version: $VER/" $ROOT/control/control
if [ ! -z "$INST_SIZE" ]; then cp $ROOT/files/root/* $ROOT/root/ -avf
echo "Installed-Size: $INST_SIZE" >> $ROOT/control/control INST_SIZE="`du -sb $ROOT/root/ | awk '{print $1}'`"
fi
sed -i "s/^Architecture.*/Architecture: all/g" $ROOT/control/control
sed -i "s/Version:.*/Version: $VER/" $ROOT/control/control
cd $ROOT/control if [ ! -z "$INST_SIZE" ]; then
chmod +x * echo "Installed-Size: $INST_SIZE" >> $ROOT/control/control
tar zcf ../control.tar.gz ./ fi
cd $ROOT
tar zcf $ROOT/data.tar.gz -C root . cd $ROOT/control
tar zcf $OUTPUTDIR/luci-app-smartdns.$VER.$FILEARCH.ipk control.tar.gz data.tar.gz debian-binary chmod +x *
tar zcf ../control.tar.gz ./
cd $ROOT
rm -fr $ROOT/ tar zcf $ROOT/data.tar.gz -C root .
tar zcf $OUTPUTDIR/luci-app-smartdns.$VER.$FILEARCH.ipk ./control.tar.gz ./data.tar.gz ./debian-binary
rm -fr $ROOT/
} }
main() main()
@@ -81,38 +97,38 @@ main()
--arch) --arch)
ARCH="$2" ARCH="$2"
shift 2;; shift 2;;
--filearch) --filearch)
FILEARCH="$2" FILEARCH="$2"
shift 2;; shift 2;;
--ver) --ver)
VER="$2" VER="$2"
shift 2;; shift 2;;
-o ) -o )
OUTPUTDIR="$2" OUTPUTDIR="$2"
shift 2;; shift 2;;
-h | --help ) -h | --help )
showhelp showhelp
return 0 return 0
shift ;; shift ;;
-- ) shift; break ;; -- ) shift; break ;;
* ) break ;; * ) break ;;
esac esac
done done
if [ -z "$ARCH" ]; then if [ -z "$ARCH" ]; then
echo "please input arch." echo "please input arch."
return 1; return 1;
fi fi
if [ -z "$FILEARCH" ]; then if [ -z "$FILEARCH" ]; then
FILEARCH=$ARCH FILEARCH=$ARCH
fi fi
if [ -z "$OUTPUTDIR" ]; then if [ -z "$OUTPUTDIR" ]; then
OUTPUTDIR=$CURR_DIR; OUTPUTDIR=$CURR_DIR;
fi fi
build build
} }
main $@ main $@

View File

@@ -1,34 +1,62 @@
#
# Copyright (c) 2018-2019 Nick Peng (pymumu@gmail.com)
# This is free software, licensed under the GNU General Public License v3.
#
include $(TOPDIR)/rules.mk include $(TOPDIR)/rules.mk
PKG_NAME:=smarttdns PKG_NAME:=smartdns
PKG_VERSION:=2018.7.1.1607 PKG_VERSION:=1.2019.28
PKG_RELEASE:=$(PKG_SOURCE_VERSION) PKG_RELEASE:=1
PKG_BUILD_DIR := $(BUILD_DIR)/$(PKG_NAME)-$(PKG_VERSION) PKG_SOURCE:=$(PKG_NAME)-$(PKG_VERSION).tar.xz
PKG_SOURCE_SUBDIR:=$(PKG_NAME)-$(PKG_VERSION)
PKG_SOURCE_PROTO:=git
PKG_SOURCE_URL:=https://www.github.com/pymumu/smartdns.git
PKG_SOURCE_VERSION:=982002e836e486fb4e360bc10e84e7e7197caf46
PKG_MIRROR_HASH:=ae889b521ecf114660ce5597af5c361f5970a7dcb75dadf61b938ba3d2baba54
PKG_MAINTAINER:=Nick Peng <pymumu@gmail.com>
PKG_LICENSE:=GPL-3.0-or-later
PKG_LICENSE_FILES:=LICENSE
PKG_BUILD_PARALLEL:=1
include $(INCLUDE_DIR)/package.mk include $(INCLUDE_DIR)/package.mk
MAKE_VARS += VER=$(PKG_VERSION)
MAKE_PATH:=src
define Package/smartdns define Package/smartdns
SECTION:=net SECTION:=net
CATEGORY:=Network CATEGORY:=Network
TITLE:=smartdns TITLE:=smartdns server
URL:=https://github.com/pymumu/smartdns DEPENDS:=+libpthread +libopenssl
MAINTAINER:=Nick Peng <pymumu@gmail.com> URL:=https://www.github.com/pymumu/smartdns/
DEPENDS:=
endef endef
define Package/smartdns/description define Package/smartdns/description
smartdns server, find fastest ip. SmartDNS is a local DNS server which accepts DNS query requests from local network clients,
get DNS query results from multiple upstream DNS servers concurrently, and returns the fastest IP to clients.
Unlike dnsmasq's all-servers, smartdns returns the fastest IP.
endef endef
define Build/Prepare define Package/smartdns/conffiles
mkdir -p $(PKG_BUILD_DIR) /etc/config/smartdns
cp -r ../* $(PKG_BUILD_DIR) /etc/smartdns/address.conf
/etc/smartdns/blacklist-ip.conf
/etc/smartdns/custom.conf
endef endef
define Package/smartdns/install define Package/smartdns/install
$(INSTALL_DIR) $(1)/usr/sbin $(INSTALL_DIR) $(1)/usr/sbin $(1)/etc/config $(1)/etc/init.d $(1)/etc/smartdns
$(INSTALL_BIN) $(PKG_BUILD_DIR)/smartdns $(1)/usr/sbin $(INSTALL_BIN) $(PKG_BUILD_DIR)/src/smartdns $(1)/usr/sbin/smartdns
$(INSTALL_BIN) $(PKG_BUILD_DIR)/package/openwrt/files/etc/init.d/smartdns $(1)/etc/init.d/smartdns
$(INSTALL_CONF) $(PKG_BUILD_DIR)/package/openwrt/address.conf $(1)/etc/smartdns/address.conf
$(INSTALL_CONF) $(PKG_BUILD_DIR)/package/openwrt/blacklist-ip.conf $(1)/etc/smartdns/blacklist-ip.conf
$(INSTALL_CONF) $(PKG_BUILD_DIR)/package/openwrt/custom.conf $(1)/etc/smartdns/custom.conf
$(INSTALL_CONF) $(PKG_BUILD_DIR)/package/openwrt/files/etc/config/smartdns $(1)/etc/config/smartdns
endef endef
$(eval $(call BuildPackage, smartdns)) $(eval $(call BuildPackage,smartdns))

View File

@@ -1,16 +1,16 @@
# Add domains which you want to force to an IP address here. # Add domains which you want to force to an IP address here.
# The example below send any host in example.com to a local webserver. # The example below send any host in example.com to a local webserver.
# address /domain/[ip|-|-4|-6|#|#4|#6] # address /domain/[ip|-|-4|-6|#|#4|#6]
# address /www.example.com/1.2.3.4, return ip 1.2.3.4 to client # address /www.example.com/1.2.3.4, return ip 1.2.3.4 to client
# address /www.example.com/-, ignore address, query from upstream, suffix 4, for ipv4, 6 for ipv6, none for all # address /www.example.com/-, ignore address, query from upstream, suffix 4, for ipv4, 6 for ipv6, none for all
# address /www.example.com/#, return SOA to client, suffix 4, for ipv4, 6 for ipv6, none for all # address /www.example.com/#, return SOA to client, suffix 4, for ipv4, 6 for ipv6, none for all
# specific ipset to domain # specific ipset to domain
# ipset /domain/[ipset|-] # ipset /domain/[ipset|-]
# ipset /www.example.com/block, set ipset with ipset name of block # ipset /www.example.com/block, set ipset with ipset name of block
# ipset /www.example.com/-, ignore this domain # ipset /www.example.com/-, ignore this domain
# specific nameserver to domain # specific nameserver to domain
# nameserver /domain/[group|-] # nameserver /domain/[group|-]
# nameserver /www.example.com/office, Set the domain name to use the appropriate server group. # nameserver /www.example.com/office, Set the domain name to use the appropriate server group.
# nameserver /www.example.com/-, ignore this domain # nameserver /www.example.com/-, ignore this domain

View File

@@ -1,4 +1,4 @@
# Add IP blacklist which you want to filtering from some DNS server here. # Add IP blacklist which you want to filtering from some DNS server here.
# The example below filtering ip from the result of DNS server which is configured with -blacklist-ip. # The example below filtering ip from the result of DNS server which is configured with -blacklist-ip.
# blacklist-ip [ip/subnet] # blacklist-ip [ip/subnet]
# blacklist-ip 254.0.0.1/16 # blacklist-ip 254.0.0.1/16

View File

@@ -1,4 +1,19 @@
#!/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
chmod +x /usr/sbin/smartdns chmod +x /usr/sbin/smartdns
chmod +x /etc/init.d/smartdns chmod +x /etc/init.d/smartdns

View File

@@ -1,4 +1,20 @@
#!/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
. ${IPKG_INSTROOT}/lib/functions.sh . ${IPKG_INSTROOT}/lib/functions.sh
default_prerm $0 $@ default_prerm $0 $@
rm /var/etc/smartdns.conf -f rm /var/etc/smartdns.conf -f

View File

@@ -1,14 +1,14 @@
# Add custom settings here. # Add custom settings here.
# set log level # set log level
# log-level [level], level=fatal, error, warn, notice, info, debug # log-level [level], level=fatal, error, warn, notice, info, debug
# log-level error # log-level error
# log-size k,m,g # log-size k,m,g
# log-size 128k # log-size 128k
# log-file /var/log/smartdns.log # log-file /var/log/smartdns.log
# log-num 2 # log-num 2
# List of hosts that supply bogus NX domain results # List of hosts that supply bogus NX domain results
# bogus-nxdomain [ip/subnet] # bogus-nxdomain [ip/subnet]

View File

@@ -1,5 +1,19 @@
#!/bin/sh /etc/rc.common #!/bin/sh /etc/rc.common
# Copyright (C) 2018 Nick Peng (pymumu@gmail.com) #
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
START=99 START=99
NAME=smartdns NAME=smartdns
@@ -8,7 +22,6 @@ SERVICE_USE_PID=1
SERVICE_WRITE_PID=1 SERVICE_WRITE_PID=1
SERVICE_DAEMONIZE=1 SERVICE_DAEMONIZE=1
SERVICE_PID_FILE="/var/run/smartdns.pid" SERVICE_PID_FILE="/var/run/smartdns.pid"
BASECONFIGFILE="/etc/smartdns/smartdns.conf"
SMARTDNS_CONF_DIR="/etc/smartdns" SMARTDNS_CONF_DIR="/etc/smartdns"
SMARTDNS_VAR_CONF_DIR="/var/etc/smartdns" SMARTDNS_VAR_CONF_DIR="/var/etc/smartdns"
SMARTDNS_CONF="$SMARTDNS_VAR_CONF_DIR/smartdns.conf" SMARTDNS_CONF="$SMARTDNS_VAR_CONF_DIR/smartdns.conf"
@@ -23,22 +36,18 @@ set_forward_dnsmasq()
{ {
local PORT="$1" local PORT="$1"
addr="127.0.0.1#$PORT" addr="127.0.0.1#$PORT"
OLD_SERVER="`uci get dhcp.@dnsmasq[0].server 2>/dev/null`" OLD_SERVER="$(uci get dhcp.@dnsmasq[0].server 2>/dev/null)"
echo $OLD_SERVER | grep "^$addr" >/dev/null 2>&1 if echo "$OLD_SERVER" | grep "^$addr" >/dev/null 2>&1; then
if [ $? -eq 0 ]; then
return return
fi fi
uci delete dhcp.@dnsmasq[0].server 2>/dev/null uci delete dhcp.@dnsmasq[0].server 2>/dev/null
uci add_list dhcp.@dnsmasq[0].server=$addr uci add_list dhcp.@dnsmasq[0].server="$addr"
for server in $OLD_SERVER; do for server in $OLD_SERVER; do
if [ "$server" = "$addr" ]; then [ "$server" = "$addr" ] && continue
continue uci add_list dhcp.@dnsmasq[0].server="$server"
fi
uci add_list dhcp.@dnsmasq[0].server=$server
done done
uci delete dhcp.@dnsmasq[0].resolvfile 2>/dev/null
uci set dhcp.@dnsmasq[0].noresolv=1 uci set dhcp.@dnsmasq[0].noresolv=1
uci commit dhcp uci commit dhcp
/etc/init.d/dnsmasq restart /etc/init.d/dnsmasq restart
} }
@@ -46,18 +55,16 @@ stop_forward_dnsmasq()
{ {
local OLD_PORT="$1" local OLD_PORT="$1"
addr="127.0.0.1#$OLD_PORT" addr="127.0.0.1#$OLD_PORT"
OLD_SERVER="`uci get dhcp.@dnsmasq[0].server 2>/dev/null`" OLD_SERVER="$(uci get dhcp.@dnsmasq[0].server 2>/dev/null)"
echo $OLD_SERVER | grep "^$addr" >/dev/null 2>&1 if ! echo "$OLD_SERVER" | grep "^$addr" >/dev/null 2>&1; then
if [ $? -ne 0 ]; then
return return
fi fi
uci del_list dhcp.@dnsmasq[0].server=$addr 2>/dev/null uci del_list dhcp.@dnsmasq[0].server="$addr" 2>/dev/null
addrlist="`uci get dhcp.@dnsmasq[0].server 2>/dev/null`" addrlist="$(uci get dhcp.@dnsmasq[0].server 2>/dev/null)"
if [ -z "$addrlist" ] ; then [ -z "$addrlist" ] && {
uci set dhcp.@dnsmasq[0].resolvfile=/tmp/resolv.conf.auto 2>/dev/null
uci delete dhcp.@dnsmasq[0].noresolv 2>/dev/null uci delete dhcp.@dnsmasq[0].noresolv 2>/dev/null
fi }
uci commit dhcp uci commit dhcp
/etc/init.d/dnsmasq restart /etc/init.d/dnsmasq restart
} }
@@ -67,26 +74,24 @@ set_iptable()
local ipv6_server=$1 local ipv6_server=$1
local tcp_server=$2 local tcp_server=$2
IPS="`ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F : '{print $2}'`" IPS="$(ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F : '{print $2}')"
for IP in $IPS for IP in $IPS
do do
if [ "$tcp_server" == "1" ]; then if [ "$tcp_server" = "1" ]; then
iptables -t nat -A PREROUTING -p tcp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT >/dev/null 2>&1 iptables -t nat -A PREROUTING -p tcp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" >/dev/null 2>&1
fi fi
iptables -t nat -A PREROUTING -p udp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT >/dev/null 2>&1 iptables -t nat -A PREROUTING -p udp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" >/dev/null 2>&1
done done
if [ "$ipv6_server" == 0 ]; then [ "$ipv6_server" = 0 ] && return
return
fi
IPS="`ifconfig | grep "inet6 addr" | grep -v " fe80::" | grep -v " ::1" | grep "Global" | awk '{print $3}'`" IPS="$(ifconfig | grep "inet6 addr" | grep -v " fe80::" | grep -v " ::1" | grep "Global" | awk '{print $3}')"
for IP in $IPS for IP in $IPS
do do
if [ "$tcp_server" == "1" ]; then if [ "$tcp_server" = "1" ]; then
ip6tables -t nat -A PREROUTING -p tcp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT >/dev/null 2>&1 ip6tables -t nat -A PREROUTING -p tcp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" >/dev/null 2>&1
fi fi
ip6tables -t nat -A PREROUTING -p udp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT >/dev/null 2>&1 ip6tables -t nat -A PREROUTING -p udp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" >/dev/null 2>&1
done done
} }
@@ -95,24 +100,22 @@ clear_iptable()
{ {
local OLD_PORT="$1" local OLD_PORT="$1"
local ipv6_server=$2 local ipv6_server=$2
IPS="`ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F : '{print $2}'`" IPS="$(ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F : '{print $2}')"
for IP in $IPS for IP in $IPS
do do
iptables -t nat -D PREROUTING -p udp -d $IP --dport 53 -j REDIRECT --to-ports $OLD_PORT >/dev/null 2>&1 iptables -t nat -D PREROUTING -p udp -d "$IP" --dport 53 -j REDIRECT --to-ports "$OLD_PORT" >/dev/null 2>&1
iptables -t nat -D PREROUTING -p tcp -d $IP --dport 53 -j REDIRECT --to-ports $OLD_PORT >/dev/null 2>&1 iptables -t nat -D PREROUTING -p tcp -d "$IP" --dport 53 -j REDIRECT --to-ports "$OLD_PORT" >/dev/null 2>&1
done done
if [ "$ipv6_server" == 0 ]; then [ "$ipv6_server" = 0 ] && return
return
fi
IPS="`ifconfig | grep "inet6 addr" | grep -v " fe80::" | grep -v " ::1" | grep "Global" | awk '{print $3}'`" IPS="$(ifconfig | grep "inet6 addr" | grep -v " fe80::" | grep -v " ::1" | grep "Global" | awk '{print $3}')"
for IP in $IPS for IP in $IPS
do do
ip6tables -t nat -D PREROUTING -p udp -d $IP --dport 53 -j REDIRECT --to-ports $OLD_PORT >/dev/null 2>&1 ip6tables -t nat -D PREROUTING -p udp -d "$IP" --dport 53 -j REDIRECT --to-ports "$OLD_PORT" >/dev/null 2>&1
ip6tables -t nat -D PREROUTING -p tcp -d $IP --dport 53 -j REDIRECT --to-ports $OLD_PORT >/dev/null 2>&1 ip6tables -t nat -D PREROUTING -p tcp -d "$IP" --dport 53 -j REDIRECT --to-ports "$OLD_PORT" >/dev/null 2>&1
done done
} }
service_triggers() { service_triggers() {
@@ -120,7 +123,6 @@ service_triggers() {
procd_add_reload_trigger smartdns procd_add_reload_trigger smartdns
} }
conf_append() conf_append()
{ {
echo "$1 $2" >> $SMARTDNS_CONF_TMP echo "$1 $2" >> $SMARTDNS_CONF_TMP
@@ -130,23 +132,16 @@ get_tz()
{ {
SET_TZ="" SET_TZ=""
if [ -e "/etc/localtime" ]; then [ -e "/etc/localtime" ] && return
return
fi
for tzfile in /etc/TZ /var/etc/TZ for tzfile in /etc/TZ /var/etc/TZ
do do
if [ ! -e "$tzfile" ]; then [ -e "$tzfile" ] || continue
continue tz="$(cat $tzfile 2>/dev/null)"
fi
tz="`cat $tzfile 2>/dev/null`"
done done
if [ -z "$tz" ]; then [ -z "$tz" ] && return
return
fi
SET_TZ=$tz SET_TZ=$tz
} }
@@ -156,22 +151,21 @@ load_server()
local ADDITIONAL_ARGS="" local ADDITIONAL_ARGS=""
local DNS_ADDRESS="" local DNS_ADDRESS=""
config_get_bool "enabled" "$section" "enabled" "1" config_get_bool enabled "$section" "enabled" "1"
config_get "port" "$section" "port" "" config_get port "$section" "port" ""
config_get "type" "$section" "type" "udp" config_get type "$section" "type" "udp"
config_get "ip" "$section" "ip" "" config_get ip "$section" "ip" ""
config_get "tls_host_verify" "$section" "tls_host_verify" "" config_get tls_host_verify "$section" "tls_host_verify" ""
config_get "host_name" "$section" "host_name" "" config_get no_check_certificate "$section" "no_check_certificate" "0"
config_get "http_host" "$section" "http_host" "" config_get host_name "$section" "host_name" ""
config_get "server_group" "$section" "server_group" "" config_get http_host "$section" "http_host" ""
config_get "blacklist_ip" "$section" "blacklist_ip" "0" config_get server_group "$section" "server_group" ""
config_get "check_edns" "$section" "check_edns" "0" config_get blacklist_ip "$section" "blacklist_ip" "0"
config_get "spki_pin" "$section" "spki_pin" "" config_get check_edns "$section" "check_edns" "0"
config_get "addition_arg" "$section" "addition_arg" "" config_get spki_pin "$section" "spki_pin" ""
config_get addition_arg "$section" "addition_arg" ""
if [ "$enabled" = "0" ]; then [ "$enabled" = "0" ] && return
return
fi
if [ -z "$ip" ] || [ -z "$type" ]; then if [ -z "$ip" ] || [ -z "$type" ]; then
return return
@@ -186,107 +180,71 @@ load_server()
SERVER="server-https" SERVER="server-https"
fi fi
if [ ! -z "`echo $ip | grep ":" | grep -v "https://"`" ]; then if echo "$ip" | grep ":" | grep -q -v "https://" >/dev/null 2>&1; then
if [ -z "`echo $ip | grep "\["`" ]; then if ! echo "$ip" | grep -q "\\[" >/dev/null 2>&1; then
ip="[$ip]" ip="[$ip]"
fi fi
fi fi
if [ ! -z "$tls_host_verify" ]; then [ -z "$tls_host_verify" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -tls-host-verify $tls_host_verify"
ADDITIONAL_ARGS="$ADDITIONAL_ARGS -tls-host-verify $tls_host_verify" [ "$no_check_certificate" = "0" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -no-check-certificate"
fi [ -z "$host_name" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -host-name $host_name"
[ -z "$http_host" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -http-host $http_host"
[ -z "$server_group" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -group $server_group"
[ "$blacklist_ip" = "0" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -blacklist-ip"
[ "$check_edns" = "0" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -check-edns"
[ -z "$spki_pin" ] || ADDITIONAL_ARGS="$ADDITIONAL_ARGS -spki-pin $spki_pin"
if [ ! -z "$host_name" ]; then if [ -z "$port" ]; then
ADDITIONAL_ARGS="$ADDITIONAL_ARGS -host-name $host_name" DNS_ADDRESS="$ip"
fi
if [ ! -z "$http_host" ]; then
ADDITIONAL_ARGS="$ADDITIONAL_ARGS -http-host $http_host"
fi
if [ ! -z "$server_group" ]; then
ADDITIONAL_ARGS="$ADDITIONAL_ARGS -group $server_group"
fi
if [ "$blacklist_ip" != "0" ]; then
ADDITIONAL_ARGS="$ADDITIONAL_ARGS -blacklist-ip"
fi
if [ "$check_edns" != "0" ]; then
ADDITIONAL_ARGS="$ADDITIONAL_ARGS -check-edns"
fi
if [ ! -z "$spki_pin" ]; then
ADDITIONAL_ARGS="$ADDITIONAL_ARGS -spki-pin $spki_pin"
fi
if [ ! -z "$port" ]; then
DNS_ADDRESS="$ip:$port"
else else
DNS_ADDRESS="$ip" DNS_ADDRESS="$ip:$port"
fi fi
if [ "$type" = "https" ]; then [ "$type" = "https" ] && DNS_ADDRESS="$ip"
DNS_ADDRESS="$ip"
fi
conf_append "$SERVER" "$DNS_ADDRESS $ADDITIONAL_ARGS $addition_arg" conf_append "$SERVER" "$DNS_ADDRESS $ADDITIONAL_ARGS $addition_arg"
} }
load_second_server() load_second_server()
{ {
local section="$1" local section="$1"
local ARGS="" local ARGS=""
local ADDR="" local ADDR=""
config_get_bool "seconddns_enabled" "$section" "seconddns_enabled" "0" config_get_bool seconddns_enabled "$section" "seconddns_enabled" "0"
if [ "$seconddns_enabled" = "0" ]; then [ "$seconddns_enabled" = "0" ] && return
return
fi
config_get "seconddns_port" "$section" "seconddns_port" "7053" config_get seconddns_port "$section" "seconddns_port" "6553"
config_get_bool "seconddns_no_speed_check" "$section" "seconddns_no_speed_check" "0" config_get_bool seconddns_no_speed_check "$section" "seconddns_no_speed_check" "0"
if [ "$seconddns_no_speed_check" = "1" ]; then [ "$seconddns_no_speed_check" = "1" ] && ARGS="$ARGS -no-speed-check"
ARGS="$ARGS -no-speed-check"
fi
config_get "seconddns_server_group" "$section" "seconddns_server_group" "" config_get seconddns_server_group "$section" "seconddns_server_group" ""
if [ ! -z "$seconddns_server_group" ]; then [ -z "$seconddns_server_group" ] || ARGS="$ARGS -group $seconddns_server_group"
ARGS="$ARGS -group $seconddns_server_group"
fi
config_get_bool "seconddns_no_rule_addr" "$section" "seconddns_no_rule_addr" "0" config_get_bool seconddns_no_rule_addr "$section" "seconddns_no_rule_addr" "0"
if [ "$seconddns_no_rule_addr" = "1" ]; then [ "$seconddns_no_rule_addr" = "1" ] && ARGS="$ARGS -no-rule-addr"
ARGS="$ARGS -no-rule-addr"
fi
config_get_bool "seconddns_no_rule_nameserver" "$section" "seconddns_no_rule_nameserver" "0" config_get_bool seconddns_no_rule_nameserver "$section" "seconddns_no_rule_nameserver" "0"
if [ "$seconddns_no_rule_nameserver" = "1" ]; then [ "$seconddns_no_rule_nameserver" = "1" ] && ARGS="$ARGS -no-rule-nameserver"
ARGS="$ARGS -no-rule-nameserver"
fi
config_get_bool "seconddns_no_rule_ipset" "$section" "seconddns_no_rule_ipset" "0" config_get_bool seconddns_no_rule_ipset "$section" "seconddns_no_rule_ipset" "0"
if [ "$seconddns_no_rule_ipset" = "1" ]; then [ "$seconddns_no_rule_ipset" = "1" ] && ARGS="$ARGS -no-rule-ipset"
ARGS="$ARGS -no-rule-ipset"
fi
config_get_bool "seconddns_no_rule_soa" "$section" "seconddns_no_rule_soa" "0" config_get_bool seconddns_no_rule_soa "$section" "seconddns_no_rule_soa" "0"
if [ "$seconddns_no_rule_soa" = "1" ]; then [ "$seconddns_no_rule_soa" = "1" ] && ARGS="$ARGS -no-rule-soa"
ARGS="$ARGS -no-rule-soa"
fi
config_get_bool "seconddns_no_dualstack_selection" "$section" "seconddns_no_dualstack_selection" "0" config_get_bool seconddns_no_dualstack_selection "$section" "seconddns_no_dualstack_selection" "0"
if [ "$seconddns_no_dualstack_selection" = "1" ]; then [ "$seconddns_no_dualstack_selection" = "1" ] && ARGS="$ARGS -no-dualstack-selection"
ARGS="$ARGS -no-dualstack-selection"
fi
config_get_bool "seconddns_no_cache" "$section" "seconddns_no_cache" "0" config_get_bool seconddns_no_cache "$section" "seconddns_no_cache" "0"
if [ "$seconddns_no_cache" = "1" ]; then [ "$seconddns_no_cache" = "1" ] && ARGS="$ARGS -no-cache"
ARGS="$ARGS -no-cache"
fi
config_get "ipv6_server" "$section" "ipv6_server" "1" config_get_bool force_aaaa_soa "$section" "force_aaaa_soa" "0"
[ "$force_aaaa_soa" = "1" ] && ARGS="$ARGS -force-aaaa-soa"
config_get ipv6_server "$section" "ipv6_server" "1"
if [ "$ipv6_server" = "1" ]; then if [ "$ipv6_server" = "1" ]; then
ADDR="[::]" ADDR="[::]"
else else
@@ -295,12 +253,10 @@ load_second_server()
conf_append "bind" "$ADDR:$seconddns_port $ARGS" conf_append "bind" "$ADDR:$seconddns_port $ARGS"
config_get_bool "seconddns_tcp_server" "$section" "seconddns_tcp_server" "1" config_get_bool "seconddns_tcp_server" "$section" "seconddns_tcp_server" "1"
if [ "$seconddns_tcp_server" = "1" ]; then [ "$seconddns_tcp_server" = "1" ] && conf_append "bind-tcp" "$ADDR:$seconddns_port $ARGS"
conf_append "bind-tcp" "$ADDR:$seconddns_port $ARGS"
fi
} }
load_service() load_service()
{ {
local section="$1" local section="$1"
args="" args=""
@@ -308,102 +264,77 @@ load_service()
mkdir -p $SMARTDNS_VAR_CONF_DIR mkdir -p $SMARTDNS_VAR_CONF_DIR
rm -f $SMARTDNS_CONF_TMP rm -f $SMARTDNS_CONF_TMP
config_get_bool "enabled" "$section" "enabled" '0' config_get_bool enabled "$section" "enabled" '0'
config_get "server_name" "$section" "server_name" "" config_get server_name "$section" "server_name" ""
if [ ! -z "$server_name" ]; then [ -z "$server_name" ] || conf_append "server-name" "$server_name"
conf_append "server-name" "$server_name"
fi
config_get "coredump" "$section" "coredump" "0" config_get coredump "$section" "coredump" "0"
if [ "$coredump" = "1" ]; then [ "$coredump" = "1" ] && COREDUMP="1"
COREDUMP="1"
fi
config_get "port" "$section" "port" "6053" config_get port "$section" "port" "6053"
config_get "ipv6_server" "$section" "ipv6_server" "1" config_get ipv6_server "$section" "ipv6_server" "1"
config_get "tcp_server" "$section" "tcp_server" "1" config_get tcp_server "$section" "tcp_server" "1"
if [ "$ipv6_server" = "1" ]; then if [ "$ipv6_server" = "1" ]; then
conf_append "bind" "[::]:$port" conf_append "bind" "[::]:$port"
else else
conf_append "bind" ":$port" conf_append "bind" ":$port"
fi fi
if [ "$tcp_server" = "1" ]; then [ "$tcp_server" = "1" ] && {
if [ "$ipv6_server" = "1" ]; then if [ "$ipv6_server" = "1" ]; then
conf_append "bind-tcp" "[::]:$port" conf_append "bind-tcp" "[::]:$port"
else else
conf_append "bind-tcp" ":$port" conf_append "bind-tcp" ":$port"
fi fi
fi }
config_get "dualstack_ip_selection" "$section" "dualstack_ip_selection" "0" config_get dualstack_ip_selection "$section" "dualstack_ip_selection" "0"
if [ "$dualstack_ip_selection" = "1" ]; then [ "$dualstack_ip_selection" = "1" ] && conf_append "dualstack-ip-selection" "yes"
conf_append "dualstack-ip-selection" "yes"
fi
config_get "prefetch_domain" "$section" "prefetch_domain" "0" config_get prefetch_domain "$section" "prefetch_domain" "0"
if [ "$prefetch_domain" = "1" ]; then [ "$prefetch_domain" = "1" ] && conf_append "prefetch-domain" "yes"
conf_append "prefetch-domain" "yes"
fi config_get serve_expired "$section" "serve_expired" "0"
[ "$serve_expired" = "1" ] && conf_append "serve-expired" "yes"
SMARTDNS_PORT="$port" SMARTDNS_PORT="$port"
config_get "cache_size" "$section" "cache_size" "" config_get cache_size "$section" "cache_size" ""
if [ ! -z "$cache_size" ]; then [ -z "$cache_size" ] || conf_append "cache-size" "$cache_size"
conf_append "cache-size" "$cache_size"
fi
config_get "rr_ttl" "$section" "rr_ttl" "" config_get rr_ttl "$section" "rr_ttl" ""
if [ ! -z "$rr_ttl" ]; then [ -z "$rr_ttl" ] || conf_append "rr-ttl" "$rr_ttl"
conf_append "rr-ttl" "$rr_ttl"
fi
config_get "rr_ttl_min" "$section" "rr_ttl_min" "" config_get rr_ttl_min "$section" "rr_ttl_min" ""
if [ ! -z "$rr_ttl_min" ]; then [ -z "$rr_ttl_min" ] || conf_append "rr-ttl-min" "$rr_ttl_min"
conf_append "rr-ttl-min" "$rr_ttl_min"
fi
config_get "rr_ttl_max" "$section" "rr_ttl_max" "" config_get rr_ttl_max "$section" "rr_ttl_max" ""
if [ ! -z "$rr_ttl_max" ]; then [ -z "$rr_ttl_max" ] || conf_append "rr-ttl-max" "$rr_ttl_max"
conf_append "rr-ttl-max" "$rr_ttl_max"
fi
config_get "log_size" "$section" "log_size" "64K" config_get log_size "$section" "log_size" "64K"
if [ ! -z "$log_size" ]; then [ -z "$log_size" ] || conf_append "log-size" "$log_size"
conf_append "log-size" "$log_size"
fi
config_get "log_num" "$section" "log_num" "1" config_get log_num "$section" "log_num" "1"
if [ ! -z "$log_num" ]; then [ -z "$log_num" ] || conf_append "log-num" "$log_num"
conf_append "log-num" "$log_num"
fi
config_get "log_level" "$section" "log_level" "error" config_get log_level "$section" "log_level" "error"
if [ ! -z "$log_level" ]; then [ -z "$log_level" ]|| conf_append "log-level" "$log_level"
conf_append "log-level" "$log_level"
fi
config_get "log_file" "$section" "log_file" "" config_get log_file "$section" "log_file" ""
if [ ! -z "$log_file" ]; then [ -z "$log_file" ] || conf_append "log-file" "$log_file"
conf_append "log-file" "$log_file"
fi
config_get "redirect" "$section" "redirect" "none" config_get redirect "$section" "redirect" "none"
config_get "old_redirect" "$section" "old_redirect" "none" config_get old_redirect "$section" "old_redirect" "none"
config_get "old_port" "$section" "old_port" "0" config_get old_port "$section" "old_port" "0"
config_get "old_enabled" "$section" "old_enabled" "0" config_get old_enabled "$section" "old_enabled" "0"
if [ "$old_redirect" != "$redirect" ] || [ "$old_port" != "$SMARTDNS_PORT" ] || [ "$old_enabled" = "1" -a "$enabled" = "0" ]; then if [ "$old_redirect" != "$redirect" ] || [ "$old_port" != "$SMARTDNS_PORT" ] || [ "$old_enabled" = "1" -a "$enabled" = "0" ]; then
if [ "$old_redirect" != "none" ]; then [ "$old_redirect" = "none" ] || {
if [ "$old_port" != "0" ]; then [ "$old_port" = "0" ] || clear_iptable "$old_port" "$ipv6_server"
clear_iptable "$old_port" "$ipv6_server" [ "$old_redirect" = "dnsmasq-upstream" ] && stop_forward_dnsmasq "$old_port"
fi }
if [ "$old_redirect" == "dnsmasq-upstream" ]; then
stop_forward_dnsmasq "$old_port"
fi
fi
fi fi
uci delete smartdns.@smartdns[0].old_redirect 2>/dev/null uci delete smartdns.@smartdns[0].old_redirect 2>/dev/null
uci delete smartdns.@smartdns[0].old_port 2>/dev/null uci delete smartdns.@smartdns[0].old_port 2>/dev/null
uci delete smartdns.@smartdns[0].old_enabled 2>/dev/null uci delete smartdns.@smartdns[0].old_enabled 2>/dev/null
@@ -424,31 +355,29 @@ load_service()
config_foreach load_server "server" config_foreach load_server "server"
echo "conf-file $ADDRESS_CONF" >> $SMARTDNS_CONF_TMP {
echo "conf-file $BLACKLIST_IP_CONF" >> $SMARTDNS_CONF_TMP echo "conf-file $ADDRESS_CONF"
echo "conf-file $CUSTOM_CONF" >> $SMARTDNS_CONF_TMP echo "conf-file $BLACKLIST_IP_CONF"
echo "conf-file $CUSTOM_CONF"
} >> $SMARTDNS_CONF_TMP
mv $SMARTDNS_CONF_TMP $SMARTDNS_CONF mv $SMARTDNS_CONF_TMP $SMARTDNS_CONF
procd_open_instance "smartdns" procd_open_instance "smartdns"
if [ "$COREDUMP" = "1" ]; then [ "$COREDUMP" = "1" ] && {
args="$args -S" args="$args -S"
procd_set_param limits core="unlimited" procd_set_param limits core="unlimited"
fi }
get_tz get_tz
if [ ! -z "$SET_TZ" ]; then [ -z "$SET_TZ" ] || procd_set_param env TZ="$SET_TZ"
procd_set_param env TZ="$SET_TZ"
fi
procd_set_param command /usr/sbin/smartdns -f -c $SMARTDNS_CONF $args procd_set_param command /usr/sbin/smartdns -f -c $SMARTDNS_CONF $args
if [ "$RESPAWN" = "1" ]; then [ "$RESPAWN" = "1" ] && procd_set_param respawn ${respawn_threshold:-3600} ${respawn_timeout:-5} ${respawn_retry:-5}
procd_set_param respawn ${respawn_threshold:-3600} ${respawn_timeout:-5} ${respawn_retry:-5}
fi
procd_set_param file "$SMARTDNS_CONF" procd_set_param file "$SMARTDNS_CONF"
procd_close_instance procd_close_instance
} }
start_service() start_service()
{ {
config_load "smartdns" config_load "smartdns"
config_foreach load_service "smartdns" config_foreach load_service "smartdns"

View File

@@ -1,5 +1,19 @@
#/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
CURR_DIR=$(cd $(dirname $0);pwd) CURR_DIR=$(cd $(dirname $0);pwd)
VER="`date +"1.%Y.%m.%d-%H%M"`" VER="`date +"1.%Y.%m.%d-%H%M"`"
@@ -16,47 +30,52 @@ showhelp()
echo "Options:" echo "Options:"
echo " -o output directory." echo " -o output directory."
echo " --arch archtecture." echo " --arch archtecture."
echo " --ver version." echo " --ver version."
echo " -h show this message." echo " -h show this message."
} }
build() build()
{ {
ROOT=/tmp/smartdns-openwrt ROOT=/tmp/smartdns-openwrt
rm -fr $ROOT rm -fr $ROOT
mkdir -p $ROOT mkdir -p $ROOT
cp $CURR_DIR/* $ROOT/ -af cp $CURR_DIR/* $ROOT/ -af
cd $ROOT/ cd $ROOT/
mkdir $ROOT/root/usr/sbin -p mkdir $ROOT/root/usr/sbin -p
mkdir $ROOT/root/etc/init.d -p mkdir $ROOT/root/etc/init.d -p
mkdir $ROOT/root/etc/smartdns/ -p mkdir $ROOT/root/etc/smartdns/ -p
cp $SMARTDNS_CONF $ROOT/root/etc/smartdns/ cp $SMARTDNS_CONF $ROOT/root/etc/smartdns/
cp $ADDRESS_CONF $ROOT/root/etc/smartdns/ cp $ADDRESS_CONF $ROOT/root/etc/smartdns/
cp $BLACKLIST_IP_CONF $ROOT/root/etc/smartdns/ cp $BLACKLIST_IP_CONF $ROOT/root/etc/smartdns/
cp $CUSTOM_CONF $ROOT/root/etc/smartdns/ cp $CUSTOM_CONF $ROOT/root/etc/smartdns/
cp $CURR_DIR/files/etc $ROOT/root/ -af cp $CURR_DIR/files/etc $ROOT/root/ -af
cp $SMARTDNS_BIN $ROOT/root/usr/sbin cp $SMARTDNS_BIN $ROOT/root/usr/sbin
if [ $? -ne 0 ]; then
echo "copy smartdns file failed."
rm -fr $ROOT/
return 1
fi
chmod +x $ROOT/root/etc/init.d/smartdns chmod +x $ROOT/root/etc/init.d/smartdns
INST_SIZE="`du -sb $ROOT/root/ | awk '{print $1}'`" INST_SIZE="`du -sb $ROOT/root/ | awk '{print $1}'`"
sed -i "s/^Architecture.*/Architecture: $ARCH/g" $ROOT/control/control sed -i "s/^Architecture.*/Architecture: $ARCH/g" $ROOT/control/control
sed -i "s/Version:.*/Version: $VER/" $ROOT/control/control sed -i "s/Version:.*/Version: $VER/" $ROOT/control/control
sed -i "s/^\(bind .*\):53/\1:6053/g" $ROOT/root/etc/smartdns/smartdns.conf sed -i "s/^\(bind .*\):53/\1:6053/g" $ROOT/root/etc/smartdns/smartdns.conf
if [ ! -z "$INST_SIZE" ]; then if [ ! -z "$INST_SIZE" ]; then
echo "Installed-Size: $INST_SIZE" >> $ROOT/control/control echo "Installed-Size: $INST_SIZE" >> $ROOT/control/control
fi fi
cd $ROOT/control cd $ROOT/control
chmod +x * chmod +x *
tar zcf ../control.tar.gz --owner=0 --group=0 ./ tar zcf ../control.tar.gz --owner=0 --group=0 ./
cd $ROOT cd $ROOT
tar zcf $ROOT/data.tar.gz -C root --owner=0 --group=0 . tar zcf $ROOT/data.tar.gz -C root --owner=0 --group=0 .
tar zcf $OUTPUTDIR/smartdns.$VER.$FILEARCH.ipk --owner=0 --group=0 control.tar.gz data.tar.gz debian-binary tar zcf $OUTPUTDIR/smartdns.$VER.$FILEARCH.ipk --owner=0 --group=0 control.tar.gz data.tar.gz debian-binary
rm -fr $ROOT/ rm -fr $ROOT/
} }
main() main()
@@ -74,38 +93,38 @@ main()
--arch) --arch)
ARCH="$2" ARCH="$2"
shift 2;; shift 2;;
--filearch) --filearch)
FILEARCH="$2" FILEARCH="$2"
shift 2;; shift 2;;
--ver) --ver)
VER="$2" VER="$2"
shift 2;; shift 2;;
-o ) -o )
OUTPUTDIR="$2" OUTPUTDIR="$2"
shift 2;; shift 2;;
-h | --help ) -h | --help )
showhelp showhelp
return 0 return 0
shift ;; shift ;;
-- ) shift; break ;; -- ) shift; break ;;
* ) break ;; * ) break ;;
esac esac
done done
if [ -z "$ARCH" ]; then if [ -z "$ARCH" ]; then
echo "please input arch." echo "please input arch."
return 1; return 1;
fi fi
if [ -z "$FILEARCH" ]; then if [ -z "$FILEARCH" ]; then
FILEARCH=$ARCH FILEARCH=$ARCH
fi fi
if [ -z "$OUTPUTDIR" ]; then if [ -z "$OUTPUTDIR" ]; then
OUTPUTDIR=$CURR_DIR; OUTPUTDIR=$CURR_DIR;
fi fi
build build
} }
main $@ main $@

View File

@@ -1,4 +1,19 @@
#! /bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
SMARTDNS_BIN=/opt/usr/sbin/smartdns SMARTDNS_BIN=/opt/usr/sbin/smartdns
SMARTDNS_CONF=/opt/etc/smartdns/smartdns.conf SMARTDNS_CONF=/opt/etc/smartdns/smartdns.conf
@@ -28,56 +43,56 @@ set_iptable()
redirect_tcp=1; redirect_tcp=1;
fi fi
IPS="`ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F: '{print $2}'`" IPS="$(ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F: '{print $2}')"
for IP in $IPS for IP in $IPS
do do
if [ $redirect_tcp -eq 1 ]; then if [ $redirect_tcp -eq 1 ]; then
iptables -t nat -A PREROUTING -p tcp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT > /dev/null 2>&1 iptables -t nat -A PREROUTING -p tcp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" > /dev/null 2>&1
fi fi
iptables -t nat -A PREROUTING -p udp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT > /dev/null 2>&1 iptables -t nat -A PREROUTING -p udp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" > /dev/null 2>&1
done done
} }
clear_iptable() clear_iptable()
{ {
IPS="`ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F: '{print $2}'`" IPS="$(ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F: '{print $2}')"
for IP in $IPS for IP in $IPS
do do
iptables -t nat -D PREROUTING -p tcp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT > /dev/null 2>&1 iptables -t nat -D PREROUTING -p tcp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" > /dev/null 2>&1
iptables -t nat -D PREROUTING -p udp -d $IP --dport 53 -j REDIRECT --to-ports $SMARTDNS_PORT > /dev/null 2>&1 iptables -t nat -D PREROUTING -p udp -d "$IP" --dport 53 -j REDIRECT --to-ports "$SMARTDNS_PORT" > /dev/null 2>&1
done done
} }
get_dnsmasq_cmd() get_dnsmasq_cmd()
{ {
CMD="`ps | grep -e '[a-zA-Z]\{0,2\} \{1,\}dnsmasq' | grep -v grep 2>/dev/null`" CMD="$(ps 2>/dev/null | grep -e '[a-zA-Z]\{0,2\} \{1,\}dnsmasq' | grep -v grep 2>/dev/null)"
if [ ! -z "$CMD" ]; then if [ ! -z "$CMD" ]; then
return return
fi fi
CMD="`ps | grep '/usr/sbin/dnsmasq' | grep -v grep 2>/dev/null`" CMD="$(ps 2>/dev/null | grep '/usr/sbin/dnsmasq' | grep -v grep 2>/dev/null)"
if [ ! -z "$CMD" ]; then if [ ! -z "$CMD" ]; then
return return
fi fi
CMD="`ps | grep 'dnsmasq' | grep -v grep 2>/dev/null`" CMD="$(ps 2>/dev/null | grep 'dnsmasq' | grep -v grep 2>/dev/null)"
if [ ! -z "$CMD" ]; then if [ ! -z "$CMD" ]; then
return return
fi fi
CMD="`ps ax | grep -e '[a-zA-Z]\{0,2\} \{1,\}dnsmasq' | grep -v grep 2>/dev/null`" CMD="$(ps ax 2>/dev/null | grep -e '[a-zA-Z]\{0,2\} \{1,\}dnsmasq' | grep -v grep 2>/dev/null)"
if [ ! -z "$CMD" ]; then if [ ! -z "$CMD" ]; then
return return
fi fi
CMD="`ps ax | grep /usr/sbin/dnsmasq | grep -v grep 2>/dev/null`" CMD="$(ps ax 2>/dev/null | grep /usr/sbin/dnsmasq | grep -v grep 2>/dev/null)"
if [ ! -z "$CMD" ]; then if [ ! -z "$CMD" ]; then
return return
fi fi
CMD="`ps ax | grep 'dnsmasq' | grep -v grep 2>/dev/null`" CMD="$(ps ax 2>/dev/null | grep 'dnsmasq' | grep -v grep 2>/dev/null)"
if [ ! -z "$CMD" ]; then if [ ! -z "$CMD" ]; then
return return
fi fi
@@ -93,15 +108,15 @@ restart_dnsmasq()
return 1 return 1
fi fi
PID=`echo "$CMD" | awk '{print $1}'` PID="$(echo "$CMD" | awk '{print $1}')"
if [ ! -d "/proc/$PID" ]; then if [ ! -d "/proc/$PID" ]; then
echo "dnsmasq is not running" echo "dnsmasq is not running"
return 1 return 1
fi fi
kill -9 $PID kill -9 "$PID"
DNSMASQ_CMD="`echo $CMD | awk '{for(i=5; i<=NF;i++)printf \$i " "}'`" DNSMASQ_CMD="$(echo "$CMD" | awk '{for(i=5; i<=NF;i++)printf $i " "}')"
$DNSMASQ_CMD $DNSMASQ_CMD
} }
@@ -109,21 +124,20 @@ restart_dnsmasq()
get_server_ip() get_server_ip()
{ {
CONF_FILE=$1 CONF_FILE=$1
IPS="`ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F: '{print $2}'`" IPS="$(ifconfig | grep "inet addr" | grep -v ":127" | grep "Bcast" | awk '{print $2}' | awk -F: '{print $2}')"
LOCAL_SERVER_IP=""
for IP in $IPS for IP in $IPS
do do
N=3 N=3
while [ $N -gt 0 ] while [ $N -gt 0 ]
do do
ADDR=`echo $IP | awk -F. "{for(i=1;i<=$N;i++)printf \\$i\".\"}"` ADDR="$(echo "$IP" | awk -F. "{for(i=1;i<="$N";i++)printf \$i\".\"}")"
grep "dhcp-range=" $CONF_FILE | grep $ADDR >/dev/null 2>&1 grep "dhcp-range=" "$CONF_FILE" | grep "$ADDR" >/dev/null 2>&1
if [ $? -eq 0 ]; then if [ $? -eq 0 ]; then
SERVER_TAG="`grep "^dhcp-range *=" $CONF_FILE | grep $ADDR | awk -F= '{print $2}' | awk -F, '{print $1}'`" SERVER_TAG="$(grep "^dhcp-range *=" "$CONF_FILE" | grep "$ADDR" | awk -F= '{print $2}' | awk -F, '{print $1}')"
LOCAL_SERVER_IP="$IP" LOCAL_SERVER_IP="$IP"
return 0 return 1
fi fi
N="`expr $N - 1`" N=$((N-1))
done done
done done
@@ -137,19 +151,20 @@ set_dnsmasq_conf()
local CONF_FILE=$1 local CONF_FILE=$1
get_server_ip $CONF_FILE get_server_ip $CONF_FILE
if [ "$LOCAL_SERVER_IP" ] && [ "$SERVER_TAG" ]; then if [ "$LOCAL_SERVER_IP" ] && [ "$SERVER_TAG" ]; then
grep "dhcp-option *=" $CONF_FILE | grep "$SERVER_TAG,6,$LOCAL_SERVER_IP" > /dev/null 2>&1 grep "dhcp-option *=" "$CONF_FILE" | grep "$SERVER_TAG,6,$LOCAL_SERVER_IP" > /dev/null 2>&1
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
sed -i "/^dhcp-option *=$SERVER_TAG,6,/d" $CONF_FILE sed -i "/^dhcp-option *=$SERVER_TAG,6,/d" "$CONF_FILE"
echo "dhcp-option=$SERVER_TAG,6,$LOCAL_SERVER_IP" >> $CONF_FILE echo "dhcp-option=$SERVER_TAG,6,$LOCAL_SERVER_IP" >> "$CONF_FILE"
RESTART_DNSMASQ=1 RESTART_DNSMASQ=1
fi fi
fi fi
grep "^port *=0" $CONF_FILE > /dev/null 2>&1 grep "^port *=0" "$CONF_FILE" > /dev/null 2>&1
if [ $? -ne 0 ]; then if [ $? -ne 0 ]; then
sed -i "/^port *=/d" $CONF_FILE sed -i "/^port *=/d" "$CONF_FILE"
echo "port=0" >> $CONF_FILE echo "port=0" >> "$CONF_FILE"
RESTART_DNSMASQ=1 RESTART_DNSMASQ=1
fi fi
} }
@@ -164,7 +179,7 @@ set_dnsmasq()
continue continue
fi fi
set_dnsmasq_conf $conf set_dnsmasq_conf "$conf"
done done
if [ $RESTART_DNSMASQ -ne 0 ]; then if [ $RESTART_DNSMASQ -ne 0 ]; then
@@ -178,18 +193,18 @@ clear_dnsmasq_conf()
local SERVER_TAG="" local SERVER_TAG=""
local CONF_FILE=$1 local CONF_FILE=$1
get_server_ip $CONF_FILE get_server_ip "$CONF_FILE"
if [ "$LOCAL_SERVER_IP" ] && [ "$SERVER_TAG" ]; then if [ "$LOCAL_SERVER_IP" ] && [ "$SERVER_TAG" ]; then
grep "dhcp-option *=" $CONF_FILE | grep "$SERVER_TAG,6,$LOCAL_SERVER_IP" > /dev/null 2>&1 grep "dhcp-option *=" "$CONF_FILE" | grep "$SERVER_TAG,6,$LOCAL_SERVER_IP" > /dev/null 2>&1
if [ $? -eq 0 ]; then if [ $? -eq 0 ]; then
sed -i "/^dhcp-option *=$SERVER_TAG,6,/d" $CONF_FILE sed -i "/^dhcp-option *=$SERVER_TAG,6,/d" "$CONF_FILE"
RESTART_DNSMASQ=1 RESTART_DNSMASQ=1
fi fi
fi fi
grep "^port *=" $CONF_FILE > /dev/null 2>&1 grep "^port *=" "$CONF_FILE" > /dev/null 2>&1
if [ $? -eq 0 ]; then if [ $? -eq 0 ]; then
sed -i "/^port *=/d" $CONF_FILE sed -i "/^port *=/d" "$CONF_FILE"
RESTART_DNSMASQ=1 RESTART_DNSMASQ=1
fi fi
} }
@@ -204,7 +219,7 @@ clear_dnsmasq()
continue continue
fi fi
clear_dnsmasq_conf $conf clear_dnsmasq_conf "$conf"
done done
if [ $RESTART_DNSMASQ -ne 0 ]; then if [ $RESTART_DNSMASQ -ne 0 ]; then
@@ -217,11 +232,11 @@ set_smartdns_port()
if [ "$SMARTDNS_WORKMODE" = "0" ]; then if [ "$SMARTDNS_WORKMODE" = "0" ]; then
return 0 return 0
elif [ "$SMARTDNS_WORKMODE" = "1" ]; then elif [ "$SMARTDNS_WORKMODE" = "1" ]; then
sed -i "s/^\(bind .*\):53 *\(.*\)\$/\1:$SMARTDNS_PORT \2/g" $SMARTDNS_CONF sed -i "s/^\\(bind .*\\):53 *\\(.*\\)\$/\\1:$SMARTDNS_PORT \\2/g" $SMARTDNS_CONF
sed -i "s/^\(bind-tcp .*\):53 *\(.*\)\$/\1:$SMARTDNS_PORT \2/g" $SMARTDNS_CONF sed -i "s/^\\(bind-tcp .*\\):53 *\\(.*\\)\$/\\1:$SMARTDNS_PORT \\2/g" $SMARTDNS_CONF
elif [ "$SMARTDNS_WORKMODE" = "2" ]; then elif [ "$SMARTDNS_WORKMODE" = "2" ]; then
sed -i "s/^\(bind .*\):$SMARTDNS_PORT *\(.*\)\$/\1:53 \2/g" $SMARTDNS_CONF sed -i "s/^\\(bind .*\\):$SMARTDNS_PORT *\\(.*\\)\$/\\1:53 \\2/g" $SMARTDNS_CONF
sed -i "s/^\(bind-tcp .*\):$SMARTDNS_PORT *\(.*\)\$/\1:53 \2/g" $SMARTDNS_CONF sed -i "s/^\\(bind-tcp .*\\):$SMARTDNS_PORT *\\(.*\\)\$/\\1:53 \\2/g" $SMARTDNS_CONF
else else
return 1 return 1
fi fi
@@ -271,7 +286,7 @@ get_tz()
continue continue
fi fi
tz="`cat $tzfile 2>/dev/null`" tz="$(cat $tzfile 2>/dev/null)"
done done
if [ -z "$tz" ]; then if [ -z "$tz" ]; then
@@ -296,7 +311,7 @@ case "$1" in
fi fi
;; ;;
status) status)
pid="`cat $SMARTDNS_PID |head -n 1 2>/dev/null`" pid="$(cat $SMARTDNS_PID |head -n 1 2>/dev/null)"
if [ -z "$pid" ]; then if [ -z "$pid" ]; then
echo "smartdns not running." echo "smartdns not running."
return 0 return 0
@@ -311,17 +326,36 @@ case "$1" in
;; ;;
stop) stop)
clear_rule clear_rule
pid="`cat $SMARTDNS_PID | head -n 1 2>/dev/null`" pid="$(cat "$SMARTDNS_PID" | head -n 1 2>/dev/null)"
if [ -z "$pid" ]; then if [ -z "$pid" ]; then
echo "smartdns not running." echo "smartdns not running."
return 0 return 0
fi fi
if [ ! -d "/proc/$pid" ]; then kill -15 "$pid" 2>/dev/null
return 0; SLEEP=`which usleep`
SLEEPTIME=200000
if [ -z "$SLEEP" ]; then
SLEEP="sleep"
SLEEPTIME=0.2
fi fi
N=30
while [ $N -gt 0 ]
do
pid="$(cat "$SMARTDNS_PID" | head -n 1 2>/dev/null)"
if [ -z "$pid" ]; then
return 0
fi
kill -9 $pid 2>/dev/null if [ ! -d "/proc/$pid" ]; then
return 0;
fi
$SLEEP $SLEEPTIME 2>/dev/null
N=$((N-1))
done
kill -9 "$pid" 2>/dev/null
;; ;;
restart) restart)
$0 stop $0 stop

View File

@@ -1,4 +1,19 @@
#! /bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
chmod +x /opt/usr/sbin/smartdns chmod +x /opt/usr/sbin/smartdns
chmod +x /opt/etc/init.d/S50smartdns chmod +x /opt/etc/init.d/S50smartdns

View File

@@ -1,3 +1,18 @@
#! /bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
/opt/etc/init.d/S50smartdns stop /opt/etc/init.d/S50smartdns stop

View File

@@ -1,5 +1,19 @@
#/bin/sh #!/bin/sh
#
# Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#
# smartdns is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# smartdns is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
CURR_DIR=$(cd $(dirname $0);pwd) CURR_DIR=$(cd $(dirname $0);pwd)
VER="`date +"1.%Y.%m.%d-%H%M"`" VER="`date +"1.%Y.%m.%d-%H%M"`"
SMARTDNS_DIR=$CURR_DIR/../../ SMARTDNS_DIR=$CURR_DIR/../../
@@ -13,39 +27,44 @@ showhelp()
echo "Options:" echo "Options:"
echo " -o output directory." echo " -o output directory."
echo " --arch archtecture." echo " --arch archtecture."
echo " --ver version." echo " --ver version."
echo " -h show this message." echo " -h show this message."
} }
build() build()
{ {
ROOT=/tmp/smartdns-optware ROOT=/tmp/smartdns-optware
rm -fr $ROOT rm -fr $ROOT
mkdir -p $ROOT mkdir -p $ROOT
cp $CURR_DIR/* $ROOT/ -af cp $CURR_DIR/* $ROOT/ -af
cd $ROOT/ cd $ROOT/
mkdir $ROOT/opt/usr/sbin -p mkdir $ROOT/opt/usr/sbin -p
mkdir $ROOT/opt/etc/init.d -p mkdir $ROOT/opt/etc/init.d -p
mkdir $ROOT/opt/etc/smartdns/ -p mkdir $ROOT/opt/etc/smartdns/ -p
cp $SMARTDNS_CONF $ROOT/opt/etc/smartdns/ cp $SMARTDNS_CONF $ROOT/opt/etc/smartdns/
cp $SMARTDNS_OPT $ROOT/opt/etc/smartdns/ cp $SMARTDNS_OPT $ROOT/opt/etc/smartdns/
cp $CURR_DIR/S50smartdns $ROOT/opt/etc/init.d/ cp $CURR_DIR/S50smartdns $ROOT/opt/etc/init.d/
cp $SMARTDNS_BIN $ROOT/opt/usr/sbin cp $SMARTDNS_BIN $ROOT/opt/usr/sbin
if [ $? -ne 0 ]; then
echo "copy smartdns file failed."
rm -fr $ROOT/
return 1
fi
sed -i "s/# *server-name smartdns/server-name smartdns/g" $ROOT/opt/etc/smartdns/smartdns.conf sed -i "s/# *server-name smartdns/server-name smartdns/g" $ROOT/opt/etc/smartdns/smartdns.conf
sed -i "s/^Architecture.*/Architecture: $ARCH/g" $ROOT/control/control sed -i "s/^Architecture.*/Architecture: $ARCH/g" $ROOT/control/control
sed -i "s/Version:.*/Version: $VER/" $ROOT/control/control sed -i "s/Version:.*/Version: $VER/" $ROOT/control/control
cd $ROOT/control cd $ROOT/control
chmod +x * chmod +x *
tar zcf ../control.tar.gz --owner=0 --group=0 ./ tar zcf ../control.tar.gz --owner=0 --group=0 ./
cd $ROOT cd $ROOT
tar zcf data.tar.gz --owner=0 --group=0 opt tar zcf data.tar.gz --owner=0 --group=0 opt
tar zcf $OUTPUTDIR/smartdns.$VER.$FILEARCH.ipk --owner=0 --group=0 control.tar.gz data.tar.gz debian-binary tar zcf $OUTPUTDIR/smartdns.$VER.$FILEARCH.ipk --owner=0 --group=0 control.tar.gz data.tar.gz debian-binary
rm -fr $ROOT/ rm -fr $ROOT/
} }
main() main()
@@ -63,38 +82,38 @@ main()
--arch) --arch)
ARCH="$2" ARCH="$2"
shift 2;; shift 2;;
--filearch) --filearch)
FILEARCH="$2" FILEARCH="$2"
shift 2;; shift 2;;
--ver) --ver)
VER="$2" VER="$2"
shift 2;; shift 2;;
-o ) -o )
OUTPUTDIR="$2" OUTPUTDIR="$2"
shift 2;; shift 2;;
-h | --help ) -h | --help )
showhelp showhelp
return 0 return 0
shift ;; shift ;;
-- ) shift; break ;; -- ) shift; break ;;
* ) break ;; * ) break ;;
esac esac
done done
if [ -z "$ARCH" ]; then if [ -z "$ARCH" ]; then
echo "please input arch." echo "please input arch."
return 1; return 1;
fi fi
if [ -z "$FILEARCH" ]; then if [ -z "$FILEARCH" ]; then
FILEARCH=$ARCH FILEARCH=$ARCH
fi fi
if [ -z "$OUTPUTDIR" ]; then if [ -z "$OUTPUTDIR" ]; then
OUTPUTDIR=$CURR_DIR; OUTPUTDIR=$CURR_DIR;
fi fi
build build
} }
main $@ main $@

View File

@@ -0,0 +1,63 @@
Name: smartdns
Version: 31
Release: 1%{?dist}
Summary: smartdns
License: GPL 3.0
URL: https://github.com/pymumu/smartdns
Source0: smartdns-Release31.tar.gz
BuildRequires: glibc
Requires: glibc
Requires: systemd
%description
A local DNS server to obtain the fastest website IP for the best Internet experience.
%prep
%setup -q -n smartdns-Release31
%build
cd src
make %{?_smp_mflags}
%install
rm -rf $RPM_BUILD_ROOT
%{__install} -D -m 755 src/smartdns $RPM_BUILD_ROOT%{_sbindir}/smartdns
%{__install} -D -m 644 etc/smartdns/smartdns.conf $RPM_BUILD_ROOT%{_sysconfdir}/smartdns/smartdns.conf
%{__install} -D -m 644 systemd/smartdns.service.in $RPM_BUILD_ROOT%{_unitdir}/smartdns.service
cat > $RPM_BUILD_ROOT%{_unitdir}/smartdns.service <<EOF
[Unit]
Description=smartdns
ConditionFileIsExecutable=/usr/sbin/smartdns
After=syslog.target network-online.target
[Service]
Type=simple
ExecStart=/usr/sbin/smartdns -c /etc/smartdns/smartdns.conf -f
PIDFile=/run/smartdns.pid
Restart=on-failure
KillMode=process
[Install]
WantedBy=multi-user.target
EOF
%files
%defattr(-,root,root,-)
%{_sbindir}/smartdns
%config(noreplace) %{_sysconfdir}/smartdns/smartdns.conf
%{_unitdir}/smartdns.service
%post
%systemd_post %{name}.service
%preun
%systemd_preun %{name}.service
%postun
%systemd_postun_with_restart %{name}.service

View File

@@ -1,27 +1,49 @@
BIN=smartdns # Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
OBJS_LIB=lib/rbtree.o lib/art.o lib/bitops.o lib/radix.o lib/conf.o #
OBJS=smartdns.o fast_ping.o dns_client.o dns_server.o dns.o util.o tlog.o dns_conf.o dns_cache.o http_parse.o $(OBJS_LIB) # smartdns is free software: you can redistribute it and/or modify
CFLAGS +=-O2 -g -Wall -Wstrict-prototypes -fno-omit-frame-pointer -Wstrict-aliasing # it under the terms of the GNU General Public License as published by
CFLAGS +=-Iinclude # the Free Software Foundation, either version 3 of the License, or
CFLAGS += -DBASE_FILE_NAME=\"$(notdir $<)\" # (at your option) any later version.
ifdef VER #
CFLAGS += -DSMARTDNS_VERION=\"$(VER)\" # smartdns is distributed in the hope that it will be useful,
endif # but WITHOUT ANY WARRANTY; without even the implied warranty of
CXXFLAGS=-O2 -g -Wall -std=c++11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
CXXFLAGS +=-Iinclude # GNU General Public License for more details.
ifeq ($(STATIC), yes) #
LDFLAGS += -lssl -lcrypto -Wl,--whole-archive -lpthread -Wl,--no-whole-archive -ldl -static # You should have received a copy of the GNU General Public License
else # along with this program. If not, see <http://www.gnu.org/licenses/>.
LDFLAGS += -lssl -lcrypto -lpthread
endif BIN=smartdns
OBJS_LIB=lib/rbtree.o lib/art.o lib/bitops.o lib/radix.o lib/conf.o
.PHONY: all OBJS=smartdns.o fast_ping.o dns_client.o dns_server.o dns.o util.o tlog.o dns_conf.o dns_cache.o http_parse.o $(OBJS_LIB)
all: $(BIN) # cflags
ifndef CFLAGS
$(BIN) : $(OBJS) CFLAGS =-O2 -g -Wall -Wstrict-prototypes -fno-omit-frame-pointer -Wstrict-aliasing
$(CC) $(OBJS) -o $@ $(LDFLAGS) endif
override CFLAGS +=-Iinclude
clean: override CFLAGS += -DBASE_FILE_NAME=\"$(notdir $<)\"
$(RM) $(OBJS) $(BIN) ifdef VER
override CFLAGS += -DSMARTDNS_VERION=\"$(VER)\"
endif
CXXFLAGS=-O2 -g -Wall -std=c++11
override CXXFLAGS +=-Iinclude
# ldflags
ifeq ($(STATIC), yes)
override LDFLAGS += -lssl -lcrypto -Wl,--whole-archive -lpthread -Wl,--no-whole-archive -ldl -static
else
override LDFLAGS += -lssl -lcrypto -lpthread
endif
.PHONY: all clean
all: $(BIN)
$(BIN) : $(OBJS)
$(CC) $(OBJS) -o $@ $(LDFLAGS)
clean:
$(RM) $(OBJS) $(BIN)

4021
src/dns.c

File diff suppressed because it is too large Load Diff

View File

@@ -1,3 +1,21 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _DNS_HEAD_H #ifndef _DNS_HEAD_H
#define _DNS_HEAD_H #define _DNS_HEAD_H
@@ -26,7 +44,10 @@ typedef enum dns_rr_type {
DNS_RRS_END, DNS_RRS_END,
} dns_rr_type; } dns_rr_type;
typedef enum dns_class { DNS_C_IN = 1, DNS_C_ANY = 255 } dns_class_t; typedef enum dns_class {
DNS_C_IN = 1, // DNS C IN
DNS_C_ANY = 255
} dns_class_t;
typedef enum dns_type { typedef enum dns_type {
DNS_T_A = 1, DNS_T_A = 1,
@@ -45,10 +66,10 @@ typedef enum dns_type {
DNS_T_ALL = 255 DNS_T_ALL = 255
} dns_type_t; } dns_type_t;
typedef enum dns_opt_code { typedef enum dns_opt_code {
DNS_OPT_T_ECS = 8, DNS_OPT_T_ECS = 8, // OPT ECS
DNS_OPT_T_TCP_KEEPALIVE = 11, DNS_OPT_T_TCP_KEEPALIVE = 11,
DNS_OPT_T_ALL = 255 DNS_OPT_T_ALL = 255
} dns_opt_code_t; } dns_opt_code_t;
typedef enum dns_opcode { typedef enum dns_opcode {
@@ -89,7 +110,7 @@ struct dns_head {
unsigned short ancount; /* number of answer entries */ unsigned short ancount; /* number of answer entries */
unsigned short nscount; /* number of authority entries */ unsigned short nscount; /* number of authority entries */
unsigned short nrcount; /* number of addititional resource entries */ unsigned short nrcount; /* number of addititional resource entries */
} __attribute__((packed)); } __attribute__((packed, aligned(2)));
struct dns_rrs { struct dns_rrs {
unsigned short next; unsigned short next;
@@ -178,7 +199,8 @@ int dns_get_A(struct dns_rrs *rrs, char *domain, int maxsize, int *ttl, unsigned
int dns_add_PTR(struct dns_packet *packet, dns_rr_type type, char *domain, int ttl, char *cname); int dns_add_PTR(struct dns_packet *packet, dns_rr_type type, char *domain, int ttl, char *cname);
int dns_get_PTR(struct dns_rrs *rrs, char *domain, int maxsize, int *ttl, char *cname, int cname_size); int dns_get_PTR(struct dns_rrs *rrs, char *domain, int maxsize, int *ttl, char *cname, int cname_size);
int dns_add_AAAA(struct dns_packet *packet, dns_rr_type type, char *domain, int ttl, unsigned char addr[DNS_RR_AAAA_LEN]); int dns_add_AAAA(struct dns_packet *packet, dns_rr_type type, char *domain, int ttl,
unsigned char addr[DNS_RR_AAAA_LEN]);
int dns_get_AAAA(struct dns_rrs *rrs, char *domain, int maxsize, int *ttl, unsigned char addr[DNS_RR_AAAA_LEN]); int dns_get_AAAA(struct dns_rrs *rrs, char *domain, int maxsize, int *ttl, unsigned char addr[DNS_RR_AAAA_LEN]);
int dns_add_SOA(struct dns_packet *packet, dns_rr_type type, char *domain, int ttl, struct dns_soa *soa); int dns_add_SOA(struct dns_packet *packet, dns_rr_type type, char *domain, int ttl, struct dns_soa *soa);
@@ -194,7 +216,8 @@ int dns_add_OPT_ECS(struct dns_packet *packet, struct dns_opt_ecs *ecs);
int dns_get_OPT_ECS(struct dns_rrs *rrs, unsigned short *opt_code, unsigned short *opt_len, struct dns_opt_ecs *ecs); int dns_get_OPT_ECS(struct dns_rrs *rrs, unsigned short *opt_code, unsigned short *opt_len, struct dns_opt_ecs *ecs);
int dns_add_OPT_TCP_KEEYALIVE(struct dns_packet *packet, unsigned short timeout); int dns_add_OPT_TCP_KEEYALIVE(struct dns_packet *packet, unsigned short timeout);
int dns_get_OPT_TCP_KEEYALIVE(struct dns_rrs *rrs, unsigned short *opt_code, unsigned short *opt_len, unsigned short *timeout); int dns_get_OPT_TCP_KEEYALIVE(struct dns_rrs *rrs, unsigned short *opt_code, unsigned short *opt_len,
unsigned short *timeout);
/* /*
* Packet operation * Packet operation
*/ */

File diff suppressed because it is too large Load Diff

View File

@@ -1,69 +1,159 @@
#ifndef _SMARTDNS_CACHE_H /*************************************************************************
#define _SMARTDNS_CACHE_H *
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#include "atomic.h" *
#include "dns.h" * smartdns is free software: you can redistribute it and/or modify
#include "hash.h" * it under the terms of the GNU General Public License as published by
#include "hashtable.h" * the Free Software Foundation, either version 3 of the License, or
#include "list.h" * (at your option) any later version.
#include <stdlib.h> *
#include <time.h> * smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
#ifdef __cpluscplus * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
extern "C" { * GNU General Public License for more details.
#endif *
* You should have received a copy of the GNU General Public License
#define DNS_CACHE_TTL_MIN 30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
struct dns_cache {
struct hlist_node node; #ifndef _SMARTDNS_CACHE_H
struct list_head list; #define _SMARTDNS_CACHE_H
struct list_head check_list;
atomic_t ref; #include "atomic.h"
char domain[DNS_MAX_CNAME_LEN]; #include "dns.h"
char cname[DNS_MAX_CNAME_LEN]; #include "hash.h"
unsigned int cname_ttl; #include "hashtable.h"
unsigned int ttl; #include "list.h"
int speed; #include <stdlib.h>
atomic_t hitnum; #include <time.h>
int hitnum_update_add;
int del_pending; #ifdef __cpluscplus
time_t insert_time; extern "C" {
dns_type_t qtype; #endif
union {
unsigned char ipv4_addr[DNS_RR_A_LEN]; #define DNS_CACHE_TTL_MIN 30
unsigned char ipv6_addr[DNS_RR_AAAA_LEN]; #define DNS_CACHE_VERSION_LEN 32
unsigned char addr[0]; #define MAGIC_NUMBER 0x6548634163536e44
}; #define MAGIC_CACHE_DATA 0x44615461
};
enum CACHE_TYPE {
int dns_cache_init(int size); CACHE_TYPE_NONE,
CACHE_TYPE_ADDR,
int dns_cache_replace(char *domain, char *cname, int cname_ttl, int ttl, dns_type_t qtype, unsigned char *addr, int addr_len, int speed); CACHE_TYPE_PACKET,
};
int dns_cache_insert(char *domain, char *cname, int cname_ttl, int ttl, dns_type_t qtype, unsigned char *addr, int addr_len, int speed);
enum CACHE_RECORD_TYPE {
struct dns_cache *dns_cache_lookup(char *domain, dns_type_t qtype); CACHE_RECORD_TYPE_ACTIVE,
CACHE_RECORD_TYPE_INACTIVE,
void dns_cache_delete(struct dns_cache *dns_cache); };
void dns_cache_get(struct dns_cache *dns_cache); struct dns_cache_data_head {
uint32_t cache_flag;
void dns_cache_release(struct dns_cache *dns_cache); enum CACHE_TYPE cache_type;
size_t size;
int dns_cache_hitnum_dec_get(struct dns_cache *dns_cache); };
void dns_cache_update(struct dns_cache *dns_cache); struct dns_cache_data {
struct dns_cache_data_head head;
typedef void dns_cache_preinvalid_callback(struct dns_cache *dns_cache); unsigned char data[0];
};
void dns_cache_invalidate(dns_cache_preinvalid_callback callback, int ttl_pre);
struct dns_cache_addr {
int dns_cache_get_ttl(struct dns_cache *dns_cache); struct dns_cache_data_head head;
struct dns_cache_addr_data {
void dns_cache_destroy(void); unsigned int cname_ttl;
char cname[DNS_MAX_CNAME_LEN];
#ifdef __cpluscplus union {
} unsigned char ipv4_addr[DNS_RR_A_LEN];
#endif unsigned char ipv6_addr[DNS_RR_AAAA_LEN];
#endif // !_SMARTDNS_CACHE_H unsigned char addr[0];
};
} addr_data;
};
struct dns_cache_packet {
struct dns_cache_data_head head;
unsigned char data[0];
};
struct dns_cache_info {
char domain[DNS_MAX_CNAME_LEN];
int ttl;
int hitnum;
int speed;
int hitnum_update_add;
time_t insert_time;
dns_type_t qtype;
};
struct dns_cache_record {
uint32_t magic;
enum CACHE_RECORD_TYPE type;
struct dns_cache_info info;
};
struct dns_cache {
struct hlist_node node;
struct list_head list;
struct list_head check_list;
atomic_t ref;
int del_pending;
struct dns_cache_info info;
struct dns_cache_data *cache_data;
};
struct dns_cache_file {
uint64_t magic;
char version[DNS_CACHE_VERSION_LEN];
uint32_t cache_number;
};
enum CACHE_TYPE dns_cache_data_type(struct dns_cache_data *cache_data);
uint32_t dns_cache_get_cache_flag(struct dns_cache_data *cache_data);
void dns_cache_data_free(struct dns_cache_data *data);
struct dns_cache_data *dns_cache_new_data_addr(uint32_t cache_flag, char *cname, int cname_ttl, unsigned char *addr,
int addr_len);
struct dns_cache_data *dns_cache_new_data_packet(uint32_t cache_flag, void *packet, size_t packet_len);
int dns_cache_init(int size, int enable_inactive, int inactive_list_expired);
int dns_cache_replace(char *domain, int ttl, dns_type_t qtype, int speed, struct dns_cache_data *cache_data);
int dns_cache_insert(char *domain, int ttl, dns_type_t qtype, int speed, struct dns_cache_data *cache_data);
struct dns_cache *dns_cache_lookup(char *domain, dns_type_t qtype);
void dns_cache_delete(struct dns_cache *dns_cache);
void dns_cache_get(struct dns_cache *dns_cache);
void dns_cache_release(struct dns_cache *dns_cache);
int dns_cache_hitnum_dec_get(struct dns_cache *dns_cache);
void dns_cache_update(struct dns_cache *dns_cache);
typedef void dns_cache_preinvalid_callback(struct dns_cache *dns_cache);
void dns_cache_invalidate(dns_cache_preinvalid_callback callback, int ttl_pre);
int dns_cache_get_ttl(struct dns_cache *dns_cache);
struct dns_cache_data *dns_cache_get_data(struct dns_cache *dns_cache);
void dns_cache_destroy(void);
int dns_cache_load(const char *file);
int dns_cache_save(const char *file);
#ifdef __cpluscplus
}
#endif
#endif // !_SMARTDNS_CACHE_H

File diff suppressed because it is too large Load Diff

View File

@@ -1,3 +1,21 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _SMART_DNS_CLIENT_H #ifndef _SMART_DNS_CLIENT_H
#define _SMART_DNS_CLIENT_H #define _SMART_DNS_CLIENT_H
@@ -34,8 +52,9 @@ int dns_client_init(void);
int dns_client_set_ecs(char *ip, int subnet); int dns_client_set_ecs(char *ip, int subnet);
/* query result notify function */ /* query result notify function */
typedef int (*dns_client_callback)(char *domain, dns_result_type rtype, unsigned int result_flag, struct dns_packet *packet, unsigned char *inpacket, typedef int (*dns_client_callback)(char *domain, dns_result_type rtype, unsigned int result_flag,
int inpacket_len, void *user_ptr); struct dns_packet *packet, unsigned char *inpacket, int inpacket_len,
void *user_ptr);
/* query domain */ /* query domain */
int dns_client_query(char *domain, int qtype, dns_client_callback callback, void *user_ptr, const char *group_name); int dns_client_query(char *domain, int qtype, dns_client_callback callback, void *user_ptr, const char *group_name);
@@ -51,6 +70,7 @@ struct client_dns_server_flag_tls {
int spi_len; int spi_len;
char hostname[DNS_MAX_CNAME_LEN]; char hostname[DNS_MAX_CNAME_LEN];
char tls_host_verify[DNS_MAX_CNAME_LEN]; char tls_host_verify[DNS_MAX_CNAME_LEN];
char skip_check_cert;
}; };
struct client_dns_server_flag_https { struct client_dns_server_flag_https {
@@ -60,6 +80,7 @@ struct client_dns_server_flag_https {
char httphost[DNS_MAX_CNAME_LEN]; char httphost[DNS_MAX_CNAME_LEN];
char path[DNS_MAX_CNAME_LEN]; char path[DNS_MAX_CNAME_LEN];
char tls_host_verify[DNS_MAX_CNAME_LEN]; char tls_host_verify[DNS_MAX_CNAME_LEN];
char skip_check_cert;
}; };
struct client_dns_server_flags { struct client_dns_server_flags {
@@ -77,7 +98,8 @@ struct client_dns_server_flags {
int dns_client_spki_decode(const char *spki, unsigned char *spki_data_out); int dns_client_spki_decode(const char *spki, unsigned char *spki_data_out);
/* add remote dns server */ /* add remote dns server */
int dns_client_add_server(char *server_ip, int port, dns_server_type_t server_type, struct client_dns_server_flags *flags); int dns_client_add_server(char *server_ip, int port, dns_server_type_t server_type,
struct client_dns_server_flags *flags);
/* remove remote dns server */ /* remove remote dns server */
int dns_client_remove_server(char *server_ip, int port, dns_server_type_t server_type); int dns_client_remove_server(char *server_ip, int port, dns_server_type_t server_type);

File diff suppressed because it is too large Load Diff

View File

@@ -1,229 +1,260 @@
#ifndef _DNS_CONF /*************************************************************************
#define _DNS_CONF *
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#include "art.h" *
#include "conf.h" * smartdns is free software: you can redistribute it and/or modify
#include "dns.h" * it under the terms of the GNU General Public License as published by
#include "dns_client.h" * the Free Software Foundation, either version 3 of the License, or
#include "hash.h" * (at your option) any later version.
#include "hashtable.h" *
#include "list.h" * smartdns is distributed in the hope that it will be useful,
#include "radix.h" * but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#ifdef __cpluscplus * GNU General Public License for more details.
extern "C" { *
#endif * You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
#define DNS_MAX_BIND_IP 16 */
#define DNS_MAX_SERVERS 64
#define DNS_MAX_SERVER_NAME_LEN 128 #ifndef _DNS_CONF
#define DNS_MAX_IPSET_NAMELEN 32 #define _DNS_CONF
#define DNS_GROUP_NAME_LEN 32
#define DNS_NAX_GROUP_NUMBER 16 #include "art.h"
#define DNS_MAX_IPLEN 64 #include "conf.h"
#define DNS_MAX_SPKI_LEN 64 #include "dns.h"
#define DNS_MAX_URL_LEN 256 #include "dns_client.h"
#define DNS_MAX_PATH 1024 #include "hash.h"
#define DEFAULT_DNS_PORT 53 #include "hashtable.h"
#define DEFAULT_DNS_TLS_PORT 853 #include "list.h"
#define DEFAULT_DNS_HTTPS_PORT 443 #include "radix.h"
#define DNS_MAX_CONF_CNAME_LEN 256
#define SMARTDNS_CONF_FILE "/etc/smartdns/smartdns.conf" #ifdef __cpluscplus
#define SMARTDNS_LOG_FILE "/var/log/smartdns.log" extern "C" {
#define SMARTDNS_AUDIT_FILE "/var/log/smartdns-audit.log" #endif
enum domain_rule { #define DNS_MAX_BIND_IP 16
DOMAIN_RULE_FLAGS = 0, #define DNS_MAX_SERVERS 64
DOMAIN_RULE_ADDRESS_IPV4, #define DNS_MAX_SERVER_NAME_LEN 128
DOMAIN_RULE_ADDRESS_IPV6, #define DNS_MAX_IPSET_NAMELEN 32
DOMAIN_RULE_IPSET, #define DNS_GROUP_NAME_LEN 32
DOMAIN_RULE_NAMESERVER, #define DNS_NAX_GROUP_NUMBER 16
DOMAIN_RULE_MAX, #define DNS_MAX_IPLEN 64
}; #define DNS_MAX_SPKI_LEN 64
#define DNS_MAX_URL_LEN 256
typedef enum { #define DNS_MAX_PATH 1024
DNS_BIND_TYPE_UDP, #define DEFAULT_DNS_PORT 53
DNS_BIND_TYPE_TCP, #define DEFAULT_DNS_TLS_PORT 853
DNS_BIND_TYPE_TLS, #define DEFAULT_DNS_HTTPS_PORT 443
} DNS_BIND_TYPE; #define DNS_MAX_CONF_CNAME_LEN 256
#define SMARTDNS_CONF_FILE "/etc/smartdns/smartdns.conf"
#define DOMAIN_CHECK_NONE 0 #define SMARTDNS_LOG_FILE "/var/log/smartdns.log"
#define DOMAIN_CHECK_ICMP 1 #define SMARTDNS_AUDIT_FILE "/var/log/smartdns-audit.log"
#define DOMAIN_CHECK_TCP 2 #define SMARTDNS_CACHE_FILE "/tmp/smartdns.cache"
#define DOMAIN_CHECK_NUM 2
enum domain_rule {
#define DOMAIN_FLAG_ADDR_SOA (1 << 0) DOMAIN_RULE_FLAGS = 0,
#define DOMAIN_FLAG_ADDR_IPV4_SOA (1 << 1) DOMAIN_RULE_ADDRESS_IPV4,
#define DOMAIN_FLAG_ADDR_IPV6_SOA (1 << 2) DOMAIN_RULE_ADDRESS_IPV6,
#define DOMAIN_FLAG_ADDR_IGN (1 << 3) DOMAIN_RULE_IPSET,
#define DOMAIN_FLAG_ADDR_IPV4_IGN (1 << 4) DOMAIN_RULE_NAMESERVER,
#define DOMAIN_FLAG_ADDR_IPV6_IGN (1 << 5) DOMAIN_RULE_CHECKSPEED,
#define DOMAIN_FLAG_IPSET_IGNORE (1 << 6) DOMAIN_RULE_MAX,
#define DOMAIN_FLAG_NAMESERVER_IGNORE (1 << 7) };
#define SERVER_FLAG_EXCLUDE_DEFAULT (1 << 0) typedef enum {
DNS_BIND_TYPE_UDP,
#define BIND_FLAG_NO_RULE_ADDR (1 << 0) DNS_BIND_TYPE_TCP,
#define BIND_FLAG_NO_RULE_NAMESERVER (1 << 1) DNS_BIND_TYPE_TLS,
#define BIND_FLAG_NO_RULE_IPSET (1 << 2) } DNS_BIND_TYPE;
#define BIND_FLAG_NO_RULE_SNIPROXY (1 << 3)
#define BIND_FLAG_NO_RULE_SOA (1 << 4) #define DOMAIN_CHECK_NONE 0
#define BIND_FLAG_NO_SPEED_CHECK (1 << 5) #define DOMAIN_CHECK_ICMP 1
#define BIND_FLAG_NO_CACHE (1 << 6) #define DOMAIN_CHECK_TCP 2
#define BIND_FLAG_NO_DUALSTACK_SELECTION (1 << 7) #define DOMAIN_CHECK_NUM 2
struct dns_rule_flags { #define DOMAIN_FLAG_ADDR_SOA (1 << 0)
unsigned int flags; #define DOMAIN_FLAG_ADDR_IPV4_SOA (1 << 1)
}; #define DOMAIN_FLAG_ADDR_IPV6_SOA (1 << 2)
#define DOMAIN_FLAG_ADDR_IGN (1 << 3)
struct dns_address_IPV4 { #define DOMAIN_FLAG_ADDR_IPV4_IGN (1 << 4)
unsigned char ipv4_addr[DNS_RR_A_LEN]; #define DOMAIN_FLAG_ADDR_IPV6_IGN (1 << 5)
}; #define DOMAIN_FLAG_IPSET_IGNORE (1 << 6)
#define DOMAIN_FLAG_NAMESERVER_IGNORE (1 << 7)
struct dns_address_IPV6 {
unsigned char ipv6_addr[DNS_RR_AAAA_LEN]; #define SERVER_FLAG_EXCLUDE_DEFAULT (1 << 0)
};
#define BIND_FLAG_NO_RULE_ADDR (1 << 0)
struct dns_ipset_name { #define BIND_FLAG_NO_RULE_NAMESERVER (1 << 1)
struct hlist_node node; #define BIND_FLAG_NO_RULE_IPSET (1 << 2)
char ipsetname[DNS_MAX_IPSET_NAMELEN]; #define BIND_FLAG_NO_RULE_SNIPROXY (1 << 3)
}; #define BIND_FLAG_NO_RULE_SOA (1 << 4)
#define BIND_FLAG_NO_SPEED_CHECK (1 << 5)
struct dns_ipset_rule { #define BIND_FLAG_NO_CACHE (1 << 6)
const char *ipsetname; #define BIND_FLAG_NO_DUALSTACK_SELECTION (1 << 7)
}; #define BIND_FLAG_FORCE_AAAA_SOA (1 << 8)
struct dns_domain_rule { struct dns_rule_flags {
void *rules[DOMAIN_RULE_MAX]; unsigned int flags;
}; };
struct dns_nameserver_rule { struct dns_address_IPV4 {
const char *group_name; unsigned char ipv4_addr[DNS_RR_A_LEN];
}; };
struct dns_server_groups { struct dns_address_IPV6 {
struct hlist_node node; unsigned char ipv6_addr[DNS_RR_AAAA_LEN];
char group_name[DNS_GROUP_NAME_LEN]; };
int server_num;
struct dns_servers *servers[DNS_MAX_SERVERS]; struct dns_ipset_name {
}; struct hlist_node node;
char ipsetname[DNS_MAX_IPSET_NAMELEN];
struct dns_domain_check_order { };
char order[DOMAIN_CHECK_NUM];
unsigned short tcp_port; struct dns_ipset_rule {
}; const char *ipsetname;
};
struct dns_group_table {
DECLARE_HASHTABLE(group, 8); struct dns_domain_rule {
}; void *rules[DOMAIN_RULE_MAX];
extern struct dns_group_table dns_group_table; };
struct dns_servers { struct dns_nameserver_rule {
char server[DNS_MAX_IPLEN]; const char *group_name;
unsigned short port; };
unsigned int result_flag;
unsigned int server_flag; struct dns_server_groups {
int ttl; struct hlist_node node;
dns_server_type_t type; char group_name[DNS_GROUP_NAME_LEN];
char spki[DNS_MAX_SPKI_LEN]; int server_num;
char hostname[DNS_MAX_CNAME_LEN]; struct dns_servers *servers[DNS_MAX_SERVERS];
char httphost[DNS_MAX_CNAME_LEN]; };
char tls_host_verify[DNS_MAX_CNAME_LEN];
char path[DNS_MAX_URL_LEN]; struct dns_domain_check_order {
}; char order[DOMAIN_CHECK_NUM];
unsigned short tcp_port;
/* ip address lists of domain */ };
struct dns_bogus_ip_address {
struct hlist_node node; struct dns_group_table {
dns_type_t addr_type; DECLARE_HASHTABLE(group, 8);
union { };
unsigned char ipv4_addr[DNS_RR_A_LEN]; extern struct dns_group_table dns_group_table;
unsigned char ipv6_addr[DNS_RR_AAAA_LEN];
unsigned char addr[0]; struct dns_servers {
}; char server[DNS_MAX_IPLEN];
}; unsigned short port;
unsigned int result_flag;
enum address_rule { unsigned int server_flag;
ADDRESS_RULE_BLACKLIST = 1, int ttl;
ADDRESS_RULE_WHITELIST = 2, dns_server_type_t type;
ADDRESS_RULE_BOGUS = 3, char skip_check_cert;
ADDRESS_RULE_IP_IGNORE = 4, char spki[DNS_MAX_SPKI_LEN];
}; char hostname[DNS_MAX_CNAME_LEN];
char httphost[DNS_MAX_CNAME_LEN];
struct dns_ip_address_rule { char tls_host_verify[DNS_MAX_CNAME_LEN];
unsigned int blacklist : 1; char path[DNS_MAX_URL_LEN];
unsigned int whitelist : 1; };
unsigned int bogus : 1;
unsigned int ip_ignore : 1; /* ip address lists of domain */
}; struct dns_bogus_ip_address {
struct hlist_node node;
struct dns_edns_client_subnet { dns_type_t addr_type;
int enable; union {
char ip[DNS_MAX_IPLEN]; unsigned char ipv4_addr[DNS_RR_A_LEN];
int subnet; unsigned char ipv6_addr[DNS_RR_AAAA_LEN];
}; unsigned char addr[0];
};
struct dns_conf_address_rule { };
radix_tree_t *ipv4;
radix_tree_t *ipv6; enum address_rule {
}; ADDRESS_RULE_BLACKLIST = 1,
ADDRESS_RULE_WHITELIST = 2,
struct dns_bind_ip { ADDRESS_RULE_BOGUS = 3,
DNS_BIND_TYPE type; ADDRESS_RULE_IP_IGNORE = 4,
uint32_t flags; };
char ip[DNS_MAX_IPLEN];
const char *group; struct dns_ip_address_rule {
}; unsigned int blacklist : 1;
unsigned int whitelist : 1;
extern struct dns_bind_ip dns_conf_bind_ip[DNS_MAX_BIND_IP]; unsigned int bogus : 1;
extern int dns_conf_bind_ip_num; unsigned int ip_ignore : 1;
};
extern int dns_conf_tcp_idle_time;
extern int dns_conf_cachesize; struct dns_edns_client_subnet {
extern int dns_conf_prefetch; int enable;
extern struct dns_servers dns_conf_servers[DNS_MAX_SERVERS]; char ip[DNS_MAX_IPLEN];
extern int dns_conf_server_num; int subnet;
};
extern int dns_conf_log_level;
extern char dns_conf_log_file[DNS_MAX_PATH]; struct dns_conf_address_rule {
extern size_t dns_conf_log_size; radix_tree_t *ipv4;
extern int dns_conf_log_num; radix_tree_t *ipv6;
};
extern struct dns_domain_check_order dns_conf_check_order;
struct dns_bind_ip {
extern struct dns_server_groups dns_conf_server_groups[DNS_NAX_GROUP_NUMBER]; DNS_BIND_TYPE type;
extern int dns_conf_server_group_num; uint32_t flags;
char ip[DNS_MAX_IPLEN];
extern int dns_conf_audit_enable; const char *group;
extern int dns_conf_audit_log_SOA; };
extern char dns_conf_audit_file[DNS_MAX_PATH];
extern size_t dns_conf_audit_size; extern struct dns_bind_ip dns_conf_bind_ip[DNS_MAX_BIND_IP];
extern int dns_conf_audit_num; extern int dns_conf_bind_ip_num;
extern char dns_conf_server_name[DNS_MAX_SERVER_NAME_LEN]; extern int dns_conf_tcp_idle_time;
extern art_tree dns_conf_domain_rule; extern int dns_conf_cachesize;
extern struct dns_conf_address_rule dns_conf_address_rule; extern int dns_conf_prefetch;
extern int dns_conf_serve_expired;
extern int dns_conf_dualstack_ip_selection; extern int dns_conf_serve_expired_ttl;
extern int dns_conf_dualstack_ip_selection_threshold; extern int dns_conf_serve_expired_reply_ttl;
extern struct dns_servers dns_conf_servers[DNS_MAX_SERVERS];
extern int dns_conf_rr_ttl; extern int dns_conf_server_num;
extern int dns_conf_rr_ttl_min;
extern int dns_conf_rr_ttl_max; extern int dns_conf_log_level;
extern int dns_conf_force_AAAA_SOA; extern char dns_conf_log_file[DNS_MAX_PATH];
extern int dns_conf_ipset_timeout_enable; extern size_t dns_conf_log_size;
extern int dns_conf_log_num;
extern struct dns_edns_client_subnet dns_conf_ipv4_ecs;
extern struct dns_edns_client_subnet dns_conf_ipv6_ecs; extern char dns_conf_ca_file[DNS_MAX_PATH];
extern char dns_conf_ca_path[DNS_MAX_PATH];
extern char dns_conf_sni_proxy_ip[DNS_MAX_IPLEN];
extern char dns_conf_cache_file[DNS_MAX_PATH];
void dns_server_load_exit(void); extern int dns_conf_cache_persist;
int dns_server_load_conf(const char *file); extern struct dns_domain_check_order dns_conf_check_order;
extern int config_addtional_file(void *data, int argc, char *argv[]); extern struct dns_server_groups dns_conf_server_groups[DNS_NAX_GROUP_NUMBER];
#ifdef __cpluscplus extern int dns_conf_server_group_num;
}
#endif extern int dns_conf_audit_enable;
#endif // !_DNS_CONF extern int dns_conf_audit_log_SOA;
extern char dns_conf_audit_file[DNS_MAX_PATH];
extern size_t dns_conf_audit_size;
extern int dns_conf_audit_num;
extern char dns_conf_server_name[DNS_MAX_SERVER_NAME_LEN];
extern art_tree dns_conf_domain_rule;
extern struct dns_conf_address_rule dns_conf_address_rule;
extern int dns_conf_dualstack_ip_selection;
extern int dns_conf_dualstack_ip_selection_threshold;
extern int dns_conf_rr_ttl;
extern int dns_conf_rr_ttl_min;
extern int dns_conf_rr_ttl_max;
extern int dns_conf_force_AAAA_SOA;
extern int dns_conf_ipset_timeout_enable;
extern struct dns_edns_client_subnet dns_conf_ipv4_ecs;
extern struct dns_edns_client_subnet dns_conf_ipv6_ecs;
extern char dns_conf_sni_proxy_ip[DNS_MAX_IPLEN];
void dns_server_load_exit(void);
int dns_server_load_conf(const char *file);
extern int config_addtional_file(void *data, int argc, char *argv[]);
#ifdef __cpluscplus
}
#endif
#endif // !_DNS_CONF

File diff suppressed because it is too large Load Diff

View File

@@ -1,7 +1,26 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _SMART_DNS_SERVER_H #ifndef _SMART_DNS_SERVER_H
#define _SMART_DNS_SERVER_H #define _SMART_DNS_SERVER_H
#include "dns.h" #include "dns.h"
#include <stdint.h>
#ifdef __cpluscplus #ifdef __cpluscplus
extern "C" { extern "C" {
@@ -18,10 +37,11 @@ void dns_server_stop(void);
void dns_server_exit(void); void dns_server_exit(void);
/* query result notify function */ /* query result notify function */
typedef int (*dns_result_callback)(char *domain, dns_rtcode_t rtcode, dns_type_t addr_type, char *ip, unsigned int ping_time, void *user_ptr); typedef int (*dns_result_callback)(char *domain, dns_rtcode_t rtcode, dns_type_t addr_type, char *ip,
unsigned int ping_time, void *user_ptr);
/* query domain */ /* query domain */
int dns_server_query(char *domain, int qtype, dns_result_callback callback, void *user_ptr); int dns_server_query(char *domain, int qtype, uint32_t server_flags, dns_result_callback callback, void *user_ptr);
#ifdef __cpluscplus #ifdef __cpluscplus
} }

File diff suppressed because it is too large Load Diff

View File

@@ -1,40 +1,60 @@
#ifndef FAST_PING_H /*************************************************************************
#define FAST_PING_H *
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#include <netdb.h> *
#include <sys/time.h> * smartdns is free software: you can redistribute it and/or modify
#ifdef __cpluscplus * it under the terms of the GNU General Public License as published by
extern "C" { * the Free Software Foundation, either version 3 of the License, or
#endif * (at your option) any later version.
*
typedef enum { * smartdns is distributed in the hope that it will be useful,
PING_TYPE_ICMP = 1, * but WITHOUT ANY WARRANTY; without even the implied warranty of
PING_TYPE_TCP = 2, * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
PING_TYPE_DNS = 3, * GNU General Public License for more details.
} PING_TYPE; *
* You should have received a copy of the GNU General Public License
typedef enum { * along with this program. If not, see <http://www.gnu.org/licenses/>.
PING_RESULT_RESPONSE = 1, */
PING_RESULT_TIMEOUT = 2,
PING_RESULT_END = 3, #ifndef FAST_PING_H
} FAST_PING_RESULT; #define FAST_PING_H
struct ping_host_struct; #include <netdb.h>
typedef void (*fast_ping_result)(struct ping_host_struct *ping_host, const char *host, FAST_PING_RESULT result, struct sockaddr *addr, socklen_t addr_len, #include <sys/time.h>
int seqno, int ttl, struct timeval *tv, void *userptr); #ifdef __cpluscplus
extern "C" {
/* start ping */ #endif
struct ping_host_struct *fast_ping_start(PING_TYPE type, const char *host, int count, int interval, int timeout, fast_ping_result ping_callback, void *userptr);
typedef enum {
/* stop ping */ PING_TYPE_ICMP = 1,
int fast_ping_stop(struct ping_host_struct *ping_host); PING_TYPE_TCP = 2,
PING_TYPE_DNS = 3,
int fast_ping_init(void); } PING_TYPE;
void fast_ping_exit(void); typedef enum {
PING_RESULT_RESPONSE = 1,
#ifdef __cpluscplus PING_RESULT_TIMEOUT = 2,
} PING_RESULT_END = 3,
#endif } FAST_PING_RESULT;
#endif // !FAST_PING_H struct ping_host_struct;
typedef void (*fast_ping_result)(struct ping_host_struct *ping_host, const char *host, FAST_PING_RESULT result,
struct sockaddr *addr, socklen_t addr_len, int seqno, int ttl, struct timeval *tv,
void *userptr);
/* start ping */
struct ping_host_struct *fast_ping_start(PING_TYPE type, const char *host, int count, int interval, int timeout,
fast_ping_result ping_callback, void *userptr);
/* stop ping */
int fast_ping_stop(struct ping_host_struct *ping_host);
int fast_ping_init(void);
void fast_ping_exit(void);
#ifdef __cpluscplus
}
#endif
#endif // !FAST_PING_H

View File

@@ -1,445 +1,462 @@
#include "http_parse.h" /*************************************************************************
#include "hash.h" *
#include "hashtable.h" * Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#include "jhash.h" *
#include "list.h" * smartdns is free software: you can redistribute it and/or modify
#include <stdlib.h> * it under the terms of the GNU General Public License as published by
#include <string.h> * the Free Software Foundation, either version 3 of the License, or
#include <unistd.h> * (at your option) any later version.
*
struct http_head_fields { * smartdns is distributed in the hope that it will be useful,
struct hlist_node node; * but WITHOUT ANY WARRANTY; without even the implied warranty of
struct list_head list; * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
char *name; *
char *value; * You should have received a copy of the GNU General Public License
}; * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
struct http_head {
HTTP_HEAD_TYPE head_type; #include "http_parse.h"
HTTP_METHOD method; #include "hash.h"
char *url; #include "hashtable.h"
char *version; #include "util.h"
int code; #include "jhash.h"
char *code_msg; #include "list.h"
int buff_size; #include <stdlib.h>
int buff_len; #include <string.h>
char *buff; #include <unistd.h>
int head_ok;
int head_len; struct http_head_fields {
char *data; struct hlist_node node;
int data_len; struct list_head list;
int expect_data_len;
struct http_head_fields field_head; char *name;
DECLARE_HASHTABLE(field_map, 4); char *value;
}; };
/* struct http_head {
* Returns: HTTP_HEAD_TYPE head_type;
* >=0 - success http data len HTTP_METHOD method;
* -1 - Incomplete request char *url;
* -2 - parse failed char *version;
*/ int code;
struct http_head *http_head_init(int buffsize) char *code_msg;
{ int buff_size;
struct http_head *http_head = NULL; int buff_len;
char *buffer = NULL; char *buff;
int head_ok;
http_head = malloc(sizeof(*http_head)); int head_len;
if (http_head == NULL) { char *data;
goto errout; int data_len;
} int expect_data_len;
memset(http_head, 0, sizeof(*http_head)); struct http_head_fields field_head;
INIT_LIST_HEAD(&http_head->field_head.list); DECLARE_HASHTABLE(field_map, 4);
hash_init(http_head->field_map); };
buffer = malloc(buffsize); /*
if (buffer == NULL) { * Returns:
goto errout; * >=0 - success http data len
} * -1 - Incomplete request
* -2 - parse failed
http_head->buff = buffer; */
http_head->buff_size = buffsize; struct http_head *http_head_init(int buffsize)
{
return http_head; struct http_head *http_head = NULL;
char *buffer = NULL;
errout:
if (buffer) { http_head = malloc(sizeof(*http_head));
free(buffer); if (http_head == NULL) {
} goto errout;
}
if (http_head) { memset(http_head, 0, sizeof(*http_head));
free(http_head); INIT_LIST_HEAD(&http_head->field_head.list);
} hash_init(http_head->field_map);
return NULL; buffer = malloc(buffsize);
} if (buffer == NULL) {
goto errout;
struct http_head_fields *http_head_first_fields(struct http_head *http_head) }
{
struct http_head_fields *first = NULL; http_head->buff = buffer;
first = list_first_entry(&http_head->field_head.list, struct http_head_fields, list); http_head->buff_size = buffsize;
if (first->name == NULL && first->value == NULL) { return http_head;
return NULL;
} errout:
if (buffer) {
return first; free(buffer);
} }
const char *http_head_get_fields_value(struct http_head *http_head, const char *name) if (http_head) {
{ free(http_head);
unsigned long key; }
struct http_head_fields *filed;
return NULL;
key = hash_string(name); }
hash_for_each_possible(http_head->field_map, filed, node, key)
{ struct http_head_fields *http_head_first_fields(struct http_head *http_head)
if (strncmp(filed->name, name, 128) == 0) { {
return filed->value; struct http_head_fields *first = NULL;
} first = list_first_entry(&http_head->field_head.list, struct http_head_fields, list);
}
if (first->name == NULL && first->value == NULL) {
return NULL; return NULL;
} }
struct http_head_fields *http_head_next_fields(struct http_head_fields *fields) return first;
{ }
struct http_head_fields *next = NULL;
next = list_next_entry(fields, list); const char *http_head_get_fields_value(struct http_head *http_head, const char *name)
{
if (next->name == NULL && next->value == NULL) { unsigned long key;
return NULL; struct http_head_fields *filed;
}
key = hash_string(name);
return next; hash_for_each_possible(http_head->field_map, filed, node, key)
} {
if (strncmp(filed->name, name, 128) == 0) {
int http_head_lookup_fields(struct http_head_fields *fields, const char **name, const char **value) return filed->value;
{ }
if (fields == NULL) { }
return -1;
} return NULL;
}
if (name) {
*name = fields->name; struct http_head_fields *http_head_next_fields(struct http_head_fields *fields)
} {
struct http_head_fields *next = NULL;
if (value) { next = list_next_entry(fields, list);
*value = fields->value;
} if (next->name == NULL && next->value == NULL) {
return NULL;
return 0; }
}
return next;
HTTP_METHOD http_head_get_method(struct http_head *http_head) }
{
return http_head->method; int http_head_lookup_fields(struct http_head_fields *fields, const char **name, const char **value)
} {
if (fields == NULL) {
const char *http_head_get_url(struct http_head *http_head) return -1;
{ }
return http_head->url;
} if (name) {
*name = fields->name;
const char *http_head_get_httpversion(struct http_head *http_head) }
{
return http_head->version; if (value) {
} *value = fields->value;
}
int http_head_get_httpcode(struct http_head *http_head)
{ return 0;
return http_head->code; }
}
HTTP_METHOD http_head_get_method(struct http_head *http_head)
char *http_head_get_httpcode_msg(struct http_head *http_head) {
{ return http_head->method;
return http_head->code_msg; }
}
const char *http_head_get_url(struct http_head *http_head)
HTTP_HEAD_TYPE http_head_get_head_type(struct http_head *http_head) {
{ return http_head->url;
return http_head->head_type; }
}
const char *http_head_get_httpversion(struct http_head *http_head)
char *http_head_get_data(struct http_head *http_head) {
{ return http_head->version;
return http_head->data; }
}
int http_head_get_httpcode(struct http_head *http_head)
int http_head_get_data_len(struct http_head *http_head) {
{ return http_head->code;
return http_head->data_len; }
}
char *http_head_get_httpcode_msg(struct http_head *http_head)
static int _http_head_add_fields(struct http_head *http_head, char *name, char *value) {
{ return http_head->code_msg;
unsigned long key = 0; }
struct http_head_fields *fields = NULL;
fields = malloc(sizeof(*fields)); HTTP_HEAD_TYPE http_head_get_head_type(struct http_head *http_head)
if (fields == NULL) { {
return -1; return http_head->head_type;
} }
memset(fields, 0, sizeof(*fields));
char *http_head_get_data(struct http_head *http_head)
fields->name = name; {
fields->value = value; return http_head->data;
}
list_add_tail(&fields->list, &http_head->field_head.list);
key = hash_string(name); int http_head_get_data_len(struct http_head *http_head)
hash_add(http_head->field_map, &fields->node, key); {
return http_head->data_len;
return 0; }
}
static int _http_head_add_fields(struct http_head *http_head, char *name, char *value)
static int _http_head_parse_response(struct http_head *http_head, char *key, char *value) {
{ unsigned long key = 0;
char *field_start = NULL; struct http_head_fields *fields = NULL;
char *tmp_ptr = NULL; fields = malloc(sizeof(*fields));
char *result = NULL; if (fields == NULL) {
char *ret_code = NULL; return -1;
}
if (strstr(key, "HTTP/") == NULL) { memset(fields, 0, sizeof(*fields));
return -1;
} fields->name = name;
fields->value = value;
for (tmp_ptr = value; *tmp_ptr != 0; tmp_ptr++) {
if (field_start == NULL) { list_add_tail(&fields->list, &http_head->field_head.list);
field_start = tmp_ptr; key = hash_string(name);
} hash_add(http_head->field_map, &fields->node, key);
if (*tmp_ptr == ' ') { return 0;
*tmp_ptr = '\0'; }
if (ret_code == NULL) {
ret_code = field_start; static int _http_head_parse_response(struct http_head *http_head, char *key, char *value)
} else if (result == NULL) { {
result = field_start; char *field_start = NULL;
break; char *tmp_ptr = NULL;
} char *ret_msg = NULL;
char *ret_code = NULL;
field_start = NULL;
} if (strstr(key, "HTTP/") == NULL) {
} return -1;
}
if (field_start && result == NULL) {
result = field_start; for (tmp_ptr = value; *tmp_ptr != 0; tmp_ptr++) {
} if (field_start == NULL) {
field_start = tmp_ptr;
if (ret_code == NULL || result == NULL) { }
return -1;
} if (*tmp_ptr != ' ') {
continue;
http_head->code = atol(ret_code); }
http_head->code_msg = result;
http_head->version = key; *tmp_ptr = '\0';
http_head->head_type = HTTP_HEAD_RESPONSE; ret_code = field_start;
ret_msg = tmp_ptr + 1;
return 0; field_start = NULL;
} break;
}
static int _http_head_parse_request(struct http_head *http_head, char *key, char *value)
{ if (ret_code == NULL || ret_msg == NULL) {
int method = HTTP_METHOD_INVALID; return -1;
char *url = NULL; }
char *version = NULL;
char *tmp_ptr = value; if (is_numeric(ret_code) != 0) {
char *field_start = NULL; return -1;
}
if (strncmp(key, "GET", sizeof("GET")) == 0) {
method = HTTP_METHOD_GET; http_head->code = atol(ret_code);
} else if (strncmp(key, "POST", sizeof("POST")) == 0) { http_head->code_msg = ret_msg;
method = HTTP_METHOD_POST; http_head->version = key;
} else if (strncmp(key, "PUT", sizeof("PUT")) == 0) { http_head->head_type = HTTP_HEAD_RESPONSE;
method = HTTP_METHOD_PUT;
} else if (strncmp(key, "DELETE", sizeof("DELETE")) == 0) { return 0;
method = HTTP_METHOD_DELETE; }
} else if (strncmp(key, "TRACE", sizeof("TRACE")) == 0) {
method = HTTP_METHOD_TRACE; static int _http_head_parse_request(struct http_head *http_head, char *key, char *value)
} else if (strncmp(key, "CONNECT", sizeof("CONNECT")) == 0) { {
method = HTTP_METHOD_CONNECT; int method = HTTP_METHOD_INVALID;
} else { char *url = NULL;
return _http_head_parse_response(http_head, key, value); char *version = NULL;
} char *tmp_ptr = value;
char *field_start = NULL;
for (tmp_ptr = value; *tmp_ptr != 0; tmp_ptr++) {
if (field_start == NULL) { if (strncmp(key, "GET", sizeof("GET")) == 0) {
field_start = tmp_ptr; method = HTTP_METHOD_GET;
} } else if (strncmp(key, "POST", sizeof("POST")) == 0) {
if (*tmp_ptr == ' ') { method = HTTP_METHOD_POST;
*tmp_ptr = '\0'; } else if (strncmp(key, "PUT", sizeof("PUT")) == 0) {
if (url == NULL) { method = HTTP_METHOD_PUT;
url = field_start; } else if (strncmp(key, "DELETE", sizeof("DELETE")) == 0) {
} method = HTTP_METHOD_DELETE;
} else if (strncmp(key, "TRACE", sizeof("TRACE")) == 0) {
field_start = NULL; method = HTTP_METHOD_TRACE;
} } else if (strncmp(key, "CONNECT", sizeof("CONNECT")) == 0) {
} method = HTTP_METHOD_CONNECT;
} else {
if (field_start && version == NULL) { return _http_head_parse_response(http_head, key, value);
version = field_start; }
}
for (tmp_ptr = value; *tmp_ptr != 0; tmp_ptr++) {
http_head->method = method; if (field_start == NULL) {
http_head->url = url; field_start = tmp_ptr;
http_head->version = version; }
http_head->head_type = HTTP_HEAD_REQUEST; if (*tmp_ptr == ' ') {
*tmp_ptr = '\0';
return 0; if (url == NULL) {
} url = field_start;
}
static int _http_head_parse(struct http_head *http_head)
{ field_start = NULL;
int i = 0; }
char *key = NULL; }
char *value = NULL;
char *data; if (field_start && version == NULL) {
int has_first_line = 0; version = field_start;
}
int inkey = 1;
int invalue = 0; http_head->method = method;
http_head->url = url;
data = http_head->buff; http_head->version = version;
for (i = 0; i < http_head->head_len; i++, data++) { http_head->head_type = HTTP_HEAD_REQUEST;
if (inkey) {
if (key == NULL && *data != ' ' && *data != '\r' && *data != '\n') { return 0;
key = data; }
continue;
} static int _http_head_parse(struct http_head *http_head)
{
if (*data == ':' || *data == ' ') { int i = 0;
*data = '\0'; char *key = NULL;
inkey = 0; char *value = NULL;
invalue = 1; char *data;
continue; int has_first_line = 0;
}
} int inkey = 1;
int invalue = 0;
if (invalue) {
if (value == NULL && *data != ' ') { data = http_head->buff;
value = data; for (i = 0; i < http_head->head_len; i++, data++) {
continue; if (inkey) {
} if (key == NULL && *data != ' ' && *data != '\r' && *data != '\n') {
key = data;
if (*data == '\r' || *data == '\n') { continue;
*data = '\0'; }
inkey = 1;
invalue = 0; if (*data == ':' || *data == ' ') {
} *data = '\0';
} inkey = 0;
invalue = 1;
if (key && value && invalue == 0) { continue;
if (has_first_line == 0) { }
if (_http_head_parse_request(http_head, key, value) != 0) { }
return -2;
} if (invalue) {
if (value == NULL && *data != ' ') {
has_first_line = 1; value = data;
} else { continue;
if (_http_head_add_fields(http_head, key, value) != 0) { }
return -2;
} if (*data == '\r' || *data == '\n') {
} *data = '\0';
inkey = 1;
key = NULL; invalue = 0;
value = NULL; }
inkey = 1; }
invalue = 0;
} if (key && value && invalue == 0) {
} if (has_first_line == 0) {
if (_http_head_parse_request(http_head, key, value) != 0) {
return 0; return -2;
} }
int http_head_parse(struct http_head *http_head, const char *data, int data_len) has_first_line = 1;
{ } else {
int i = 0; if (_http_head_add_fields(http_head, key, value) != 0) {
char *buff_end = NULL; return -2;
int left_size = 0; }
int process_data_len = 0; }
left_size = http_head->buff_size - http_head->buff_len; key = NULL;
value = NULL;
if (left_size < data_len) { inkey = 1;
return -3; invalue = 0;
} }
}
buff_end = http_head->buff + http_head->buff_len;
if (http_head->head_ok == 0) { return 0;
for (i = 0; i < data_len; i++, data++) { }
*(buff_end + i) = *data;
if (*data == '\n') { int http_head_parse(struct http_head *http_head, const char *data, int data_len)
if (http_head->buff_len + i < 2) { {
continue; int i = 0;
} char *buff_end = NULL;
int left_size = 0;
if (*(buff_end + i - 2) == '\n') { int process_data_len = 0;
http_head->head_ok = 1;
http_head->head_len = http_head->buff_len + i - 2; left_size = http_head->buff_size - http_head->buff_len;
i++;
buff_end += i; if (left_size < data_len) {
data_len -= i; return -3;
data++; }
if (_http_head_parse(http_head) != 0) {
return -2; buff_end = http_head->buff + http_head->buff_len;
} if (http_head->head_ok == 0) {
for (i = 0; i < data_len; i++, data++) {
const char *content_len = NULL; *(buff_end + i) = *data;
content_len = http_head_get_fields_value(http_head, "Content-Length"); if (*data == '\n') {
if (content_len) { if (http_head->buff_len + i < 2) {
http_head->expect_data_len = atol(content_len); continue;
} else { }
http_head->expect_data_len = 0;
} if (*(buff_end + i - 2) == '\n') {
http_head->head_ok = 1;
if (http_head->expect_data_len < 0) { http_head->head_len = http_head->buff_len + i - 2;
return -2; i++;
} buff_end += i;
data_len -= i;
break; data++;
} if (_http_head_parse(http_head) != 0) {
} return -2;
} }
process_data_len += i; const char *content_len = NULL;
if (http_head->head_ok == 0) { content_len = http_head_get_fields_value(http_head, "Content-Length");
// Read data again */ if (content_len) {
http_head->buff_len += process_data_len; http_head->expect_data_len = atol(content_len);
return -1; } else {
} http_head->expect_data_len = 0;
} }
if (http_head->head_ok == 1) { if (http_head->expect_data_len < 0) {
int get_data_len = (http_head->expect_data_len > data_len) ? data_len : http_head->expect_data_len; return -2;
if (http_head->data == NULL) { }
http_head->data = buff_end;
} break;
}
memcpy(buff_end, data, get_data_len); }
process_data_len += get_data_len; }
http_head->data_len += get_data_len;
} process_data_len += i;
if (http_head->head_ok == 0) {
http_head->buff_len += process_data_len; // Read data again */
if (http_head->data_len < http_head->expect_data_len) { http_head->buff_len += process_data_len;
return -1; return -1;
} }
}
return process_data_len;
} if (http_head->head_ok == 1) {
int get_data_len = (http_head->expect_data_len > data_len) ? data_len : http_head->expect_data_len;
void http_head_destroy(struct http_head *http_head) if (http_head->data == NULL) {
{ http_head->data = buff_end;
struct http_head_fields *fields, *tmp; }
list_for_each_entry_safe(fields, tmp, &http_head->field_head.list, list) memcpy(buff_end, data, get_data_len);
{ process_data_len += get_data_len;
list_del(&fields->list); http_head->data_len += get_data_len;
free(fields); }
}
http_head->buff_len += process_data_len;
if (http_head->buff) { if (http_head->data_len < http_head->expect_data_len) {
free(http_head->buff); return -1;
} }
free(http_head); return process_data_len;
} }
void http_head_destroy(struct http_head *http_head)
{
struct http_head_fields *fields, *tmp;
list_for_each_entry_safe(fields, tmp, &http_head->field_head.list, list)
{
list_del(&fields->list);
free(fields);
}
if (http_head->buff) {
free(http_head->buff);
}
free(http_head);
}

View File

@@ -1,69 +1,87 @@
#ifndef HTTP_PARSER_H /*************************************************************************
#define HTTP_PARSER_H *
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#ifdef __cpluscplus *
extern "C" { * smartdns is free software: you can redistribute it and/or modify
#endif * it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
struct http_head; * (at your option) any later version.
struct http_head_fields; *
* smartdns is distributed in the hope that it will be useful,
typedef enum HTTP_METHOD { * but WITHOUT ANY WARRANTY; without even the implied warranty of
HTTP_METHOD_INVALID = 0, * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
HTTP_METHOD_GET, * GNU General Public License for more details.
HTTP_METHOD_HEAD, *
HTTP_METHOD_POST, * You should have received a copy of the GNU General Public License
HTTP_METHOD_PUT, * along with this program. If not, see <http://www.gnu.org/licenses/>.
HTTP_METHOD_DELETE, */
HTTP_METHOD_TRACE,
HTTP_METHOD_CONNECT, #ifndef HTTP_PARSER_H
} HTTP_METHOD; #define HTTP_PARSER_H
typedef enum HTTP_HEAD_TYPE { #ifdef __cpluscplus
HTTP_HEAD_INVALID = 0, extern "C" {
HTTP_HEAD_REQUEST = 1, #endif
HTTP_HEAD_RESPONSE = 2,
} HTTP_HEAD_TYPE; struct http_head;
struct http_head_fields;
struct http_head *http_head_init(int buffsize);
typedef enum HTTP_METHOD {
HTTP_HEAD_TYPE http_head_get_head_type(struct http_head *http_head); HTTP_METHOD_INVALID = 0,
HTTP_METHOD_GET,
HTTP_METHOD http_head_get_method(struct http_head *http_head); HTTP_METHOD_HEAD,
HTTP_METHOD_POST,
const char *http_head_get_url(struct http_head *http_head); HTTP_METHOD_PUT,
HTTP_METHOD_DELETE,
const char *http_head_get_httpversion(struct http_head *http_head); HTTP_METHOD_TRACE,
HTTP_METHOD_CONNECT,
int http_head_get_httpcode(struct http_head *http_head); } HTTP_METHOD;
char *http_head_get_httpcode_msg(struct http_head *http_head); typedef enum HTTP_HEAD_TYPE {
HTTP_HEAD_INVALID = 0,
char *http_head_get_data(struct http_head *http_head); HTTP_HEAD_REQUEST = 1,
HTTP_HEAD_RESPONSE = 2,
int http_head_get_data_len(struct http_head *http_head); } HTTP_HEAD_TYPE;
struct http_head_fields *http_head_first_fields(struct http_head *http_head); struct http_head *http_head_init(int buffsize);
struct http_head_fields *http_head_next_fields(struct http_head_fields *fields); HTTP_HEAD_TYPE http_head_get_head_type(struct http_head *http_head);
const char *http_head_get_fields_value(struct http_head *http_head, const char *name); HTTP_METHOD http_head_get_method(struct http_head *http_head);
int http_head_lookup_fields(struct http_head_fields *fields, const char **name, const char **value); const char *http_head_get_url(struct http_head *http_head);
/* const char *http_head_get_httpversion(struct http_head *http_head);
* Returns:
* >=0 - success http data len int http_head_get_httpcode(struct http_head *http_head);
* -1 - Incomplete request
* -2 - parse failed char *http_head_get_httpcode_msg(struct http_head *http_head);
* -3 - buffer is small
*/ char *http_head_get_data(struct http_head *http_head);
int http_head_parse(struct http_head *http_head, const char *data, int data_len);
int http_head_get_data_len(struct http_head *http_head);
void http_head_destroy(struct http_head *http_head);
struct http_head_fields *http_head_first_fields(struct http_head *http_head);
#ifdef __cpluscplus
} struct http_head_fields *http_head_next_fields(struct http_head_fields *fields);
#endif
const char *http_head_get_fields_value(struct http_head *http_head, const char *name);
#endif // !HTTP_PARSER_H
int http_head_lookup_fields(struct http_head_fields *fields, const char **name, const char **value);
/*
* Returns:
* >=0 - success http data len
* -1 - Incomplete request
* -2 - parse failed
* -3 - buffer is small
*/
int http_head_parse(struct http_head *http_head, const char *data, int data_len);
void http_head_destroy(struct http_head *http_head);
#ifdef __cpluscplus
}
#endif
#endif // !HTTP_PARSER_H

View File

@@ -1,251 +1,251 @@
/* /*
Copyright (c) 2012, Armon Dadgar Copyright (c) 2012, Armon Dadgar
All rights reserved. All rights reserved.
Redistribution and use in source and binary forms, with or without Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met: modification, are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright * Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer. notice, this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright * Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in the notice, this list of conditions and the following disclaimer in the
documentation and/or other materials provided with the distribution. documentation and/or other materials provided with the distribution.
* Neither the name of the organization nor the * Neither the name of the organization nor the
names of its contributors may be used to endorse or promote products names of its contributors may be used to endorse or promote products
derived from this software without specific prior written permission. derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL ARMON DADGAR BE LIABLE FOR ANY DISCLAIMED. IN NO EVENT SHALL ARMON DADGAR BE LIABLE FOR ANY
DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/ */
#include <stdint.h> #include <stdint.h>
#ifndef ART_H #ifndef ART_H
#define ART_H #define ART_H
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
#endif #endif
#define NODE4 1 #define NODE4 1
#define NODE16 2 #define NODE16 2
#define NODE48 3 #define NODE48 3
#define NODE256 4 #define NODE256 4
#define MAX_PREFIX_LEN 10 #define MAX_PREFIX_LEN 10
#if defined(__GNUC__) && !defined(__clang__) #if defined(__GNUC__) && !defined(__clang__)
# if __STDC_VERSION__ >= 199901L && 402 == (__GNUC__ * 100 + __GNUC_MINOR__) # if __STDC_VERSION__ >= 199901L && 402 == (__GNUC__ * 100 + __GNUC_MINOR__)
/* /*
* GCC 4.2.2's C99 inline keyword support is pretty broken; avoid. Introduced in * GCC 4.2.2's C99 inline keyword support is pretty broken; avoid. Introduced in
* GCC 4.2.something, fixed in 4.3.0. So checking for specific major.minor of * GCC 4.2.something, fixed in 4.3.0. So checking for specific major.minor of
* 4.2 is fine. * 4.2 is fine.
*/ */
# define BROKEN_GCC_C99_INLINE # define BROKEN_GCC_C99_INLINE
# endif # endif
#endif #endif
typedef int(*art_callback)(void *data, const unsigned char *key, uint32_t key_len, void *value); typedef int(*art_callback)(void *data, const unsigned char *key, uint32_t key_len, void *value);
/** /**
* This struct is included as part * This struct is included as part
* of all the various node sizes * of all the various node sizes
*/ */
typedef struct { typedef struct {
uint8_t type; uint8_t type;
uint8_t num_children; uint8_t num_children;
uint32_t partial_len; uint32_t partial_len;
unsigned char partial[MAX_PREFIX_LEN]; unsigned char partial[MAX_PREFIX_LEN];
} art_node; } art_node;
/** /**
* Small node with only 4 children * Small node with only 4 children
*/ */
typedef struct { typedef struct {
art_node n; art_node n;
unsigned char keys[4]; unsigned char keys[4];
art_node *children[4]; art_node *children[4];
} art_node4; } art_node4;
/** /**
* Node with 16 children * Node with 16 children
*/ */
typedef struct { typedef struct {
art_node n; art_node n;
unsigned char keys[16]; unsigned char keys[16];
art_node *children[16]; art_node *children[16];
} art_node16; } art_node16;
/** /**
* Node with 48 children, but * Node with 48 children, but
* a full 256 byte field. * a full 256 byte field.
*/ */
typedef struct { typedef struct {
art_node n; art_node n;
unsigned char keys[256]; unsigned char keys[256];
art_node *children[48]; art_node *children[48];
} art_node48; } art_node48;
/** /**
* Full node with 256 children * Full node with 256 children
*/ */
typedef struct { typedef struct {
art_node n; art_node n;
art_node *children[256]; art_node *children[256];
} art_node256; } art_node256;
/** /**
* Represents a leaf. These are * Represents a leaf. These are
* of arbitrary size, as they include the key. * of arbitrary size, as they include the key.
*/ */
typedef struct { typedef struct {
void *value; void *value;
uint32_t key_len; uint32_t key_len;
unsigned char key[0]; unsigned char key[0];
} art_leaf; } art_leaf;
/** /**
* Main struct, points to root. * Main struct, points to root.
*/ */
typedef struct { typedef struct {
art_node *root; art_node *root;
uint64_t size; uint64_t size;
} art_tree; } art_tree;
/** /**
* Initializes an ART tree * Initializes an ART tree
* @return 0 on success. * @return 0 on success.
*/ */
int art_tree_init(art_tree *t); int art_tree_init(art_tree *t);
/** /**
* DEPRECATED * DEPRECATED
* Initializes an ART tree * Initializes an ART tree
* @return 0 on success. * @return 0 on success.
*/ */
#define init_art_tree(...) art_tree_init(__VA_ARGS__) #define init_art_tree(...) art_tree_init(__VA_ARGS__)
/** /**
* Destroys an ART tree * Destroys an ART tree
* @return 0 on success. * @return 0 on success.
*/ */
int art_tree_destroy(art_tree *t); int art_tree_destroy(art_tree *t);
/** /**
* DEPRECATED * DEPRECATED
* Initializes an ART tree * Initializes an ART tree
* @return 0 on success. * @return 0 on success.
*/ */
#define destroy_art_tree(...) art_tree_destroy(__VA_ARGS__) #define destroy_art_tree(...) art_tree_destroy(__VA_ARGS__)
/** /**
* Returns the size of the ART tree. * Returns the size of the ART tree.
*/ */
#ifdef BROKEN_GCC_C99_INLINE #ifdef BROKEN_GCC_C99_INLINE
# define art_size(t) ((t)->size) # define art_size(t) ((t)->size)
#else #else
static inline uint64_t art_size(art_tree *t) { static inline uint64_t art_size(art_tree *t) {
return t->size; return t->size;
} }
#endif #endif
/** /**
* Inserts a new value into the ART tree * Inserts a new value into the ART tree
* @arg t The tree * @arg t The tree
* @arg key The key * @arg key The key
* @arg key_len The length of the key * @arg key_len The length of the key
* @arg value Opaque value. * @arg value Opaque value.
* @return NULL if the item was newly inserted, otherwise * @return NULL if the item was newly inserted, otherwise
* the old value pointer is returned. * the old value pointer is returned.
*/ */
void* art_insert(art_tree *t, const unsigned char *key, int key_len, void *value); void* art_insert(art_tree *t, const unsigned char *key, int key_len, void *value);
/** /**
* Deletes a value from the ART tree * Deletes a value from the ART tree
* @arg t The tree * @arg t The tree
* @arg key The key * @arg key The key
* @arg key_len The length of the key * @arg key_len The length of the key
* @return NULL if the item was not found, otherwise * @return NULL if the item was not found, otherwise
* the value pointer is returned. * the value pointer is returned.
*/ */
void* art_delete(art_tree *t, const unsigned char *key, int key_len); void* art_delete(art_tree *t, const unsigned char *key, int key_len);
/** /**
* Searches for a value in the ART tree * Searches for a value in the ART tree
* @arg t The tree * @arg t The tree
* @arg key The key * @arg key The key
* @arg key_len The length of the key * @arg key_len The length of the key
* @return NULL if the item was not found, otherwise * @return NULL if the item was not found, otherwise
* the value pointer is returned. * the value pointer is returned.
*/ */
void* art_search(const art_tree *t, const unsigned char *key, int key_len); void* art_search(const art_tree *t, const unsigned char *key, int key_len);
/** /**
* Searches substring for a value in the ART tree * Searches substring for a value in the ART tree
* @arg t The tree * @arg t The tree
* @arg str The key * @arg str The key
* @arg str_len The length of the key * @arg str_len The length of the key
* @return NULL if the item was not found, otherwise * @return NULL if the item was not found, otherwise
* the value pointer is returned. * the value pointer is returned.
*/ */
void *art_substring(const art_tree *t, const unsigned char *str, int str_len, unsigned char *key, int *key_len); void *art_substring(const art_tree *t, const unsigned char *str, int str_len, unsigned char *key, int *key_len);
/** /**
* Wakk substring for a value in the ART tree * Wakk substring for a value in the ART tree
* @arg t The tree * @arg t The tree
* @arg str The key * @arg str The key
* @arg str_len The length of the key * @arg str_len The length of the key
* @return NULL if the item was not found, otherwise * @return NULL if the item was not found, otherwise
* the value pointer is returned. * the value pointer is returned.
*/ */
typedef int (*walk_func)(unsigned char *key, uint32_t key_len, void *value, void *arg); typedef int (*walk_func)(unsigned char *key, uint32_t key_len, void *value, void *arg);
void art_substring_walk(const art_tree *t, const unsigned char *str, int str_len, walk_func func, void *arg); void art_substring_walk(const art_tree *t, const unsigned char *str, int str_len, walk_func func, void *arg);
/** /**
* Returns the minimum valued leaf * Returns the minimum valued leaf
* @return The minimum leaf or NULL * @return The minimum leaf or NULL
*/ */
art_leaf* art_minimum(art_tree *t); art_leaf* art_minimum(art_tree *t);
/** /**
* Returns the maximum valued leaf * Returns the maximum valued leaf
* @return The maximum leaf or NULL * @return The maximum leaf or NULL
*/ */
art_leaf* art_maximum(art_tree *t); art_leaf* art_maximum(art_tree *t);
/** /**
* Iterates through the entries pairs in the map, * Iterates through the entries pairs in the map,
* invoking a callback for each. The call back gets a * invoking a callback for each. The call back gets a
* key, value for each and returns an integer stop value. * key, value for each and returns an integer stop value.
* If the callback returns non-zero, then the iteration stops. * If the callback returns non-zero, then the iteration stops.
* @arg t The tree to iterate over * @arg t The tree to iterate over
* @arg cb The callback function to invoke * @arg cb The callback function to invoke
* @arg data Opaque handle passed to the callback * @arg data Opaque handle passed to the callback
* @return 0 on success, or the return of the callback. * @return 0 on success, or the return of the callback.
*/ */
int art_iter(art_tree *t, art_callback cb, void *data); int art_iter(art_tree *t, art_callback cb, void *data);
/** /**
* Iterates through the entries pairs in the map, * Iterates through the entries pairs in the map,
* invoking a callback for each that matches a given prefix. * invoking a callback for each that matches a given prefix.
* The call back gets a key, value for each and returns an integer stop value. * The call back gets a key, value for each and returns an integer stop value.
* If the callback returns non-zero, then the iteration stops. * If the callback returns non-zero, then the iteration stops.
* @arg t The tree to iterate over * @arg t The tree to iterate over
* @arg prefix The prefix of keys to read * @arg prefix The prefix of keys to read
* @arg prefix_len The length of the prefix * @arg prefix_len The length of the prefix
* @arg cb The callback function to invoke * @arg cb The callback function to invoke
* @arg data Opaque handle passed to the callback * @arg data Opaque handle passed to the callback
* @return 0 on success, or the return of the callback. * @return 0 on success, or the return of the callback.
*/ */
int art_iter_prefix(art_tree *t, const unsigned char *prefix, int prefix_len, art_callback cb, void *data); int art_iter_prefix(art_tree *t, const unsigned char *prefix, int prefix_len, art_callback cb, void *data);
#ifdef __cplusplus #ifdef __cplusplus
} }
#endif #endif
#endif #endif

View File

@@ -1,12 +1,25 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _GENERIC_ATOMIC_H #ifndef _GENERIC_ATOMIC_H
#define _GENERIC_ATOMIC_H #define _GENERIC_ATOMIC_H
/* Check GCC version, just to be safe */
#if !defined(__GNUC__) || (__GNUC__ < 4) || (__GNUC_MINOR__ < 1)
# error atomic.h works only with GCC newer than version 4.1
#endif /* GNUC >= 4.1 */
/** /**
* Atomic type. * Atomic type.
*/ */

View File

@@ -1,3 +1,22 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _PERF_BITOPS_H #ifndef _PERF_BITOPS_H
#define _PERF_BITOPS_H #define _PERF_BITOPS_H

View File

@@ -1,3 +1,22 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _GENERIC_BITOPS_H_ #ifndef _GENERIC_BITOPS_H_
#define _GENERIC_BITOPS_H_ #define _GENERIC_BITOPS_H_

View File

@@ -1,123 +1,141 @@
#ifndef _GENERIC_CONF_H /*************************************************************************
#define _GENERIC_CONF_H *
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#include <unistd.h> *
* smartdns is free software: you can redistribute it and/or modify
#define MAX_LINE_LEN 1024 * it under the terms of the GNU General Public License as published by
#define MAX_KEY_LEN 64 * the Free Software Foundation, either version 3 of the License, or
#define CONF_INT_MAX (~(1 << 31)) * (at your option) any later version.
#define CONF_INT_MIN (1 << 31) *
* smartdns is distributed in the hope that it will be useful,
#define CONF_RET_OK 0 * but WITHOUT ANY WARRANTY; without even the implied warranty of
#define CONF_RET_ERR -1 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#define CONF_RET_WARN -2 * GNU General Public License for more details.
#define CONF_RET_NOENT -3 *
* You should have received a copy of the GNU General Public License
struct config_item { * along with this program. If not, see <http://www.gnu.org/licenses/>.
const char *item; */
int (*item_func)(const char *item, void *data, int argc, char *argv[]);
void *data; #ifndef _GENERIC_CONF_H
}; #define _GENERIC_CONF_H
struct config_item_custom { #include <unistd.h>
void *custom_data;
int (*custom_func)(void *data, int argc, char *argv[]); #define MAX_LINE_LEN 1024
}; #define MAX_KEY_LEN 64
#define CONF_INT_MAX (~(1 << 31))
struct config_item_int { #define CONF_INT_MIN (1 << 31)
int *data;
int min; #define CONF_RET_OK 0
int max; #define CONF_RET_ERR -1
}; #define CONF_RET_WARN -2
#define CONF_RET_NOENT -3
struct config_item_string {
char *data; struct config_item {
size_t size; const char *item;
}; int (*item_func)(const char *item, void *data, int argc, char *argv[]);
void *data;
struct config_item_yesno { };
int *data;
}; struct config_item_custom {
void *custom_data;
struct config_item_size { int (*custom_func)(void *data, int argc, char *argv[]);
size_t *data; };
size_t min;
size_t max; struct config_item_int {
}; int *data;
int min;
#define CONF_INT(key, value, min_value, max_value) \ int max;
{ \ };
key, conf_int, &(struct config_item_int) \
{ \ struct config_item_string {
.data = value, .min = min_value, .max = max_value \ char *data;
} \ size_t size;
} };
#define CONF_STRING(key, value, len_value) \
{ \ struct config_item_yesno {
key, conf_string, &(struct config_item_string) \ int *data;
{ \ };
.data = value, .size = len_value \
} \ struct config_item_size {
} size_t *data;
#define CONF_YESNO(key, value) \ size_t min;
{ \ size_t max;
key, conf_yesno, &(struct config_item_yesno) \ };
{ \
.data = value \ #define CONF_INT(key, value, min_value, max_value) \
} \ { \
} key, conf_int, &(struct config_item_int) \
#define CONF_SIZE(key, value, min_value, max_value) \ { \
{ \ .data = value, .min = min_value, .max = max_value \
key, conf_size, &(struct config_item_size) \ } \
{ \ }
.data = value, .min = min_value, .max = max_value \ #define CONF_STRING(key, value, len_value) \
} \ { \
} key, conf_string, &(struct config_item_string) \
/* { \
* func: int (*func)(void *data, int argc, char *argv[]); .data = value, .size = len_value \
*/ } \
#define CONF_CUSTOM(key, func, data) \ }
{ \ #define CONF_YESNO(key, value) \
key, conf_custom, &(struct config_item_custom) \ { \
{ \ key, conf_yesno, &(struct config_item_yesno) \
.custom_data = data, .custom_func = func \ { \
} \ .data = value \
} } \
}
#define CONF_END() \ #define CONF_SIZE(key, value, min_value, max_value) \
{ \ { \
NULL, NULL, NULL \ key, conf_size, &(struct config_item_size) \
} { \
.data = value, .min = min_value, .max = max_value \
extern int conf_custom(const char *item, void *data, int argc, char *argv[]); } \
}
extern int conf_int(const char *item, void *data, int argc, char *argv[]); /*
* func: int (*func)(void *data, int argc, char *argv[]);
extern int conf_string(const char *item, void *data, int argc, char *argv[]); */
#define CONF_CUSTOM(key, func, data) \
extern int conf_yesno(const char *item, void *data, int argc, char *argv[]); { \
key, conf_custom, &(struct config_item_custom) \
extern int conf_size(const char *item, void *data, int argc, char *argv[]); { \
.custom_data = data, .custom_func = func \
/* } \
* Example: }
* int num = 0;
* #define CONF_END() \
* struct config_item itmes [] = { { \
* CONF_INT("CONF_NUM", &num, -1, 10), NULL, NULL, NULL \
* CONF_END(); }
* }
* extern int conf_custom(const char *item, void *data, int argc, char *argv[]);
* load_conf(file, items);
* extern int conf_int(const char *item, void *data, int argc, char *argv[]);
*/
extern int conf_string(const char *item, void *data, int argc, char *argv[]);
typedef int(conf_error_handler)(const char *file, int lineno, int ret);
extern int conf_yesno(const char *item, void *data, int argc, char *argv[]);
int load_conf(const char *file, struct config_item items[], conf_error_handler handler);
extern int conf_size(const char *item, void *data, int argc, char *argv[]);
void load_exit(void);
/*
const char *conf_get_conf_file(void); * Example:
* int num = 0;
#endif // !_GENERIC_CONF_H *
* struct config_item itmes [] = {
* CONF_INT("CONF_NUM", &num, -1, 10),
* CONF_END();
* }
*
* load_conf(file, items);
*
*/
typedef int(conf_error_handler)(const char *file, int lineno, int ret);
int load_conf(const char *file, struct config_item items[], conf_error_handler handler);
void load_exit(void);
const char *conf_get_conf_file(void);
#endif // !_GENERIC_CONF_H

View File

@@ -1,5 +1,24 @@
#ifndef _TOOLS_LINUX_ASM_GENERIC_BITOPS_FIND_H_ /*************************************************************************
#define _TOOLS_LINUX_ASM_GENERIC_BITOPS_FIND_H_ *
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _TOOLS_GENERIC_BITOPS_FIND_H_
#define _TOOLS_GENERIC_BITOPS_FIND_H_
#ifndef find_next_bit #ifndef find_next_bit
/** /**
@@ -74,4 +93,4 @@ extern unsigned long find_first_bit(const unsigned long *addr,
unsigned long find_first_zero_bit(const unsigned long *addr, unsigned long size); unsigned long find_first_zero_bit(const unsigned long *addr, unsigned long size);
#endif #endif
#endif /*_TOOLS_LINUX_ASM_GENERIC_BITOPS_FIND_H_ */ #endif /*_TOOLS_GENERIC_BITOPS_FIND_H_ */

View File

@@ -1,3 +1,22 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _TOOLS_LINUX_COMPILER_H_ #ifndef _TOOLS_LINUX_COMPILER_H_
#define _TOOLS_LINUX_COMPILER_H_ #define _TOOLS_LINUX_COMPILER_H_

View File

@@ -1,3 +1,22 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _GENERIC_HASH_H #ifndef _GENERIC_HASH_H
#define _GENERIC_HASH_H #define _GENERIC_HASH_H

View File

@@ -1,6 +1,19 @@
/* /*************************************************************************
* Statically sized hash table implementation *
* (C) 2012 Sasha Levin <levinsasha928@gmail.com> * Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/ */
#ifndef _GENERIC_HASHTABLE_H #ifndef _GENERIC_HASHTABLE_H

View File

@@ -1,3 +1,21 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _JHASH_H #ifndef _JHASH_H
#define _JHASH_H #define _JHASH_H

View File

@@ -1,3 +1,20 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _GENERIC_LIST_H #ifndef _GENERIC_LIST_H
#define _GENERIC_LIST_H #define _GENERIC_LIST_H

View File

@@ -1,162 +1,162 @@
/* /*
* Copyright (c) 1999-2000 * Copyright (c) 1999-2000
* *
* The Regents of the University of Michigan ("The Regents") and * The Regents of the University of Michigan ("The Regents") and
* Merit Network, Inc. All rights reserved. Redistribution and use * Merit Network, Inc. All rights reserved. Redistribution and use
* in source and binary forms, with or without modification, are * in source and binary forms, with or without modification, are
* permitted provided that the following conditions are met: * permitted provided that the following conditions are met:
* *
* 1. Redistributions of source code must retain the above * 1. Redistributions of source code must retain the above
* copyright notice, this list of conditions and the * copyright notice, this list of conditions and the
* following disclaimer. * following disclaimer.
* *
* 2. Redistributions in binary form must reproduce the above * 2. Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the * copyright notice, this list of conditions and the
* following disclaimer in the documentation and/or other * following disclaimer in the documentation and/or other
* materials provided with the distribution. * materials provided with the distribution.
* *
* 3. All advertising materials mentioning features or use of * 3. All advertising materials mentioning features or use of
* this software must display the following acknowledgement: * this software must display the following acknowledgement:
* *
* This product includes software developed by the University of * This product includes software developed by the University of
* Michigan, Merit Network, Inc., and their contributors. * Michigan, Merit Network, Inc., and their contributors.
* *
* 4. Neither the name of the University, Merit Network, nor the * 4. Neither the name of the University, Merit Network, nor the
* names of their contributors may be used to endorse or * names of their contributors may be used to endorse or
* promote products derived from this software without * promote products derived from this software without
* specific prior written permission. * specific prior written permission.
* *
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS "AS IS" * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
* PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL TH E REGENTS * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL TH E REGENTS
* OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HO WEVER CAUSED * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HO WEVER CAUSED
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE. * POSSIBILITY OF SUCH DAMAGE.
*/ */
/* /*
* Portions Copyright (c) 2004,2005 Damien Miller <djm@mindrot.org> * Portions Copyright (c) 2004,2005 Damien Miller <djm@mindrot.org>
* *
* Permission to use, copy, modify, and distribute this software for any * Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above * purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies. * copyright notice and this permission notice appear in all copies.
* *
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/ */
/* $Id: radix.h,v 1.9 2007/10/24 06:03:08 djm Exp $ */ /* $Id: radix.h,v 1.9 2007/10/24 06:03:08 djm Exp $ */
#ifndef _RADIX_H #ifndef _RADIX_H
#define _RADIX_H #define _RADIX_H
#if defined(_MSC_VER) #if defined(_MSC_VER)
#include <winsock2.h> #include <winsock2.h>
#include <ws2tcpip.h> #include <ws2tcpip.h>
#else #else
# include <sys/types.h> # include <sys/types.h>
# include <sys/socket.h> # include <sys/socket.h>
# include <netinet/in.h> # include <netinet/in.h>
# include <arpa/inet.h> # include <arpa/inet.h>
# include <netdb.h> # include <netdb.h>
#endif #endif
#if defined(_MSC_VER) #if defined(_MSC_VER)
# define snprintf _snprintf # define snprintf _snprintf
typedef unsigned __int8 u_int8_t; typedef unsigned __int8 u_int8_t;
typedef unsigned __int16 u_int16_t; typedef unsigned __int16 u_int16_t;
typedef unsigned __int32 u_int32_t; typedef unsigned __int32 u_int32_t;
const char *inet_ntop(int af, const void *src, char *dst, size_t size); const char *inet_ntop(int af, const void *src, char *dst, size_t size);
size_t strlcpy(char *dst, const char *src, size_t size); size_t strlcpy(char *dst, const char *src, size_t size);
#endif #endif
/* /*
* Originally from MRT include/mrt.h * Originally from MRT include/mrt.h
* $MRTId: mrt.h,v 1.1.1.1 2000/08/14 18:46:10 labovit Exp $ * $MRTId: mrt.h,v 1.1.1.1 2000/08/14 18:46:10 labovit Exp $
*/ */
typedef struct _prefix_t { typedef struct _prefix_t {
u_int family; /* AF_INET | AF_INET6 */ u_int family; /* AF_INET | AF_INET6 */
u_int bitlen; /* same as mask? */ u_int bitlen; /* same as mask? */
int ref_count; /* reference count */ int ref_count; /* reference count */
union { union {
struct in_addr sin; struct in_addr sin;
struct in6_addr sin6; struct in6_addr sin6;
} add; } add;
} prefix_t; } prefix_t;
void Deref_Prefix(prefix_t *prefix); void Deref_Prefix(prefix_t *prefix);
/* /*
* Originally from MRT include/radix.h * Originally from MRT include/radix.h
* $MRTId: radix.h,v 1.1.1.1 2000/08/14 18:46:10 labovit Exp $ * $MRTId: radix.h,v 1.1.1.1 2000/08/14 18:46:10 labovit Exp $
*/ */
typedef struct _radix_node_t { typedef struct _radix_node_t {
u_int bit; /* flag if this node used */ u_int bit; /* flag if this node used */
prefix_t *prefix; /* who we are in radix tree */ prefix_t *prefix; /* who we are in radix tree */
struct _radix_node_t *l, *r; /* left and right children */ struct _radix_node_t *l, *r; /* left and right children */
struct _radix_node_t *parent; /* may be used */ struct _radix_node_t *parent; /* may be used */
void *data; /* pointer to data */ void *data; /* pointer to data */
} radix_node_t; } radix_node_t;
typedef struct _radix_tree_t { typedef struct _radix_tree_t {
radix_node_t *head; radix_node_t *head;
u_int maxbits; /* for IP, 32 bit addresses */ u_int maxbits; /* for IP, 32 bit addresses */
int num_active_node; /* for debug purpose */ int num_active_node; /* for debug purpose */
} radix_tree_t; } radix_tree_t;
/* Type of callback function */ /* Type of callback function */
typedef void (*rdx_cb_t)(radix_node_t *, void *); typedef void (*rdx_cb_t)(radix_node_t *, void *);
radix_tree_t *New_Radix(void); radix_tree_t *New_Radix(void);
void Destroy_Radix(radix_tree_t *radix, rdx_cb_t func, void *cbctx); void Destroy_Radix(radix_tree_t *radix, rdx_cb_t func, void *cbctx);
radix_node_t *radix_lookup(radix_tree_t *radix, prefix_t *prefix); radix_node_t *radix_lookup(radix_tree_t *radix, prefix_t *prefix);
void radix_remove(radix_tree_t *radix, radix_node_t *node); void radix_remove(radix_tree_t *radix, radix_node_t *node);
radix_node_t *radix_search_exact(radix_tree_t *radix, prefix_t *prefix); radix_node_t *radix_search_exact(radix_tree_t *radix, prefix_t *prefix);
radix_node_t *radix_search_best(radix_tree_t *radix, prefix_t *prefix); radix_node_t *radix_search_best(radix_tree_t *radix, prefix_t *prefix);
void radix_process(radix_tree_t *radix, rdx_cb_t func, void *cbctx); void radix_process(radix_tree_t *radix, rdx_cb_t func, void *cbctx);
#define RADIX_MAXBITS 128 #define RADIX_MAXBITS 128
#define RADIX_WALK(Xhead, Xnode) \ #define RADIX_WALK(Xhead, Xnode) \
do { \ do { \
radix_node_t *Xstack[RADIX_MAXBITS+1]; \ radix_node_t *Xstack[RADIX_MAXBITS+1]; \
radix_node_t **Xsp = Xstack; \ radix_node_t **Xsp = Xstack; \
radix_node_t *Xrn = (Xhead); \ radix_node_t *Xrn = (Xhead); \
while ((Xnode = Xrn)) { \ while ((Xnode = Xrn)) { \
if (Xnode->prefix) if (Xnode->prefix)
#define RADIX_WALK_END \ #define RADIX_WALK_END \
if (Xrn->l) { \ if (Xrn->l) { \
if (Xrn->r) { \ if (Xrn->r) { \
*Xsp++ = Xrn->r; \ *Xsp++ = Xrn->r; \
} \ } \
Xrn = Xrn->l; \ Xrn = Xrn->l; \
} else if (Xrn->r) { \ } else if (Xrn->r) { \
Xrn = Xrn->r; \ Xrn = Xrn->r; \
} else if (Xsp != Xstack) { \ } else if (Xsp != Xstack) { \
Xrn = *(--Xsp); \ Xrn = *(--Xsp); \
} else { \ } else { \
Xrn = (radix_node_t *) 0; \ Xrn = (radix_node_t *) 0; \
} \ } \
} \ } \
} while (0) } while (0)
/* Local additions */ /* Local additions */
prefix_t *prefix_pton(const char *string, long len, prefix_t *prefix, const char **errmsg); prefix_t *prefix_pton(const char *string, long len, prefix_t *prefix, const char **errmsg);
prefix_t *prefix_from_blob(unsigned char *blob, int len, int prefixlen, prefix_t *prefix); prefix_t *prefix_from_blob(unsigned char *blob, int len, int prefixlen, prefix_t *prefix);
const char *prefix_addr_ntop(prefix_t *prefix, char *buf, size_t len); const char *prefix_addr_ntop(prefix_t *prefix, char *buf, size_t len);
const char *prefix_ntop(prefix_t *prefix, char *buf, size_t len); const char *prefix_ntop(prefix_t *prefix, char *buf, size_t len);
#endif /* _RADIX_H */ #endif /* _RADIX_H */

View File

@@ -1,288 +1,305 @@
/*************************************************************************
#ifndef _GENERIC_RBTREE_H *
#define _GENERIC_RBTREE_H * Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
#include <stddef.h> * smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
struct rb_node { * the Free Software Foundation, either version 3 of the License, or
unsigned long __rb_parent_color; * (at your option) any later version.
struct rb_node *rb_right; *
struct rb_node *rb_left; * smartdns is distributed in the hope that it will be useful,
} __attribute__((aligned(sizeof(long)))); * but WITHOUT ANY WARRANTY; without even the implied warranty of
/* The alignment might seem pointless, but allegedly CRIS needs it */ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
struct rb_root { *
struct rb_node *rb_node; * You should have received a copy of the GNU General Public License
}; * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#define rb_parent(r) ((struct rb_node *)((r)->__rb_parent_color & ~3))
#ifndef _GENERIC_RBTREE_H
#define RB_ROOT \ #define _GENERIC_RBTREE_H
(struct rb_root) \
{ \ #include <stddef.h>
NULL, \
} struct rb_node {
#define rb_entry(ptr, type, member) container_of(ptr, type, member) unsigned long __rb_parent_color;
struct rb_node *rb_right;
#define RB_EMPTY_ROOT(root) ((root)->rb_node == NULL) struct rb_node *rb_left;
} __attribute__((aligned(sizeof(long))));
/* 'empty' nodes are nodes that are known not to be inserted in an rbtree */ /* The alignment might seem pointless, but allegedly CRIS needs it */
#define RB_EMPTY_NODE(node) ((node)->__rb_parent_color == (unsigned long)(node))
#define RB_CLEAR_NODE(node) ((node)->__rb_parent_color = (unsigned long)(node)) struct rb_root {
struct rb_node *rb_node;
extern void rb_insert_color(struct rb_node *, struct rb_root *); };
extern void rb_erase(struct rb_node *, struct rb_root *);
#define rb_parent(r) ((struct rb_node *)((r)->__rb_parent_color & ~3))
/* Find logical next and previous nodes in a tree */
extern struct rb_node *rb_next(const struct rb_node *); #define RB_ROOT \
extern struct rb_node *rb_prev(const struct rb_node *); (struct rb_root) \
extern struct rb_node *rb_first(const struct rb_root *); { \
extern struct rb_node *rb_last(const struct rb_root *); NULL, \
}
/* Postorder iteration - always visit the parent after its children */ #define rb_entry(ptr, type, member) container_of(ptr, type, member)
extern struct rb_node *rb_first_postorder(const struct rb_root *);
extern struct rb_node *rb_next_postorder(const struct rb_node *); #define RB_EMPTY_ROOT(root) ((root)->rb_node == NULL)
/* Fast replacement of a single node without remove/rebalance/add/rebalance */ /* 'empty' nodes are nodes that are known not to be inserted in an rbtree */
extern void rb_replace_node(struct rb_node *victim, struct rb_node *new_node, struct rb_root *root); #define RB_EMPTY_NODE(node) ((node)->__rb_parent_color == (unsigned long)(node))
#define RB_CLEAR_NODE(node) ((node)->__rb_parent_color = (unsigned long)(node))
static inline void rb_link_node(struct rb_node *node, struct rb_node *parent, struct rb_node **rb_link)
{ extern void rb_insert_color(struct rb_node *, struct rb_root *);
node->__rb_parent_color = (unsigned long)parent; extern void rb_erase(struct rb_node *, struct rb_root *);
node->rb_left = node->rb_right = NULL;
/* Find logical next and previous nodes in a tree */
*rb_link = node; extern struct rb_node *rb_next(const struct rb_node *);
} extern struct rb_node *rb_prev(const struct rb_node *);
extern struct rb_node *rb_first(const struct rb_root *);
#define rb_entry_safe(ptr, type, member) \ extern struct rb_node *rb_last(const struct rb_root *);
({ \
typeof(ptr) ____ptr = (ptr); \ /* Postorder iteration - always visit the parent after its children */
____ptr ? rb_entry(____ptr, type, member) : NULL; \ extern struct rb_node *rb_first_postorder(const struct rb_root *);
}) extern struct rb_node *rb_next_postorder(const struct rb_node *);
/* /* Fast replacement of a single node without remove/rebalance/add/rebalance */
* Handy for checking that we are not deleting an entry that is extern void rb_replace_node(struct rb_node *victim, struct rb_node *new_node, struct rb_root *root);
* already in a list, found in block/{blk-throttle,cfq-iosched}.c,
* probably should be moved to lib/rbtree.c... static inline void rb_link_node(struct rb_node *node, struct rb_node *parent, struct rb_node **rb_link)
*/ {
static inline void rb_erase_init(struct rb_node *n, struct rb_root *root) node->__rb_parent_color = (unsigned long)parent;
{ node->rb_left = node->rb_right = NULL;
rb_erase(n, root);
RB_CLEAR_NODE(n); *rb_link = node;
} }
#define rb_entry_safe(ptr, type, member) \
/* ({ \
* Please note - only struct rb_augment_callbacks and the prototypes for typeof(ptr) ____ptr = (ptr); \
* rb_insert_augmented() and rb_erase_augmented() are intended to be public. ____ptr ? rb_entry(____ptr, type, member) : NULL; \
* The rest are implementation details you are not expected to depend on. })
*
* See Documentation/rbtree.txt for documentation and samples. /*
*/ * Handy for checking that we are not deleting an entry that is
* already in a list, found in block/{blk-throttle,cfq-iosched}.c,
struct rb_augment_callbacks { * probably should be moved to lib/rbtree.c...
void (*propagate)(struct rb_node *node, struct rb_node *stop); */
void (*copy)(struct rb_node *old_node, struct rb_node *new_node); static inline void rb_erase_init(struct rb_node *n, struct rb_root *root)
void (*rotate)(struct rb_node *old_node, struct rb_node *new_node); {
}; rb_erase(n, root);
RB_CLEAR_NODE(n);
extern void __rb_insert_augmented(struct rb_node *node, struct rb_root *root, }
void (*augment_rotate)(struct rb_node *old_node, struct rb_node *new_node));
/*
* Fixup the rbtree and update the augmented information when rebalancing. /*
* * Please note - only struct rb_augment_callbacks and the prototypes for
* On insertion, the user must update the augmented information on the path * rb_insert_augmented() and rb_erase_augmented() are intended to be public.
* leading to the inserted node, then call rb_link_node() as usual and * The rest are implementation details you are not expected to depend on.
* rb_augment_inserted() instead of the usual rb_insert_color() call. *
* If rb_augment_inserted() rebalances the rbtree, it will callback into * See Documentation/rbtree.txt for documentation and samples.
* a user provided function to update the augmented information on the */
* affected subtrees.
*/ struct rb_augment_callbacks {
static inline void void (*propagate)(struct rb_node *node, struct rb_node *stop);
rb_insert_augmented(struct rb_node *node, struct rb_root *root, void (*copy)(struct rb_node *old_node, struct rb_node *new_node);
const struct rb_augment_callbacks *augment) void (*rotate)(struct rb_node *old_node, struct rb_node *new_node);
{ };
__rb_insert_augmented(node, root, augment->rotate);
} extern void __rb_insert_augmented(struct rb_node *node, struct rb_root *root,
void (*augment_rotate)(struct rb_node *old_node, struct rb_node *new_node));
#define RB_DECLARE_CALLBACKS(rbstatic, rbname, rbstruct, rbfield, \ /*
rbtype, rbaugmented, rbcompute) \ * Fixup the rbtree and update the augmented information when rebalancing.
static inline void \ *
rbname ## _propagate(struct rb_node *rb, struct rb_node *stop) \ * On insertion, the user must update the augmented information on the path
{ \ * leading to the inserted node, then call rb_link_node() as usual and
while (rb != stop) { \ * rb_augment_inserted() instead of the usual rb_insert_color() call.
rbstruct *node = rb_entry(rb, rbstruct, rbfield); \ * If rb_augment_inserted() rebalances the rbtree, it will callback into
rbtype augmented = rbcompute(node); \ * a user provided function to update the augmented information on the
if (node->rbaugmented == augmented) \ * affected subtrees.
break; \ */
node->rbaugmented = augmented; \ static inline void
rb = rb_parent(&node->rbfield); \ rb_insert_augmented(struct rb_node *node, struct rb_root *root,
} \ const struct rb_augment_callbacks *augment)
} \ {
static inline void \ __rb_insert_augmented(node, root, augment->rotate);
rbname ## _copy(struct rb_node *rb_old, struct rb_node *rb_new) \ }
{ \
rbstruct *old_node = rb_entry(rb_old, rbstruct, rbfield); \ #define RB_DECLARE_CALLBACKS(rbstatic, rbname, rbstruct, rbfield, \
rbstruct *new_node = rb_entry(rb_new, rbstruct, rbfield); \ rbtype, rbaugmented, rbcompute) \
new_node->rbaugmented = old_node->rbaugmented; \ static inline void \
} \ rbname ## _propagate(struct rb_node *rb, struct rb_node *stop) \
static void \ { \
rbname ## _rotate(struct rb_node *rb_old, struct rb_node *rb_new) \ while (rb != stop) { \
{ \ rbstruct *node = rb_entry(rb, rbstruct, rbfield); \
rbstruct *old_node = rb_entry(rb_old, rbstruct, rbfield); \ rbtype augmented = rbcompute(node); \
rbstruct *new_node = rb_entry(rb_new, rbstruct, rbfield); \ if (node->rbaugmented == augmented) \
new_node->rbaugmented = old_node->rbaugmented; \ break; \
old_node->rbaugmented = rbcompute(old_node); \ node->rbaugmented = augmented; \
} \ rb = rb_parent(&node->rbfield); \
rbstatic const struct rb_augment_callbacks rbname = { \ } \
rbname ## _propagate, rbname ## _copy, rbname ## _rotate \ } \
}; static inline void \
rbname ## _copy(struct rb_node *rb_old, struct rb_node *rb_new) \
{ \
#define RB_RED 0 rbstruct *old_node = rb_entry(rb_old, rbstruct, rbfield); \
#define RB_BLACK 1 rbstruct *new_node = rb_entry(rb_new, rbstruct, rbfield); \
new_node->rbaugmented = old_node->rbaugmented; \
#define __rb_parent(pc) ((struct rb_node *)(pc & ~3)) } \
static void \
#define __rb_color(pc) ((pc) & 1) rbname ## _rotate(struct rb_node *rb_old, struct rb_node *rb_new) \
#define __rb_is_black(pc) __rb_color(pc) { \
#define __rb_is_red(pc) (!__rb_color(pc)) rbstruct *old_node = rb_entry(rb_old, rbstruct, rbfield); \
#define rb_color(rb) __rb_color((rb)->__rb_parent_color) rbstruct *new_node = rb_entry(rb_new, rbstruct, rbfield); \
#define rb_is_red(rb) __rb_is_red((rb)->__rb_parent_color) new_node->rbaugmented = old_node->rbaugmented; \
#define rb_is_black(rb) __rb_is_black((rb)->__rb_parent_color) old_node->rbaugmented = rbcompute(old_node); \
} \
static inline void rb_set_parent(struct rb_node *rb, struct rb_node *p) rbstatic const struct rb_augment_callbacks rbname = { \
{ rbname ## _propagate, rbname ## _copy, rbname ## _rotate \
rb->__rb_parent_color = rb_color(rb) | (unsigned long)p; };
}
static inline void rb_set_parent_color(struct rb_node *rb, #define RB_RED 0
struct rb_node *p, int color) #define RB_BLACK 1
{
rb->__rb_parent_color = (unsigned long)p | color; #define __rb_parent(pc) ((struct rb_node *)(pc & ~3))
}
#define __rb_color(pc) ((pc) & 1)
static inline void #define __rb_is_black(pc) __rb_color(pc)
__rb_change_child(struct rb_node *old_node, struct rb_node *new_node, #define __rb_is_red(pc) (!__rb_color(pc))
struct rb_node *parent, struct rb_root *root) #define rb_color(rb) __rb_color((rb)->__rb_parent_color)
{ #define rb_is_red(rb) __rb_is_red((rb)->__rb_parent_color)
if (parent) { #define rb_is_black(rb) __rb_is_black((rb)->__rb_parent_color)
if (parent->rb_left == old_node)
parent->rb_left = new_node; static inline void rb_set_parent(struct rb_node *rb, struct rb_node *p)
else {
parent->rb_right = new_node; rb->__rb_parent_color = rb_color(rb) | (unsigned long)p;
} else }
root->rb_node = new_node;
} static inline void rb_set_parent_color(struct rb_node *rb,
struct rb_node *p, int color)
extern void __rb_erase_color(struct rb_node *parent, struct rb_root *root, {
void (*augment_rotate)(struct rb_node *old_node, struct rb_node *new_node)); rb->__rb_parent_color = (unsigned long)p | color;
}
static inline struct rb_node *
__rb_erase_augmented(struct rb_node *node, struct rb_root *root, static inline void
const struct rb_augment_callbacks *augment) __rb_change_child(struct rb_node *old_node, struct rb_node *new_node,
{ struct rb_node *parent, struct rb_root *root)
struct rb_node *child = node->rb_right, *tmp = node->rb_left; {
struct rb_node *parent, *rebalance; if (parent) {
unsigned long pc; if (parent->rb_left == old_node)
parent->rb_left = new_node;
if (!tmp) { else
/* parent->rb_right = new_node;
* Case 1: node to erase has no more than 1 child (easy!) } else
* root->rb_node = new_node;
* Note that if there is one child it must be red due to 5) }
* and node must be black due to 4). We adjust colors locally
* so as to bypass __rb_erase_color() later on. extern void __rb_erase_color(struct rb_node *parent, struct rb_root *root,
*/ void (*augment_rotate)(struct rb_node *old_node, struct rb_node *new_node));
pc = node->__rb_parent_color;
parent = __rb_parent(pc); static inline struct rb_node *
__rb_change_child(node, child, parent, root); __rb_erase_augmented(struct rb_node *node, struct rb_root *root,
if (child) { const struct rb_augment_callbacks *augment)
child->__rb_parent_color = pc; {
rebalance = NULL; struct rb_node *child = node->rb_right, *tmp = node->rb_left;
} else struct rb_node *parent, *rebalance;
rebalance = __rb_is_black(pc) ? parent : NULL; unsigned long pc;
tmp = parent;
} else if (!child) { if (!tmp) {
/* Still case 1, but this time the child is node->rb_left */ /*
tmp->__rb_parent_color = pc = node->__rb_parent_color; * Case 1: node to erase has no more than 1 child (easy!)
parent = __rb_parent(pc); *
__rb_change_child(node, tmp, parent, root); * Note that if there is one child it must be red due to 5)
rebalance = NULL; * and node must be black due to 4). We adjust colors locally
tmp = parent; * so as to bypass __rb_erase_color() later on.
} else { */
struct rb_node *successor = child, *child2; pc = node->__rb_parent_color;
tmp = child->rb_left; parent = __rb_parent(pc);
if (!tmp) { __rb_change_child(node, child, parent, root);
/* if (child) {
* Case 2: node's successor is its right child child->__rb_parent_color = pc;
* rebalance = NULL;
* (n) (s) } else
* / \ / \ rebalance = __rb_is_black(pc) ? parent : NULL;
* (x) (s) -> (x) (c) tmp = parent;
* \ } else if (!child) {
* (c) /* Still case 1, but this time the child is node->rb_left */
*/ tmp->__rb_parent_color = pc = node->__rb_parent_color;
parent = successor; parent = __rb_parent(pc);
child2 = successor->rb_right; __rb_change_child(node, tmp, parent, root);
augment->copy(node, successor); rebalance = NULL;
} else { tmp = parent;
/* } else {
* Case 3: node's successor is leftmost under struct rb_node *successor = child, *child2;
* node's right child subtree tmp = child->rb_left;
* if (!tmp) {
* (n) (s) /*
* / \ / \ * Case 2: node's successor is its right child
* (x) (y) -> (x) (y) *
* / / * (n) (s)
* (p) (p) * / \ / \
* / / * (x) (s) -> (x) (c)
* (s) (c) * \
* \ * (c)
* (c) */
*/ parent = successor;
do { child2 = successor->rb_right;
parent = successor; augment->copy(node, successor);
successor = tmp; } else {
tmp = tmp->rb_left; /*
} while (tmp); * Case 3: node's successor is leftmost under
parent->rb_left = child2 = successor->rb_right; * node's right child subtree
successor->rb_right = child; *
rb_set_parent(child, successor); * (n) (s)
augment->copy(node, successor); * / \ / \
augment->propagate(parent, successor); * (x) (y) -> (x) (y)
} * / /
* (p) (p)
successor->rb_left = tmp = node->rb_left; * / /
rb_set_parent(tmp, successor); * (s) (c)
* \
pc = node->__rb_parent_color; * (c)
tmp = __rb_parent(pc); */
__rb_change_child(node, successor, tmp, root); do {
if (child2) { parent = successor;
successor->__rb_parent_color = pc; successor = tmp;
rb_set_parent_color(child2, parent, RB_BLACK); tmp = tmp->rb_left;
rebalance = NULL; } while (tmp);
} else { parent->rb_left = child2 = successor->rb_right;
unsigned long pc2 = successor->__rb_parent_color; successor->rb_right = child;
successor->__rb_parent_color = pc; rb_set_parent(child, successor);
rebalance = __rb_is_black(pc2) ? parent : NULL; augment->copy(node, successor);
} augment->propagate(parent, successor);
tmp = successor; }
}
successor->rb_left = tmp = node->rb_left;
augment->propagate(tmp, NULL); rb_set_parent(tmp, successor);
return rebalance;
} pc = node->__rb_parent_color;
tmp = __rb_parent(pc);
static inline void __rb_change_child(node, successor, tmp, root);
rb_erase_augmented(struct rb_node *node, struct rb_root *root, if (child2) {
const struct rb_augment_callbacks *augment) successor->__rb_parent_color = pc;
{ rb_set_parent_color(child2, parent, RB_BLACK);
struct rb_node *rebalance = __rb_erase_augmented(node, root, augment); rebalance = NULL;
if (rebalance) } else {
__rb_erase_color(rebalance, root, augment->rotate); unsigned long pc2 = successor->__rb_parent_color;
} successor->__rb_parent_color = pc;
rebalance = __rb_is_black(pc2) ? parent : NULL;
#endif /* _GENERIC_RBTREE_H */ }
tmp = successor;
}
augment->propagate(tmp, NULL);
return rebalance;
}
static inline void
rb_erase_augmented(struct rb_node *node, struct rb_root *root,
const struct rb_augment_callbacks *augment)
{
struct rb_node *rebalance = __rb_erase_augmented(node, root, augment);
if (rebalance)
__rb_erase_color(rebalance, root, augment->rotate);
}
#endif /* _GENERIC_RBTREE_H */

View File

@@ -1,24 +1,42 @@
#ifndef _GENERIC_STRING_UITL_H /*************************************************************************
#define _GENERIC_STRING_UITL_H *
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#include <stddef.h> *
#include <string.h> * smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
static inline char *safe_strncpy(char *dest, const char *src, size_t n) * the Free Software Foundation, either version 3 of the License, or
{ * (at your option) any later version.
#if __GNUC__ > 7 *
#pragma GCC diagnostic push * smartdns is distributed in the hope that it will be useful,
#pragma GCC diagnostic ignored "-Wstringop-truncation" * but WITHOUT ANY WARRANTY; without even the implied warranty of
#endif * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
char *ret = strncpy(dest, src, n - 1); * GNU General Public License for more details.
if (n > 0) { *
dest[n - 1] = '\0'; * You should have received a copy of the GNU General Public License
} * along with this program. If not, see <http://www.gnu.org/licenses/>.
#if __GNUC__ > 7 */
#pragma GCC diagnostic pop
#endif #ifndef _GENERIC_STRING_UITL_H
return ret; #define _GENERIC_STRING_UITL_H
}
#include <stddef.h>
#include <string.h>
#endif
static inline char *safe_strncpy(char *dest, const char *src, size_t n)
{
#if __GNUC__ > 7
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wstringop-truncation"
#endif
char *ret = strncpy(dest, src, n - 1);
if (n > 0) {
dest[n - 1] = '\0';
}
#if __GNUC__ > 7
#pragma GCC diagnostic pop
#endif
return ret;
}
#endif

File diff suppressed because it is too large Load Diff

View File

@@ -1,3 +1,20 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "bitmap.h" #include "bitmap.h"
#include "bitops.h" #include "bitops.h"

View File

@@ -1,293 +1,311 @@
#include "conf.h" /*************************************************************************
#include <getopt.h> *
#include <stdio.h> * Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
#include <stdlib.h> *
#include <string.h> * smartdns is free software: you can redistribute it and/or modify
#include <unistd.h> * it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
static const char *currrent_conf_file = NULL; * (at your option) any later version.
*
const char *conf_get_conf_file(void) * smartdns is distributed in the hope that it will be useful,
{ * but WITHOUT ANY WARRANTY; without even the implied warranty of
return currrent_conf_file; * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
} * GNU General Public License for more details.
*
int conf_custom(const char *item, void *data, int argc, char *argv[]) * You should have received a copy of the GNU General Public License
{ * along with this program. If not, see <http://www.gnu.org/licenses/>.
struct config_item_custom *item_custom = data; */
return item_custom->custom_func(item_custom->custom_data, argc, argv);
} #include "conf.h"
#include <getopt.h>
int conf_int(const char *item, void *data, int argc, char *argv[]) #include <stdio.h>
{ #include <stdlib.h>
struct config_item_int *item_int = data; #include <string.h>
int value = 0; #include <unistd.h>
if (argc < 2) {
return -1; static const char *currrent_conf_file = NULL;
}
const char *conf_get_conf_file(void)
value = atoi(argv[1]); {
return currrent_conf_file;
if (value < item_int->min) { }
value = item_int->min;
} else if (value > item_int->max) { int conf_custom(const char *item, void *data, int argc, char *argv[])
value = item_int->max; {
} struct config_item_custom *item_custom = data;
return item_custom->custom_func(item_custom->custom_data, argc, argv);
*(item_int->data) = value; }
return 0; int conf_int(const char *item, void *data, int argc, char *argv[])
} {
struct config_item_int *item_int = data;
int conf_string(const char *item, void *data, int argc, char *argv[]) int value = 0;
{ if (argc < 2) {
struct config_item_string *item_string = data; return -1;
}
if (argc < 2) {
return -1; value = atoi(argv[1]);
}
if (value < item_int->min) {
strncpy(item_string->data, argv[1], item_string->size); value = item_int->min;
} else if (value > item_int->max) {
return 0; value = item_int->max;
} }
int conf_yesno(const char *item, void *data, int argc, char *argv[]) *(item_int->data) = value;
{
struct config_item_yesno *item_yesno = data; return 0;
int yes = 0; }
if (argc < 2) { int conf_string(const char *item, void *data, int argc, char *argv[])
return -1; {
} struct config_item_string *item_string = data;
char *value = argv[1]; if (argc < 2) {
if (strncmp("auto", value, sizeof("auto")) == 0 || strncmp("AUTO", value, sizeof("AUTO")) == 0) { return -1;
return 0; }
}
strncpy(item_string->data, argv[1], item_string->size);
if (strncmp("yes", value, sizeof("yes")) == 0 || strncmp("YES", value, sizeof("YES")) == 0) {
yes = 1; return 0;
} else if (strncmp("no", value, sizeof("no")) == 0 || strncmp("NO", value, sizeof("NO")) == 0) { }
yes = 0;
} int conf_yesno(const char *item, void *data, int argc, char *argv[])
{
*(item_yesno->data) = yes; struct config_item_yesno *item_yesno = data;
int yes = 0;
return 0;
} if (argc < 2) {
return -1;
int conf_size(const char *item, void *data, int argc, char *argv[]) }
{
/* read dns cache size */ char *value = argv[1];
int base = 1; if (strncmp("auto", value, sizeof("auto")) == 0 || strncmp("AUTO", value, sizeof("AUTO")) == 0) {
size_t size = 0; return 0;
int num = 0; }
struct config_item_size *item_size = data;
char *value = argv[1]; if (strncmp("yes", value, sizeof("yes")) == 0 || strncmp("YES", value, sizeof("YES")) == 0) {
yes = 1;
if (strstr(value, "k") || strstr(value, "K")) { } else if (strncmp("no", value, sizeof("no")) == 0 || strncmp("NO", value, sizeof("NO")) == 0) {
base = 1024; yes = 0;
} else if (strstr(value, "m") || strstr(value, "M")) { }
base = 1024 * 1024;
} else if (strstr(value, "g") || strstr(value, "G")) { *(item_yesno->data) = yes;
base = 1024 * 1024 * 1024;
} return 0;
}
num = atoi(value);
if (num < 0) { int conf_size(const char *item, void *data, int argc, char *argv[])
return -1; {
} /* read dns cache size */
int base = 1;
size = num * base; size_t size = 0;
if (size > item_size->max) { int num = 0;
size = item_size->max; struct config_item_size *item_size = data;
} else if (size < item_size->min) { char *value = argv[1];
size = item_size->min;
} if (strstr(value, "k") || strstr(value, "K")) {
base = 1024;
*(item_size->data) = size; } else if (strstr(value, "m") || strstr(value, "M")) {
base = 1024 * 1024;
return 0; } else if (strstr(value, "g") || strstr(value, "G")) {
} base = 1024 * 1024 * 1024;
}
void conf_getopt_reset(void)
{ num = atoi(value);
static struct option long_options[] = {{"-", 0, 0, 0}, {0, 0, 0, 0}}; if (num < 0) {
int argc = 2; return -1;
char *argv[3] = {"reset", "", 0}; }
optind = 0; size = num * base;
opterr = 0; if (size > item_size->max) {
optopt = 0; size = item_size->max;
getopt_long(argc, argv, "", long_options, NULL); } else if (size < item_size->min) {
optind = 0; size = item_size->min;
opterr = 0; }
optopt = 0;
} *(item_size->data) = size;
int conf_parse_args(char *key, char *value, int *argc, char **argv) return 0;
{ }
char *start = NULL;
char *ptr = value; void conf_getopt_reset(void)
int count = 0; {
int sep_flag = ' '; static struct option long_options[] = {{"-", 0, 0, 0}, {0, 0, 0, 0}};
int argc = 2;
argv[0] = key; char *argv[3] = {"reset", "", 0};
count++;
optind = 0;
while (*ptr != '\0') { opterr = 0;
if (*ptr == '\\') { optopt = 0;
char *tmp = ptr + 1; getopt_long(argc, argv, "", long_options, NULL);
while (*tmp != '\0') { optind = 0;
*(tmp - 1) = *tmp; opterr = 0;
tmp++; optopt = 0;
} }
ptr++;
continue; int conf_parse_args(char *key, char *value, int *argc, char **argv)
} {
char *start = NULL;
if (*ptr == '"' && start == NULL) { char *ptr = value;
sep_flag = '"'; int count = 0;
start = NULL; int sep_flag = ' ';
}
argv[0] = key;
if (*ptr != sep_flag && start == NULL) { count++;
start = ptr;
ptr++; while (*ptr != '\0') {
continue; if (*ptr == '\\') {
} char *tmp = ptr + 1;
while (*tmp != '\0') {
if (*ptr == sep_flag && start == NULL) { *(tmp - 1) = *tmp;
ptr++; tmp++;
continue; }
} ptr++;
continue;
if (*ptr == sep_flag && start) { }
argv[count] = start;
*ptr = '\0'; if (*ptr == '"' && start == NULL) {
ptr++; sep_flag = '"';
count++; start = NULL;
sep_flag = ' '; }
start = NULL;
continue; if (*ptr != sep_flag && start == NULL) {
} start = ptr;
ptr++;
ptr++; continue;
} }
if (start != ptr && start) { if (*ptr == sep_flag && start == NULL) {
argv[count] = start; ptr++;
count++; continue;
} }
*argc = count; if (*ptr == sep_flag && start) {
argv[count] = 0; argv[count] = start;
*ptr = '\0';
return 0; ptr++;
} count++;
sep_flag = ' ';
void load_exit(void) start = NULL;
{ continue;
return; }
}
ptr++;
int load_conf_printf(const char *file, int lineno, int ret) }
{
if (ret != CONF_RET_OK) { if (start != ptr && start) {
printf("process config file '%s' failed at line %d.", file, lineno); argv[count] = start;
if (ret == CONF_RET_ERR || ret == CONF_RET_NOENT) { count++;
return -1; }
}
*argc = count;
return 0; argv[count] = 0;
}
return 0;
return 0; }
}
void load_exit(void)
int load_conf_file(const char *file, struct config_item *items, conf_error_handler handler) {
{ return;
FILE *fp = NULL; }
char line[MAX_LINE_LEN];
char key[MAX_KEY_LEN]; int load_conf_printf(const char *file, int lineno, int ret)
char value[MAX_LINE_LEN]; {
int filed_num = 0; if (ret != CONF_RET_OK) {
int i; printf("process config file '%s' failed at line %d.", file, lineno);
int argc; if (ret == CONF_RET_ERR || ret == CONF_RET_NOENT) {
char *argv[1024]; return -1;
int ret = 0; }
int call_ret = 0;
int line_no = 0; return 0;
}
if (handler == NULL) {
handler = load_conf_printf; return 0;
} }
fp = fopen(file, "r"); int load_conf_file(const char *file, struct config_item *items, conf_error_handler handler)
if (fp == NULL) { {
return -1; FILE *fp = NULL;
} char line[MAX_LINE_LEN];
char key[MAX_KEY_LEN];
line_no = 0; char value[MAX_LINE_LEN];
while (fgets(line, MAX_LINE_LEN, fp)) { int filed_num = 0;
line_no++; int i;
filed_num = sscanf(line, "%63s %1023[^\r\n]s", key, value); int argc;
if (filed_num <= 0) { char *argv[1024];
continue; int ret = 0;
} int call_ret = 0;
int line_no = 0;
/* comment, skip */
if (key[0] == '#') { if (handler == NULL) {
continue; handler = load_conf_printf;
} }
/* if field format is not key = value, error */ fp = fopen(file, "r");
if (filed_num != 2) { if (fp == NULL) {
goto errout; return -1;
} }
for (i = 0;; i++) { line_no = 0;
if (items[i].item == NULL) { while (fgets(line, MAX_LINE_LEN, fp)) {
handler(file, line_no, CONF_RET_NOENT); line_no++;
break; filed_num = sscanf(line, "%63s %1023[^\r\n]s", key, value);
} if (filed_num <= 0) {
continue;
if (strncmp(items[i].item, key, MAX_KEY_LEN) != 0) { }
continue;
} /* comment, skip */
if (key[0] == '#') {
if (conf_parse_args(key, value, &argc, argv) != 0) { continue;
continue; }
}
/* if field format is not key = value, error */
conf_getopt_reset(); if (filed_num != 2) {
/* call item function */ goto errout;
currrent_conf_file = file; }
call_ret = items[i].item_func(items[i].item, items[i].data, argc, argv);
ret = handler(file, line_no, call_ret); for (i = 0;; i++) {
if (ret != 0) { if (items[i].item == NULL) {
conf_getopt_reset(); handler(file, line_no, CONF_RET_NOENT);
goto errout; break;
} }
conf_getopt_reset(); if (strncmp(items[i].item, key, MAX_KEY_LEN) != 0) {
continue;
break; }
}
} if (conf_parse_args(key, value, &argc, argv) != 0) {
continue;
fclose(fp); }
return 0; conf_getopt_reset();
errout: /* call item function */
if (fp) { currrent_conf_file = file;
fclose(fp); call_ret = items[i].item_func(items[i].item, items[i].data, argc, argv);
} ret = handler(file, line_no, call_ret);
return -1; if (ret != 0) {
} conf_getopt_reset();
goto errout;
int load_conf(const char *file, struct config_item items[], conf_error_handler handler) }
{
return load_conf_file(file, items, handler); conf_getopt_reset();
}
break;
}
}
fclose(fp);
return 0;
errout:
if (fp) {
fclose(fp);
}
return -1;
}
int load_conf(const char *file, struct config_item items[], conf_error_handler handler)
{
return load_conf_file(file, items, handler);
}

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

View File

@@ -0,0 +1,17 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

View File

@@ -1,443 +1,463 @@
/************************************************************************* /*************************************************************************
* *
* Copyright (C) 2018 Ruilin Peng (Nick) <pymumu@gmail.com>. * Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
* *
* smartdns is free software: you can redistribute it and/or modify * smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by * it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or * the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version. * (at your option) any later version.
* *
* smartdns is distributed in the hope that it will be useful, * smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of * but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details. * GNU General Public License for more details.
* *
* You should have received a copy of the GNU General Public License * You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>. * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/ */
#define _GNU_SOURCE #define _GNU_SOURCE
#include "art.h" #include "art.h"
#include "atomic.h" #include "atomic.h"
#include "dns_client.h" #include "dns_client.h"
#include "dns_conf.h" #include "dns_conf.h"
#include "dns_server.h" #include "dns_server.h"
#include "fast_ping.h" #include "fast_ping.h"
#include "hashtable.h" #include "hashtable.h"
#include "list.h" #include "list.h"
#include "rbtree.h" #include "rbtree.h"
#include "tlog.h" #include "tlog.h"
#include "util.h" #include "util.h"
#include <errno.h> #include <errno.h>
#include <fcntl.h> #include <fcntl.h>
#include <libgen.h> #include <libgen.h>
#include <openssl/err.h> #include <openssl/err.h>
#include <openssl/ssl.h> #include <openssl/ssl.h>
#include <signal.h> #include <signal.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <sys/types.h> #include <sys/types.h>
#include <ucontext.h> #include <ucontext.h>
#define RESOLVE_FILE "/etc/resolv.conf" #define RESOLVE_FILE "/etc/resolv.conf"
#define MAX_LINE_LEN 1024 #define MAX_LINE_LEN 1024
#define MAX_KEY_LEN 64 #define MAX_KEY_LEN 64
#define SMARTDNS_PID_FILE "/var/run/smartdns.pid" #define SMARTDNS_PID_FILE "/var/run/smartdns.pid"
#define TMP_BUFF_LEN_32 32 #define TMP_BUFF_LEN_32 32
static int verbose_screen; static int verbose_screen;
static void _help(void) static void _help(void)
{ {
/* clang-format off */ /* clang-format off */
char *help = "" char *help = ""
"Usage: smartdns [OPTION]...\n" "Usage: smartdns [OPTION]...\n"
"Start smartdns server.\n" "Start smartdns server.\n"
" -f run forground.\n" " -f run forground.\n"
" -c [conf] config file.\n" " -c [conf] config file.\n"
" -p [pid] pid file path\n" " -p [pid] pid file path\n"
" -S ignore segment fault signal.\n" " -S ignore segment fault signal.\n"
" -x verbose screen.\n" " -x verbose screen.\n"
" -v dispaly version.\n" " -v dispaly version.\n"
" -h show this help message.\n" " -h show this help message.\n"
"Online help: http://pymumu.github.io/smartdns\n" "Online help: http://pymumu.github.io/smartdns\n"
"Copyright (C) Nick Peng <pymumu@gmail.com>\n" "Copyright (C) Nick Peng <pymumu@gmail.com>\n"
; ;
/* clang-format on */ /* clang-format on */
printf("%s", help); printf("%s", help);
} }
static void _show_version(void) static void _show_version(void)
{ {
char str_ver[256] = {0}; char str_ver[256] = {0};
#ifdef SMARTDNS_VERION #ifdef SMARTDNS_VERION
const char *ver = SMARTDNS_VERION; const char *ver = SMARTDNS_VERION;
snprintf(str_ver, sizeof(str_ver), "%s", ver); snprintf(str_ver, sizeof(str_ver), "%s", ver);
#else #else
struct tm tm; struct tm tm;
get_compiled_time(&tm); get_compiled_time(&tm);
snprintf(str_ver, sizeof(str_ver), "1.%.4d%.2d%.2d-%.2d%.2d", tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday, tm.tm_hour, tm.tm_min); snprintf(str_ver, sizeof(str_ver), "1.%.4d%.2d%.2d-%.2d%.2d", tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday,
#endif tm.tm_hour, tm.tm_min);
printf("smartdns %s\n", str_ver); #endif
} printf("smartdns %s\n", str_ver);
}
static int _smartdns_load_from_resolv(void)
{ static int _smartdns_load_from_resolv(void)
FILE *fp = NULL; {
char line[MAX_LINE_LEN]; FILE *fp = NULL;
char key[MAX_KEY_LEN]; char line[MAX_LINE_LEN];
char value[MAX_LINE_LEN]; char key[MAX_KEY_LEN];
char ns_ip[DNS_MAX_IPLEN]; char value[MAX_LINE_LEN];
int port = PORT_NOT_DEFINED; char ns_ip[DNS_MAX_IPLEN];
int ret = -1; int port = PORT_NOT_DEFINED;
int ret = -1;
int filed_num = 0;
int line_num = 0; int filed_num = 0;
int line_num = 0;
fp = fopen(RESOLVE_FILE, "r");
if (fp == NULL) { fp = fopen(RESOLVE_FILE, "r");
tlog(TLOG_ERROR, "open %s failed, %s", RESOLVE_FILE, strerror(errno)); if (fp == NULL) {
return -1; tlog(TLOG_ERROR, "open %s failed, %s", RESOLVE_FILE, strerror(errno));
} return -1;
}
while (fgets(line, MAX_LINE_LEN, fp)) {
line_num++; while (fgets(line, MAX_LINE_LEN, fp)) {
filed_num = sscanf(line, "%63s %1023[^\r\n]s", key, value); line_num++;
filed_num = sscanf(line, "%63s %1023[^\r\n]s", key, value);
if (filed_num != 2) {
continue; if (filed_num != 2) {
} continue;
}
if (strncmp(key, "nameserver", MAX_KEY_LEN) != 0) {
continue; if (strncmp(key, "nameserver", MAX_KEY_LEN) != 0) {
} continue;
}
if (parse_ip(value, ns_ip, &port) != 0) {
continue; if (parse_ip(value, ns_ip, &port) != 0) {
} continue;
}
if (port == PORT_NOT_DEFINED) {
port = DEFAULT_DNS_PORT; if (port == PORT_NOT_DEFINED) {
} port = DEFAULT_DNS_PORT;
}
safe_strncpy(dns_conf_servers[dns_conf_server_num].server, ns_ip, DNS_MAX_IPLEN);
dns_conf_servers[dns_conf_server_num].port = port; safe_strncpy(dns_conf_servers[dns_conf_server_num].server, ns_ip, DNS_MAX_IPLEN);
dns_conf_servers[dns_conf_server_num].type = DNS_SERVER_UDP; dns_conf_servers[dns_conf_server_num].port = port;
dns_conf_server_num++; dns_conf_servers[dns_conf_server_num].type = DNS_SERVER_UDP;
ret = 0; dns_conf_server_num++;
} ret = 0;
}
fclose(fp);
fclose(fp);
return ret;
} return ret;
}
static int _smartdns_add_servers(void)
{ static int _smartdns_add_servers(void)
int i = 0; {
int j = 0; int i = 0;
int ret = 0; int j = 0;
struct dns_server_groups *group = NULL; int ret = 0;
struct dns_servers *server = NULL; struct dns_server_groups *group = NULL;
struct client_dns_server_flags flags; struct dns_servers *server = NULL;
struct client_dns_server_flags flags;
for (i = 0; i < dns_conf_server_num; i++) {
memset(&flags, 0, sizeof(flags)); for (i = 0; i < dns_conf_server_num; i++) {
switch (dns_conf_servers[i].type) { memset(&flags, 0, sizeof(flags));
case DNS_SERVER_UDP: { switch (dns_conf_servers[i].type) {
struct client_dns_server_flag_udp *flag_udp = &flags.udp; case DNS_SERVER_UDP: {
flag_udp->ttl = dns_conf_servers[i].ttl; struct client_dns_server_flag_udp *flag_udp = &flags.udp;
} break; flag_udp->ttl = dns_conf_servers[i].ttl;
case DNS_SERVER_HTTPS: { } break;
struct client_dns_server_flag_https *flag_http = &flags.https; case DNS_SERVER_HTTPS: {
flag_http->spi_len = dns_client_spki_decode(dns_conf_servers[i].spki, (unsigned char *)flag_http->spki); struct client_dns_server_flag_https *flag_http = &flags.https;
safe_strncpy(flag_http->hostname, dns_conf_servers[i].hostname, sizeof(flag_http->hostname)); flag_http->spi_len = dns_client_spki_decode(dns_conf_servers[i].spki, (unsigned char *)flag_http->spki);
safe_strncpy(flag_http->path, dns_conf_servers[i].path, sizeof(flag_http->path)); safe_strncpy(flag_http->hostname, dns_conf_servers[i].hostname, sizeof(flag_http->hostname));
safe_strncpy(flag_http->httphost, dns_conf_servers[i].httphost, sizeof(flag_http->httphost)); safe_strncpy(flag_http->path, dns_conf_servers[i].path, sizeof(flag_http->path));
safe_strncpy(flag_http->tls_host_verify, dns_conf_servers[i].tls_host_verify, sizeof(flag_http->tls_host_verify)); safe_strncpy(flag_http->httphost, dns_conf_servers[i].httphost, sizeof(flag_http->httphost));
} break; safe_strncpy(flag_http->tls_host_verify, dns_conf_servers[i].tls_host_verify,
case DNS_SERVER_TLS: { sizeof(flag_http->tls_host_verify));
struct client_dns_server_flag_tls *flag_tls = &flags.tls; flag_http->skip_check_cert = dns_conf_servers[i].skip_check_cert;
flag_tls->spi_len = dns_client_spki_decode(dns_conf_servers[i].spki, (unsigned char *)flag_tls->spki); } break;
safe_strncpy(flag_tls->hostname, dns_conf_servers[i].hostname, sizeof(flag_tls->hostname)); case DNS_SERVER_TLS: {
safe_strncpy(flag_tls->tls_host_verify, dns_conf_servers[i].tls_host_verify, sizeof(flag_tls->tls_host_verify)); struct client_dns_server_flag_tls *flag_tls = &flags.tls;
} break; flag_tls->spi_len = dns_client_spki_decode(dns_conf_servers[i].spki, (unsigned char *)flag_tls->spki);
case DNS_SERVER_TCP: safe_strncpy(flag_tls->hostname, dns_conf_servers[i].hostname, sizeof(flag_tls->hostname));
break; safe_strncpy(flag_tls->tls_host_verify, dns_conf_servers[i].tls_host_verify,
default: sizeof(flag_tls->tls_host_verify));
return -1; flag_tls->skip_check_cert = dns_conf_servers[i].skip_check_cert;
break;
} } break;
case DNS_SERVER_TCP:
flags.type = dns_conf_servers[i].type; break;
flags.server_flag = dns_conf_servers[i].server_flag; default:
flags.result_flag = dns_conf_servers[i].result_flag; return -1;
ret = dns_client_add_server(dns_conf_servers[i].server, dns_conf_servers[i].port, dns_conf_servers[i].type, &flags); break;
if (ret != 0) { }
tlog(TLOG_ERROR, "add server failed, %s:%d", dns_conf_servers[i].server, dns_conf_servers[i].port);
return -1; flags.type = dns_conf_servers[i].type;
} flags.server_flag = dns_conf_servers[i].server_flag;
} flags.result_flag = dns_conf_servers[i].result_flag;
ret = dns_client_add_server(dns_conf_servers[i].server, dns_conf_servers[i].port, dns_conf_servers[i].type,
hash_for_each(dns_group_table.group, i, group, node) &flags);
{ if (ret != 0) {
ret = dns_client_add_group(group->group_name); tlog(TLOG_ERROR, "add server failed, %s:%d", dns_conf_servers[i].server, dns_conf_servers[i].port);
if (ret != 0) { return -1;
tlog(TLOG_ERROR, "add group failed, %s", group->group_name); }
return -1; }
}
hash_for_each(dns_group_table.group, i, group, node)
for (j = 0; j < group->server_num; j++) { {
server = group->servers[j]; ret = dns_client_add_group(group->group_name);
if (server == NULL) { if (ret != 0) {
continue; tlog(TLOG_ERROR, "add group failed, %s", group->group_name);
} return -1;
ret = dns_client_add_to_group(group->group_name, server->server, server->port, server->type); }
if (ret != 0) {
tlog(TLOG_ERROR, "add server %s to group %s failed", server->server, group->group_name); for (j = 0; j < group->server_num; j++) {
return -1; server = group->servers[j];
} if (server == NULL) {
} continue;
} }
ret = dns_client_add_to_group(group->group_name, server->server, server->port, server->type);
return 0; if (ret != 0) {
} tlog(TLOG_ERROR, "add server %s to group %s failed", server->server, group->group_name);
return -1;
static int _smartdns_set_ecs_ip(void) }
{ }
int ret = 0; }
if (dns_conf_ipv4_ecs.enable) {
ret |= dns_client_set_ecs(dns_conf_ipv4_ecs.ip, dns_conf_ipv4_ecs.subnet); return 0;
} }
if (dns_conf_ipv6_ecs.enable) { static int _smartdns_set_ecs_ip(void)
ret |= dns_client_set_ecs(dns_conf_ipv6_ecs.ip, dns_conf_ipv6_ecs.subnet); {
} int ret = 0;
if (dns_conf_ipv4_ecs.enable) {
return ret; ret |= dns_client_set_ecs(dns_conf_ipv4_ecs.ip, dns_conf_ipv4_ecs.subnet);
} }
static int _smartdns_init_ssl(void) if (dns_conf_ipv6_ecs.enable) {
{ ret |= dns_client_set_ecs(dns_conf_ipv6_ecs.ip, dns_conf_ipv6_ecs.subnet);
SSL_load_error_strings(); }
SSL_library_init();
OpenSSL_add_all_algorithms(); return ret;
SSL_CRYPTO_thread_setup(); }
return 0; static int _smartdns_init_ssl(void)
} {
#if OPENSSL_API_COMPAT < 0x10100000L
static int _smartdns_destroy_ssl(void) SSL_load_error_strings();
{ SSL_library_init();
SSL_CRYPTO_thread_cleanup(); OpenSSL_add_all_algorithms();
ERR_free_strings(); SSL_CRYPTO_thread_setup();
EVP_cleanup(); #endif
return 0;
return 0; }
}
static int _smartdns_destroy_ssl(void)
static int _smartdns_init(void) {
{ #if OPENSSL_API_COMPAT < 0x10100000L
int ret; SSL_CRYPTO_thread_cleanup();
char *logfile = SMARTDNS_LOG_FILE; ERR_free_strings();
EVP_cleanup();
if (dns_conf_log_file[0] != 0) { #endif
logfile = dns_conf_log_file; return 0;
} }
ret = tlog_init(logfile, dns_conf_log_size, dns_conf_log_num, 0, 0); static int _smartdns_init(void)
if (ret != 0) { {
tlog(TLOG_ERROR, "start tlog failed.\n"); int ret;
goto errout; char *logfile = SMARTDNS_LOG_FILE;
}
if (dns_conf_log_file[0] != 0) {
tlog_setlogscreen(verbose_screen); logfile = dns_conf_log_file;
tlog_setlevel(dns_conf_log_level); }
tlog(TLOG_NOTICE, "smartdns starting...(Copyright (C) Nick Peng <pymumu@gmail.com>, build:%s %s)", __DATE__, __TIME__); ret = tlog_init(logfile, dns_conf_log_size, dns_conf_log_num, 0, 0);
if (ret != 0) {
if (_smartdns_init_ssl() != 0) { tlog(TLOG_ERROR, "start tlog failed.\n");
tlog(TLOG_ERROR, "init ssl failed."); goto errout;
goto errout; }
}
tlog_setlogscreen(verbose_screen);
if (dns_conf_server_num <= 0) { tlog_setlevel(dns_conf_log_level);
if (_smartdns_load_from_resolv() != 0) {
tlog(TLOG_ERROR, "load dns from resolv failed."); tlog(TLOG_NOTICE, "smartdns starting...(Copyright (C) Nick Peng <pymumu@gmail.com>, build:%s %s)", __DATE__,
goto errout; __TIME__);
}
} if (_smartdns_init_ssl() != 0) {
tlog(TLOG_ERROR, "init ssl failed.");
ret = fast_ping_init(); goto errout;
if (ret != 0) { }
tlog(TLOG_ERROR, "start ping failed.\n");
goto errout; if (dns_conf_server_num <= 0) {
} if (_smartdns_load_from_resolv() != 0) {
tlog(TLOG_ERROR, "load dns from resolv failed.");
ret = dns_server_init(); goto errout;
if (ret != 0) { }
tlog(TLOG_ERROR, "start dns server failed.\n"); }
goto errout;
} ret = fast_ping_init();
if (ret != 0) {
ret = dns_client_init(); tlog(TLOG_ERROR, "start ping failed.\n");
if (ret != 0) { goto errout;
tlog(TLOG_ERROR, "start dns client failed.\n"); }
goto errout;
} ret = dns_server_init();
ret = _smartdns_add_servers(); if (ret != 0) {
if (ret != 0) { tlog(TLOG_ERROR, "start dns server failed.\n");
tlog(TLOG_ERROR, "add servers failed."); goto errout;
goto errout; }
}
ret = dns_client_init();
ret = _smartdns_set_ecs_ip(); if (ret != 0) {
if (ret != 0) { tlog(TLOG_ERROR, "start dns client failed.\n");
tlog(TLOG_WARN, "set ecs ip address failed."); goto errout;
} }
ret = _smartdns_add_servers();
return 0; if (ret != 0) {
errout: tlog(TLOG_ERROR, "add servers failed.");
goto errout;
return -1; }
}
ret = _smartdns_set_ecs_ip();
static int _smartdns_run(void) if (ret != 0) {
{ tlog(TLOG_WARN, "set ecs ip address failed.");
return dns_server_run(); }
}
return 0;
static void _smartdns_exit(void) errout:
{
dns_server_exit(); return -1;
dns_client_exit(); }
fast_ping_exit();
_smartdns_destroy_ssl(); static int _smartdns_run(void)
tlog_exit(); {
dns_server_load_exit(); return dns_server_run();
} }
static void _sig_exit(int signo) static void _smartdns_exit(void)
{ {
dns_server_stop(); dns_server_exit();
} dns_client_exit();
fast_ping_exit();
static void _sig_error_exit(int signo, siginfo_t *siginfo, void *ct) _smartdns_destroy_ssl();
{ tlog_exit();
unsigned long PC = 0; dns_server_load_exit();
ucontext_t *context = ct; }
#if defined(__i386__)
int *pgregs = (int *)(&(context->uc_mcontext.gregs)); static void _sig_exit(int signo)
PC = pgregs[REG_EIP]; {
#elif defined(__x86_64__) dns_server_stop();
int *pgregs = (int *)(&(context->uc_mcontext.gregs)); }
PC = pgregs[REG_RIP];
#elif defined(__arm__) static void _sig_error_exit(int signo, siginfo_t *siginfo, void *ct)
PC = context->uc_mcontext.arm_pc; {
#elif defined(__aarch64__) unsigned long PC = 0;
PC = context->uc_mcontext.pc; ucontext_t *context = ct;
#elif defined(__mips__) const char *arch = NULL;
PC = context->uc_mcontext.pc; #if defined(__i386__)
#endif int *pgregs = (int *)(&(context->uc_mcontext.gregs));
tlog(TLOG_FATAL, "process exit with signal %d, code = %d, errno = %d, pid = %d, self = %d, pc = %#lx, addr = %#lx, build(%s %s)\n", signo, siginfo->si_code, PC = pgregs[REG_EIP];
siginfo->si_errno, siginfo->si_pid, getpid(), PC, (unsigned long)siginfo->si_addr, __DATE__, __TIME__); arch = "i386";
#elif defined(__x86_64__)
sleep(1); int *pgregs = (int *)(&(context->uc_mcontext.gregs));
_exit(0); PC = pgregs[REG_RIP];
} arch = "x86_64";
#elif defined(__arm__)
static int sig_list[] = {SIGSEGV, SIGABRT, SIGBUS, SIGILL, SIGFPE}; PC = context->uc_mcontext.arm_pc;
arch = "arm";
static int sig_num = sizeof(sig_list) / sizeof(int); #elif defined(__aarch64__)
PC = context->uc_mcontext.pc;
static void _reg_signal(void) arch = "arm64";
{ #elif defined(__mips__)
struct sigaction act, old; PC = context->uc_mcontext.pc;
int i = 0; arch = "mips";
act.sa_sigaction = _sig_error_exit; #endif
sigemptyset(&act.sa_mask); tlog(TLOG_FATAL,
act.sa_flags = SA_RESTART | SA_SIGINFO; "process exit with signal %d, code = %d, errno = %d, pid = %d, self = %d, pc = %#lx, addr = %#lx, build(%s "
"%s %s)\n",
for (i = 0; i < sig_num; i++) { signo, siginfo->si_code, siginfo->si_errno, siginfo->si_pid, getpid(), PC, (unsigned long)siginfo->si_addr,
sigaction(sig_list[i], &act, &old); __DATE__, __TIME__, arch);
}
} sleep(1);
_exit(0);
int main(int argc, char *argv[]) }
{
int ret; static int sig_list[] = {SIGSEGV, SIGABRT, SIGBUS, SIGILL, SIGFPE};
int is_forground = 0;
int opt; static int sig_num = sizeof(sig_list) / sizeof(int);
char config_file[MAX_LINE_LEN];
char pid_file[MAX_LINE_LEN]; static void _reg_signal(void)
int signal_ignore = 0; {
struct sigaction act, old;
safe_strncpy(config_file, SMARTDNS_CONF_FILE, MAX_LINE_LEN); int i = 0;
safe_strncpy(pid_file, SMARTDNS_PID_FILE, MAX_LINE_LEN); act.sa_sigaction = _sig_error_exit;
sigemptyset(&act.sa_mask);
while ((opt = getopt(argc, argv, "fhc:p:Svx")) != -1) { act.sa_flags = SA_RESTART | SA_SIGINFO;
switch (opt) {
case 'f': for (i = 0; i < sig_num; i++) {
is_forground = 1; sigaction(sig_list[i], &act, &old);
break; }
case 'c': }
snprintf(config_file, sizeof(config_file), "%s", optarg);
break; int main(int argc, char *argv[])
case 'p': {
snprintf(pid_file, sizeof(pid_file), "%s", optarg); int ret;
break; int is_forground = 0;
case 'S': int opt;
signal_ignore = 1; char config_file[MAX_LINE_LEN];
break; char pid_file[MAX_LINE_LEN];
case 'x': int signal_ignore = 0;
verbose_screen = 1;
break; safe_strncpy(config_file, SMARTDNS_CONF_FILE, MAX_LINE_LEN);
case 'v': safe_strncpy(pid_file, SMARTDNS_PID_FILE, MAX_LINE_LEN);
_show_version();
return 0; while ((opt = getopt(argc, argv, "fhc:p:Svx")) != -1) {
break; switch (opt) {
case 'h': case 'f':
_help(); is_forground = 1;
return 1; break;
} case 'c':
} snprintf(config_file, sizeof(config_file), "%s", optarg);
break;
if (is_forground == 0) { case 'p':
if (daemon(0, 0) < 0) { snprintf(pid_file, sizeof(pid_file), "%s", optarg);
fprintf(stderr, "run daemon process failed, %s\n", strerror(errno)); break;
return 1; case 'S':
} signal_ignore = 1;
} break;
case 'x':
if (signal_ignore == 0) { verbose_screen = 1;
_reg_signal(); break;
} case 'v':
_show_version();
if (dns_server_load_conf(config_file) != 0) { return 0;
fprintf(stderr, "load config failed.\n"); break;
goto errout; case 'h':
} _help();
return 1;
if (create_pid_file(pid_file) != 0) { }
goto errout; }
}
if (is_forground == 0) {
ret = _smartdns_init(); if (daemon(0, 0) < 0) {
if (ret != 0) { fprintf(stderr, "run daemon process failed, %s\n", strerror(errno));
usleep(100000); return 1;
goto errout; }
} }
signal(SIGINT, _sig_exit); if (signal_ignore == 0) {
signal(SIGPIPE, SIG_IGN); _reg_signal();
atexit(_smartdns_exit); }
return _smartdns_run(); if (create_pid_file(pid_file) != 0) {
goto errout;
errout: }
return 1; signal(SIGPIPE, SIG_IGN);
} if (dns_server_load_conf(config_file) != 0) {
fprintf(stderr, "load config failed.\n");
goto errout;
}
ret = _smartdns_init();
if (ret != 0) {
usleep(100000);
goto errout;
}
signal(SIGINT, _sig_exit);
signal(SIGTERM, _sig_exit);
atexit(_smartdns_exit);
return _smartdns_run();
errout:
return 1;
}

View File

@@ -1,6 +1,6 @@
/* /*
* tinylog * tinylog
* Copyright (C) 2018-2019 Nick Peng <pymumu@gmail.com> * Copyright (C) 2018-2020 Nick Peng <pymumu@gmail.com>
* https://github.com/pymumu/tinylog * https://github.com/pymumu/tinylog
*/ */
#ifndef _GNU_SOURCE #ifndef _GNU_SOURCE
@@ -94,7 +94,7 @@ struct tlog {
}; };
struct tlog_segment_log_head { struct tlog_segment_log_head {
struct tlog_info info; struct tlog_loginfo info;
unsigned short len; unsigned short len;
char data[0]; char data[0];
} __attribute__((packed)); } __attribute__((packed));
@@ -117,7 +117,7 @@ struct count_log {
}; };
struct tlog_info_inter { struct tlog_info_inter {
struct tlog_info info; struct tlog_loginfo info;
void *userptr; void *userptr;
}; };
@@ -166,6 +166,10 @@ static int _tlog_mkdir(const char *path)
if (access(path, F_OK) == 0) { if (access(path, F_OK) == 0) {
return 0; return 0;
} }
while(*path == ' ' && *path != '\0') {
path++;
}
strncpy(path_c, path, sizeof(path_c) - 1); strncpy(path_c, path, sizeof(path_c) - 1);
path_c[sizeof(path_c) - 1] = '\0'; path_c[sizeof(path_c) - 1] = '\0';
@@ -181,6 +185,11 @@ static int _tlog_mkdir(const char *path)
continue; continue;
} }
if (path_end == path_c) {
path_end++;
continue;
}
str = *path_end; str = *path_end;
*path_end = '\0'; *path_end = '\0';
if (access(path_c, F_OK) == 0) { if (access(path_c, F_OK) == 0) {
@@ -203,8 +212,8 @@ static int _tlog_mkdir(const char *path)
static struct tm *_tlog_localtime(time_t *timep, struct tm *tm) static struct tm *_tlog_localtime(time_t *timep, struct tm *tm)
{ {
static time_t last_time = {0}; static time_t last_time;
static struct tm last_tm = {0}; static struct tm last_tm;
/* localtime_r has a global timezone lock, it's about 8 times slower than gmtime /* localtime_r has a global timezone lock, it's about 8 times slower than gmtime
* this code is used to speed up localtime_r call. * this code is used to speed up localtime_r call.
@@ -297,11 +306,14 @@ void *tlog_get_private(tlog_log *log)
return log->private_data; return log->private_data;
} }
static int _tlog_format(char *buff, int maxlen, struct tlog_info *info, void *userptr, const char *format, va_list ap) static int _tlog_format(char *buff, int maxlen, struct tlog_loginfo *info, void *userptr, const char *format, va_list ap)
{ {
int len = 0; int len = 0;
int total_len = 0; int total_len = 0;
struct tlog_time *tm = &info->time; struct tlog_time *tm = &info->time;
void* unused __attribute__ ((unused));
unused = userptr;
if (tlog.root->multi_log) { if (tlog.root->multi_log) {
/* format prefix */ /* format prefix */
@@ -315,7 +327,7 @@ static int _tlog_format(char *buff, int maxlen, struct tlog_info *info, void *us
tlog_get_level_string(info->level), info->file, info->line); tlog_get_level_string(info->level), info->file, info->line);
} }
if (len < 0 || len == maxlen) { if (len < 0 || len >= maxlen) {
return -1; return -1;
} }
buff += len; buff += len;
@@ -340,9 +352,9 @@ static int _tlog_root_log_buffer(char *buff, int maxlen, void *userptr, const ch
int log_len = 0; int log_len = 0;
struct tlog_info_inter *info_inter = (struct tlog_info_inter *)userptr; struct tlog_info_inter *info_inter = (struct tlog_info_inter *)userptr;
struct tlog_segment_log_head *log_head = NULL; struct tlog_segment_log_head *log_head = NULL;
int max_format_len = 0; int max_format_len = 0;
if (tlog_format == NULL) { if (tlog_format == NULL) {
return -1; return -1;
} }
@@ -352,29 +364,32 @@ static int _tlog_root_log_buffer(char *buff, int maxlen, void *userptr, const ch
memcpy(&log_head->info, &info_inter->info, sizeof(log_head->info)); memcpy(&log_head->info, &info_inter->info, sizeof(log_head->info));
} }
max_format_len = maxlen - len - 2; max_format_len = maxlen - len - 2;
log_len = tlog_format(buff + len, max_format_len, &info_inter->info, info_inter->userptr, format, ap); buff[maxlen - 1] = 0;
if (log_len < 0) { log_len = tlog_format(buff + len, max_format_len, &info_inter->info, info_inter->userptr, format, ap);
if (log_len < 0) {
return -1; return -1;
} else if (log_len >= max_format_len) { } else if (log_len >= max_format_len) {
buff[max_format_len - 1] = '.'; buff[len + max_format_len - 2] = '.';
buff[max_format_len - 2] = '.'; buff[len + max_format_len - 3] = '.';
buff[max_format_len - 3] = '.'; buff[len + max_format_len - 4] = '.';
log_len = max_format_len; log_len = max_format_len - 1;
} }
len += log_len; len += log_len;
/* add new line character*/ /* add new line character*/
if (*(buff + len - 1) != '\n' && len + 1 < maxlen - len) { if (*(buff + len - 1) != '\n' && len + 1 < maxlen - 1) {
*(buff + len) = '\n'; *(buff + len) = '\n';
len++; len++;
log_len++; log_len++;
} }
if (tlog.root->segment_log && len + 1 < maxlen - len) { if (tlog.root->segment_log) {
*(buff + len) = '\0'; if (len + 1 < maxlen - 1) {
len++; *(buff + len) = '\0';
log_len++; len++;
log_len++;
}
log_head->len = log_len; log_head->len = log_len;
} }
@@ -385,6 +400,9 @@ static int _tlog_print_buffer(char *buff, int maxlen, void *userptr, const char
{ {
int len; int len;
int total_len = 0; int total_len = 0;
void* unused __attribute__ ((unused));
unused = userptr;
/* format log message */ /* format log message */
len = vsnprintf(buff, maxlen, format, ap); len = vsnprintf(buff, maxlen, format, ap);
@@ -452,12 +470,12 @@ static int _tlog_vprintf(struct tlog_log *log, vprint_callback print_callback, v
if (len <= 0) { if (len <= 0) {
return -1; return -1;
} else if (len >= TLOG_MAX_LINE_LEN) { } else if (len >= TLOG_MAX_LINE_LEN) {
strncpy(buff, "[LOG TOO LONG, DISCARD]\n", sizeof(buff)); strncpy(buff, "[LOG TOO LONG, DISCARD]\n", sizeof(buff));
buff[sizeof(buff) - 1] = '\0'; buff[sizeof(buff) - 1] = '\0';
len = strnlen(buff, sizeof(buff)); len = strnlen(buff, sizeof(buff));
} }
pthread_mutex_lock(&tlog.lock); pthread_mutex_lock(&tlog.lock);
do { do {
if (log->end == log->start) { if (log->end == log->start) {
if (log->ext_end == 0) { if (log->ext_end == 0) {
@@ -547,8 +565,9 @@ int tlog_printf(struct tlog_log *log, const char *format, ...)
static int _tlog_early_print(const char *format, va_list ap) static int _tlog_early_print(const char *format, va_list ap)
{ {
char log_buf[TLOG_MAX_LINE_LEN]; char log_buf[TLOG_MAX_LINE_LEN];
int len = 0; size_t len = 0;
int out_len = 0; size_t out_len = 0;
int unused __attribute__ ((unused));
if (tlog_disable_early_print) { if (tlog_disable_early_print) {
return 0; return 0;
@@ -562,9 +581,9 @@ static int _tlog_early_print(const char *format, va_list ap)
out_len = sizeof(log_buf); out_len = sizeof(log_buf);
} }
write(STDOUT_FILENO, log_buf, out_len); unused = write(STDOUT_FILENO, log_buf, out_len);
if (log_buf[out_len - 1] != '\n') { if (log_buf[out_len - 1] != '\n') {
write(STDOUT_FILENO, "\n", 1); unused = write(STDOUT_FILENO, "\n", 1);
} }
return len; return len;
@@ -647,6 +666,7 @@ static int _tlog_list_dir(const char *path, list_callback callback, void *userpt
DIR *dir = NULL; DIR *dir = NULL;
struct dirent *ent; struct dirent *ent;
int ret = 0; int ret = 0;
const char* unused __attribute__ ((unused)) = path;
dir = opendir(path); dir = opendir(path);
if (dir == NULL) { if (dir == NULL) {
@@ -679,6 +699,7 @@ static int _tlog_count_log_callback(const char *path, struct dirent *entry, void
struct count_log *count_log = (struct count_log *)userptr; struct count_log *count_log = (struct count_log *)userptr;
struct tlog_log *log = count_log->log; struct tlog_log *log = count_log->log;
char logname[TLOG_LOG_NAME_LEN * 2]; char logname[TLOG_LOG_NAME_LEN * 2];
const char* unused __attribute__ ((unused)) = path;
if (strstr(entry->d_name, log->suffix) == NULL) { if (strstr(entry->d_name, log->suffix) == NULL) {
return 0; return 0;
@@ -1002,9 +1023,10 @@ static int _tlog_archive_log(struct tlog_log *log)
} }
} }
static int _tlog_write(struct tlog_log *log, char *buff, int bufflen) static int _tlog_write(struct tlog_log *log, const char *buff, int bufflen)
{ {
int len; int len;
int unused __attribute__ ((unused));
if (bufflen <= 0) { if (bufflen <= 0) {
return 0; return 0;
@@ -1012,7 +1034,7 @@ static int _tlog_write(struct tlog_log *log, char *buff, int bufflen)
/* output log to screen */ /* output log to screen */
if (log->logscreen) { if (log->logscreen) {
write(STDOUT_FILENO, buff, bufflen); unused = write(STDOUT_FILENO, buff, bufflen);
} }
/* if log file size exceeds threshold, start to compress */ /* if log file size exceeds threshold, start to compress */
@@ -1024,7 +1046,7 @@ static int _tlog_write(struct tlog_log *log, char *buff, int bufflen)
if (log->filesize < lseek(log->fd, 0, SEEK_END) && log->multi_log == 0) { if (log->filesize < lseek(log->fd, 0, SEEK_END) && log->multi_log == 0) {
const char *msg = "[Auto enable multi-process write mode, log may be lost, please enable multi-process write mode manually]\n"; const char *msg = "[Auto enable multi-process write mode, log may be lost, please enable multi-process write mode manually]\n";
log->multi_log = 1; log->multi_log = 1;
write(log->fd, msg, strlen(msg)); unused = write(log->fd, msg, strlen(msg));
} }
close(log->fd); close(log->fd);
log->fd = -1; log->fd = -1;
@@ -1080,7 +1102,7 @@ static int _tlog_write(struct tlog_log *log, char *buff, int bufflen)
return len; return len;
} }
int tlog_write(struct tlog_log *log, char *buff, int bufflen) int tlog_write(struct tlog_log *log, const char *buff, int bufflen)
{ {
return _tlog_write(log, buff, bufflen); return _tlog_write(log, buff, bufflen);
} }
@@ -1294,11 +1316,15 @@ static void _tlog_work_write(struct tlog_log *log, int log_len, int log_extlen,
} }
} }
static int _tlog_root_write_log(struct tlog_log *log, char *buff, int bufflen) static int _tlog_root_write_log(struct tlog_log *log, const char *buff, int bufflen)
{ {
struct tlog_segment_log_head *head = NULL; struct tlog_segment_log_head *head = NULL;
static struct tlog_segment_log_head empty_info; static struct tlog_segment_log_head empty_info;
if (tlog.output_func == NULL) { if (tlog.output_func == NULL) {
if (log->segment_log) {
head = (struct tlog_segment_log_head *)buff;
return _tlog_write(log, head->data, head->len);
}
return _tlog_write(log, buff, bufflen); return _tlog_write(log, buff, bufflen);
} }
@@ -1319,7 +1345,10 @@ static void *_tlog_work(void *arg)
int log_dropped = 0; int log_dropped = 0;
struct tlog_log *log = NULL; struct tlog_log *log = NULL;
struct tlog_log *loop_log = NULL; struct tlog_log *loop_log = NULL;
void* unused __attribute__ ((unused));
unused = arg;
while (1) { while (1) {
log_len = 0; log_len = 0;
log_extlen = 0; log_extlen = 0;
@@ -1348,7 +1377,7 @@ static void *_tlog_work(void *arg)
log = _tlog_wait_log_locked(log); log = _tlog_wait_log_locked(log);
if (log == NULL) { if (log == NULL) {
pthread_mutex_unlock(&tlog.lock); pthread_mutex_unlock(&tlog.lock);
if (errno != ETIMEDOUT) { if (errno != ETIMEDOUT && tlog.run) {
sleep(1); sleep(1);
} }
continue; continue;
@@ -1484,6 +1513,11 @@ int tlog_setlevel(tlog_level level)
return 0; return 0;
} }
tlog_level tlog_getlevel(void)
{
return tlog_set_level;
}
tlog_log *tlog_open(const char *logfile, int maxlogsize, int maxlogcount, int buffsize, unsigned int flag) tlog_log *tlog_open(const char *logfile, int maxlogsize, int maxlogcount, int buffsize, unsigned int flag)
{ {
struct tlog_log *log = NULL; struct tlog_log *log = NULL;

View File

@@ -1,6 +1,6 @@
/* /*
* tinylog * tinylog
* Copyright (C) 2018-2019 Ruilin Peng (Nick) <pymumu@gmail.com> * Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>
* https://github.com/pymumu/tinylog * https://github.com/pymumu/tinylog
*/ */
@@ -9,6 +9,11 @@
#include <stdarg.h> #include <stdarg.h>
#ifdef __cplusplus #ifdef __cplusplus
#include <string>
#include <memory>
#include <sstream>
#include <iostream>
#include <functional>
extern "C" { extern "C" {
#endif /*__cplusplus */ #endif /*__cplusplus */
@@ -24,13 +29,13 @@ typedef enum {
struct tlog_time { struct tlog_time {
int year; int year;
int mon; unsigned int usec;
int mday; unsigned char mon;
int hour; unsigned char mday;
int min; unsigned char hour;
int sec; unsigned char min;
int usec; unsigned char sec;
}; } __attribute__((packed));
#ifndef TLOG_MAX_LINE_LEN #ifndef TLOG_MAX_LINE_LEN
#define TLOG_MAX_LINE_LEN (1024) #define TLOG_MAX_LINE_LEN (1024)
@@ -55,7 +60,7 @@ struct tlog_time {
/* enable log to screen */ /* enable log to screen */
#define TLOG_SCREEN (1 << 4) #define TLOG_SCREEN (1 << 4)
struct tlog_info { struct tlog_loginfo {
tlog_level level; tlog_level level;
const char *file; const char *file;
const char *func; const char *func;
@@ -83,6 +88,9 @@ extern int tlog_write_log(char *buff, int bufflen);
/* set log level */ /* set log level */
extern int tlog_setlevel(tlog_level level); extern int tlog_setlevel(tlog_level level);
/* get log level */
extern tlog_level tlog_getlevel(void);
/* enalbe log to screen */ /* enalbe log to screen */
extern void tlog_setlogscreen(int enable); extern void tlog_setlogscreen(int enable);
@@ -113,13 +121,13 @@ steps:
read _tlog_format for example. read _tlog_format for example.
*/ */
typedef int (*tlog_format_func)(char *buff, int maxlen, struct tlog_info *info, void *userptr, const char *format, va_list ap); typedef int (*tlog_format_func)(char *buff, int maxlen, struct tlog_loginfo *info, void *userptr, const char *format, va_list ap);
extern int tlog_reg_format_func(tlog_format_func func); extern int tlog_reg_format_func(tlog_format_func func);
/* register log output callback /* register log output callback
Note: info is invalid when flag TLOG_SEGMENT is not set. Note: info is invalid when flag TLOG_SEGMENT is not set.
*/ */
typedef int (*tlog_log_output_func)(struct tlog_info *info, char *buff, int bufflen, void *private_data); typedef int (*tlog_log_output_func)(struct tlog_loginfo *info, const char *buff, int bufflen, void *private_data);
extern int tlog_reg_log_output_func(tlog_log_output_func output, void *private_data); extern int tlog_reg_log_output_func(tlog_log_output_func output, void *private_data);
struct tlog_log; struct tlog_log;
@@ -132,11 +140,11 @@ maxlogcount: Number of archived logs.
buffsize: Buffer size, zero for default (128K) buffsize: Buffer size, zero for default (128K)
flag: read tlog flags flag: read tlog flags
return: log stream handler. return: log stream handler.
*/ */
extern tlog_log *tlog_open(const char *logfile, int maxlogsize, int maxlogcount, int buffsize, unsigned int flag); extern tlog_log *tlog_open(const char *logfile, int maxlogsize, int maxlogcount, int buffsize, unsigned int flag);
/* write buff to log file */ /* write buff to log file */
extern int tlog_write(struct tlog_log *log, char *buff, int bufflen); extern int tlog_write(struct tlog_log *log, const char *buff, int bufflen);
/* close log stream */ /* close log stream */
extern void tlog_close(tlog_log *log); extern void tlog_close(tlog_log *log);
@@ -160,7 +168,7 @@ extern int tlog_vprintf(tlog_log *log, const char *format, va_list ap);
extern void tlog_logscreen(tlog_log *log, int enable); extern void tlog_logscreen(tlog_log *log, int enable);
/* register output callback */ /* register output callback */
typedef int (*tlog_output_func)(struct tlog_log *log, char *buff, int bufflen); typedef int (*tlog_output_func)(struct tlog_log *log, const char *buff, int bufflen);
extern int tlog_reg_output_func(tlog_log *log, tlog_output_func output); extern int tlog_reg_output_func(tlog_log *log, tlog_output_func output);
/* set private data */ /* set private data */
@@ -173,6 +181,65 @@ extern void *tlog_get_private(tlog_log *log);
extern int tlog_localtime(struct tlog_time *tm); extern int tlog_localtime(struct tlog_time *tm);
#ifdef __cplusplus #ifdef __cplusplus
} class Tlog {
#endif /*__cplusplus */ using Stream = std::ostringstream;
using Buffer = std::unique_ptr<Stream, std::function<void(Stream*)>>;
public:
Tlog(){}
~Tlog(){}
static Tlog &Instance() {
static Tlog logger;
return logger;
}
Buffer LogStream(tlog_level level, const char *file, int line, const char *func, void *userptr) {
return Buffer(new Stream, [=](Stream *st) {
tlog_ext(level, file, line, func, userptr, "%s", st->str().c_str());
delete st;
});
}
};
class TlogOut {
using Stream = std::ostringstream;
using Buffer = std::unique_ptr<Stream, std::function<void(Stream*)>>;
public:
TlogOut(){}
~TlogOut(){}
static TlogOut &Instance() {
static TlogOut logger;
return logger;
}
Buffer Out(tlog_log *log) {
return Buffer(new Stream, [=](Stream *st) {
tlog_printf(log, "%s", st->str().c_str());
delete st;
});
}
};
#define Tlog_logger (Tlog::Instance())
#define Tlog_stream(level) if (tlog_getlevel() <= level) *Tlog_logger.LogStream(level, BASE_FILE_NAME, __LINE__, __func__, NULL)
#define tlog_debug Tlog_stream(TLOG_DEBUG)
#define tlog_info Tlog_stream(TLOG_INFO)
#define tlog_notice Tlog_stream(TLOG_NOTICE)
#define tlog_warn Tlog_stream(TLOG_WARN)
#define tlog_error Tlog_stream(TLOG_ERROR)
#define tlog_fatal Tlog_stream(TLOG_FATAL)
#define Tlog_out_logger (TlogOut::Instance())
#define tlog_out(stream) (*Tlog_out_logger.Out(stream))
} /*__cplusplus */
#else
#define tlog_debug(...) tlog(TLOG_DEBUG, ##__VA_ARGS__)
#define tlog_info(...) tlog(TLOG_INFO, ##__VA_ARGS__)
#define tlog_notice(...) tlog(TLOG_NOTICE, ##__VA_ARGS__)
#define tlog_warn(...) tlog(TLOG_WARN, ##__VA_ARGS__)
#define tlog_error(...) tlog(TLOG_ERROR, ##__VA_ARGS__)
#define tlog_fatal(...) tlog(TLOG_FATAL, ##__VA_ARGS__)
#endif
#endif // !TLOG_H #endif // !TLOG_H

View File

@@ -1,3 +1,21 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _GNU_SOURCE #ifndef _GNU_SOURCE
#define _GNU_SOURCE #define _GNU_SOURCE
#endif #endif
@@ -6,19 +24,21 @@
#include <arpa/inet.h> #include <arpa/inet.h>
#include <errno.h> #include <errno.h>
#include <fcntl.h> #include <fcntl.h>
#include <linux/netlink.h> #include <inttypes.h>
#include <linux/capability.h> #include <linux/capability.h>
#include <linux/netlink.h>
#include <netinet/tcp.h>
#include <openssl/crypto.h> #include <openssl/crypto.h>
#include <openssl/ssl.h> #include <openssl/ssl.h>
#include <pthread.h> #include <pthread.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <sys/prctl.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <sys/statvfs.h>
#include <sys/types.h> #include <sys/types.h>
#include <time.h> #include <time.h>
#include <unistd.h> #include <unistd.h>
#include <inttypes.h>
#include <sys/prctl.h>
#define TMP_BUFF_LEN_32 32 #define TMP_BUFF_LEN_32 32
@@ -266,14 +286,14 @@ int parse_ip(const char *value, char *ip, int *port)
return 0; return 0;
} }
static int _check_is_ipv4(const char *ip) static int _check_is_ipv4(const char *ip)
{ {
const char *ptr = ip; const char *ptr = ip;
char c = 0; char c = 0;
int dot_num = 0; int dot_num = 0;
int dig_num = 0; int dig_num = 0;
while ( (c = *ptr++) != '\0') { while ((c = *ptr++) != '\0') {
if (c == '.') { if (c == '.') {
dot_num++; dot_num++;
dig_num = 0; dig_num = 0;
@@ -307,7 +327,7 @@ static int _check_is_ipv6(const char *ip)
int colon_num = 0; int colon_num = 0;
int dig_num = 0; int dig_num = 0;
while ( (c = *ptr++) != '\0') { while ((c = *ptr++) != '\0') {
if (c == '[' || c == ']') { if (c == '[' || c == ']') {
continue; continue;
} }
@@ -401,7 +421,7 @@ int parse_uri(char *value, char *scheme, char *host, int *port, char *path)
if (path) { if (path) {
strncpy(path, process_ptr, PATH_MAX); strncpy(path, process_ptr, PATH_MAX);
} }
return 0; return 0;
} }
@@ -437,8 +457,8 @@ char *reverse_string(char *output, const char *input, int len, int to_lower_case
if (to_lower_case) { if (to_lower_case) {
if (*output >= 'A' && *output <= 'Z') { if (*output >= 'A' && *output <= 'Z') {
/* To lower case */ /* To lower case */
*output = *output + 32; *output = *output + 32;
} }
} }
output++; output++;
len--; len--;
@@ -483,7 +503,8 @@ static int _ipset_support_timeout(const char *ipsetname)
return -1; return -1;
} }
static int _ipset_operate(const char *ipsetname, const unsigned char addr[], int addr_len, unsigned long timeout, int operate) static int _ipset_operate(const char *ipsetname, const unsigned char addr[], int addr_len, unsigned long timeout,
int operate)
{ {
struct nlmsghdr *netlink_head; struct nlmsghdr *netlink_head;
struct ipset_netlink_msg *netlink_msg; struct ipset_netlink_msg *netlink_msg;
@@ -541,7 +562,9 @@ static int _ipset_operate(const char *ipsetname, const unsigned char addr[], int
netlink_head->nlmsg_len += NETLINK_ALIGN(sizeof(struct ipset_netlink_attr)); netlink_head->nlmsg_len += NETLINK_ALIGN(sizeof(struct ipset_netlink_attr));
nested[1]->type = NLA_F_NESTED | IPSET_ATTR_IP; nested[1]->type = NLA_F_NESTED | IPSET_ATTR_IP;
_ipset_add_attr(netlink_head, (af == AF_INET ? IPSET_ATTR_IPADDR_IPV4 : IPSET_ATTR_IPADDR_IPV6) | NLA_F_NET_BYTEORDER, addr_len, addr); _ipset_add_attr(netlink_head,
(af == AF_INET ? IPSET_ATTR_IPADDR_IPV4 : IPSET_ATTR_IPADDR_IPV6) | NLA_F_NET_BYTEORDER, addr_len,
addr);
nested[1]->len = (void *)buffer + NETLINK_ALIGN(netlink_head->nlmsg_len) - (void *)nested[1]; nested[1]->len = (void *)buffer + NETLINK_ALIGN(netlink_head->nlmsg_len) - (void *)nested[1];
if (timeout > 0 && _ipset_support_timeout(ipsetname) == 0) { if (timeout > 0 && _ipset_support_timeout(ipsetname) == 0) {
@@ -668,6 +691,7 @@ errout:
return -1; return -1;
} }
#if OPENSSL_API_COMPAT < 0x10100000
#define THREAD_STACK_SIZE (16 * 1024) #define THREAD_STACK_SIZE (16 * 1024)
static pthread_mutex_t *lock_cs; static pthread_mutex_t *lock_cs;
static long *lock_count; static long *lock_count;
@@ -724,6 +748,7 @@ void SSL_CRYPTO_thread_cleanup(void)
OPENSSL_free(lock_cs); OPENSSL_free(lock_cs);
OPENSSL_free(lock_count); OPENSSL_free(lock_count);
} }
#endif
#define SERVER_NAME_LEN 256 #define SERVER_NAME_LEN 256
#define TLS_HEADER_LEN 5 #define TLS_HEADER_LEN 5
@@ -904,25 +929,35 @@ static int parse_server_name_extension(const char *data, size_t data_len, char *
return -2; return -2;
} }
void get_compiled_time(struct tm *tm) void get_compiled_time(struct tm *tm)
{ {
char s_month[5]; char s_month[5];
int month, day, year; int month, day, year;
int hour, min, sec; int hour, min, sec;
static const char *month_names = "JanFebMarAprMayJunJulAugSepOctNovDec"; static const char *month_names = "JanFebMarAprMayJunJulAugSepOctNovDec";
sscanf(__DATE__, "%5s %d %d", s_month, &day, &year); sscanf(__DATE__, "%5s %d %d", s_month, &day, &year);
month = (strstr(month_names, s_month) - month_names) / 3; month = (strstr(month_names, s_month) - month_names) / 3;
sscanf(__TIME__, "%d:%d:%d", &hour, &min, &sec); sscanf(__TIME__, "%d:%d:%d", &hour, &min, &sec);
tm->tm_year = year - 1900; tm->tm_year = year - 1900;
tm->tm_mon = month; tm->tm_mon = month;
tm->tm_mday = day; tm->tm_mday = day;
tm->tm_isdst = -1; tm->tm_isdst = -1;
tm->tm_hour = hour; tm->tm_hour = hour;
tm->tm_min = min; tm->tm_min = min;
tm->tm_sec = sec; tm->tm_sec = sec;
} }
int is_numeric(const char *str)
{
while (*str != '\0') {
if (*str < '0' || *str > '9')
return -1;
str++;
}
return 0;
}
int has_network_raw_cap(void) int has_network_raw_cap(void)
{ {
int fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP); int fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
@@ -933,3 +968,44 @@ int has_network_raw_cap(void)
close(fd); close(fd);
return 1; return 1;
} }
int set_sock_keepalive(int fd, int keepidle, int keepinterval, int keepcnt)
{
const int yes = 1;
if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &yes, sizeof(yes)) != 0) {
return -1;
}
setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, &keepidle, sizeof(keepidle));
setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, &keepinterval, sizeof(keepinterval));
setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, &keepcnt, sizeof(keepcnt));
return 0;
}
int set_sock_lingertime(int fd, int time)
{
struct linger l;
l.l_onoff = 1;
l.l_linger = 0;
if (setsockopt(fd, SOL_SOCKET, SO_LINGER, (const char *)&l, sizeof(l)) != 0) {
return -1;
}
return 0;
}
uint64_t get_free_space(const char *path)
{
uint64_t size = 0;
struct statvfs buf;
if (statvfs(path, &buf) != 0) {
return 0;
}
size = (uint64_t)buf.f_frsize * buf.f_bavail;
return size;
}

View File

@@ -1,16 +1,47 @@
/*************************************************************************
*
* Copyright (C) 2018-2020 Ruilin Peng (Nick) <pymumu@gmail.com>.
*
* smartdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* smartdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef SMART_DNS_UTIL_H #ifndef SMART_DNS_UTIL_H
#define SMART_DNS_UTIL_H #define SMART_DNS_UTIL_H
#include "stringutil.h"
#include <netdb.h> #include <netdb.h>
#include <time.h> #include <time.h>
#include "stringutil.h"
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
#endif /*__cplusplus */ #endif /*__cplusplus */
#ifndef TCP_FASTOPEN
#define TCP_FASTOPEN 23
#endif
#ifndef TCP_FASTOPEN_CONNECT
#define TCP_FASTOPEN_CONNECT 30
#endif
#ifndef TCP_THIN_LINEAR_TIMEOUTS
#define TCP_THIN_LINEAR_TIMEOUTS 16
#endif
#ifndef TCP_THIN_DUPACK
#define TCP_THIN_DUPACK 17
#endif
#define PORT_NOT_DEFINED -1 #define PORT_NOT_DEFINED -1
#define MAX_IP_LEN 64 #define MAX_IP_LEN 64
@@ -67,8 +98,16 @@ int parse_tls_header(const char *data, size_t data_len, char *hostname, const ch
void get_compiled_time(struct tm *tm); void get_compiled_time(struct tm *tm);
int is_numeric(const char *str);
int has_network_raw_cap(void); int has_network_raw_cap(void);
int set_sock_keepalive(int fd, int keepidle, int keepinterval, int keepcnt);
int set_sock_lingertime(int fd, int time);
uint64_t get_free_space(const char *path);
#ifdef __cplusplus #ifdef __cplusplus
} }
#endif /*__cplusplus */ #endif /*__cplusplus */

View File

@@ -1,17 +1,18 @@
[Unit] [Unit]
Description=smart dns server Description=smartdns server
After=network.target After=network.target
StartLimitBurst=0
StartLimitIntervalSec=60
TimeoutStopSec=5
[Service] [Service]
Type=forking Type=forking
PIDFile=/var/run/smartdns.pid PIDFile=@RUNSTATEDIR@/smartdns.pid
EnvironmentFile=/etc/default/smartdns EnvironmentFile=@SYSCONFDIR@/default/smartdns
ExecStart=/usr/sbin/smartdns $SMART_DNS_OPTS ExecStart=@SBINDIR@/smartdns -p @RUNSTATEDIR@/smartdns.pid $SMART_DNS_OPTS
KillMode=process KillMode=process
Restart=always Restart=always
RestartSec=2 RestartSec=2
StartLimitBurst=0
StartLimitIntervalSec=60
[Install] [Install]
WantedBy=multi-user.target WantedBy=multi-user.target