mirror of https://github.com/grpc/grpc.git
Merge pull request #14610 from apolcyn/python_dig
Remove the C++ test dependency on dig and dnsutilspull/14611/head
apolcyn
7 years ago
committed by
GitHub
commit
44dac6050a
10 changed files with 186 additions and 52 deletions
@ -0,0 +1,50 @@ |
|||||||
|
# Copyright 2017 gRPC authors. |
||||||
|
# |
||||||
|
# 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. |
||||||
|
|
||||||
|
package( |
||||||
|
default_visibility = ["//visibility:public"], |
||||||
|
features = [ |
||||||
|
"-layering_check", |
||||||
|
"-parse_headers", |
||||||
|
], |
||||||
|
) |
||||||
|
|
||||||
|
licenses(["notice"]) # Apache v2 |
||||||
|
|
||||||
|
load("//bazel:grpc_build_system.bzl", "grpc_py_binary") |
||||||
|
|
||||||
|
grpc_py_binary( |
||||||
|
name = "dns_server", |
||||||
|
srcs = ["dns_server.py"], |
||||||
|
testonly = True, |
||||||
|
external_deps = [ |
||||||
|
"twisted", |
||||||
|
"yaml", |
||||||
|
] |
||||||
|
) |
||||||
|
|
||||||
|
grpc_py_binary( |
||||||
|
name = "dns_resolver", |
||||||
|
srcs = ["dns_resolver.py"], |
||||||
|
testonly = True, |
||||||
|
external_deps = [ |
||||||
|
"twisted", |
||||||
|
] |
||||||
|
) |
||||||
|
|
||||||
|
grpc_py_binary( |
||||||
|
name = "tcp_connect", |
||||||
|
srcs = ["tcp_connect.py"], |
||||||
|
testonly = True, |
||||||
|
) |
||||||
@ -0,0 +1,48 @@ |
|||||||
|
#!/usr/bin/env python2.7 |
||||||
|
# Copyright 2015 gRPC authors. |
||||||
|
# |
||||||
|
# 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. |
||||||
|
|
||||||
|
"""Makes DNS queries for A records to specified servers""" |
||||||
|
|
||||||
|
import argparse |
||||||
|
import signal |
||||||
|
import twisted.internet.task as task |
||||||
|
import twisted.names.client as client |
||||||
|
|
||||||
|
def main(): |
||||||
|
argp = argparse.ArgumentParser(description='Make DNS queries for A records') |
||||||
|
argp.add_argument('-s', '--server_host', default='127.0.0.1', type=str, |
||||||
|
help='Host for DNS server to listen on for TCP and UDP.') |
||||||
|
argp.add_argument('-p', '--server_port', default=53, type=int, |
||||||
|
help='Port that the DNS server is listening on.') |
||||||
|
argp.add_argument('-n', '--qname', default=None, type=str, |
||||||
|
help=('Name of the record to query for. ')) |
||||||
|
argp.add_argument('-t', '--timeout', default=1, type=int, |
||||||
|
help=('Force process exit after this number of seconds.')) |
||||||
|
args = argp.parse_args() |
||||||
|
signal.alarm(args.timeout) |
||||||
|
def OnResolverResultAvailable(result): |
||||||
|
answers, authority, additional = result |
||||||
|
for a in answers: |
||||||
|
print(a.payload) |
||||||
|
def BeginQuery(reactor, qname): |
||||||
|
servers = [(args.server_host, args.server_port)] |
||||||
|
resolver = client.Resolver(servers=servers) |
||||||
|
deferred_result = resolver.lookupAddress(args.qname) |
||||||
|
deferred_result.addCallback(OnResolverResultAvailable) |
||||||
|
return deferred_result |
||||||
|
task.react(BeginQuery, [args.qname]) |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
main() |
||||||
@ -0,0 +1,35 @@ |
|||||||
|
#!/usr/bin/env python2.7 |
||||||
|
# Copyright 2015 gRPC authors. |
||||||
|
# |
||||||
|
# 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. |
||||||
|
|
||||||
|
"""Opens a TCP connection to a specified server and then exits.""" |
||||||
|
|
||||||
|
import argparse |
||||||
|
import signal |
||||||
|
import socket |
||||||
|
|
||||||
|
def main(): |
||||||
|
argp = argparse.ArgumentParser(description='Open a TCP handshake to a server') |
||||||
|
argp.add_argument('-s', '--server_host', default=None, type=str, |
||||||
|
help='Server host name or IP.') |
||||||
|
argp.add_argument('-p', '--server_port', default=0, type=int, |
||||||
|
help='Port that the server is listening on.') |
||||||
|
argp.add_argument('-t', '--timeout', default=1, type=int, |
||||||
|
help='Force process exit after this number of seconds.') |
||||||
|
args = argp.parse_args() |
||||||
|
signal.alarm(args.timeout) |
||||||
|
socket.create_connection([args.server_host, args.server_port]) |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
main() |
||||||
Loading…
Reference in new issue