mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 16:55:23 +00:00
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
|
package oracle
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/protocols/common/protocolstate"
|
||
|
|
)
|
||
|
|
|
||
|
|
// oracleCustomDialer implements the dialer interface expected by go-ora
|
||
|
|
type oracleCustomDialer struct {
|
||
|
|
executionId string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *oracleCustomDialer) dialWithCtx(ctx context.Context, network, address string) (net.Conn, error) {
|
||
|
|
dialers := protocolstate.GetDialersWithId(o.executionId)
|
||
|
|
if dialers == nil {
|
||
|
|
return nil, fmt.Errorf("dialers not initialized for %s", o.executionId)
|
||
|
|
}
|
||
|
|
if !protocolstate.IsHostAllowed(o.executionId, address) {
|
||
|
|
// host is not valid according to network policy
|
||
|
|
return nil, protocolstate.ErrHostDenied.Msgf(address)
|
||
|
|
}
|
||
|
|
return dialers.Fastdialer.Dial(ctx, network, address)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *oracleCustomDialer) Dial(network, address string) (net.Conn, error) {
|
||
|
|
return o.dialWithCtx(context.TODO(), network, address)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *oracleCustomDialer) DialTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
return o.dialWithCtx(ctx, network, address)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *oracleCustomDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
|
||
|
|
return o.dialWithCtx(ctx, network, address)
|
||
|
|
}
|