Class: NicInfo::Main
- Inherits:
-
Object
- Object
- NicInfo::Main
- Defined in:
- lib/nicinfo/nicinfo_main.rb
Overview
The main class for the nicinfo command.
Instance Method Summary collapse
- #cache_self_references(json_data) ⇒ Object
-
#create_resource_url(args, queryType) ⇒ Object
Creates a query type.
- #display_rdap_query(json_data, show_help = true) ⇒ Object
- #do_rdap_query ⇒ Object
- #eval_json_value(json_value, json_data) ⇒ Object
-
#get(url, try, expect_rdap = true) ⇒ Object
Do an HTTP GET with the path.
-
#get_file_via_http(url, file_name, try) ⇒ Object
Do an HTTP GET of a file.
- #get_iana_files ⇒ Object
-
#get_query_type_from_result(json_data) ⇒ Object
Looks at the returned JSON and attempts to match that to a query type.
- #get_query_type_from_url(url) ⇒ Object
-
#guess_query_value_type(args) ⇒ Object
Evaluates the args and guesses at the type of query.
- #handle_error_response(res) ⇒ Object
- #help ⇒ Object
-
#initialize(args, config = nil) ⇒ Main
constructor
A new instance of Main.
- #inspect_rdap_compliance(json) ⇒ Object
- #make_rdap_url(base_url, resource_path) ⇒ Object
- #run ⇒ Object
- #show_conformance_messages ⇒ Object
- #show_helpful_messages(json_data, data_tree) ⇒ Object
- #show_search_results_truncated(json_data) ⇒ Object
Constructor Details
#initialize(args, config = nil) ⇒ Main
Returns a new instance of Main.
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 |
# File 'lib/nicinfo/nicinfo_main.rb', line 74 def initialize args, config = nil if config @config = config else @config = NicInfo::Config.new(NicInfo::Config::formulate_app_data_dir()) end @config..require_query = true @opts = OptionParser.new do |opts| opts. = "Usage: nicinfo [options] QUERY_VALUE" opts.version = NicInfo::VERSION opts.separator "" opts.separator "Query Options:" opts.on("-t", "--type TYPE", "Specify type of the query value.", " ip4addr - IPv4 address", " ip6addr - IPv6 address", " ip4cidr - IPv4 cidr block", " ip6cidr - IPv6 cidr block", " asnumber - autonomous system number", " domain - domain name", " entityhandle - handle or id of a contact, organization, registrar or other entity", " nameserver - fully qualified domain name of a nameserver", " result - result from a previous query", " esbyname - entity search by name", " dsbyname - domain search by name", " dsbynsname - domain search by nameserver name", " dsbynsip - domain search by nameserver IP address", " nsbyname - nameserver search by nameserver name", " nsbyip - nameserver search by IP address", " trace - trace route", " url - RDAP URL", " help - server help") do |type| uptype = type.upcase raise OptionParser::InvalidArgument, type.to_s unless QueryType.has_value?(uptype) @config..query_type = uptype @config..require_query = false if uptype == "HELP" end opts.on("-r", "--reverse", "Creates a reverse DNS name from an IP address. ") do |reverse| @config..reverse_ip = true end opts.on("-b", "--base (or bootstrap) URL", "The base URL of the RDAP Service.", "When set, the internal bootstrap is bypassed.") do |url| @config.config[ NicInfo::BOOTSTRAP][ NicInfo::BOOTSTRAP_URL ] = url end opts.separator "" opts.separator "Cache Options:" opts.on("--cache-expiry SECONDS", "Age in seconds of items in the cache to be considered expired.") do |s| @config.config[ NicInfo::CACHE ][ NicInfo::CACHE_EXPIRY ] = s end opts.on("--cache YES|NO|TRUE|FALSE", "Controls if the cache is used or not.") do |cc| @config.config[ NicInfo::CACHE ][ NicInfo::USE_CACHE ] = false if cc =~ /no|false/i @config.config[ NicInfo::CACHE ][ NicInfo::USE_CACHE ] = true if cc =~ /yes|true/i raise OptionParser::InvalidArgument, cc.to_s unless cc =~ /yes|no|true|false/i end opts.on("--empty-cache", "Empties the cache of all files regardless of eviction policy.") do |cc| @config..empty_cache = true @config..require_query = false end opts.on("--demo", "Populates the cache with demonstration results.") do |cc| @config..demo = true @config..require_query = false end opts.separator "" opts.separator "Output Options:" opts.on( "--messages MESSAGE_LEVEL", "Specify the message level", " none - no messages are to be output", " some - some messages but not all", " all - all messages to be outupt" ) do |m| @config.logger. = m.to_s.upcase begin @config.logger. rescue raise OptionParser::InvalidArgument, m.to_s end end opts.on( "--messages-out FILE", "FILE where messages will be written." ) do |f| @config.logger. = File.open( f, "w+" ) end opts.on( "--data DATA_AMOUNT", "Specify the amount of data", " terse - enough data to identify the object", " normal - normal view of data on objects", " extra - all data about the object" ) do |d| @config.logger.data_amount = d.to_s.upcase begin @config.logger.validate_data_amount rescue raise OptionParser::InvalidArgument, d.to_s end end opts.on( "--data-out FILE", "FILE where data will be written." ) do |f| @config.logger.data_out = File.open( f, "w+" ) end opts.on( "--pager YES|NO|TRUE|FALSE", "Turns the pager on and off." ) do |pager| @config.logger.pager = false if pager =~ /no|false/i @config.logger.pager = true if pager =~ /yes|true/i raise OptionParser::InvalidArgument, pager.to_s unless pager =~ /yes|no|true|false/i end opts.on( "--color-scheme DARK|LIGHT|NONE", "Determines color scheme to use:", " dark - for terminals with dark backgrounds", " light - for terminals with light backgrounds", " none - turn off colors" ) do |cs| @config.logger.color_scheme = cs.to_s.upcase raise OptionParser::InvalidArgument, cs.to_s unless cs =~ /dark|light|none/i end opts.on( "-V", "Equivalent to --messages all and --data extra" ) do |v| @config.logger.data_amount = NicInfo::DataAmount::EXTRA_DATA @config.logger. = NicInfo::MessageLevel::ALL_MESSAGES end opts.on( "-Q", "Equivalent to --messages none and --data extra and --pager false" ) do |q| @config.logger.data_amount = NicInfo::DataAmount::EXTRA_DATA @config.logger. = NicInfo::MessageLevel::NO_MESSAGES @config.logger.pager = false end opts.on( "--json", "Output raw JSON response." ) do |json| @config..output_json = true end opts.on( "--jv VALUE", "Outputs a specific JSON value." ) do |value| unless @config..json_values @config..json_values = Array.new end @config..json_values << value end opts.separator "" opts.separator "Security Options:" opts.on( "--try-insecure YES|NO|TRUE|FALSE", "Try HTTP if HTTPS fails" ) do |try_insecure| @config.config[ NicInfo::SECURITY ][ NicInfo::TRY_INSECURE ] = false if try_insecure =~ /no|false/i @config.config[ NicInfo::SECURITY ][ NicInfo::TRY_INSECURE ] = true if try_insecure =~ /yes|true/i raise OptionsParser::InvalidArgument, try_insecure.to_s unless try_insecure =~/yes|no|true|false/i end opts.separator "" opts.separator "General Options:" opts.on( "-h", "--help", "Show this message" ) do @config..help = true @config..require_query = false end opts.on( "--reset", "Reset configuration to defaults" ) do @config..reset_config = true @config..require_query = false end opts.on( "--iana", "Download RDAP bootstrap files from IANA" ) do @config..get_iana_files = true @config..require_query = false end end begin @opts.parse!(args) rescue OptionParser::InvalidOption => e puts e. puts "use -h for help" exit rescue OptionParser::InvalidArgument => e puts e. puts "use -h for help" exit rescue puts "Unable to parse command line options" puts "use -h for help" exit end @config..argv = args end |
Instance Method Details
#cache_self_references(json_data) ⇒ Object
950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 |
# File 'lib/nicinfo/nicinfo_main.rb', line 950 def cache_self_references json_data links = NicInfo::get_links json_data, @config if links self_link = NicInfo.get_self_link links if self_link pretty = JSON::pretty_generate( json_data ) @cache.create( self_link, pretty ) end end entities = NicInfo::get_entitites json_data entities.each do |entity| cache_self_references( entity ) end if entities nameservers = NicInfo::get_nameservers json_data nameservers.each do |ns| cache_self_references( ns ) end if nameservers ds_data_objs = NicInfo::get_ds_data_objs json_data ds_data_objs.each do |ds| cache_self_references( ds ) end if ds_data_objs key_data_objs = NicInfo::get_key_data_objs json_data key_data_objs.each do |key| cache_self_references( key ) end if key_data_objs end |
#create_resource_url(args, queryType) ⇒ Object
Creates a query type
853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 |
# File 'lib/nicinfo/nicinfo_main.rb', line 853 def create_resource_url(args, queryType) path = "" case queryType when QueryType::BY_IP4_ADDR path << "ip/" << args[0] when QueryType::BY_IP6_ADDR path << "ip/" << args[0] when QueryType::BY_IP4_CIDR path << "ip/" << args[0] when QueryType::BY_IP6_CIDR path << "ip/" << args[0] when QueryType::BY_AS_NUMBER path << "autnum/" << args[0] when QueryType::BY_NAMESERVER path << "nameserver/" << args[0] when QueryType::BY_DOMAIN path << "domain/" << args[0] when QueryType::BY_RESULT tree = @config.load_as_yaml(NicInfo::ARININFO_LASTTREE_YAML) path = tree.find_rest_ref(args[0]) raise ArgumentError.new("Unable to find result for " + args[0]) unless path when QueryType::BY_ENTITY_HANDLE path << "entity/" << URI.escape( args[ 0 ] ) when QueryType::SRCH_ENTITY_BY_NAME case args.length when 1 path << "entities?fn=" << URI.escape( args[ 0 ] ) when 2 path << "entities?fn=" << URI.escape( args[ 0 ] + " " + args[ 1 ] ) when 3 path << "entities?fn=" << URI.escape( args[ 0 ] + " " + args[ 1 ] + " " + args[ 2 ] ) end when QueryType::SRCH_DOMAIN_BY_NAME path << "domains?name=" << args[ 0 ] when QueryType::SRCH_DOMAIN_BY_NSNAME path << "domains?nsLdhName=" << args[ 0 ] when QueryType::SRCH_DOMAIN_BY_NSIP path << "domains?nsIp=" << args[ 0 ] when QueryType::SRCH_NS_BY_NAME path << "nameservers?name=" << args[ 0 ] when QueryType::SRCH_NS_BY_IP path << "nameservers?ip=" << args[ 0 ] when QueryType::BY_SERVER_HELP path << "help" else raise ArgumentError.new("Unable to create a resource URL for " + queryType) end return path end |
#display_rdap_query(json_data, show_help = true) ⇒ Object
633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 |
# File 'lib/nicinfo/nicinfo_main.rb', line 633 def display_rdap_query json_data, show_help = true if @config..output_json @config.logger.raw( DataAmount::TERSE_DATA, json_data, false ) elsif @config..json_values @config..json_values.each do |value| @config.logger.raw( DataAmount::TERSE_DATA, eval_json_value( value, json_data), false ) end else @config.factory.new_notices.display_notices json_data, @config..query_type == QueryType::BY_SERVER_HELP if @config..query_type != QueryType::BY_SERVER_HELP result_type = get_query_type_from_result( json_data ) if result_type != nil if result_type != @config..query_type @config.logger.mesg( "Query type is " + @config..query_type + ". Result type is " + result_type + "." ) else @config.logger.mesg( "Result type is " + result_type + "." ) end @config..query_type = result_type elsif json_data[ "errorCode" ] == nil @config.conf_msgs << "Response has no result type." end data_tree = DataTree.new( ) case @config..query_type when QueryType::BY_IP4_ADDR NicInfo::display_ip( json_data, @config, data_tree ) when QueryType::BY_IP6_ADDR NicInfo::display_ip( json_data, @config, data_tree ) when QueryType::BY_IP4_CIDR NicInfo::display_ip( json_data, @config, data_tree ) when QueryType::BY_IP6_CIDR NicInfo::display_ip( json_data, @config, data_tree ) when QueryType::BY_IP NicInfo::display_ip( json_data, @config, data_tree ) when QueryType::BY_AS_NUMBER NicInfo::display_autnum( json_data, @config, data_tree ) when "NicInfo::DsData" NicInfo::display_ds_data( json_data, @config, data_tree ) when "NicInfo::KeyData" NicInfo::display_key_data( json_data, @config, data_tree ) when QueryType::BY_DOMAIN NicInfo::display_domain( json_data, @config, data_tree ) when QueryType::BY_NAMESERVER NicInfo::display_ns( json_data, @config, data_tree ) when QueryType::BY_ENTITY_HANDLE NicInfo::display_entity( json_data, @config, data_tree ) when QueryType::SRCH_DOMAINS NicInfo::display_domains( json_data, @config, data_tree ) when QueryType::SRCH_DOMAIN_BY_NAME NicInfo::display_domains( json_data, @config, data_tree ) when QueryType::SRCH_DOMAIN_BY_NSNAME NicInfo::display_domains( json_data, @config, data_tree ) when QueryType::SRCH_DOMAIN_BY_NSIP NicInfo::display_domains( json_data, @config, data_tree ) when QueryType::SRCH_ENTITY_BY_NAME NicInfo::display_entities( json_data, @config, data_tree ) when QueryType::SRCH_NS NicInfo::display_nameservers( json_data, @config, data_tree ) when QueryType::SRCH_NS_BY_NAME NicInfo::display_nameservers( json_data, @config, data_tree ) when QueryType::SRCH_NS_BY_IP NicInfo::display_nameservers( json_data, @config, data_tree ) end @config.save_as_yaml( NicInfo::LASTTREE_YAML, data_tree ) if !data_tree.empty? show_search_results_truncated json_data json_data, data_tree if show_help end end @config.logger.end_run end |
#do_rdap_query ⇒ Object
532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 |
# File 'lib/nicinfo/nicinfo_main.rb', line 532 def do_rdap_query retval = nil if @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] == nil && !@config..url bootstrap = Bootstrap.new( @config ) qtype = @config..query_type if qtype == QueryType::BY_SERVER_HELP qtype = guess_query_value_type( @config..argv ) end case qtype when QueryType::BY_IP4_ADDR @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_ip( @config..argv[ 0 ] ) when QueryType::BY_IP6_ADDR @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_ip( @config..argv[ 0 ] ) when QueryType::BY_IP4_CIDR @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_ip( @config..argv[ 0 ] ) when QueryType::BY_IP6_CIDR @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_ip( @config..argv[ 0 ] ) when QueryType::BY_AS_NUMBER @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_as( @config..argv[ 0 ] ) when QueryType::BY_DOMAIN @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_domain( @config..argv[ 0 ] ) when QueryType::BY_NAMESERVER @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_domain( @config..argv[ 0 ] ) when QueryType::BY_ENTITY_HANDLE @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_entity( @config..argv[ 0 ] ) when QueryType::SRCH_ENTITY_BY_NAME @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::ENTITY_ROOT_URL ] when QueryType::SRCH_DOMAIN_BY_NAME @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_domain( @config..argv[ 0 ] ) when QueryType::SRCH_DOMAIN_BY_NSNAME @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_domain( @config..argv[ 0 ] ) when QueryType::SRCH_DOMAIN_BY_NSIP @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::DOMAIN_ROOT_URL ] when QueryType::SRCH_NS_BY_NAME @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_domain( @config..argv[ 0 ] ) when QueryType::SRCH_NS_BY_IP @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = bootstrap.find_url_by_ip( @config..argv[ 0 ] ) else @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::HELP_ROOT_URL ] end end begin rdap_url = nil unless @config..url path = create_resource_url(@config..argv, @config..query_type) rdap_url = make_rdap_url(@config.config[NicInfo::BOOTSTRAP][NicInfo::BOOTSTRAP_URL], path) else rdap_url = @config..argv[0] end data = get( rdap_url, 0 ) json_data = JSON.load data if (ec = json_data[ NicInfo::NICINFO_DEMO_ERROR ]) != nil res = MyHTTPResponse.new( "1.1", ec, "Demo Exception" ) res["content-type"] = NicInfo::RDAP_CONTENT_TYPE res.body=data raise Net::HTTPServerException.new( "Demo Exception", res ) end inspect_rdap_compliance json_data cache_self_references json_data retval = json_data rescue JSON::ParserError => a @config.logger.mesg( "Server returned invalid JSON!", NicInfo::AttentionType::ERROR ) rescue SocketError => a @config.logger.mesg(a., NicInfo::AttentionType::ERROR ) rescue ArgumentError => a @config.logger.mesg(a., NicInfo::AttentionType::ERROR ) rescue Net::HTTPServerException => e case e.response.code when "200" @config.logger.mesg( e., NicInfo::AttentionType::SUCCESS ) when "401" @config.logger.mesg("Authorization is required.", NicInfo::AttentionType::ERROR ) handle_error_response e.response when "404" @config.logger.mesg("Query yielded no results.", NicInfo::AttentionType::INFO ) handle_error_response e.response else @config.logger.mesg("Error #{e.response.code}.", NicInfo::AttentionType::ERROR ) handle_error_response e.response end @config.logger.trace("Server response code was " + e.response.code) rescue Net::HTTPFatalError => e case e.response.code when "500" @config.logger.mesg("RDAP server is reporting an internal error.", NicInfo::AttentionType::ERROR ) handle_error_response e.response when "501" @config.logger.mesg("RDAP server does not implement the query.", NicInfo::AttentionType::ERROR ) handle_error_response e.response when "503" @config.logger.mesg("RDAP server is reporting that it is unavailable.", NicInfo::AttentionType::ERROR ) handle_error_response e.response else @config.logger.mesg("Error #{e.response.code}.", NicInfo::AttentionType::ERROR ) handle_error_response e.response end @config.logger.trace("Server response code was " + e.response.code) end return retval end |
#eval_json_value(json_value, json_data) ⇒ Object
935 936 937 938 939 940 941 942 943 944 945 946 947 948 |
# File 'lib/nicinfo/nicinfo_main.rb', line 935 def eval_json_value json_value, json_data appended_code = String.new values = json_value.split( "." ) values.each do |value| i = Integer( value ) rescue false if i appended_code << "[#{i}]" else appended_code << "[\"#{value}\"]" end end code = "json_data#{appended_code}" return eval( code ) end |
#get(url, try, expect_rdap = true) ⇒ Object
Do an HTTP GET with the path.
297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 |
# File 'lib/nicinfo/nicinfo_main.rb', line 297 def get url, try, expect_rdap = true data = @cache.get(url) if data == nil @config.logger.trace("Issuing GET for " + url) uri = URI.parse( URI::encode( url ) ) req = Net::HTTP::Get.new(uri.request_uri) req["User-Agent"] = NicInfo::VERSION_LABEL req["Accept"] = NicInfo::RDAP_CONTENT_TYPE + ", " + NicInfo::JSON_CONTENT_TYPE req["Connection"] = "close" http = Net::HTTP.new( uri.host, uri.port ) if uri.scheme == "https" http.use_ssl=true http.verify_mode=OpenSSL::SSL::VERIFY_NONE end begin res = http.start do |http_req| http_req.request(req) end rescue OpenSSL::SSL::SSLError => e if @config.config[ NicInfo::SECURITY ][ NicInfo::TRY_INSECURE ] @config.logger.mesg( "Secure connection failed. Trying insecure connection." ) uri.scheme = "http" return get( uri.to_s, try, expect_rdap ) else raise e end end case res when Net::HTTPSuccess content_type = res[ "content-type" ].downcase if expect_rdap unless content_type.include?(NicInfo::RDAP_CONTENT_TYPE) or content_type.include?(NicInfo::JSON_CONTENT_TYPE) raise Net::HTTPServerException.new("Bad Content Type", res) end if content_type.include? NicInfo::JSON_CONTENT_TYPE @config.conf_msgs << "Server responded with non-RDAP content type but it is JSON" end end data = res.body @cache.create_or_update(url, data) else if res.code == "301" or res.code == "302" or res.code == "303" or res.code == "307" or res.code == "308" res.error! if try >= 5 location = res["location"] return get( location, try + 1, expect_rdap) end res.error! end #end case end #end if return data end |
#get_file_via_http(url, file_name, try) ⇒ Object
Do an HTTP GET of a file
357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 |
# File 'lib/nicinfo/nicinfo_main.rb', line 357 def get_file_via_http url, file_name, try @config.logger.trace("Downloading " + url + " to " + file_name ) uri = URI.parse( URI::encode( url ) ) req = Net::HTTP::Get.new(uri.request_uri) req["User-Agent"] = NicInfo::VERSION_LABEL req["Accept"] = NicInfo::JSON_CONTENT_TYPE req["Connection"] = "close" http = Net::HTTP.new( uri.host, uri.port ) if uri.scheme == "https" http.use_ssl=true http.verify_mode=OpenSSL::SSL::VERIFY_NONE end res = http.start do |http_req| http_req.request(req) end case res when Net::HTTPSuccess File.write(file_name, res.body) else if res.code == "301" or res.code == "302" or res.code == "303" or res.code == "307" or res.code == "308" res.error! if try >= 5 location = res["location"] return get_file_via_http( location, file_name, try + 1) end res.error! end end |
#get_iana_files ⇒ Object
524 525 526 527 528 529 530 |
# File 'lib/nicinfo/nicinfo_main.rb', line 524 def get_iana_files get_file_via_http("http://data.iana.org/rdap/asn.json", File.join(@config.rdap_bootstrap_dir, "asn.json"), 0) get_file_via_http("http://data.iana.org/rdap/ipv4.json", File.join(@config.rdap_bootstrap_dir, "ipv4.json"), 0) get_file_via_http("http://data.iana.org/rdap/ipv6.json", File.join(@config.rdap_bootstrap_dir, "ipv6.json"), 0) get_file_via_http("http://data.iana.org/rdap/dns.json", File.join(@config.rdap_bootstrap_dir, "dns.json"), 0) @config.set_bsfiles_last_update_time end |
#get_query_type_from_result(json_data) ⇒ Object
Looks at the returned JSON and attempts to match that to a query type.
754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 |
# File 'lib/nicinfo/nicinfo_main.rb', line 754 def get_query_type_from_result( json_data ) retval = nil object_class_name = json_data[ "objectClassName" ] if object_class_name != nil case object_class_name when "domain" retval = QueryType::BY_DOMAIN when "ip network" retval = QueryType::BY_IP when "entity" retval = QueryType::BY_ENTITY_HANDLE when "autnum" retval = QueryType::BY_AS_NUMBER when "nameserver" retval = QueryType::BY_NAMESERVER end end if json_data[ "domainSearchResults" ] retval = QueryType::SRCH_DOMAINS elsif json_data[ "nameserverSearchResults" ] retval = QueryType::SRCH_NS elsif json_data[ "entitySearchResults" ] retval = QueryType::SRCH_ENTITY_BY_NAME end return retval end |
#get_query_type_from_url(url) ⇒ Object
905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 |
# File 'lib/nicinfo/nicinfo_main.rb', line 905 def get_query_type_from_url url queryType = nil case url when /.*\/ip\/.*/ # covers all IP cases queryType = QueryType::BY_IP when /.*\/autnum\/.*/ queryType = QueryType::BY_AS_NUMBER when /.*\/nameserver\/.*/ queryType = QueryType::BY_NAMESERVER when /.*\/domain\/.*/ queryType = QueryType::BY_DOMAIN when /.*\/entity\/.*/ queryType = QueryType::BY_ENTITY_HANDLE when /.*\/entities.*/ queryType = QueryType::SRCH_ENTITY_BY_NAME when /.*\/domains.*/ # covers all domain searches queryType = QueryType::SRCH_DOMAIN when /.*\/nameservers.*/ # covers all nameserver searches queryType = QueryType::SRCH_NS when /.*\/help.*/ queryType = QueryType::BY_SERVER_HELP else raise ArgumentError.new( "Unable to determine query type from url '#{url}'" ) end return queryType end |
#guess_query_value_type(args) ⇒ Object
Evaluates the args and guesses at the type of query. Args is an array of strings, most likely what is left over after parsing ARGV
784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 |
# File 'lib/nicinfo/nicinfo_main.rb', line 784 def guess_query_value_type(args) retval = nil if args.length() == 1 case args[0] when NicInfo::URL_REGEX retval = QueryType::BY_URL when NicInfo::IPV4_REGEX retval = QueryType::BY_IP4_ADDR when NicInfo::IPV6_REGEX retval = QueryType::BY_IP6_ADDR when NicInfo::IPV6_HEXCOMPRESS_REGEX retval = QueryType::BY_IP6_ADDR when NicInfo::AS_REGEX retval = QueryType::BY_AS_NUMBER when NicInfo::ASN_REGEX old = args[0] args[0] = args[0].sub(/^AS/i, "") @config.logger.trace("Interpretting " + old + " as autonomous system number " + args[0]) retval = QueryType::BY_AS_NUMBER when NicInfo::IP4_ARPA retval = QueryType::BY_DOMAIN when NicInfo::IP6_ARPA retval = QueryType::BY_DOMAIN when /(.*)\/\d/ ip = $+ if ip =~ NicInfo::IPV4_REGEX retval = QueryType::BY_IP4_CIDR elsif ip =~ NicInfo::IPV6_REGEX || ip =~ NicInfo::IPV6_HEXCOMPRESS_REGEX retval = QueryType::BY_IP6_CIDR end when NicInfo::DATA_TREE_ADDR_REGEX retval = QueryType::BY_RESULT when NicInfo::NS_REGEX retval = QueryType::BY_NAMESERVER when NicInfo::DOMAIN_REGEX retval = QueryType::BY_DOMAIN when NicInfo::ENTITY_REGEX retval = QueryType::BY_ENTITY_HANDLE else last_name = args[ 0 ].sub( /\*/, '' ).upcase if NicInfo::is_last_name( last_name ) retval = QueryType::SRCH_ENTITY_BY_NAME end end elsif args.length() == 2 last_name = args[ 1 ].sub( /\*/, '' ).upcase first_name = args[ 0 ].sub( /\*/, '' ).upcase if NicInfo::is_last_name(last_name) && (NicInfo::is_male_name(first_name) || NicInfo::is_female_name(first_name)) retval = QueryType::SRCH_ENTITY_BY_NAME end elsif args.length() == 3 last_name = args[ 2 ].sub( /\*/, '' ).upcase first_name = args[ 0 ].sub( /\*/, '' ).upcase if NicInfo::is_last_name(last_name) && (NicInfo::is_male_name(first_name) || NicInfo::is_female_name(first_name)) retval = QueryType::SRCH_ENTITY_BY_NAME end end return retval end |
#handle_error_response(res) ⇒ Object
704 705 706 707 708 709 710 711 |
# File 'lib/nicinfo/nicinfo_main.rb', line 704 def handle_error_response (res) if res["content-type"] == NicInfo::RDAP_CONTENT_TYPE && res.body && res.body.to_s.size > 0 json_data = JSON.load( res.body ) inspect_rdap_compliance json_data @config.factory.new_notices.display_notices json_data, true @config.factory.new_error_code.display_error_code( json_data ) end end |
#help ⇒ Object
724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 |
# File 'lib/nicinfo/nicinfo_main.rb', line 724 def help puts NicInfo::VERSION_LABEL puts NicInfo::COPYRIGHT puts <<HELP_SUMMARY SYNOPSIS nicinfo [OPTIONS] QUERY_VALUE SUMMARY NicInfo is a Registry Data Access Protocol (RDAP) client capable of querying RDAP servers containing IP address, Autonomous System, and Domain name information. The general usage is "nicinfo QUERY_VALUE" where the QUERY_VALUE is an IP address, autonomous system number, or domain name. The type of query to perform is implicitly determined but maybe explicitly set using the -t option. When the QUERY_VALUE is simply a dot or period character (e.g. "."), the IP address of the client is implied. Given the type of query to perform, this program will attempt to use the most appropriate RDAP server it can determine, and follow referrals from that server if necessary. HELP_SUMMARY puts @opts.help puts EXTENDED_HELP exit end |
#inspect_rdap_compliance(json) ⇒ Object
713 714 715 716 717 718 719 720 721 722 |
# File 'lib/nicinfo/nicinfo_main.rb', line 713 def inspect_rdap_compliance json rdap_conformance = json[ "rdapConformance" ] if rdap_conformance rdap_conformance.each do |conformance| @config.logger.trace( "Server conforms to #{conformance}", NicInfo::AttentionType::SECONDARY ) end else @config.conf_msgs << "Response has no RDAP Conformance level specified." end end |
#make_rdap_url(base_url, resource_path) ⇒ Object
289 290 291 292 293 294 |
# File 'lib/nicinfo/nicinfo_main.rb', line 289 def make_rdap_url( base_url, resource_path ) unless base_url.end_with?("/") base_url << "/" end base_url << resource_path end |
#run ⇒ Object
388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 |
# File 'lib/nicinfo/nicinfo_main.rb', line 388 def run @config.logger.run_pager @config.logger.mesg(NicInfo::VERSION_LABEL, NicInfo::AttentionType::PRIMARY ) @config.setup_workspace @config.check_config_version @cache = Cache.new(@config) @cache.clean if @config.config[ NicInfo::CACHE ][ NicInfo::CLEAN_CACHE ] if @config..empty_cache @cache.empty end if @config..get_iana_files get_iana_files else check_bsfiles_age = @config.check_bsfiles_age? update_bsfiles = @config.update_bsfiles?( check_bsfiles_age ) if update_bsfiles @config.logger.mesg( "IANA RDAP bootstrap files are old and need to be updated.", NicInfo::AttentionType::ERROR ) get_iana_files elsif check_bsfiles_age @config.logger.mesg( "IANA RDAP bootstrap files are old. Update them with --iana option", NicInfo::AttentionType::ERROR ) end end if @config..demo @config.logger.mesg( "Populating cache with demonstration results", NicInfo::AttentionType::INFO ) @config.logger.mesg( "Try the following demonstration queries:", NicInfo::AttentionType::INFO ) demo_dir = File.join( File.dirname( __FILE__ ), NicInfo::DEMO_DIR ) demo_files = Dir::entries( demo_dir ) demo_files.each do |file| df = File.join( demo_dir, file ) if File.file?( df ) demo_data = File.read( df ) json_data = JSON.load demo_data demo_url = json_data[ NicInfo::NICINFO_DEMO_URL ] demo_hint = json_data[ NicInfo::NICINFO_DEMO_HINT ] @cache.create( demo_url, demo_data ) @config.logger.mesg( " " + demo_hint, NicInfo::AttentionType::INFO ) end end end if @config..help help() end if @config..argv == nil || @config..argv == [] && !@config..query_type unless @config..require_query exit else help end end if @config..argv[0] == '.' @config.logger.mesg( "Obtaining current IP Address...") data = get("https://stat.ripe.net/data/whats-my-ip/data.json", 0, false ) json_data = JSON.load(data) if json_data["data"] == nil || json_data["data"]["ip"] == nil @config.logger.mesg("Server repsonded with unknown JSON") @config.logger.mesg("Unable to determine your IP Address. You must specify it.") exit elsif @config.logger.mesg("Your IP address is " + json_data["data"]["ip"], NicInfo::AttentionType::SUCCESS ) @config..argv[0] = json_data["data"]["ip"] end end if @config..query_type == nil @config..query_type = guess_query_value_type(@config..argv) if (@config..query_type == QueryType::BY_IP4_ADDR || @config..query_type == QueryType::BY_IP6_ADDR ) && @config..reverse_ip == true ip = IPAddr.new( @config..argv[ 0 ] ) @config..argv[ 0 ] = ip.reverse.split( "\." )[ 1..-1 ].join( "." ) if ip.ipv4? @config..argv[ 0 ] = ip.reverse.split( "\." )[ 24..-1 ].join( "." ) if ip.ipv6? @config.logger.mesg( "Query value changed to " + @config..argv[ 0 ] ) @config..query_type = QueryType::BY_DOMAIN @config..externally_queriable = true elsif @config..query_type == QueryType::BY_RESULT data_tree = @config.load_as_yaml( NicInfo::LASTTREE_YAML ) node = data_tree.find_node( @config..argv[ 0 ] ) if node and node.rest_ref @config..argv[ 0 ] = node.rest_ref @config..url = true if node.data_type @config..query_type = node.data_type @config..externally_queriable = false elsif node.rest_ref @config..query_type = get_query_type_from_url( node.rest_ref ) @config..externally_queriable = true end else @config.logger.mesg( "#{@config..argv[0]} is not retrievable.") exit end elsif @config..query_type == QueryType::BY_URL @config..url = @config..argv[0] @config..query_type = get_query_type_from_url( @config..url ) else @config..externally_queriable = true end if @config..query_type == nil @config.logger.mesg("Unable to guess type of query. You must specify it.") exit else @config.logger.trace("Assuming query value is " + @config..query_type) end end #determine if this will be a single query or multiple qtype = @config..query_type case qtype when QueryType::TRACE ips = NicInfo.traceroute @config..argv[ 0 ], @config if ips.empty? @config.logger.mesg "Trace route yeilded no data" else ips.each do |ip| @config..query_type = QueryType::BY_IP4_ADDR @config..argv[ 0 ] = ip @config.config[ NicInfo::BOOTSTRAP ][ NicInfo::BOOTSTRAP_URL ] = nil json_data = do_rdap_query display_rdap_query( json_data, false ) end end else json_data = do_rdap_query display_rdap_query( json_data, true ) if json_data end end |
#show_conformance_messages ⇒ Object
977 978 979 980 981 982 983 984 985 |
# File 'lib/nicinfo/nicinfo_main.rb', line 977 def return if @config.conf_msgs.size == 0 @config.logger.mesg( "** WARNING: There are problems in the response that might cause some data to discarded. **", NicInfo::AttentionType::ERROR ) i = 1 @config.conf_msgs.each do |msg| @config.logger.trace( "#{i} : #{msg}", NicInfo::AttentionType::ERROR ) i = i + 1 end end |
#show_helpful_messages(json_data, data_tree) ⇒ Object
999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 |
# File 'lib/nicinfo/nicinfo_main.rb', line 999 def json_data, data_tree arg = @config..argv[0] case @config..query_type when QueryType::BY_IP4_ADDR @config.logger.mesg("Use \"nicinfo -r #{arg}\" to see reverse DNS information."); when QueryType::BY_IP6_ADDR @config.logger.mesg("Use \"nicinfo -r #{arg}\" to see reverse DNS information."); when QueryType::BY_AS_NUMBER @config.logger.mesg("Use \"nicinfo #{arg}\" or \"nicinfo as#{arg}\" for autnums."); end unless data_tree.empty? @config.logger.mesg("Use \"nicinfo 1=\" to show #{data_tree.roots.first}") unless data_tree.roots.first.empty? children = data_tree.roots.first.children @config.logger.mesg("Use \"nicinfo 1.1=\" to show #{children.first}") if children.first.rest_ref if children.first != children.last len = children.length @config.logger.mesg("Use \"nicinfo 1.#{len}=\" to show #{children.last}") if children.last.rest_ref end end end self_link = NicInfo.get_self_link( NicInfo.get_links( json_data, @config ) ) @config.logger.mesg("Use \"nicinfo #{self_link}\" to directly query this resource in the future.") if self_link and @config..externally_queriable @config.logger.mesg('Use "nicinfo -h" for help.') end |
#show_search_results_truncated(json_data) ⇒ Object
987 988 989 990 991 992 993 994 995 996 997 |
# File 'lib/nicinfo/nicinfo_main.rb', line 987 def show_search_results_truncated json_data truncated = json_data[ "searchResultsTruncated" ] if truncated.instance_of?(TrueClass) || truncated.instance_of?(FalseClass) if truncated @config.logger.mesg( "Results have been truncated by the server.", NicInfo::AttentionType::INFO ) end end if truncated != nil @config.conf_msgs << "'searchResultsTruncated' is not part of the RDAP specification and was removed before standardization." end end |