jcifs.smb.SmbTreeHandleImpl Maven / Gradle / Ivy
/*
* © 2017 AgNO3 Gmbh & Co. KG
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
package jcifs.smb;
import java.util.Set;
import java.util.concurrent.atomic.AtomicLong;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import jcifs.CIFSException;
import jcifs.Configuration;
import jcifs.RuntimeCIFSException;
import jcifs.SmbTreeHandle;
import jcifs.internal.CommonServerMessageBlockRequest;
import jcifs.internal.CommonServerMessageBlockResponse;
import jcifs.internal.SmbNegotiationResponse;
import jcifs.internal.smb1.com.SmbComNegotiateResponse;
/**
* @author mbechler
*
*/
class SmbTreeHandleImpl implements SmbTreeHandleInternal {
private static final Logger log = LoggerFactory.getLogger(SmbTreeHandleImpl.class);
private final SmbResourceLocatorImpl resourceLoc;
private final SmbTreeConnection treeConnection;
private final AtomicLong usageCount = new AtomicLong(1);
/**
* @param resourceLoc
* @param treeConnection
*/
public SmbTreeHandleImpl ( SmbResourceLocatorImpl resourceLoc, SmbTreeConnection treeConnection ) {
this.resourceLoc = resourceLoc;
this.treeConnection = treeConnection.acquire();
}
@Override
public SmbSessionImpl getSession () {
return this.treeConnection.getSession();
}
@Override
public void ensureDFSResolved () throws CIFSException {
this.treeConnection.ensureDFSResolved(this.resourceLoc);
}
@Override
public boolean hasCapability ( int cap ) throws SmbException {
return this.treeConnection.hasCapability(cap);
}
/**
* {@inheritDoc}
*
* @see jcifs.SmbTreeHandle#isConnected()
*/
@Override
public boolean isConnected () {
return this.treeConnection.isConnected();
}
/**
* {@inheritDoc}
*
* @see jcifs.SmbTreeHandle#getConfig()
*/
@Override
public Configuration getConfig () {
return this.treeConnection.getConfig();
}
/**
* @return the currently connected tree id
*/
public long getTreeId () {
return this.treeConnection.getTreeId();
}
/**
*
* @param req
* @param params
* @return response
* @throws CIFSException
*/
public T send ( jcifs.internal.Request req, RequestParam... params ) throws CIFSException {
return send(req, null, params);
}
/**
* @param request
* @param response
* @param params
* @return response
* @throws CIFSException
*/
public T send ( CommonServerMessageBlockRequest request, T response, RequestParam... params )
throws CIFSException {
return this.treeConnection.send(this.resourceLoc, request, response, params);
}
/**
*
* @param request
* @param response
* @param params
* @return response
* @throws CIFSException
*/
public T send ( CommonServerMessageBlockRequest request, T response, Set params )
throws CIFSException {
return this.treeConnection.send(this.resourceLoc, request, response, params);
}
/**
* {@inheritDoc}
*
* @see jcifs.SmbTreeHandle#close()
*/
@Override
public synchronized void close () {
release();
}
/**
* @return tree handle with increased usage count
*/
public SmbTreeHandleImpl acquire () {
if ( this.usageCount.incrementAndGet() == 1 ) {
this.treeConnection.acquire();
}
return this;
}
@Override
public void release () {
long us = this.usageCount.decrementAndGet();
if ( us == 0 ) {
this.treeConnection.release();
}
else if ( us < 0 ) {
throw new RuntimeCIFSException("Usage count dropped below zero");
}
}
/**
* {@inheritDoc}
*
* @see java.lang.Object#finalize()
*/
@Override
protected void finalize () throws Throwable {
if ( this.usageCount.get() != 0 ) {
log.warn("Tree handle was not properly released " + this.resourceLoc.getURL());
}
}
/**
* {@inheritDoc}
*
* @see jcifs.SmbTreeHandle#getRemoteHostName()
*/
@Override
public String getRemoteHostName () {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
return transport.getRemoteHostName();
}
}
/**
* {@inheritDoc}
*
* @throws SmbException
*
* @see jcifs.SmbTreeHandle#getServerTimeZoneOffset()
*/
@Override
public long getServerTimeZoneOffset () throws SmbException {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
SmbNegotiationResponse nego = transport.getNegotiateResponse();
if ( nego instanceof SmbComNegotiateResponse ) {
return ( (SmbComNegotiateResponse) nego ).getServerData().serverTimeZone * 1000 * 60L;
}
return 0;
}
}
/**
* {@inheritDoc}
*
* @throws SmbException
*
* @see jcifs.SmbTreeHandle#getOEMDomainName()
*/
@Override
public String getOEMDomainName () throws SmbException {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
SmbNegotiationResponse nego = transport.getNegotiateResponse();
if ( nego instanceof SmbComNegotiateResponse ) {
return ( (SmbComNegotiateResponse) nego ).getServerData().oemDomainName;
}
return null;
}
}
/**
* {@inheritDoc}
*
* @see jcifs.SmbTreeHandle#getTreeType()
*/
@Override
public int getTreeType () {
return this.treeConnection.getTreeType();
}
/**
* {@inheritDoc}
*
* @see jcifs.SmbTreeHandle#getConnectedShare()
*/
@Override
public String getConnectedShare () {
return this.treeConnection.getConnectedShare();
}
/**
*
* {@inheritDoc}
*
* @see jcifs.SmbTreeHandle#isSameTree(jcifs.SmbTreeHandle)
*/
@Override
public boolean isSameTree ( SmbTreeHandle th ) {
if ( ! ( th instanceof SmbTreeHandleImpl ) ) {
return false;
}
return this.treeConnection.isSame( ( (SmbTreeHandleImpl) th ).treeConnection);
}
@Override
public int getSendBufferSize () throws SmbException {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
return transport.getNegotiateResponse().getSendBufferSize();
}
}
@Override
public int getReceiveBufferSize () throws SmbException {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
return transport.getNegotiateResponse().getReceiveBufferSize();
}
}
@Override
public int getMaximumBufferSize () throws SmbException {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
return transport.getNegotiateResponse().getTransactionBufferSize();
}
}
@Override
public boolean areSignaturesActive () throws SmbException {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
return transport.getNegotiateResponse().isSigningNegotiated();
}
}
/**
* @return whether this tree handle uses SMB2
*/
@Override
public boolean isSMB2 () {
try ( SmbSessionImpl session = this.treeConnection.getSession();
SmbTransportImpl transport = session.getTransport() ) {
return transport.isSMB2();
}
catch ( SmbException e ) {
log.debug("Failed to connect for determining SMB2 support", e);
return false;
}
}
}