// DialContext is a function to make custom Dial for internode communications
type DialContext func(ctx context.Context, network, address string) (net.Conn, error)

// NewInternodeDialContext setups a custom dialer for internode communication
func NewInternodeDialContext(dialTimeout time.Duration, opts TCPOptions) DialContext {
	return func(ctx context.Context, network, addr string) (net.Conn, error) {
		dialer := &net.Dialer{
			Timeout: dialTimeout,

	// Part size.
	partSize = int64(partSizeFlt)
	if partSize == 0 {
		return minPartSize, nil
	}
	return partSize, nil
}

func (m *warmBackendMinIO) PutWithMeta(ctx context.Context, object string, r io.Reader, length int64, meta map[string]string) (remoteVersionID, error) {
	partSize, err := optimalPartSize(length)
	if err != nil {
		return remoteVersionID(""), err
	}

    // Test that we properly fall back to TCP when RDMA is not available
    CIFSContext context = getTestContext();
    context.getConfig().setProperty("jcifs.smb.client.useRDMA", "true");
    context.getConfig().setProperty("jcifs.smb.client.rdmaProvider", "nonexistent");
    
    SmbFile file = new SmbFile("smb://server/share/test.txt", context);
    
    // Should work even if RDMA provider is not available

def test_dependency_contextvars():
    """
    Check that custom middlewares don't affect the contextvar context for dependencies.

    The code before yield and the code after yield should be run in the same contextvar
    context, so that request_state_context_var.reset(contextvar_token).

    If they are run in a different context, that raises an error.
    """
    response = client.get("/user")
    assert response.json() == "deadpond"

            RequestTrace trace,
            String context,
            ArtifactResolver resolver,
            VersionRangeResolver versionRangeResolver,
            RemoteRepositoryManager remoteRepositoryManager,
            List<RemoteRepository> repositories) {
        this.session = session;
        this.trace = trace;
        this.context = context;
        this.resolver = resolver;

pkg testing, method (*B) Chdir(string) #62516
pkg testing, method (*B) Context() context.Context #36532
pkg testing, method (*B) Loop() bool #61515
pkg testing, method (*F) Chdir(string) #62516
pkg testing, method (*F) Context() context.Context #36532
pkg testing, method (*T) Chdir(string) #62516
pkg testing, method (*T) Context() context.Context #36532
pkg testing, type TB interface, Chdir(string) #62516

    }

    /**
     * Create a lease break acknowledgment
     *
     * @param context CIFS context
     * @param leaseKey lease key
     * @param leaseState acknowledged lease state
     */
    public Smb2LeaseBreakAcknowledgment(CIFSContext context, Smb2LeaseKey leaseKey, int leaseState) {
        this(context.getConfig(), leaseKey, leaseState);
    }

    /**
     * Gets the lease key for this acknowledgment

     * @param sessionKey
     * @param label
     * @param context
     */
    private static byte[] derive(final byte[] sessionKey, final byte[] label, final byte[] context) {
        final KDFCounterBytesGenerator gen = new KDFCounterBytesGenerator(new HMac(new SHA256Digest()));

        final int r = 32;
        final byte[] suffix = new byte[label.length + context.length + 5];
        // per bouncycastle

     *
     * @param queryContext the query context containing the query string
     * @param context a consumer for additional query context customization
     * @throws InvalidQueryException if the query string cannot be parsed
     */
    public void buildBaseQuery(final QueryContext queryContext, final Consumer<QueryContext> context) {
        try {

    private final CIFSContext context;
    private final ReentrantReadWriteLock lock = new ReentrantReadWriteLock();

    private volatile boolean shutdown = false;

    /**
     * Create a new persistent handle manager
     * @param context the CIFS context
     */
    public PersistentHandleManager(CIFSContext context) {
        this.context = context;
        this.handles = new ConcurrentHashMap<>();