pax_global_header00006660000000000000000000000064130102232250014500gustar00rootroot0000000000000052 comment=285f23be6bc07764b7cc7544b5b48234e4a63e03 prometheus-blackbox-exporter-0.3.0+ds1/000077500000000000000000000000001301022322500200075ustar00rootroot00000000000000prometheus-blackbox-exporter-0.3.0+ds1/.dockerignore000066400000000000000000000000511301022322500224570ustar00rootroot00000000000000.build/ .tarballs/ !.build/linux-amd64/ prometheus-blackbox-exporter-0.3.0+ds1/.gitignore000066400000000000000000000005161301022322500220010ustar00rootroot00000000000000# Compiled Object files, Static and Dynamic libs (Shared Objects) *.o *.a *.so # Folders _obj _test # Architecture specific extensions/prefixes *.[568vq] [568vq].out *.cgo1.go *.cgo2.c _cgo_defun.c _cgo_gotypes.go _cgo_export.* _testmain.go *.exe dependencies-stamp /blackbox_exporter /.build /.release /.tarballs .deps *.tar.gz prometheus-blackbox-exporter-0.3.0+ds1/.promu.yml000066400000000000000000000012221301022322500217470ustar00rootroot00000000000000repository: path: github.com/prometheus/blackbox_exporter build: flags: -a -tags netgo ldflags: | -X {{repoPath}}/vendor/github.com/prometheus/common/version.Version={{.Version}} -X {{repoPath}}/vendor/github.com/prometheus/common/version.Revision={{.Revision}} -X {{repoPath}}/vendor/github.com/prometheus/common/version.Branch={{.Branch}} -X {{repoPath}}/vendor/github.com/prometheus/common/version.BuildUser={{user}}@{{host}} -X {{repoPath}}/vendor/github.com/prometheus/common/version.BuildDate={{date "20060102-15:04:05"}} tarball: files: - blackbox.yml - LICENSE - NOTICE prometheus-blackbox-exporter-0.3.0+ds1/.travis.yml000066400000000000000000000000771301022322500221240ustar00rootroot00000000000000sudo: false language: go go: - 1.5.4 - 1.6.2 script: - make prometheus-blackbox-exporter-0.3.0+ds1/AUTHORS.md000066400000000000000000000005611301022322500214600ustar00rootroot00000000000000The Prometheus project was started by Matt T. Proud (emeritus) and Julius Volz in 2012. Maintainers of this repository: * Brian Brazil The following individuals have contributed code to this repository (listed in alphabetical order): * Brian Brazil * Michael Stapelberg prometheus-blackbox-exporter-0.3.0+ds1/Dockerfile000066400000000000000000000005401301022322500220000ustar00rootroot00000000000000FROM quay.io/prometheus/busybox:latest MAINTAINER The Prometheus Authors COPY blackbox_exporter /bin/blackbox_exporter COPY blackbox.yml /etc/blackbox_exporter/config.yml EXPOSE 9115 ENTRYPOINT [ "/bin/blackbox_exporter" ] CMD [ "-config.file=/etc/blackbox_exporter/config.yml" ] prometheus-blackbox-exporter-0.3.0+ds1/LICENSE000066400000000000000000000261351301022322500210230ustar00rootroot00000000000000 Apache License Version 2.0, January 2004 http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 1. Definitions. "License" shall mean the terms and conditions for use, reproduction, and distribution as defined by Sections 1 through 9 of this document. "Licensor" shall mean the copyright owner or entity authorized by the copyright owner that is granting the License. "Legal Entity" shall mean the union of the acting entity and all other entities that control, are controlled by, or are under common control with that entity. For the purposes of this definition, "control" means (i) the power, direct or indirect, to cause the direction or management of such entity, whether by contract or otherwise, or (ii) ownership of fifty percent (50%) or more of the outstanding shares, or (iii) beneficial ownership of such entity. "You" (or "Your") shall mean an individual or Legal Entity exercising permissions granted by this License. "Source" form shall mean the preferred form for making modifications, including but not limited to software source code, documentation source, and configuration files. "Object" form shall mean any form resulting from mechanical transformation or translation of a Source form, including but not limited to compiled object code, generated documentation, and conversions to other media types. "Work" shall mean the work of authorship, whether in Source or Object form, made available under the License, as indicated by a copyright notice that is included in or attached to the work (an example is provided in the Appendix below). "Derivative Works" shall mean any work, whether in Source or Object form, that is based on (or derived from) the Work and for which the editorial revisions, annotations, elaborations, or other modifications represent, as a whole, an original work of authorship. For the purposes of this License, Derivative Works shall not include works that remain separable from, or merely link (or bind by name) to the interfaces of, the Work and Derivative Works thereof. "Contribution" shall mean any work of authorship, including the original version of the Work and any modifications or additions to that Work or Derivative Works thereof, that is intentionally submitted to Licensor for inclusion in the Work by the copyright owner or by an individual or Legal Entity authorized to submit on behalf of the copyright owner. For the purposes of this definition, "submitted" means any form of electronic, verbal, or written communication sent to the Licensor or its representatives, including but not limited to communication on electronic mailing lists, source code control systems, and issue tracking systems that are managed by, or on behalf of, the Licensor for the purpose of discussing and improving the Work, but excluding communication that is conspicuously marked or otherwise designated in writing by the copyright owner as "Not a Contribution." "Contributor" shall mean Licensor and any individual or Legal Entity on behalf of whom a Contribution has been received by Licensor and subsequently incorporated within the Work. 2. Grant of Copyright License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable copyright license to reproduce, prepare Derivative Works of, publicly display, publicly perform, sublicense, and distribute the Work and such Derivative Works in Source or Object form. 3. Grant of Patent License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable (except as stated in this section) patent license to make, have made, use, offer to sell, sell, import, and otherwise transfer the Work, where such license applies only to those patent claims licensable by such Contributor that are necessarily infringed by their Contribution(s) alone or by combination of their Contribution(s) with the Work to which such Contribution(s) was submitted. If You institute patent litigation against any entity (including a cross-claim or counterclaim in a lawsuit) alleging that the Work or a Contribution incorporated within the Work constitutes direct or contributory patent infringement, then any patent licenses granted to You under this License for that Work shall terminate as of the date such litigation is filed. 4. Redistribution. You may reproduce and distribute copies of the Work or Derivative Works thereof in any medium, with or without modifications, and in Source or Object form, provided that You meet the following conditions: (a) You must give any other recipients of the Work or Derivative Works a copy of this License; and (b) You must cause any modified files to carry prominent notices stating that You changed the files; and (c) You must retain, in the Source form of any Derivative Works that You distribute, all copyright, patent, trademark, and attribution notices from the Source form of the Work, excluding those notices that do not pertain to any part of the Derivative Works; and (d) If the Work includes a "NOTICE" text file as part of its distribution, then any Derivative Works that You distribute must include a readable copy of the attribution notices contained within such NOTICE file, excluding those notices that do not pertain to any part of the Derivative Works, in at least one of the following places: within a NOTICE text file distributed as part of the Derivative Works; within the Source form or documentation, if provided along with the Derivative Works; or, within a display generated by the Derivative Works, if and wherever such third-party notices normally appear. The contents of the NOTICE file are for informational purposes only and do not modify the License. You may add Your own attribution notices within Derivative Works that You distribute, alongside or as an addendum to the NOTICE text from the Work, provided that such additional attribution notices cannot be construed as modifying the License. You may add Your own copyright statement to Your modifications and may provide additional or different license terms and conditions for use, reproduction, or distribution of Your modifications, or for any such Derivative Works as a whole, provided Your use, reproduction, and distribution of the Work otherwise complies with the conditions stated in this License. 5. Submission of Contributions. Unless You explicitly state otherwise, any Contribution intentionally submitted for inclusion in the Work by You to the Licensor shall be under the terms and conditions of this License, without any additional terms or conditions. Notwithstanding the above, nothing herein shall supersede or modify the terms of any separate license agreement you may have executed with Licensor regarding such Contributions. 6. Trademarks. This License does not grant permission to use the trade names, trademarks, service marks, or product names of the Licensor, except as required for reasonable and customary use in describing the origin of the Work and reproducing the content of the NOTICE file. 7. Disclaimer of Warranty. Unless required by applicable law or agreed to in writing, Licensor provides the Work (and each Contributor provides its Contributions) on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied, including, without limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A PARTICULAR PURPOSE. You are solely responsible for determining the appropriateness of using or redistributing the Work and assume any risks associated with Your exercise of permissions under this License. 8. Limitation of Liability. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise, unless required by applicable law (such as deliberate and grossly negligent acts) or agreed to in writing, shall any Contributor be liable to You for damages, including any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this License or out of the use or inability to use the Work (including but not limited to damages for loss of goodwill, work stoppage, computer failure or malfunction, or any and all other commercial damages or losses), even if such Contributor has been advised of the possibility of such damages. 9. Accepting Warranty or Additional Liability. While redistributing the Work or Derivative Works thereof, You may choose to offer, and charge a fee for, acceptance of support, warranty, indemnity, or other liability obligations and/or rights consistent with this License. However, in accepting such obligations, You may act only on Your own behalf and on Your sole responsibility, not on behalf of any other Contributor, and only if You agree to indemnify, defend, and hold each Contributor harmless for any liability incurred by, or claims asserted against, such Contributor by reason of your accepting any such warranty or additional liability. END OF TERMS AND CONDITIONS APPENDIX: How to apply the Apache License to your work. To apply the Apache License to your work, attach the following boilerplate notice, with the fields enclosed by brackets "[]" replaced with your own identifying information. (Don't include the brackets!) The text should be enclosed in the appropriate comment syntax for the file format. We also recommend that a file or class name and description of purpose be included on the same "printed page" as the copyright notice for easier identification within third-party archives. Copyright [yyyy] [name of copyright owner] 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. prometheus-blackbox-exporter-0.3.0+ds1/Makefile000066400000000000000000000033311301022322500214470ustar00rootroot00000000000000# Copyright 2016 The Prometheus 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. GO := GO15VENDOREXPERIMENT=1 go PROMU := $(GOPATH)/bin/promu pkgs = $(shell $(GO) list ./... | grep -v /vendor/) PREFIX ?= $(shell pwd) BIN_DIR ?= $(shell pwd) DOCKER_IMAGE_NAME ?= blackbox-exporter DOCKER_IMAGE_TAG ?= $(subst /,-,$(shell git rev-parse --abbrev-ref HEAD)) all: format build test style: @echo ">> checking code style" @! gofmt -d $(shell find . -path ./vendor -prune -o -name '*.go' -print) | grep '^' test: @echo ">> running tests" @$(GO) test -short $(pkgs) format: @echo ">> formatting code" @$(GO) fmt $(pkgs) vet: @echo ">> vetting code" @$(GO) vet $(pkgs) build: promu @echo ">> building binaries" @$(PROMU) build --prefix $(PREFIX) tarball: promu @echo ">> building release tarball" @$(PROMU) tarball --prefix $(PREFIX) $(BIN_DIR) docker: @echo ">> building docker image" @docker build -t "$(DOCKER_IMAGE_NAME):$(DOCKER_IMAGE_TAG)" . promu: @GOOS=$(shell uname -s | tr A-Z a-z) \ GOARCH=$(subst x86_64,amd64,$(patsubst i%86,386,$(shell uname -m))) \ $(GO) get -u github.com/prometheus/promu .PHONY: all style format build test vet tarball docker promu prometheus-blackbox-exporter-0.3.0+ds1/NOTICE000066400000000000000000000001361301022322500207130ustar00rootroot00000000000000The Blackbox exporter for blackbox probing metrics Copyright 2012-2016 The Prometheus Authors prometheus-blackbox-exporter-0.3.0+ds1/README.md000066400000000000000000000104661301022322500212750ustar00rootroot00000000000000# Blackbox exporter [![Build Status](https://travis-ci.org/prometheus/blackbox_exporter.svg)][travis] [![CircleCI](https://circleci.com/gh/prometheus/blackbox_exporter/tree/master.svg?style=shield)][circleci] [![Docker Repository on Quay](https://quay.io/repository/prometheus/blackbox-exporter/status)][quay] [![Docker Pulls](https://img.shields.io/docker/pulls/prom/blackbox-exporter.svg?maxAge=604800)][hub] The blackbox exporter allows blackbox probing of endpoints over HTTP, HTTPS, DNS, TCP and ICMP. ## Building and running ### Local Build make ./blackbox_exporter Visiting [http://localhost:9115/probe?target=google.com&module=http_2xx](http://localhost:9115/probe?target=google.com&module=http_2xx) will return metrics for a HTTP probe against google.com. ### Building with Docker docker build -t blackbox_exporter . docker run -d -p 9115:9115 --name blackbox_exporter -v `pwd`:/config blackbox_exporter -config.file=/config/blackbox.yml ## Configuration A configuration showing all options is below: ``` modules: http_2xx_example: prober: http timeout: 5s http: valid_status_codes: [] # Defaults to 2xx method: GET headers: Host: vhost.example.com Accept-Language: en-US no_follow_redirects: false fail_if_ssl: false fail_if_not_ssl: false fail_if_matches_regexp: - "Could not connect to database" fail_if_not_matches_regexp: - "Download the latest version here" tls_config: insecure_skip_verify: false protocol: "tcp" # accepts "tcp/tcp4/tcp6", defaults to "tcp" preferred_ip_protocol: "ip4" # used for "tcp", defaults to "ip6" tcp_connect_v4_example: prober: tcp timeout: 5s tcp: protocol: "tcp4" irc_banner_example: prober: tcp timeout: 5s tcp: query_response: - send: "NICK prober" - send: "USER prober prober prober :prober" - expect: "PING :([^ ]+)" send: "PONG ${1}" - expect: "^:[^ ]+ 001" icmp_example: prober: icmp timeout: 5s icmp: protocol: "icmp" preferred_ip_protocol: "ip4" dns_udp_example: prober: dns timeout: 5s dns: query_name: "www.prometheus.io" query_type: "A" valid_rcodes: - NOERROR validate_answer_rrs: fail_if_matches_regexp: - ".*127.0.0.1" fail_if_not_matches_regexp: - "www.prometheus.io.\t300\tIN\tA\t127.0.0.1" validate_authority_rrs: fail_if_matches_regexp: - ".*127.0.0.1" validate_additional_rrs: fail_if_matches_regexp: - ".*127.0.0.1" dns_tcp_example: prober: dns dns: protocol: "tcp" # accepts "tcp/tcp4/tcp6/udp/udp4/udp6", defaults to "udp" preferred_ip_protocol: "ip4" # used for "udp/tcp", defaults to "ip6" query_name: "www.prometheus.io" ``` HTTP, HTTPS (via the `http` prober), DNS, TCP socket and ICMP (v4 only, see permissions section) are currently supported. Additional modules can be defined to meet your needs. ## Prometheus Configuration The blackbox exporter needs to be passed the target as a parameter, this can be done with relabelling. Example config: ``` scrape_configs: - job_name: 'blackbox' metrics_path: /probe params: module: [http_2xx] # Look for a HTTP 200 response. static_configs: - targets: - prometheus.io # Target to probe relabel_configs: - source_labels: [__address__] regex: (.*)(:80)? target_label: __param_target replacement: ${1} - source_labels: [__param_target] regex: (.*) target_label: instance replacement: ${1} - source_labels: [] regex: .* target_label: __address__ replacement: 127.0.0.1:9115 # Blackbox exporter. ``` ## Permissions The ICMP probe requires elevated privileges to function: * *Windows*: Administrator privileges are required. * *Linux*: root user _or_ `CAP_NET_RAW` capability is required. * Can be set by executing `setcap cap_net_raw+ep blackbox_exporter` * *BSD / OS X*: root user is required. [circleci]: https://circleci.com/gh/prometheus/blackbox_exporter [hub]: https://hub.docker.com/r/prom/blackbox-exporter/ [travis]: https://travis-ci.org/prometheus/blackbox_exporter [quay]: https://quay.io/repository/prometheus/blackbox-exporter prometheus-blackbox-exporter-0.3.0+ds1/VERSION000066400000000000000000000000061301022322500210530ustar00rootroot000000000000000.3.0 prometheus-blackbox-exporter-0.3.0+ds1/blackbox.yml000066400000000000000000000013371301022322500223230ustar00rootroot00000000000000modules: http_2xx: prober: http timeout: 5s http: http_post_2xx: prober: http timeout: 5s http: method: POST tcp_connect: prober: tcp timeout: 5s pop3s_banner: prober: tcp tcp: query_response: - expect: "^+OK" tls: true tls_config: insecure_skip_verify: false ssh_banner: prober: tcp timeout: 5s tcp: query_response: - expect: "^SSH-2.0-" irc_banner: prober: tcp timeout: 5s tcp: query_response: - send: "NICK prober" - send: "USER prober prober prober :prober" - expect: "PING :([^ ]+)" send: "PONG ${1}" - expect: "^:[^ ]+ 001" icmp: prober: icmp timeout: 5s prometheus-blackbox-exporter-0.3.0+ds1/circle.yml000066400000000000000000000045001301022322500217720ustar00rootroot00000000000000machine: environment: DOCKER_IMAGE_NAME: prom/blackbox-exporter QUAY_IMAGE_NAME: quay.io/prometheus/blackbox-exporter DOCKER_TEST_IMAGE_NAME: quay.io/prometheus/golang-builder:1.6-base REPO_PATH: github.com/prometheus/blackbox_exporter pre: - sudo curl -L -o /usr/bin/docker 'https://s3-external-1.amazonaws.com/circle-downloads/docker-1.9.1-circleci' - sudo chmod 0755 /usr/bin/docker - sudo curl -L 'https://github.com/aktau/github-release/releases/download/v0.6.2/linux-amd64-github-release.tar.bz2' | tar xvjf - --strip-components 3 -C $HOME/bin services: - docker dependencies: pre: - make promu - docker info override: - promu crossbuild - ln -s .build/linux-amd64/blackbox_exporter blackbox_exporter - | if [ -n "$CIRCLE_TAG" ]; then make docker DOCKER_IMAGE_NAME=$DOCKER_IMAGE_NAME DOCKER_IMAGE_TAG=$CIRCLE_TAG make docker DOCKER_IMAGE_NAME=$QUAY_IMAGE_NAME DOCKER_IMAGE_TAG=$CIRCLE_TAG else make docker DOCKER_IMAGE_NAME=$DOCKER_IMAGE_NAME make docker DOCKER_IMAGE_NAME=$QUAY_IMAGE_NAME fi post: - mkdir $CIRCLE_ARTIFACTS/binaries/ && cp -a .build/* $CIRCLE_ARTIFACTS/binaries/ - docker images test: override: - docker run --rm -t -v "$(pwd):/app" "${DOCKER_TEST_IMAGE_NAME}" -i "${REPO_PATH}" -T deployment: hub_branch: branch: master owner: prometheus commands: - docker login -e $DOCKER_EMAIL -u $DOCKER_LOGIN -p $DOCKER_PASSWORD - docker login -e $QUAY_EMAIL -u $QUAY_LOGIN -p $QUAY_PASSWORD quay.io - docker push $DOCKER_IMAGE_NAME - docker push $QUAY_IMAGE_NAME hub_tag: tag: /^v[0-9]+(\.[0-9]+){2}(-.+|[^-.]*)$/ owner: prometheus commands: - promu crossbuild tarballs - promu release .tarballs - mkdir $CIRCLE_ARTIFACTS/releases/ && cp -a .tarballs/* $CIRCLE_ARTIFACTS/releases/ - docker login -e $DOCKER_EMAIL -u $DOCKER_LOGIN -p $DOCKER_PASSWORD - docker login -e $QUAY_EMAIL -u $QUAY_LOGIN -p $QUAY_PASSWORD quay.io - | if [[ "$CIRCLE_TAG" =~ ^v[0-9]+(\.[0-9]+){2}$ ]]; then docker tag "$DOCKER_IMAGE_NAME:$CIRCLE_TAG" "$DOCKER_IMAGE_NAME:latest" docker tag "$QUAY_IMAGE_NAME:$CIRCLE_TAG" "$QUAY_IMAGE_NAME:latest" fi - docker push $DOCKER_IMAGE_NAME - docker push $QUAY_IMAGE_NAME prometheus-blackbox-exporter-0.3.0+ds1/dns.go000066400000000000000000000116511301022322500211260ustar00rootroot00000000000000// Copyright 2016 The Prometheus 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 main import ( "fmt" "net" "net/http" "regexp" "github.com/miekg/dns" "github.com/prometheus/common/log" ) // validRRs checks a slice of RRs received from the server against a DNSRRValidator. func validRRs(rrs *[]dns.RR, v *DNSRRValidator) bool { // Fail the probe if there are no RRs of a given type, but a regexp match is required // (i.e. FailIfNotMatchesRegexp is set). if len(*rrs) == 0 && len(v.FailIfNotMatchesRegexp) > 0 { return false } for _, rr := range *rrs { log.Debugf("Validating RR: %q", rr) for _, re := range v.FailIfMatchesRegexp { match, err := regexp.MatchString(re, rr.String()) if err != nil { log.Errorf("Error matching regexp %q: %s", re, err) return false } if match { return false } } for _, re := range v.FailIfNotMatchesRegexp { match, err := regexp.MatchString(re, rr.String()) if err != nil { log.Errorf("Error matching regexp %q: %s", re, err) return false } if !match { return false } } } return true } // validRcode checks rcode in the response against a list of valid rcodes. func validRcode(rcode int, valid []string) bool { var validRcodes []int // If no list of valid rcodes is specified, only NOERROR is considered valid. if valid == nil { validRcodes = append(validRcodes, dns.StringToRcode["NOERROR"]) } else { for _, rcode := range valid { rc, ok := dns.StringToRcode[rcode] if !ok { log.Errorf("Invalid rcode %v. Existing rcodes: %v", rcode, dns.RcodeToString) return false } validRcodes = append(validRcodes, rc) } } for _, rc := range validRcodes { if rcode == rc { return true } } log.Debugf("%s (%d) is not one of the valid rcodes (%v)", dns.RcodeToString[rcode], rcode, validRcodes) return false } func probeDNS(target string, w http.ResponseWriter, module Module) bool { var numAnswer, numAuthority, numAdditional int var dialProtocol, fallbackProtocol string defer func() { // These metrics can be used to build additional alerting based on the number of replies. // They should be returned even in case of errors. fmt.Fprintf(w, "probe_dns_answer_rrs %d\n", numAnswer) fmt.Fprintf(w, "probe_dns_authority_rrs %d\n", numAuthority) fmt.Fprintf(w, "probe_dns_additional_rrs %d\n", numAdditional) }() if module.DNS.Protocol == "" { module.DNS.Protocol = "udp" } if (module.DNS.Protocol == "tcp" || module.DNS.Protocol == "udp") && module.DNS.PreferredIpProtocol == "" { module.DNS.PreferredIpProtocol = "ip6" } if module.DNS.PreferredIpProtocol == "ip6" { fallbackProtocol = "ip4" } else { fallbackProtocol = "ip6" } dialProtocol = module.DNS.Protocol if module.DNS.Protocol == "udp" || module.DNS.Protocol == "tcp" { target_address, _, _ := net.SplitHostPort(target) ip, err := net.ResolveIPAddr(module.DNS.PreferredIpProtocol, target_address) if err != nil { ip, err = net.ResolveIPAddr(fallbackProtocol, target_address) if err != nil { return false } } if ip.IP.To4() == nil { dialProtocol = module.DNS.Protocol + "6" } else { dialProtocol = module.DNS.Protocol + "4" } } if dialProtocol[len(dialProtocol)-1] == '6' { fmt.Fprintf(w, "probe_ip_protocol 6\n") } else { fmt.Fprintf(w, "probe_ip_protocol 4\n") } client := new(dns.Client) client.Net = dialProtocol client.Timeout = module.Timeout qt := dns.TypeANY if module.DNS.QueryType != "" { var ok bool qt, ok = dns.StringToType[module.DNS.QueryType] if !ok { log.Errorf("Invalid type %v. Existing types: %v", module.DNS.QueryType, dns.TypeToString) return false } } msg := new(dns.Msg) msg.SetQuestion(dns.Fqdn(module.DNS.QueryName), qt) response, _, err := client.Exchange(msg, target) if err != nil { log.Warnf("Error while sending a DNS query: %s", err) return false } log.Debugf("Got response: %#v", response) numAnswer, numAuthority, numAdditional = len(response.Answer), len(response.Ns), len(response.Extra) if !validRcode(response.Rcode, module.DNS.ValidRcodes) { return false } if !validRRs(&response.Answer, &module.DNS.ValidateAnswer) { log.Debugf("Answer RRs validation failed") return false } if !validRRs(&response.Ns, &module.DNS.ValidateAuthority) { log.Debugf("Authority RRs validation failed") return false } if !validRRs(&response.Extra, &module.DNS.ValidateAdditional) { log.Debugf("Additional RRs validation failed") return false } return true } prometheus-blackbox-exporter-0.3.0+ds1/dns_test.go000066400000000000000000000266471301022322500222000ustar00rootroot00000000000000// Copyright 2016 The Prometheus 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 main import ( "net" "net/http/httptest" "runtime" "strings" "testing" "time" "github.com/miekg/dns" ) var PROTOCOLS = [...]string{"udp", "tcp"} // startDNSServer starts a DNS server with a given handler function on a random port. // Returns the Server object itself as well as the net.Addr corresponding to the server port. func startDNSServer(protocol string, handler func(dns.ResponseWriter, *dns.Msg)) (*dns.Server, net.Addr) { h := dns.NewServeMux() h.HandleFunc(".", handler) server := &dns.Server{Addr: ":0", Net: protocol, Handler: h} go server.ListenAndServe() // Wait until PacketConn becomes available, but give up after 1 second. for i := 0; server.PacketConn == nil && i < 200; i++ { if protocol == "tcp" && server.Listener != nil { break } if protocol == "udp" && server.PacketConn != nil { break } time.Sleep(5 * time.Millisecond) } if protocol == "tcp" { return server, server.Listener.Addr() } return server, server.PacketConn.LocalAddr() } func recursiveDNSHandler(w dns.ResponseWriter, r *dns.Msg) { m := new(dns.Msg) m.SetReply(r) answers := []string{ "example.com. 3600 IN A 127.0.0.1", "example.com. 3600 IN A 127.0.0.2", } for _, rr := range answers { a, err := dns.NewRR(rr) if err != nil { panic(err) } m.Answer = append(m.Answer, a) } if err := w.WriteMsg(m); err != nil { panic(err) } } func TestRecursiveDNSResponse(t *testing.T) { tests := []struct { Probe DNSProbe ShouldSucceed bool }{ { DNSProbe{ QueryName: "example.com", }, true, }, { DNSProbe{ QueryName: "example.com", ValidRcodes: []string{"SERVFAIL", "NXDOMAIN"}, }, false, }, { DNSProbe{ QueryName: "example.com", ValidateAnswer: DNSRRValidator{ FailIfMatchesRegexp: []string{".*7200.*"}, FailIfNotMatchesRegexp: []string{".*3600.*"}, }, }, true, }, { DNSProbe{ QueryName: "example.com", ValidateAuthority: DNSRRValidator{ FailIfMatchesRegexp: []string{".*7200.*"}, }, }, true, }, { DNSProbe{ QueryName: "example.com", ValidateAdditional: DNSRRValidator{ FailIfNotMatchesRegexp: []string{".*3600.*"}, }, }, false, }, } expectedOutput := []string{ "probe_dns_answer_rrs 2\n", "probe_dns_authority_rrs 0\n", "probe_dns_additional_rrs 0\n", } for _, protocol := range PROTOCOLS { server, addr := startDNSServer(protocol, recursiveDNSHandler) defer server.Shutdown() for i, test := range tests { test.Probe.Protocol = protocol recorder := httptest.NewRecorder() result := probeDNS(addr.String(), recorder, Module{Timeout: time.Second, DNS: test.Probe}) if result != test.ShouldSucceed { t.Fatalf("Test %d had unexpected result: %v", i, result) } body := recorder.Body.String() for _, line := range expectedOutput { if !strings.Contains(body, line) { t.Fatalf("Did not find expected output in test %d: %q", i, line) } } } } } func authoritativeDNSHandler(w dns.ResponseWriter, r *dns.Msg) { m := new(dns.Msg) m.SetReply(r) a, err := dns.NewRR("example.com. 3600 IN A 127.0.0.1") if err != nil { panic(err) } m.Answer = append(m.Answer, a) authority := []string{ "example.com. 7200 IN NS ns1.isp.net.", "example.com. 7200 IN NS ns2.isp.net.", } for _, rr := range authority { a, err := dns.NewRR(rr) if err != nil { panic(err) } m.Ns = append(m.Ns, a) } additional := []string{ "ns1.isp.net. 7200 IN A 127.0.0.1", "ns1.isp.net. 7200 IN AAAA ::1", "ns2.isp.net. 7200 IN A 127.0.0.2", } for _, rr := range additional { a, err := dns.NewRR(rr) if err != nil { panic(err) } m.Extra = append(m.Extra, a) } if err := w.WriteMsg(m); err != nil { panic(err) } } func TestAuthoritativeDNSResponse(t *testing.T) { tests := []struct { Probe DNSProbe ShouldSucceed bool }{ { DNSProbe{ QueryName: "example.com", }, true, }, { DNSProbe{ QueryName: "example.com", ValidRcodes: []string{"SERVFAIL", "NXDOMAIN"}, }, false, }, { DNSProbe{ QueryName: "example.com", ValidateAnswer: DNSRRValidator{ FailIfMatchesRegexp: []string{".*3600.*"}, FailIfNotMatchesRegexp: []string{".*3600.*"}, }, }, false, }, { DNSProbe{ QueryName: "example.com", ValidateAnswer: DNSRRValidator{ FailIfMatchesRegexp: []string{".*7200.*"}, FailIfNotMatchesRegexp: []string{".*7200.*"}, }, }, false, }, { DNSProbe{ QueryName: "example.com", ValidateAuthority: DNSRRValidator{ FailIfNotMatchesRegexp: []string{"ns.*.isp.net"}, }, }, true, }, { DNSProbe{ QueryName: "example.com", ValidateAdditional: DNSRRValidator{ FailIfNotMatchesRegexp: []string{"^ns.*.isp"}, }, }, true, }, { DNSProbe{ QueryName: "example.com", ValidateAdditional: DNSRRValidator{ FailIfMatchesRegexp: []string{"^ns.*.isp"}, }, }, false, }, } expectedOutput := []string{ "probe_dns_answer_rrs 1\n", "probe_dns_authority_rrs 2\n", "probe_dns_additional_rrs 3\n", } for _, protocol := range PROTOCOLS { server, addr := startDNSServer(protocol, authoritativeDNSHandler) defer server.Shutdown() for i, test := range tests { test.Probe.Protocol = protocol recorder := httptest.NewRecorder() result := probeDNS(addr.String(), recorder, Module{Timeout: time.Second, DNS: test.Probe}) if result != test.ShouldSucceed { t.Fatalf("Test %d had unexpected result: %v", i, result) } body := recorder.Body.String() for _, line := range expectedOutput { if !strings.Contains(body, line) { t.Fatalf("Did not find expected output in test %d: %q", i, line) } } } } } func TestServfailDNSResponse(t *testing.T) { tests := []struct { Probe DNSProbe ShouldSucceed bool }{ { DNSProbe{ QueryName: "example.com", }, false, }, { DNSProbe{ QueryName: "example.com", ValidRcodes: []string{"SERVFAIL", "NXDOMAIN"}, }, true, }, { DNSProbe{ QueryName: "example.com", QueryType: "NOT_A_VALID_QUERY_TYPE", }, false, }, { DNSProbe{ QueryName: "example.com", ValidRcodes: []string{"NOT_A_VALID_RCODE"}, }, false, }, } expectedOutput := []string{ "probe_dns_answer_rrs 0\n", "probe_dns_authority_rrs 0\n", "probe_dns_additional_rrs 0\n", } for _, protocol := range PROTOCOLS { // dns.HandleFailed returns SERVFAIL on everything server, addr := startDNSServer(protocol, dns.HandleFailed) defer server.Shutdown() for i, test := range tests { test.Probe.Protocol = protocol recorder := httptest.NewRecorder() result := probeDNS(addr.String(), recorder, Module{Timeout: time.Second, DNS: test.Probe}) if result != test.ShouldSucceed { t.Fatalf("Test %d had unexpected result: %v", i, result) } body := recorder.Body.String() for _, line := range expectedOutput { if !strings.Contains(body, line) { t.Fatalf("Did not find expected output in test %d: %q", i, line) } } } } } func TestDNSProtocol(t *testing.T) { // This test assumes that listening "tcp" listens both IPv6 and IPv4 traffic and // localhost resolves to both 127.0.0.1 and ::1. we must skip the test if either // of these isn't true. This should be true for modern Linux systems. if runtime.GOOS == "dragonfly" || runtime.GOOS == "openbsd" { t.Skip("IPv6 socket isn't able to accept IPv4 traffic in the system.") } _, err := net.ResolveIPAddr("ip6", "localhost") if err != nil { t.Skip("\"localhost\" doesn't resolve to ::1.") } for _, protocol := range PROTOCOLS { server, addr := startDNSServer(protocol, recursiveDNSHandler) defer server.Shutdown() _, port, _ := net.SplitHostPort(addr.String()) // Force IPv4 module := Module{ Timeout: time.Second, DNS: DNSProbe{ QueryName: "example.com", Protocol: protocol + "4", }, } recorder := httptest.NewRecorder() result := probeDNS(net.JoinHostPort("localhost", port), recorder, module) body := recorder.Body.String() if !result { t.Fatalf("DNS protocol: \"%v4\" connection test failed, expected success.", protocol) } if !strings.Contains(body, "probe_ip_protocol 4\n") { t.Fatalf("Expected IPv4, got %s", body) } // Force IPv6 module = Module{ Timeout: time.Second, DNS: DNSProbe{ QueryName: "example.com", Protocol: protocol + "6", }, } recorder = httptest.NewRecorder() result = probeDNS(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("DNS protocol: \"%v6\" connection test failed, expected success.", protocol) } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } // Prefer IPv6 module = Module{ Timeout: time.Second, DNS: DNSProbe{ QueryName: "example.com", Protocol: protocol, PreferredIpProtocol: "ip6", }, } recorder = httptest.NewRecorder() result = probeDNS(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("DNS protocol: \"%v\", preferred \"ip6\" connection test failed, expected success.", protocol) } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } // Prefer IPv4 module = Module{ Timeout: time.Second, DNS: DNSProbe{ QueryName: "example.com", Protocol: protocol, PreferredIpProtocol: "ip4", }, } recorder = httptest.NewRecorder() result = probeDNS(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("DNS protocol: \"%v\", preferred \"ip4\" connection test failed, expected success.", protocol) } if !strings.Contains(body, "probe_ip_protocol 4\n") { t.Fatalf("Expected IPv4, got %s", body) } // Prefer none module = Module{ Timeout: time.Second, DNS: DNSProbe{ QueryName: "example.com", Protocol: protocol, }, } recorder = httptest.NewRecorder() result = probeDNS(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("DNS protocol: \"%v\" connection test failed, expected success.", protocol) } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } // No protocol module = Module{ Timeout: time.Second, DNS: DNSProbe{ QueryName: "example.com", }, } recorder = httptest.NewRecorder() result = probeDNS(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if protocol == "udp" { if !result { t.Fatalf("DNS test connection with protocol unspecified failed, expected success.", protocol) } } else { if result { t.Fatalf("DNS test connection with protocol unspecified succeeded, expected failure.", protocol) } } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } } } prometheus-blackbox-exporter-0.3.0+ds1/http.go000066400000000000000000000117041301022322500213200ustar00rootroot00000000000000// Copyright 2016 The Prometheus 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 main import ( "errors" "fmt" "io" "io/ioutil" "net" "net/http" "net/url" "regexp" "strings" "github.com/prometheus/common/log" ) func matchRegularExpressions(reader io.Reader, config HTTPProbe) bool { body, err := ioutil.ReadAll(reader) if err != nil { log.Errorf("Error reading HTTP body: %s", err) return false } for _, expression := range config.FailIfMatchesRegexp { re, err := regexp.Compile(expression) if err != nil { log.Errorf("Could not compile expression %q as regular expression: %s", expression, err) return false } if re.Match(body) { return false } } for _, expression := range config.FailIfNotMatchesRegexp { re, err := regexp.Compile(expression) if err != nil { log.Errorf("Could not compile expression %q as regular expression: %s", expression, err) return false } if !re.Match(body) { return false } } return true } func probeHTTP(target string, w http.ResponseWriter, module Module) (success bool) { var isSSL, redirects int var dialProtocol, fallbackProtocol string config := module.HTTP if module.HTTP.Protocol == "" { module.HTTP.Protocol = "tcp" } if module.HTTP.Protocol == "tcp" && module.HTTP.PreferredIpProtocol == "" { module.HTTP.PreferredIpProtocol = "ip6" } if module.HTTP.PreferredIpProtocol == "ip6" { fallbackProtocol = "ip4" } else { fallbackProtocol = "ip6" } if !strings.HasPrefix(target, "http://") && !strings.HasPrefix(target, "https://") { target = "http://" + target } dialProtocol = module.HTTP.Protocol if module.HTTP.Protocol == "tcp" { target_url, err := url.Parse(target) if err != nil { return false } target_host, _, err := net.SplitHostPort(target_url.Host) // If split fails, assuming it's a hostname without port part if err != nil { target_host = target_url.Host } ip, err := net.ResolveIPAddr(module.HTTP.PreferredIpProtocol, target_host) if err != nil { ip, err = net.ResolveIPAddr(fallbackProtocol, target_host) if err != nil { return false } } if ip.IP.To4() == nil { dialProtocol = "tcp6" } else { dialProtocol = "tcp4" } } if dialProtocol == "tcp6" { fmt.Fprintf(w, "probe_ip_protocol 6\n") } else { fmt.Fprintf(w, "probe_ip_protocol 4\n") } client := &http.Client{ Timeout: module.Timeout, } tlsconfig, err := module.HTTP.TLSConfig.GenerateConfig() if err != nil { log.Errorf("Error generating TLS config: %s", err) return false } dial := func(network, address string) (net.Conn, error) { return net.Dial(dialProtocol, address) } client.Transport = &http.Transport{ TLSClientConfig: tlsconfig, Dial: dial, } client.CheckRedirect = func(_ *http.Request, via []*http.Request) error { redirects = len(via) if redirects > 10 || config.NoFollowRedirects { return errors.New("Don't follow redirects") } return nil } if config.Method == "" { config.Method = "GET" } request, err := http.NewRequest(config.Method, target, nil) if err != nil { log.Errorf("Error creating request for target %s: %s", target, err) return } for key, value := range config.Headers { if strings.Title(key) == "Host" { request.Host = value continue } request.Header.Set(key, value) } resp, err := client.Do(request) // Err won't be nil if redirects were turned off. See https://github.com/golang/go/issues/3795 if err != nil && resp == nil { log.Warnf("Error for HTTP request to %s: %s", target, err) } else { defer resp.Body.Close() if len(config.ValidStatusCodes) != 0 { for _, code := range config.ValidStatusCodes { if resp.StatusCode == code { success = true break } } } else if 200 <= resp.StatusCode && resp.StatusCode < 300 { success = true } if success && (len(config.FailIfMatchesRegexp) > 0 || len(config.FailIfNotMatchesRegexp) > 0) { success = matchRegularExpressions(resp.Body, config) } } if resp == nil { resp = &http.Response{} } if resp.TLS != nil { isSSL = 1 fmt.Fprintf(w, "probe_ssl_earliest_cert_expiry %f\n", float64(getEarliestCertExpiry(resp.TLS).UnixNano())/1e9) if config.FailIfSSL { success = false } } else if config.FailIfNotSSL { success = false } fmt.Fprintf(w, "probe_http_status_code %d\n", resp.StatusCode) fmt.Fprintf(w, "probe_http_content_length %d\n", resp.ContentLength) fmt.Fprintf(w, "probe_http_redirects %d\n", redirects) fmt.Fprintf(w, "probe_http_ssl %d\n", isSSL) return } prometheus-blackbox-exporter-0.3.0+ds1/http_test.go000066400000000000000000000247131301022322500223630ustar00rootroot00000000000000// Copyright 2015 The Prometheus 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 main import ( "fmt" "github.com/prometheus/common/config" "net/http" "net/http/httptest" "strings" "testing" "time" ) func TestHTTPStatusCodes(t *testing.T) { tests := []struct { StatusCode int ValidStatusCodes []int ShouldSucceed bool }{ {200, []int{}, true}, {201, []int{}, true}, {299, []int{}, true}, {300, []int{}, false}, {404, []int{}, false}, {404, []int{200, 404}, true}, {200, []int{200, 404}, true}, {201, []int{200, 404}, false}, {404, []int{404}, true}, {200, []int{404}, false}, } for i, test := range tests { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(test.StatusCode) })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{ValidStatusCodes: test.ValidStatusCodes}}) body := recorder.Body.String() if result != test.ShouldSucceed { t.Fatalf("Test %d had unexpected result: %s", i, body) } } } func TestRedirectFollowed(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path == "/" { http.Redirect(w, r, "/noredirect", http.StatusFound) } })) defer ts.Close() // Follow redirect, should succeed with 200. recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{}}) body := recorder.Body.String() if !result { t.Fatalf("Redirect test failed unexpectedly, got %s", body) } if !strings.Contains(body, "probe_http_redirects 1\n") { t.Fatalf("Expected one redirect, got %s", body) } } func TestRedirectNotFollowed(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { http.Redirect(w, r, "/noredirect", http.StatusFound) })) defer ts.Close() // Follow redirect, should succeed with 200. recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{NoFollowRedirects: true, ValidStatusCodes: []int{302}}}) body := recorder.Body.String() if !result { t.Fatalf("Redirect test failed unexpectedly, got %s", body) } } func TestPost(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.Method != "POST" { w.WriteHeader(http.StatusBadRequest) } })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{Method: "POST"}}) body := recorder.Body.String() if !result { t.Fatalf("Post test failed unexpectedly, got %s", body) } } func TestFailIfNotSSL(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfNotSSL: true}}) body := recorder.Body.String() if result { t.Fatalf("Fail if not SSL test suceeded unexpectedly, got %s", body) } if !strings.Contains(body, "probe_http_ssl 0\n") { t.Fatalf("Expected HTTP without SSL, got %s", body) } } func TestFailIfMatchesRegexp(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Bad news: could not connect to database server") })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfMatchesRegexp: []string{"could not connect to database"}}}) body := recorder.Body.String() if result { t.Fatalf("Regexp test succeeded unexpectedly, got %s", body) } ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Download the latest version here") })) defer ts.Close() recorder = httptest.NewRecorder() result = probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfMatchesRegexp: []string{"could not connect to database"}}}) body = recorder.Body.String() if !result { t.Fatalf("Regexp test failed unexpectedly, got %s", body) } // With multiple regexps configured, verify that any matching regexp causes // the probe to fail, but probes succeed when no regexp matches. ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "internal error") })) defer ts.Close() recorder = httptest.NewRecorder() result = probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfMatchesRegexp: []string{"could not connect to database", "internal error"}}}) body = recorder.Body.String() if result { t.Fatalf("Regexp test succeeded unexpectedly, got %s", body) } ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "hello world") })) defer ts.Close() recorder = httptest.NewRecorder() result = probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfMatchesRegexp: []string{"could not connect to database", "internal error"}}}) body = recorder.Body.String() if !result { t.Fatalf("Regexp test failed unexpectedly, got %s", body) } } func TestFailIfNotMatchesRegexp(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Bad news: could not connect to database server") })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfNotMatchesRegexp: []string{"Download the latest version here"}}}) body := recorder.Body.String() if result { t.Fatalf("Regexp test succeeded unexpectedly, got %s", body) } ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Download the latest version here") })) defer ts.Close() recorder = httptest.NewRecorder() result = probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfNotMatchesRegexp: []string{"Download the latest version here"}}}) body = recorder.Body.String() if !result { t.Fatalf("Regexp test failed unexpectedly, got %s", body) } // With multiple regexps configured, verify that any non-matching regexp // causes the probe to fail, but probes succeed when all regexps match. ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Download the latest version here") })) defer ts.Close() recorder = httptest.NewRecorder() result = probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfNotMatchesRegexp: []string{"Download the latest version here", "Copyright 2015"}}}) body = recorder.Body.String() if result { t.Fatalf("Regexp test succeeded unexpectedly, got %s", body) } ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Download the latest version here. Copyright 2015 Test Inc.") })) defer ts.Close() recorder = httptest.NewRecorder() result = probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{FailIfNotMatchesRegexp: []string{"Download the latest version here", "Copyright 2015"}}}) body = recorder.Body.String() if !result { t.Fatalf("Regexp test failed unexpectedly, got %s", body) } } func TestHTTPHeaders(t *testing.T) { headers := map[string]string{ "Host": "my-secret-vhost.com", "User-Agent": "unsuspicious user", "Accept-Language": "en-US", } ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { for key, value := range headers { if strings.Title(key) == "Host" { if r.Host != value { t.Errorf("Unexpected host: expected %q, got %q.", value, r.Host) } continue } if got := r.Header.Get(key); got != value { t.Errorf("Unexpected value of header %q: expected %q, got %q", key, value, got) } } w.WriteHeader(http.StatusOK) })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{ Headers: headers, }}) if !result { t.Fatalf("Probe failed unexpectedly.") } } func TestFailIfSelfSignedCA(t *testing.T) { ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{ TLSConfig: config.TLSConfig{InsecureSkipVerify: false}, }}) body := recorder.Body.String() if result { t.Fatalf("Fail if selfsigned CA test suceeded unexpectedly, got %s", body) } if !strings.Contains(body, "probe_http_ssl 0\n") { t.Fatalf("Expected HTTP without SSL because of CA failure, got %s", body) } } func TestSucceedIfSelfSignedCA(t *testing.T) { ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{ TLSConfig: config.TLSConfig{InsecureSkipVerify: true}, }}) body := recorder.Body.String() if !result { t.Fatalf("Fail if (not strict) selfsigned CA test fails unexpectedly, got %s", body) } if !strings.Contains(body, "probe_http_ssl 1\n") { t.Fatalf("Expected HTTP with SSL, got %s", body) } } func TestTLSConfigIsIgnoredForPlainHTTP(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { })) defer ts.Close() recorder := httptest.NewRecorder() result := probeHTTP(ts.URL, recorder, Module{Timeout: time.Second, HTTP: HTTPProbe{ TLSConfig: config.TLSConfig{InsecureSkipVerify: false}, }}) body := recorder.Body.String() if !result { t.Fatalf("Fail if InsecureSkipVerify affects simple http fails unexpectedly, got %s", body) } if !strings.Contains(body, "probe_http_ssl 0\n") { t.Fatalf("Expected HTTP without SSL, got %s", body) } } prometheus-blackbox-exporter-0.3.0+ds1/icmp.go000066400000000000000000000077721301022322500213030ustar00rootroot00000000000000// Copyright 2016 The Prometheus 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 main import ( "bytes" "fmt" "golang.org/x/net/icmp" "golang.org/x/net/ipv4" "golang.org/x/net/ipv6" "net" "net/http" "os" "sync" "time" "github.com/prometheus/common/log" ) var ( icmpSequence uint16 icmpSequenceMutex sync.Mutex ) func getICMPSequence() uint16 { icmpSequenceMutex.Lock() defer icmpSequenceMutex.Unlock() icmpSequence += 1 return icmpSequence } func probeICMP(target string, w http.ResponseWriter, module Module) (success bool) { var ( socket *icmp.PacketConn requestType icmp.Type replyType icmp.Type fallbackProtocol string ) deadline := time.Now().Add(module.Timeout) // Defaults to IPv4 to be compatible with older versions if module.ICMP.Protocol == "" { module.ICMP.Protocol = "icmp" } // In case of ICMP prefer IPv6 by default if module.ICMP.Protocol == "icmp" && module.ICMP.PreferredIpProtocol == "" { module.ICMP.PreferredIpProtocol = "ip6" } if module.ICMP.Protocol == "icmp4" { module.ICMP.PreferredIpProtocol = "ip4" fallbackProtocol = "" } else if module.ICMP.Protocol == "icmp6" { module.ICMP.PreferredIpProtocol = "ip6" fallbackProtocol = "" } else if module.ICMP.PreferredIpProtocol == "ip6" { fallbackProtocol = "ip4" } else { fallbackProtocol = "ip6" } ip, err := net.ResolveIPAddr(module.ICMP.PreferredIpProtocol, target) if err != nil && fallbackProtocol != "" { ip, err = net.ResolveIPAddr(fallbackProtocol, target) } if err != nil { log.Errorf("Error resolving address %s: %s", target, err) } if ip.IP.To4() == nil { requestType = ipv6.ICMPTypeEchoRequest replyType = ipv6.ICMPTypeEchoReply socket, err = icmp.ListenPacket("ip6:ipv6-icmp", "::") fmt.Fprintf(w, "probe_ip_protocol 6\n") } else { requestType = ipv4.ICMPTypeEcho replyType = ipv4.ICMPTypeEchoReply socket, err = icmp.ListenPacket("ip4:icmp", "0.0.0.0") fmt.Fprintf(w, "probe_ip_protocol 4\n") } if err != nil { log.Errorf("Error listening to socket: %s", err) return } defer socket.Close() if err != nil { log.Errorf("Error resolving address %s: %s", target, err) return } seq := getICMPSequence() pid := os.Getpid() & 0xffff wm := icmp.Message{ Type: requestType, Code: 0, Body: &icmp.Echo{ ID: pid, Seq: int(seq), Data: []byte("Prometheus Blackbox Exporter"), }, } wb, err := wm.Marshal(nil) if err != nil { log.Errorf("Error marshalling packet for %s: %s", target, err) return } if _, err := socket.WriteTo(wb, ip); err != nil { log.Errorf("Error writing to socker for %s: %s", target, err) return } // Reply should be the same except for the message type. wm.Type = replyType wb, err = wm.Marshal(nil) if err != nil { log.Errorf("Error marshalling packet for %s: %s", target, err) return } rb := make([]byte, 1500) if err := socket.SetReadDeadline(deadline); err != nil { log.Errorf("Error setting socket deadline for %s: %s", target, err) return } for { n, peer, err := socket.ReadFrom(rb) if err != nil { if nerr, ok := err.(net.Error); ok && nerr.Timeout() { log.Infof("Timeout reading from socket for %s: %s", target, err) return } log.Errorf("Error reading from socket for %s: %s", target, err) continue } if peer.String() != ip.String() { continue } if replyType == ipv6.ICMPTypeEchoReply { // Clear checksum to make comparison succeed. rb[2] = 0 rb[3] = 0 } if bytes.Compare(rb[:n], wb) == 0 { success = true return } } return } prometheus-blackbox-exporter-0.3.0+ds1/main.go000066400000000000000000000136301301022322500212650ustar00rootroot00000000000000// Copyright 2016 The Prometheus 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 main import ( "flag" "fmt" "io/ioutil" "net/http" "os" "time" "gopkg.in/yaml.v2" "github.com/prometheus/client_golang/prometheus" "github.com/prometheus/common/config" "github.com/prometheus/common/log" "github.com/prometheus/common/version" ) var ( configFile = flag.String("config.file", "blackbox.yml", "Blackbox exporter configuration file.") listenAddress = flag.String("web.listen-address", ":9115", "The address to listen on for HTTP requests.") showVersion = flag.Bool("version", false, "Print version information.") ) type Config struct { Modules map[string]Module `yaml:"modules"` } type Module struct { Prober string `yaml:"prober"` Timeout time.Duration `yaml:"timeout"` HTTP HTTPProbe `yaml:"http"` TCP TCPProbe `yaml:"tcp"` ICMP ICMPProbe `yaml:"icmp"` DNS DNSProbe `yaml:"dns"` } type HTTPProbe struct { // Defaults to 2xx. ValidStatusCodes []int `yaml:"valid_status_codes"` NoFollowRedirects bool `yaml:"no_follow_redirects"` FailIfSSL bool `yaml:"fail_if_ssl"` FailIfNotSSL bool `yaml:"fail_if_not_ssl"` Method string `yaml:"method"` Headers map[string]string `yaml:"headers"` FailIfMatchesRegexp []string `yaml:"fail_if_matches_regexp"` FailIfNotMatchesRegexp []string `yaml:"fail_if_not_matches_regexp"` TLSConfig config.TLSConfig `yaml:"tls_config"` Protocol string `yaml:"protocol"` // Defaults to "tcp". PreferredIpProtocol string `yaml:"preferred_ip_protocol"` // Defaults to "ip6". } type QueryResponse struct { Expect string `yaml:"expect"` Send string `yaml:"send"` } type TCPProbe struct { QueryResponse []QueryResponse `yaml:"query_response"` TLS bool `yaml:"tls"` TLSConfig config.TLSConfig `yaml:"tls_config"` Protocol string `yaml:"protocol"` // Defaults to "tcp". PreferredIpProtocol string `yaml:"preferred_ip_protocol"` // Defaults to "ip6". } type ICMPProbe struct { Protocol string `yaml:"protocol"` // Defaults to "icmp4". PreferredIpProtocol string `yaml:"preferred_ip_protocol"` // Defaults to "ip6". } type DNSProbe struct { Protocol string `yaml:"protocol"` // Defaults to "udp". QueryName string `yaml:"query_name"` QueryType string `yaml:"query_type"` // Defaults to ANY. ValidRcodes []string `yaml:"valid_rcodes"` // Defaults to NOERROR. ValidateAnswer DNSRRValidator `yaml:"validate_answer_rrs"` ValidateAuthority DNSRRValidator `yaml:"validate_authority_rrs"` ValidateAdditional DNSRRValidator `yaml:"validate_additional_rrs"` PreferredIpProtocol string `yaml:"preferred_ip_protocol"` // Defaults to "ip6". } type DNSRRValidator struct { FailIfMatchesRegexp []string `yaml:"fail_if_matches_regexp"` FailIfNotMatchesRegexp []string `yaml:"fail_if_not_matches_regexp"` } var Probers = map[string]func(string, http.ResponseWriter, Module) bool{ "http": probeHTTP, "tcp": probeTCP, "icmp": probeICMP, "dns": probeDNS, } func probeHandler(w http.ResponseWriter, r *http.Request, config *Config) { params := r.URL.Query() target := params.Get("target") moduleName := params.Get("module") if target == "" { http.Error(w, "Target parameter is missing", 400) return } if moduleName == "" { moduleName = "http_2xx" } module, ok := config.Modules[moduleName] if !ok { http.Error(w, fmt.Sprintf("Unkown module %s", moduleName), 400) return } prober, ok := Probers[module.Prober] if !ok { http.Error(w, fmt.Sprintf("Unkown prober %s", module.Prober), 400) return } start := time.Now() success := prober(target, w, module) fmt.Fprintf(w, "probe_duration_seconds %f\n", float64(time.Now().Sub(start))/1e9) if success { fmt.Fprintf(w, "probe_success %d\n", 1) } else { fmt.Fprintf(w, "probe_success %d\n", 0) } } func init() { prometheus.MustRegister(version.NewCollector("blackbox_exporter")) } func main() { flag.Parse() if *showVersion { fmt.Fprintln(os.Stdout, version.Print("blackbox_exporter")) os.Exit(0) } log.Infoln("Starting blackbox_exporter", version.Info()) log.Infoln("Build context", version.BuildContext()) yamlFile, err := ioutil.ReadFile(*configFile) if err != nil { log.Fatalf("Error reading config file: %s", err) } config := Config{} err = yaml.Unmarshal(yamlFile, &config) if err != nil { log.Fatalf("Error parsing config file: %s", err) } http.Handle("/metrics", prometheus.Handler()) http.HandleFunc("/probe", func(w http.ResponseWriter, r *http.Request) { probeHandler(w, r, &config) }) http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte(` Blackbox Exporter

Blackbox Exporter

Probe prometheus.io for http_2xx

Metrics

`)) }) log.Infoln("Listening on", *listenAddress) if err := http.ListenAndServe(*listenAddress, nil); err != nil { log.Fatalf("Error starting HTTP server: %s", err) } } prometheus-blackbox-exporter-0.3.0+ds1/tcp.go000066400000000000000000000070421301022322500211270ustar00rootroot00000000000000// Copyright 2016 The Prometheus 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 main import ( "bufio" "crypto/tls" "fmt" "net" "net/http" "regexp" "time" "github.com/prometheus/common/log" ) func dialTCP(target string, w http.ResponseWriter, module Module) (net.Conn, error) { var dialProtocol, fallbackProtocol string dialer := &net.Dialer{Timeout: module.Timeout} if module.TCP.Protocol == "" { module.TCP.Protocol = "tcp" } if module.TCP.Protocol == "tcp" && module.TCP.PreferredIpProtocol == "" { module.TCP.PreferredIpProtocol = "ip6" } if module.TCP.PreferredIpProtocol == "ip6" { fallbackProtocol = "ip4" } else { fallbackProtocol = "ip6" } dialProtocol = module.TCP.Protocol if module.TCP.Protocol == "tcp" { target_address, _, err := net.SplitHostPort(target) ip, err := net.ResolveIPAddr(module.TCP.PreferredIpProtocol, target_address) if err != nil { ip, err = net.ResolveIPAddr(fallbackProtocol, target_address) if err != nil { return nil, err } } if ip.IP.To4() == nil { dialProtocol = "tcp6" } else { dialProtocol = "tcp4" } } if dialProtocol == "tcp6" { fmt.Fprintf(w, "probe_ip_protocol 6\n") } else { fmt.Fprintf(w, "probe_ip_protocol 4\n") } if !module.TCP.TLS { return dialer.Dial(dialProtocol, target) } config, err := module.TCP.TLSConfig.GenerateConfig() if err != nil { return nil, err } return tls.DialWithDialer(dialer, dialProtocol, target, config) } func probeTCP(target string, w http.ResponseWriter, module Module) bool { deadline := time.Now().Add(module.Timeout) conn, err := dialTCP(target, w, module) if err != nil { return false } defer conn.Close() // Set a deadline to prevent the following code from blocking forever. // If a deadline cannot be set, better fail the probe by returning an error // now rather than blocking forever. if err := conn.SetDeadline(deadline); err != nil { return false } if module.TCP.TLS { state := conn.(*tls.Conn).ConnectionState() fmt.Fprintf(w, "probe_ssl_earliest_cert_expiry %f\n", float64(getEarliestCertExpiry(&state).UnixNano())/1e9) } scanner := bufio.NewScanner(conn) for _, qr := range module.TCP.QueryResponse { log.Debugf("Processing query response entry %+v", qr) send := qr.Send if qr.Expect != "" { re, err := regexp.Compile(qr.Expect) if err != nil { log.Errorf("Could not compile %q into regular expression: %v", qr.Expect, err) return false } var match []int // Read lines until one of them matches the configured regexp. for scanner.Scan() { log.Debugf("read %q\n", scanner.Text()) match = re.FindSubmatchIndex(scanner.Bytes()) if match != nil { log.Debugf("regexp %q matched %q", re, scanner.Text()) break } } if scanner.Err() != nil { return false } if match == nil { return false } send = string(re.Expand(nil, []byte(send), scanner.Bytes(), match)) } if send != "" { log.Debugf("Sending %q", send) if _, err := fmt.Fprintf(conn, "%s\n", send); err != nil { return false } } } return true } prometheus-blackbox-exporter-0.3.0+ds1/tcp_test.go000066400000000000000000000167031301022322500221720ustar00rootroot00000000000000// Copyright 2015 The Prometheus 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 main import ( "fmt" "net" "net/http/httptest" "runtime" "strings" "testing" "time" ) func TestTCPConnection(t *testing.T) { ln, err := net.Listen("tcp", "localhost:0") if err != nil { t.Fatalf("Error listening on socket: %s", err) } defer ln.Close() ch := make(chan (struct{})) go func() { conn, err := ln.Accept() if err != nil { t.Fatalf("Error accepting on socket: %s", err) } conn.Close() ch <- struct{}{} }() recorder := httptest.NewRecorder() if !probeTCP(ln.Addr().String(), recorder, Module{Timeout: time.Second}) { t.Fatalf("TCP module failed, expected success.") } <-ch } func TestTCPConnectionFails(t *testing.T) { // Invalid port number. recorder := httptest.NewRecorder() if probeTCP(":0", recorder, Module{Timeout: time.Second}) { t.Fatalf("TCP module suceeded, expected failure.") } } func TestTCPConnectionQueryResponseIRC(t *testing.T) { ln, err := net.Listen("tcp", "localhost:0") if err != nil { t.Fatalf("Error listening on socket: %s", err) } defer ln.Close() module := Module{ Timeout: time.Second, TCP: TCPProbe{ QueryResponse: []QueryResponse{ QueryResponse{Send: "NICK prober"}, QueryResponse{Send: "USER prober prober prober :prober"}, QueryResponse{Expect: "^:[^ ]+ 001"}, }, }, } ch := make(chan (struct{})) go func() { conn, err := ln.Accept() if err != nil { t.Fatalf("Error accepting on socket: %s", err) } fmt.Fprintf(conn, ":ircd.localhost NOTICE AUTH :*** Looking up your hostname...\n") var nick, user, mode, unused, realname string fmt.Fscanf(conn, "NICK %s", &nick) fmt.Fscanf(conn, "USER %s %s %s :%s", &user, &mode, &unused, &realname) fmt.Fprintf(conn, ":ircd.localhost 001 %s :Welcome to IRC!\n", nick) conn.Close() ch <- struct{}{} }() recorder := httptest.NewRecorder() if !probeTCP(ln.Addr().String(), recorder, module) { t.Fatalf("TCP module failed, expected success.") } <-ch go func() { conn, err := ln.Accept() if err != nil { t.Fatalf("Error accepting on socket: %s", err) } fmt.Fprintf(conn, ":ircd.localhost NOTICE AUTH :*** Looking up your hostname...\n") var nick, user, mode, unused, realname string fmt.Fscanf(conn, "NICK %s", &nick) fmt.Fscanf(conn, "USER %s %s %s :%s", &user, &mode, &unused, &realname) fmt.Fprintf(conn, "ERROR: Your IP address has been blacklisted.\n") conn.Close() ch <- struct{}{} }() if probeTCP(ln.Addr().String(), recorder, module) { t.Fatalf("TCP module succeeded, expected failure.") } <-ch } func TestTCPConnectionQueryResponseMatching(t *testing.T) { ln, err := net.Listen("tcp", "localhost:0") if err != nil { t.Fatalf("Error listening on socket: %s", err) } defer ln.Close() module := Module{ Timeout: time.Second, TCP: TCPProbe{ QueryResponse: []QueryResponse{ QueryResponse{ Expect: "SSH-2.0-(OpenSSH_6.9p1) Debian-2", Send: "CONFIRM ${1}", }, }, }, } ch := make(chan string) go func() { conn, err := ln.Accept() if err != nil { t.Fatalf("Error accepting on socket: %s", err) } conn.SetDeadline(time.Now().Add(1 * time.Second)) fmt.Fprintf(conn, "SSH-2.0-OpenSSH_6.9p1 Debian-2\n") var version string fmt.Fscanf(conn, "CONFIRM %s", &version) conn.Close() ch <- version }() recorder := httptest.NewRecorder() if !probeTCP(ln.Addr().String(), recorder, module) { t.Fatalf("TCP module failed, expected success.") } if got, want := <-ch, "OpenSSH_6.9p1"; got != want { t.Fatalf("Read unexpected version: got %q, want %q", got, want) } } func TestTCPConnectionProtocol(t *testing.T) { // This test assumes that listening "tcp" listens both IPv6 and IPv4 traffic and // localhost resolves to both 127.0.0.1 and ::1. we must skip the test if either // of these isn't true. This should be true for modern Linux systems. if runtime.GOOS == "dragonfly" || runtime.GOOS == "openbsd" { t.Skip("IPv6 socket isn't able to accept IPv4 traffic in the system.") } _, err := net.ResolveIPAddr("ip6", "localhost") if err != nil { t.Skip("\"localhost\" doesn't resolve to ::1.") } ln, err := net.Listen("tcp", ":0") if err != nil { t.Fatalf("Error listening on socket: %s", err) } defer ln.Close() _, port, _ := net.SplitHostPort(ln.Addr().String()) // Force IPv4 module := Module{ Timeout: time.Second, TCP: TCPProbe{ Protocol: "tcp4", }, } recorder := httptest.NewRecorder() result := probeTCP(net.JoinHostPort("localhost", port), recorder, module) body := recorder.Body.String() if !result { t.Fatalf("TCP protocol: \"tcp4\" connection test failed, expected success.") } if !strings.Contains(body, "probe_ip_protocol 4\n") { t.Fatalf("Expected IPv4, got %s", body) } // Force IPv6 module = Module{ Timeout: time.Second, TCP: TCPProbe{ Protocol: "tcp6", }, } recorder = httptest.NewRecorder() result = probeTCP(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("TCP protocol: \"tcp6\" connection test failed, expected success.") } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } // Prefer IPv4 module = Module{ Timeout: time.Second, TCP: TCPProbe{ Protocol: "tcp", PreferredIpProtocol: "ip4", }, } recorder = httptest.NewRecorder() result = probeTCP(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("TCP protocol: \"tcp\", prefer: \"ip4\" connection test failed, expected success.") } if !strings.Contains(body, "probe_ip_protocol 4\n") { t.Fatalf("Expected IPv4, got %s", body) } // Prefer IPv6 module = Module{ Timeout: time.Second, TCP: TCPProbe{ Protocol: "tcp", PreferredIpProtocol: "ip6", }, } recorder = httptest.NewRecorder() result = probeTCP(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("TCP protocol: \"tcp\", prefer: \"ip6\" connection test failed, expected success.") } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } // Prefer nothing module = Module{ Timeout: time.Second, TCP: TCPProbe{ Protocol: "tcp", }, } recorder = httptest.NewRecorder() result = probeTCP(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("TCP protocol: \"tcp\" connection test failed, expected success.") } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } // No protocol module = Module{ Timeout: time.Second, TCP: TCPProbe{}, } recorder = httptest.NewRecorder() result = probeTCP(net.JoinHostPort("localhost", port), recorder, module) body = recorder.Body.String() if !result { t.Fatalf("TCP connection test with protocol unspecified failed, expected success.") } if !strings.Contains(body, "probe_ip_protocol 6\n") { t.Fatalf("Expected IPv6, got %s", body) } } prometheus-blackbox-exporter-0.3.0+ds1/tls.go000066400000000000000000000016321301022322500211420ustar00rootroot00000000000000// Copyright 2016 The Prometheus 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 main import ( "crypto/tls" "time" ) func getEarliestCertExpiry(state *tls.ConnectionState) time.Time { earliest := time.Time{} for _, cert := range state.PeerCertificates { if (earliest.IsZero() || cert.NotAfter.Before(earliest)) && !cert.NotAfter.IsZero() { earliest = cert.NotAfter } } return earliest }