@@ -47,8 +47,7 @@ import (
47
47
48
48
"golang.org/x/build/buildlet"
49
49
"golang.org/x/build/dashboard"
50
- "golang.org/x/build/revdial"
51
- revdialv2 "golang.org/x/build/revdial/v2"
50
+ "golang.org/x/build/revdial/v2"
52
51
"golang.org/x/build/types"
53
52
)
54
53
@@ -502,14 +501,14 @@ func handleReverse(w http.ResponseWriter, r *http.Request) {
502
501
hostType = r .Header .Get ("X-Go-Host-Type" )
503
502
buildKey = r .Header .Get ("X-Go-Builder-Key" )
504
503
buildletVersion = r .Header .Get ("X-Go-Builder-Version" )
505
- revDialVersion = r .Header .Get ("X-Revdial-Version" )
506
504
hostname = r .Header .Get ("X-Go-Builder-Hostname" )
507
505
)
508
506
509
- switch revDialVersion {
507
+ switch r . Header . Get ( "X-Revdial-Version" ) {
510
508
case "" :
511
509
// Old.
512
- revDialVersion = "1"
510
+ http .Error (w , "buildlet binary is too old" , http .StatusBadRequest )
511
+ return
513
512
case "2" :
514
513
// Current.
515
514
default :
@@ -532,7 +531,7 @@ func handleReverse(w http.ResponseWriter, r *http.Request) {
532
531
return
533
532
}
534
533
535
- conn , bufrw , err := w .(http.Hijacker ).Hijack ()
534
+ conn , _ , err := w .(http.Hijacker ).Hijack ()
536
535
if err != nil {
537
536
http .Error (w , err .Error (), http .StatusInternalServerError )
538
537
return
@@ -544,24 +543,12 @@ func handleReverse(w http.ResponseWriter, r *http.Request) {
544
543
return
545
544
}
546
545
547
- log .Printf ("Registering reverse buildlet %q (%s) for host type %v; buildletVersion=%v; revDialVersion=%v " ,
548
- hostname , r .RemoteAddr , hostType , buildletVersion , revDialVersion )
546
+ log .Printf ("Registering reverse buildlet %q (%s) for host type %v; buildletVersion=%v" ,
547
+ hostname , r .RemoteAddr , hostType , buildletVersion )
549
548
550
- var dialer func (context.Context ) (net.Conn , error )
551
- var revDialerDone <- chan struct {}
552
- switch revDialVersion {
553
- case "1" :
554
- revDialer := revdial .NewDialer (bufrw , conn )
555
- revDialerDone = revDialer .Done ()
556
- dialer = func (ctx context.Context ) (net.Conn , error ) {
557
- // ignoring context.
558
- return revDialer .Dial ()
559
- }
560
- case "2" :
561
- revDialer := revdialv2 .NewDialer (conn , "/revdial" )
562
- revDialerDone = revDialer .Done ()
563
- dialer = revDialer .Dial
564
- }
549
+ revDialer := revdial .NewDialer (conn , "/revdial" )
550
+ revDialerDone := revDialer .Done ()
551
+ dialer := revDialer .Dial
565
552
566
553
client := buildlet .NewClient (hostname , buildlet .NoKeyPair )
567
554
client .SetHTTPClient (& http.Client {
0 commit comments