|
@@ -0,0 +1,155 @@
|
|
|
+/**
|
|
|
+ * Licensed to the Apache Software Foundation (ASF) under one
|
|
|
+ * or more contributor license agreements. See the NOTICE file
|
|
|
+ * distributed with this work for additional information
|
|
|
+ * regarding copyright ownership. The ASF licenses this file
|
|
|
+ * to you under the Apache License, Version 2.0 (the
|
|
|
+ * "License"); you may not use this file except in compliance
|
|
|
+ * with the License. You may obtain a copy of the License at
|
|
|
+ *
|
|
|
+ * http://www.apache.org/licenses/LICENSE-2.0
|
|
|
+ *
|
|
|
+ * Unless required by applicable law or agreed to in writing, software
|
|
|
+ * distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
+ * See the License for the specific language governing permissions and
|
|
|
+ * limitations under the License.
|
|
|
+ */
|
|
|
+package org.apache.hadoop.hdfs.server.namenode;
|
|
|
+
|
|
|
+import java.io.ByteArrayOutputStream;
|
|
|
+import java.io.IOException;
|
|
|
+import java.util.ArrayList;
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
+import org.apache.hadoop.fs.XAttr;
|
|
|
+import org.apache.hadoop.hdfs.XAttrHelper;
|
|
|
+
|
|
|
+import com.google.common.base.Preconditions;
|
|
|
+import com.google.common.primitives.Ints;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Class to pack XAttrs into byte[].<br>
|
|
|
+ * For each XAttr:<br>
|
|
|
+ * The first 4 bytes represents XAttr namespace and name<br>
|
|
|
+ * [0:3) - XAttr namespace<br>
|
|
|
+ * [3:32) - The name of the entry, which is an ID that points to a
|
|
|
+ * string in map<br>
|
|
|
+ * The following two bytes represents the length of XAttr value<br>
|
|
|
+ * The remaining bytes is the XAttr value<br>
|
|
|
+ */
|
|
|
+class XAttrFormat {
|
|
|
+ private static final int XATTR_NAMESPACE_MASK = (1 << 3) - 1;
|
|
|
+ private static final int XATTR_NAMESPACE_OFFSET = 29;
|
|
|
+ private static final int XATTR_NAME_MASK = (1 << 29) - 1;
|
|
|
+ private static final int XATTR_NAME_ID_MAX = 1 << 29;
|
|
|
+ private static final int XATTR_VALUE_LEN_MAX = 1 << 16;
|
|
|
+ private static final XAttr.NameSpace[] XATTR_NAMESPACE_VALUES =
|
|
|
+ XAttr.NameSpace.values();
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Unpack byte[] to XAttrs.
|
|
|
+ *
|
|
|
+ * @param attrs the packed bytes of XAttrs
|
|
|
+ * @return XAttrs list
|
|
|
+ */
|
|
|
+ static List<XAttr> toXAttrs(byte[] attrs) {
|
|
|
+ List<XAttr> xAttrs = new ArrayList<>();
|
|
|
+ if (attrs == null || attrs.length == 0) {
|
|
|
+ return xAttrs;
|
|
|
+ }
|
|
|
+ for (int i = 0; i < attrs.length;) {
|
|
|
+ XAttr.Builder builder = new XAttr.Builder();
|
|
|
+ // big-endian
|
|
|
+ int v = Ints.fromBytes(attrs[i++], attrs[i++], attrs[i++], attrs[i++]);
|
|
|
+ int ns = (v >> XATTR_NAMESPACE_OFFSET) & XATTR_NAMESPACE_MASK;
|
|
|
+ int nid = v & XATTR_NAME_MASK;
|
|
|
+ builder.setNameSpace(XATTR_NAMESPACE_VALUES[ns]);
|
|
|
+ builder.setName(XAttrStorage.getName(nid));
|
|
|
+ int vlen = (attrs[i++] << 8) | attrs[i++];
|
|
|
+ if (vlen > 0) {
|
|
|
+ byte[] value = new byte[vlen];
|
|
|
+ System.arraycopy(attrs, i, value, 0, vlen);
|
|
|
+ builder.setValue(value);
|
|
|
+ i += vlen;
|
|
|
+ }
|
|
|
+ xAttrs.add(builder.build());
|
|
|
+ }
|
|
|
+ return xAttrs;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Get XAttr by name with prefix.
|
|
|
+ * Will unpack the byte[] until find the specific XAttr
|
|
|
+ *
|
|
|
+ * @param attrs the packed bytes of XAttrs
|
|
|
+ * @param prefixedName the XAttr name with prefix
|
|
|
+ * @return the XAttr
|
|
|
+ */
|
|
|
+ static XAttr getXAttr(byte[] attrs, String prefixedName) {
|
|
|
+ if (prefixedName == null || attrs == null) {
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+
|
|
|
+ XAttr xAttr = XAttrHelper.buildXAttr(prefixedName);
|
|
|
+ for (int i = 0; i < attrs.length;) {
|
|
|
+ // big-endian
|
|
|
+ int v = Ints.fromBytes(attrs[i++], attrs[i++], attrs[i++], attrs[i++]);
|
|
|
+ int ns = (v >> XATTR_NAMESPACE_OFFSET) & XATTR_NAMESPACE_MASK;
|
|
|
+ int nid = v & XATTR_NAME_MASK;
|
|
|
+ XAttr.NameSpace namespace = XATTR_NAMESPACE_VALUES[ns];
|
|
|
+ String name = XAttrStorage.getName(nid);
|
|
|
+ int vlen = (attrs[i++] << 8) | attrs[i++];
|
|
|
+ if (xAttr.getNameSpace() == namespace &&
|
|
|
+ xAttr.getName().equals(name)) {
|
|
|
+ if (vlen > 0) {
|
|
|
+ byte[] value = new byte[vlen];
|
|
|
+ System.arraycopy(attrs, i, value, 0, vlen);
|
|
|
+ return new XAttr.Builder().setNameSpace(namespace).
|
|
|
+ setName(name).setValue(value).build();
|
|
|
+ }
|
|
|
+ return xAttr;
|
|
|
+ }
|
|
|
+ i += vlen;
|
|
|
+ }
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Pack the XAttrs to byte[].
|
|
|
+ *
|
|
|
+ * @param xAttrs the XAttrs
|
|
|
+ * @return the packed bytes
|
|
|
+ */
|
|
|
+ static byte[] toBytes(List<XAttr> xAttrs) {
|
|
|
+ if (xAttrs == null || xAttrs.isEmpty()) {
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+ ByteArrayOutputStream out = new ByteArrayOutputStream();
|
|
|
+ try {
|
|
|
+ for (XAttr a : xAttrs) {
|
|
|
+ int nsOrd = a.getNameSpace().ordinal();
|
|
|
+ Preconditions.checkArgument(nsOrd < 8, "Too many namespaces.");
|
|
|
+ int nid = XAttrStorage.getNameSerialNumber(a.getName());
|
|
|
+ Preconditions.checkArgument(nid < XATTR_NAME_ID_MAX,
|
|
|
+ "Too large serial number of the xattr name");
|
|
|
+
|
|
|
+ // big-endian
|
|
|
+ int v = ((nsOrd & XATTR_NAMESPACE_MASK) << XATTR_NAMESPACE_OFFSET)
|
|
|
+ | (nid & XATTR_NAME_MASK);
|
|
|
+ out.write(Ints.toByteArray(v));
|
|
|
+ int vlen = a.getValue() == null ? 0 : a.getValue().length;
|
|
|
+ Preconditions.checkArgument(vlen < XATTR_VALUE_LEN_MAX,
|
|
|
+ "The length of xAttr values is too long.");
|
|
|
+ out.write((byte)(vlen >> 8));
|
|
|
+ out.write((byte)(vlen));
|
|
|
+ if (vlen > 0) {
|
|
|
+ out.write(a.getValue());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (IOException e) {
|
|
|
+ // in fact, no exception
|
|
|
+ }
|
|
|
+ return out.toByteArray();
|
|
|
+ }
|
|
|
+}
|