Block Token 原理分析

介绍

文件权限检查由NameNode执行，而不是DataNode执行。 默认情况下，任何客户端都可以访问只有其块ID的任何块。 为了解决这个问题，Hadoop引入了块访问令牌的概念。 块访问令牌由NameNode生成，并在DataNode端进行合法性验证。块访问令牌作为Hadoop数据传输协议的一部分或通过HTTP查询参数来呈现。一个典型的应用场景如下：一个客户端向NameNode发送文件读请求，NameNode验证该用户具有文件读权限后，将文件对应的所有数据块的ID、位置以及数据块访问令牌发送给客户端；当客户端需要读取某个数据块时，将数据块ID和数据块访问令牌发送给对应的DataNode。由于NameNode已经通过心跳将密钥发送给各个DataNode，因此DataNode可以对数据块进行安全验证，而只有通过安全验证的访问请求才可以获取数据块。

Block Token产生和验证的过程如下：

（1）Namenode经过对客户的身份验证和访问权限验证之后，返回块位置以及块访问令牌。

（2）客户端给Datanode发送块ID以及块访问令牌请求数据。

（3）Datanode经过验证块访问令牌之后返回给客户端请求的数据。

源码分析

用户使用Block Token访问数据的流程图如下：

Block Token的产生

由代码追踪可知Block Token是调用BlockPoolTokenSecretManager类产生的，但实际产生Block Token的操作是由BlockTokenSecretManager类执行的，该类与BlockPoolTokenSecretManager类的关系如下：

BlockPoolTokenSecretManager包含BlockTokenSecretManager,并且每一个blockPool对应一个BlockTokenSecretManager

　　实际用map存储对应关系：

 private final Map<String, BlockTokenSecretManager> map =
     new HashMap<String, BlockTokenSecretManager>();

所以先调用BlockPoolTokenSecretManager类的方法获取BlockPoolId找到对应的BlockTokenSecretManager。

/**
   * See {@link BlockTokenSecretManager#generateToken(ExtendedBlock, EnumSet)}
   */
  public Token<BlockTokenIdentifier> generateToken(ExtendedBlock b,
      EnumSet<AccessMode> of) throws IOException {
    return get(b.getBlockPoolId()).generateToken(b, of);
  }

进入实际产生BlockToken的方法：

  /** Generate an block token for current user */
   public Token<BlockTokenIdentifier> generateToken(ExtendedBlock block,
       EnumSet<AccessMode> modes) throws IOException {
     UserGroupInformation ugi = UserGroupInformation.getCurrentUser();
     String userID = (ugi == null ? null : ugi.getShortUserName());
     LOG.info("sqy test!"+"ugi="+ugi+",userID="+userID);
     return generateToken(userID, block, modes);
   }

   /** Generate a block token for a specified user */
   public Token<BlockTokenIdentifier> generateToken(String userId,
       ExtendedBlock block, EnumSet<AccessMode> modes) throws IOException {
     BlockTokenIdentifier id = new BlockTokenIdentifier(userId, block
         .getBlockPoolId(), block.getBlockId(), modes);
     return new Token<BlockTokenIdentifier>(id, this);
   }

经过Kerberos和权限检查之后，Namenode就需要返回给用户块信息了。下面只讲主要的实现方法，调用BlockManager#createLocatedBlock()，该方法主要做了两件事：创建LocatedBlock和产生BlockToken。

（1）进入BlockManager类

  private LocatedBlock createLocatedBlock(final BlockInfoContiguous blk, final long pos,
     final BlockTokenSecretManager.AccessMode mode) throws IOException {
     //获取BlockID、BlockPoolID、userID、位置pos
     final LocatedBlock lb = createLocatedBlock(blk, pos);
     if (mode != null) {
       //设置块令牌
       setBlockToken(lb, mode);
     }
     return lb;
   }

由于我们这里主要讲块访问令牌的建立使用过程，创建LocatedBlock获取块信息这部分就不展开讲了。mode=AccessMode.READ，进入到setBlockToken方法中。

  public void setBlockToken(final LocatedBlock b,
       final BlockTokenSecretManager.AccessMode mode) throws IOException {
     // 如果开启BlockToken认证功能，这里是在hdfs-site.xml文件中配置的。
     if (isBlockTokenEnabled()) {
       // Use cached UGI if serving RPC calls.
       b.setBlockToken(blockTokenSecretManager.generateToken(
           NameNode.getRemoteUser().getShortUserName(),
           b.getBlock(), EnumSet.of(mode)));
     }
   }

（2）进入BlockTokenSecretManager类

在BlockManager类中调用generateToken方法创建令牌之前，会先创建BlockTokenSecretManager类的实例对象blockTokenSecretManager，设置块访问密钥更新间隔时间、块访问令牌的生命周期、加密算法以及生成了密钥。

/** Generate a block token for a specified user */
  public Token<BlockTokenIdentifier> generateToken(String userId,
      ExtendedBlock block, EnumSet<AccessMode> modes) throws IOException {
    //生成TokenID
    BlockTokenIdentifier id = new BlockTokenIdentifier(userId, block
        .getBlockPoolId(), block.getBlockId(), modes);
    //返回块访问令牌
    return new Token<BlockTokenIdentifier>(id, this);
  }

产生返回的块令牌信息的实现：

  /**
    * Construct a token given a token identifier and a secret manager for the
    * type of the token identifier.
    * @param id the token identifier
    * @param mgr the secret manager
    */
   public Token(T id, SecretManager<T> mgr) {
     password = mgr.createPassword(id); // 设置令牌过期时间和keyId
     identifier = id.getBytes();
     kind = id.getKind();
     service = new Text();
   }

Block Token的验证

由之前的理论知识可知，namenode返回给用户块访问令牌，用户根据块信息和块访问令牌去datanode请求文件信息。因此块访问令牌的验证是在datanode发生的，根据代码追踪可知是在DataXceiver类，该类执行了各种block操作处理方法，而在readBlock、writeBlock中就包含了Block Token的验证操作。下面以readBlock方法为例来就行说明。

（1）进入DataXceiver类

 public void readBlock(final ExtendedBlock block,
       final Token<BlockTokenIdentifier> blockToken,
       final String clientName,
       final long blockOffset,
       final long length,
       final boolean sendChecksum,
       final CachingStrategy cachingStrategy) throws IOException {
     previousOpClientName = clientName;
     long read = 0;
     updateCurrentThreadName("Sending block " + block);
     OutputStream baseStream = getOutputStream();
     DataOutputStream out = getBufferedOutputStream();
     //进行Token READ访问模式的验证
     checkAccess(out, true, block, blockToken,
         Op.READ_BLOCK, BlockTokenSecretManager.AccessMode.READ);
   ......

进入到checkAccess方法中。

 private void checkAccess(OutputStream out, final boolean reply,
       final ExtendedBlock blk,
       final Token<BlockTokenIdentifier> t,
       final Op op,
       final BlockTokenSecretManager.AccessMode mode) throws IOException {
     checkAndWaitForBP(blk);
     //判断是否启用BlockToken验证
     if (datanode.isBlockTokenEnabled) {
       if (LOG.isDebugEnabled()) {
         LOG.debug("Checking block access token for block '" + blk.getBlockId()
             + "' with mode '" + mode + "'");
       }
       try {
         //进行BlockToken验证
         datanode.blockPoolTokenSecretManager.checkAccess(t, null, blk, mode);
       } catch(InvalidToken e) {
       .....

（2）进入BlockPoolTokenSecretManager类

  public Token<BlockTokenIdentifier> generateToken(ExtendedBlock b,
       EnumSet<AccessMode> of) throws IOException {
     return get(b.getBlockPoolId()).generateToken(b, of);
   }

（3）进入BlockTokenSecretManager类

 public void checkAccess(Token<BlockTokenIdentifier> token, String userId,
       ExtendedBlock block, AccessMode mode) throws InvalidToken {
     BlockTokenIdentifier id = new BlockTokenIdentifier();
     try {
       //从输入流读取参数到tokenID，对其反序列化
       id.readFields(new DataInputStream(new ByteArrayInputStream(token
           .getIdentifier())));
     } catch (IOException e) {
       throw new InvalidToken(
           "Unable to de-serialize block token identifier for user=" + userId
               + ", block=" + block + ", access mode=" + mode);
     }
     //验证块令牌中的相关信息（userID、blockID、BlockPoolID、ExpiryDate、mode） 
     checkAccess(id, userId, block, mode); 
    .....

结论

以上就是Block Token产生、验证的整个过程。不过是否开启Block Token验证是需要在hdfs-site.xml文件中配置的，默认是false。

    dfs.block.access.token.enable