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