@@ -1898,61 +1898,62 @@ func TestCVE202133195(t *testing.T) {
1898
1898
// Change the default resolver to match our manipulated resolver
1899
1899
originalDefault := DefaultResolver
1900
1900
DefaultResolver = & r
1901
- defer func () {
1902
- DefaultResolver = originalDefault
1903
- }()
1901
+ defer func () { DefaultResolver = originalDefault }()
1902
+ // Redirect host file lookups.
1903
+ defer func (orig string ) { testHookHostsPath = orig }(testHookHostsPath )
1904
+ testHookHostsPath = "testdata/hosts"
1904
1905
1905
1906
_ , err := r .LookupCNAME (context .Background (), "golang.org" )
1906
1907
if expected := "lookup golang.org: CNAME target is invalid" ; err == nil || err .Error () != expected {
1907
- t .Errorf ("Resolver.LookupCNAME returned unexpected error, got %q, want %q" , err . Error () , expected )
1908
+ t .Errorf ("Resolver.LookupCNAME returned unexpected error, got %q, want %q" , err , expected )
1908
1909
}
1909
1910
_ , err = LookupCNAME ("golang.org" )
1910
1911
if expected := "lookup golang.org: CNAME target is invalid" ; err == nil || err .Error () != expected {
1911
- t .Errorf ("LookupCNAME returned unexpected error, got %q, want %q" , err . Error () , expected )
1912
+ t .Errorf ("LookupCNAME returned unexpected error, got %q, want %q" , err , expected )
1912
1913
}
1913
1914
1914
1915
_ , _ , err = r .LookupSRV (context .Background (), "target" , "tcp" , "golang.org" )
1915
1916
if expected := "lookup golang.org: SRV target is invalid" ; err == nil || err .Error () != expected {
1916
- t .Errorf ("Resolver.LookupSRV returned unexpected error, got %q, want %q" , err . Error () , expected )
1917
+ t .Errorf ("Resolver.LookupSRV returned unexpected error, got %q, want %q" , err , expected )
1917
1918
}
1918
1919
_ , _ , err = LookupSRV ("target" , "tcp" , "golang.org" )
1919
1920
if expected := "lookup golang.org: SRV target is invalid" ; err == nil || err .Error () != expected {
1920
- t .Errorf ("LookupSRV returned unexpected error, got %q, want %q" , err . Error () , expected )
1921
+ t .Errorf ("LookupSRV returned unexpected error, got %q, want %q" , err , expected )
1921
1922
}
1922
1923
1923
1924
_ , _ , err = r .LookupSRV (context .Background (), "hdr" , "tcp" , "golang.org" )
1924
1925
if expected := "lookup golang.org: SRV header name is invalid" ; err == nil || err .Error () != expected {
1925
- t .Errorf ("Resolver.LookupSRV returned unexpected error, got %q, want %q" , err . Error () , expected )
1926
+ t .Errorf ("Resolver.LookupSRV returned unexpected error, got %q, want %q" , err , expected )
1926
1927
}
1927
1928
_ , _ , err = LookupSRV ("hdr" , "tcp" , "golang.org" )
1928
1929
if expected := "lookup golang.org: SRV header name is invalid" ; err == nil || err .Error () != expected {
1929
- t .Errorf ("LookupSRV returned unexpected error, got %q, want %q" , err . Error () , expected )
1930
+ t .Errorf ("LookupSRV returned unexpected error, got %q, want %q" , err , expected )
1930
1931
}
1931
1932
1932
1933
_ , err = r .LookupMX (context .Background (), "golang.org" )
1933
1934
if expected := "lookup golang.org: MX target is invalid" ; err == nil || err .Error () != expected {
1934
- t .Errorf ("Resolver.LookupMX returned unexpected error, got %q, want %q" , err . Error () , expected )
1935
+ t .Errorf ("Resolver.LookupMX returned unexpected error, got %q, want %q" , err , expected )
1935
1936
}
1936
1937
_ , err = LookupMX ("golang.org" )
1937
1938
if expected := "lookup golang.org: MX target is invalid" ; err == nil || err .Error () != expected {
1938
- t .Errorf ("LookupMX returned unexpected error, got %q, want %q" , err . Error () , expected )
1939
+ t .Errorf ("LookupMX returned unexpected error, got %q, want %q" , err , expected )
1939
1940
}
1940
1941
1941
1942
_ , err = r .LookupNS (context .Background (), "golang.org" )
1942
1943
if expected := "lookup golang.org: NS target is invalid" ; err == nil || err .Error () != expected {
1943
- t .Errorf ("Resolver.LookupNS returned unexpected error, got %q, want %q" , err . Error () , expected )
1944
+ t .Errorf ("Resolver.LookupNS returned unexpected error, got %q, want %q" , err , expected )
1944
1945
}
1945
1946
_ , err = LookupNS ("golang.org" )
1946
1947
if expected := "lookup golang.org: NS target is invalid" ; err == nil || err .Error () != expected {
1947
- t .Errorf ("LookupNS returned unexpected error, got %q, want %q" , err . Error () , expected )
1948
+ t .Errorf ("LookupNS returned unexpected error, got %q, want %q" , err , expected )
1948
1949
}
1949
1950
1950
- _ , err = r .LookupAddr (context .Background (), "1.2.3.4 " )
1951
- if expected := "lookup 1.2.3.4 : PTR target is invalid" ; err == nil || err .Error () != expected {
1952
- t .Errorf ("Resolver.LookupAddr returned unexpected error, got %q, want %q" , err . Error () , expected )
1951
+ _ , err = r .LookupAddr (context .Background (), "192.0.2.42 " )
1952
+ if expected := "lookup 192.0.2.42 : PTR target is invalid" ; err == nil || err .Error () != expected {
1953
+ t .Errorf ("Resolver.LookupAddr returned unexpected error, got %q, want %q" , err , expected )
1953
1954
}
1954
- _ , err = LookupAddr ("1.2.3.4 " )
1955
- if expected := "lookup 1.2.3.4 : PTR target is invalid" ; err == nil || err .Error () != expected {
1956
- t .Errorf ("LookupAddr returned unexpected error, got %q, want %q" , err . Error () , expected )
1955
+ _ , err = LookupAddr ("192.0.2.42 " )
1956
+ if expected := "lookup 192.0.2.42 : PTR target is invalid" ; err == nil || err .Error () != expected {
1957
+ t .Errorf ("LookupAddr returned unexpected error, got %q, want %q" , err , expected )
1957
1958
}
1958
1959
}
0 commit comments