Examples with ByteArrayInputStream java.io.ByteArrayInputStream used on opensource projects

Example 71 with ByteArrayInputStream

use of java.io.ByteArrayInputStream in project hadoop by apache.

the class ZKDelegationTokenSecretManager method getTokenInfoFromZK.

private DelegationTokenInformation getTokenInfoFromZK(TokenIdent ident, boolean quiet) throws IOException {
    String nodePath = getNodePath(ZK_DTSM_TOKENS_ROOT, DELEGATION_TOKEN_PREFIX + ident.getSequenceNumber());
    try {
        byte[] data = zkClient.getData().forPath(nodePath);
        if ((data == null) || (data.length == 0)) {
            return null;
        }
        ByteArrayInputStream bin = new ByteArrayInputStream(data);
        DataInputStream din = new DataInputStream(bin);
        createIdentifier().readFields(din);
        long renewDate = din.readLong();
        int pwdLen = din.readInt();
        byte[] password = new byte[pwdLen];
        int numRead = din.read(password, 0, pwdLen);
        if (numRead > -1) {
            DelegationTokenInformation tokenInfo = new DelegationTokenInformation(renewDate, password);
            return tokenInfo;
        }
    } catch (KeeperException.NoNodeException e) {
        if (!quiet) {
            LOG.error("No node in path [" + nodePath + "]");
        }
    } catch (Exception ex) {
        throw new IOException(ex);
    }
    return null;
}

Example 72 with ByteArrayInputStream

use of java.io.ByteArrayInputStream in project hadoop by apache.

the class ZKDelegationTokenSecretManager method cancelToken.

@Override
public synchronized TokenIdent cancelToken(Token<TokenIdent> token, String canceller) throws IOException {
    ByteArrayInputStream buf = new ByteArrayInputStream(token.getIdentifier());
    DataInputStream in = new DataInputStream(buf);
    TokenIdent id = createIdentifier();
    id.readFields(in);
    syncLocalCacheWithZk(id);
    return super.cancelToken(token, canceller);
}

Example 73 with ByteArrayInputStream

use of java.io.ByteArrayInputStream in project hadoop by apache.

the class SaslRpcServer method getIdentifier.

public static <T extends TokenIdentifier> T getIdentifier(String id, SecretManager<T> secretManager) throws InvalidToken {
    byte[] tokenId = decodeIdentifier(id);
    T tokenIdentifier = secretManager.createIdentifier();
    try {
        tokenIdentifier.readFields(new DataInputStream(new ByteArrayInputStream(tokenId)));
    } catch (IOException e) {
        throw (InvalidToken) new InvalidToken("Can't de-serialize tokenIdentifier").initCause(e);
    }
    return tokenIdentifier;
}

Example 74 with ByteArrayInputStream

use of java.io.ByteArrayInputStream in project hadoop by apache.

the class AbstractDelegationTokenSecretManager method cancelToken.

/**
   * Cancel a token by removing it from cache.
   * @return Identifier of the canceled token
   * @throws InvalidToken for invalid token
   * @throws AccessControlException if the user isn't allowed to cancel
   */
public synchronized TokenIdent cancelToken(Token<TokenIdent> token, String canceller) throws IOException {
    ByteArrayInputStream buf = new ByteArrayInputStream(token.getIdentifier());
    DataInputStream in = new DataInputStream(buf);
    TokenIdent id = createIdentifier();
    id.readFields(in);
    LOG.info("Token cancellation requested for identifier: " + formatTokenId(id));
    if (id.getUser() == null) {
        throw new InvalidToken("Token with no owner " + formatTokenId(id));
    }
    String owner = id.getUser().getUserName();
    Text renewer = id.getRenewer();
    HadoopKerberosName cancelerKrbName = new HadoopKerberosName(canceller);
    String cancelerShortName = cancelerKrbName.getShortName();
    if (!canceller.equals(owner) && (renewer == null || renewer.toString().isEmpty() || !cancelerShortName.equals(renewer.toString()))) {
        throw new AccessControlException(canceller + " is not authorized to cancel the token " + formatTokenId(id));
    }
    DelegationTokenInformation info = currentTokens.remove(id);
    if (info == null) {
        throw new InvalidToken("Token not found " + formatTokenId(id));
    }
    removeStoredToken(id);
    return id;
}

Example 75 with ByteArrayInputStream

use of java.io.ByteArrayInputStream in project hadoop by apache.

the class DelegationTokenManager method decodeToken.

private static DelegationTokenIdentifier decodeToken(Token<DelegationTokenIdentifier> token, Text tokenKind) throws IOException {
    ByteArrayInputStream buf = new ByteArrayInputStream(token.getIdentifier());
    DataInputStream dis = new DataInputStream(buf);
    DelegationTokenIdentifier id = new DelegationTokenIdentifier(tokenKind);
    id.readFields(dis);
    dis.close();
    return id;
}