tikhomirov@36: /* tikhomirov@532: * Copyright (c) 2011-2013 TMate Software Ltd tikhomirov@74: * tikhomirov@74: * This program is free software; you can redistribute it and/or modify tikhomirov@74: * it under the terms of the GNU General Public License as published by tikhomirov@74: * the Free Software Foundation; version 2 of the License. tikhomirov@74: * tikhomirov@74: * This program is distributed in the hope that it will be useful, tikhomirov@74: * but WITHOUT ANY WARRANTY; without even the implied warranty of tikhomirov@74: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the tikhomirov@74: * GNU General Public License for more details. tikhomirov@74: * tikhomirov@74: * For information on how to redistribute this software under tikhomirov@74: * the terms of a license other than GNU General Public License tikhomirov@102: * contact TMate Software at support@hg4j.com tikhomirov@36: */ tikhomirov@74: package org.tmatesoft.hg.repo; tikhomirov@36: tikhomirov@36: import java.io.File; tikhomirov@645: import java.io.FileInputStream; tikhomirov@36: import java.io.IOException; tikhomirov@512: import java.util.ConcurrentModificationException; tikhomirov@36: tikhomirov@645: import org.tmatesoft.hg.core.HgIOException; tikhomirov@74: import org.tmatesoft.hg.core.Nodeid; tikhomirov@357: import org.tmatesoft.hg.core.SessionContext; tikhomirov@157: import org.tmatesoft.hg.internal.ByteArrayChannel; tikhomirov@157: import org.tmatesoft.hg.internal.ByteArrayDataAccess; tikhomirov@512: import org.tmatesoft.hg.internal.Callback; tikhomirov@74: import org.tmatesoft.hg.internal.DataAccess; tikhomirov@74: import org.tmatesoft.hg.internal.DataAccessProvider; tikhomirov@645: import org.tmatesoft.hg.internal.DataSerializer; tikhomirov@74: import org.tmatesoft.hg.internal.DigestHelper; tikhomirov@358: import org.tmatesoft.hg.internal.Experimental; tikhomirov@645: import org.tmatesoft.hg.internal.FileUtils; tikhomirov@169: import org.tmatesoft.hg.internal.InflaterDataAccess; tikhomirov@526: import org.tmatesoft.hg.internal.Internals; tikhomirov@512: import org.tmatesoft.hg.internal.Lifecycle; tikhomirov@329: import org.tmatesoft.hg.internal.Patch; tikhomirov@154: import org.tmatesoft.hg.repo.HgChangelog.RawChangeset; tikhomirov@512: import org.tmatesoft.hg.util.Adaptable; tikhomirov@157: import org.tmatesoft.hg.util.CancelledException; tikhomirov@74: tikhomirov@36: /** tikhomirov@423: * WORK IN PROGRESS tikhomirov@423: * tikhomirov@36: * @see http://mercurial.selenic.com/wiki/BundleFormat tikhomirov@169: * tikhomirov@74: * @author Artem Tikhomirov tikhomirov@74: * @author TMate Software Ltd. tikhomirov@36: */ tikhomirov@467: @Experimental(reason="API is not stable") tikhomirov@36: public class HgBundle { tikhomirov@36: tikhomirov@36: private final File bundleFile; tikhomirov@36: private final DataAccessProvider accessProvider; tikhomirov@645: private final SessionContext ctx; tikhomirov@512: private Lifecycle.BasicCallback flowControl; tikhomirov@36: tikhomirov@645: HgBundle(SessionContext sessionContext, DataAccessProvider dap, File bundle) { tikhomirov@645: ctx = sessionContext; tikhomirov@36: accessProvider = dap; tikhomirov@36: bundleFile = bundle; tikhomirov@36: } tikhomirov@36: tikhomirov@169: private DataAccess getDataStream() throws IOException { tikhomirov@606: DataAccess da = accessProvider.createReader(bundleFile, false); tikhomirov@169: byte[] signature = new byte[6]; tikhomirov@169: if (da.length() > 6) { tikhomirov@169: da.readBytes(signature, 0, 6); tikhomirov@169: if (signature[0] == 'H' && signature[1] == 'G' && signature[2] == '1' && signature[3] == '0') { tikhomirov@169: if (signature[4] == 'G' && signature[5] == 'Z') { tikhomirov@169: return new InflaterDataAccess(da, 6, da.length() - 6); tikhomirov@169: } tikhomirov@169: if (signature[4] == 'B' && signature[5] == 'Z') { tikhomirov@526: throw Internals.notImplemented(); tikhomirov@169: } tikhomirov@169: if (signature[4] != 'U' || signature[5] != 'N') { tikhomirov@423: throw new HgInvalidStateException(String.format("Bad bundle signature: %s", String.valueOf(signature))); tikhomirov@169: } tikhomirov@169: // "...UN", fall-through tikhomirov@169: } else { tikhomirov@169: da.reset(); tikhomirov@39: } tikhomirov@169: } tikhomirov@169: return da; tikhomirov@169: } tikhomirov@169: tikhomirov@186: private int uses = 0; tikhomirov@186: public HgBundle link() { tikhomirov@186: uses++; tikhomirov@186: return this; tikhomirov@186: } tikhomirov@186: public void unlink() { tikhomirov@186: uses--; tikhomirov@186: if (uses == 0 && bundleFile != null) { tikhomirov@186: bundleFile.deleteOnExit(); tikhomirov@186: } tikhomirov@186: } tikhomirov@186: public boolean inUse() { tikhomirov@186: return uses > 0; tikhomirov@186: } tikhomirov@186: tikhomirov@182: /** tikhomirov@182: * Get changes recorded in the bundle that are missing from the supplied repository. tikhomirov@182: * @param hgRepo repository that shall possess base revision for this bundle tikhomirov@182: * @param inspector callback to get each changeset found tikhomirov@182: */ tikhomirov@423: public void changes(final HgRepository hgRepo, final HgChangelog.Inspector inspector) throws HgRuntimeException { tikhomirov@182: Inspector bundleInsp = new Inspector() { tikhomirov@169: DigestHelper dh = new DigestHelper(); tikhomirov@169: boolean emptyChangelog = true; tikhomirov@169: private DataAccess prevRevContent; tikhomirov@182: private int revisionIndex; tikhomirov@169: tikhomirov@169: public void changelogStart() { tikhomirov@169: emptyChangelog = true; tikhomirov@182: revisionIndex = 0; tikhomirov@169: } tikhomirov@169: tikhomirov@169: public void changelogEnd() { tikhomirov@169: if (emptyChangelog) { tikhomirov@169: throw new IllegalStateException("No changelog group in the bundle"); // XXX perhaps, just be silent and/or log? tikhomirov@42: } tikhomirov@37: } tikhomirov@169: tikhomirov@169: /* tikhomirov@169: * Despite that BundleFormat wiki says: "Each Changelog entry patches the result of all previous patches tikhomirov@169: * (the previous, or parent patch of a given patch p is the patch that has a node equal to p's p1 field)", tikhomirov@169: * it seems not to hold true. Instead, each entry patches previous one, regardless of whether the one tikhomirov@169: * before is its parent (i.e. ge.firstParent()) or not. tikhomirov@169: * tikhomirov@169: Actual state in the changelog.i tikhomirov@169: Index Offset Flags Packed Actual Base Rev Link Rev Parent1 Parent2 nodeid tikhomirov@169: 50: 9212 0 209 329 48 50 49 -1 f1db8610da62a3e0beb8d360556ee1fd6eb9885e tikhomirov@169: 51: 9421 0 278 688 48 51 50 -1 9429c7bd1920fab164a9d2b621d38d57bcb49ae0 tikhomirov@169: 52: 9699 0 154 179 52 52 50 -1 30bd389788464287cee22ccff54c330a4b715de5 tikhomirov@169: 53: 9853 0 133 204 52 53 51 52 a6f39e595b2b54f56304470269a936ead77f5725 tikhomirov@169: 54: 9986 0 156 182 54 54 52 -1 fd4f2c98995beb051070630c272a9be87bef617d tikhomirov@169: tikhomirov@169: Excerpt from bundle (nodeid, p1, p2, cs): tikhomirov@169: f1db8610da62a3e0beb8d360556ee1fd6eb9885e 26e3eeaa39623de552b45ee1f55c14f36460f220 0000000000000000000000000000000000000000 f1db8610da62a3e0beb8d360556ee1fd6eb9885e; patches:4 tikhomirov@169: 9429c7bd1920fab164a9d2b621d38d57bcb49ae0 f1db8610da62a3e0beb8d360556ee1fd6eb9885e 0000000000000000000000000000000000000000 9429c7bd1920fab164a9d2b621d38d57bcb49ae0; patches:3 tikhomirov@169: > 30bd389788464287cee22ccff54c330a4b715de5 f1db8610da62a3e0beb8d360556ee1fd6eb9885e 0000000000000000000000000000000000000000 30bd389788464287cee22ccff54c330a4b715de5; patches:3 tikhomirov@169: a6f39e595b2b54f56304470269a936ead77f5725 9429c7bd1920fab164a9d2b621d38d57bcb49ae0 30bd389788464287cee22ccff54c330a4b715de5 a6f39e595b2b54f56304470269a936ead77f5725; patches:3 tikhomirov@169: fd4f2c98995beb051070630c272a9be87bef617d 30bd389788464287cee22ccff54c330a4b715de5 0000000000000000000000000000000000000000 fd4f2c98995beb051070630c272a9be87bef617d; patches:3 tikhomirov@169: tikhomirov@169: To recreate 30bd..e5, one have to take content of 9429..e0, not its p1 f1db..5e tikhomirov@169: */ tikhomirov@628: public boolean element(GroupElement ge) throws HgRuntimeException { tikhomirov@169: emptyChangelog = false; tikhomirov@169: HgChangelog changelog = hgRepo.getChangelog(); tikhomirov@169: try { tikhomirov@169: if (prevRevContent == null) { tikhomirov@274: if (ge.firstParent().isNull() && ge.secondParent().isNull()) { tikhomirov@169: prevRevContent = new ByteArrayDataAccess(new byte[0]); tikhomirov@169: } else { tikhomirov@169: final Nodeid base = ge.firstParent(); tikhomirov@169: if (!changelog.isKnown(base) /*only first parent, that's Bundle contract*/) { tikhomirov@169: throw new IllegalStateException(String.format("Revision %s needs a parent %s, which is missing in the supplied repo %s", ge.node().shortNotation(), base.shortNotation(), hgRepo.toString())); tikhomirov@169: } tikhomirov@169: ByteArrayChannel bac = new ByteArrayChannel(); tikhomirov@418: changelog.rawContent(base, bac); // TODO post-1.0 get DataAccess directly, to avoid tikhomirov@169: // extra byte[] (inside ByteArrayChannel) duplication just for the sake of subsequent ByteArrayDataChannel wrap. tikhomirov@169: prevRevContent = new ByteArrayDataAccess(bac.toArray()); tikhomirov@169: } tikhomirov@169: } tikhomirov@169: // tikhomirov@169: byte[] csetContent = ge.apply(prevRevContent); tikhomirov@169: dh = dh.sha1(ge.firstParent(), ge.secondParent(), csetContent); // XXX ge may give me access to byte[] content of nodeid directly, perhaps, I don't need DH to be friend of Nodeid? tikhomirov@169: if (!ge.node().equalsTo(dh.asBinary())) { tikhomirov@423: throw new HgInvalidStateException(String.format("Integrity check failed on %s, node: %s", bundleFile, ge.node().shortNotation())); tikhomirov@169: } tikhomirov@169: ByteArrayDataAccess csetDataAccess = new ByteArrayDataAccess(csetContent); tikhomirov@169: RawChangeset cs = RawChangeset.parse(csetDataAccess); tikhomirov@182: inspector.next(revisionIndex++, ge.node(), cs); tikhomirov@170: prevRevContent.done(); tikhomirov@169: prevRevContent = csetDataAccess.reset(); tikhomirov@169: } catch (CancelledException ex) { tikhomirov@169: return false; tikhomirov@423: } catch (IOException ex) { tikhomirov@423: throw new HgInvalidFileException("Invalid bundle file", ex, bundleFile); // TODO post-1.0 revisit exception handling tikhomirov@427: } catch (HgInvalidDataFormatException ex) { tikhomirov@423: throw new HgInvalidControlFileException("Invalid bundle file", ex, bundleFile); tikhomirov@169: } tikhomirov@169: return true; tikhomirov@169: } tikhomirov@169: tikhomirov@169: public void manifestStart() {} tikhomirov@169: public void manifestEnd() {} tikhomirov@169: public void fileStart(String name) {} tikhomirov@169: public void fileEnd(String name) {} tikhomirov@169: tikhomirov@169: }; tikhomirov@423: inspectChangelog(bundleInsp); tikhomirov@169: } tikhomirov@169: tikhomirov@169: // callback to minimize amount of Strings and Nodeids instantiated tikhomirov@512: @Callback tikhomirov@169: public interface Inspector { tikhomirov@628: void changelogStart() throws HgRuntimeException; tikhomirov@169: tikhomirov@628: void changelogEnd() throws HgRuntimeException; tikhomirov@169: tikhomirov@628: void manifestStart() throws HgRuntimeException; tikhomirov@169: tikhomirov@628: void manifestEnd() throws HgRuntimeException; tikhomirov@169: tikhomirov@628: void fileStart(String name) throws HgRuntimeException; tikhomirov@628: tikhomirov@628: void fileEnd(String name) throws HgRuntimeException; tikhomirov@169: tikhomirov@169: /** tikhomirov@170: * XXX desperately need exceptions here tikhomirov@170: * @param element data element, instance might be reused, don't keep a reference to it or its raw data tikhomirov@169: * @return true to continue tikhomirov@169: */ tikhomirov@628: boolean element(GroupElement element) throws HgRuntimeException; tikhomirov@169: } tikhomirov@169: tikhomirov@423: /** tikhomirov@423: * @param inspector callback to visit changelog entries tikhomirov@423: * @throws HgRuntimeException subclass thereof to indicate issues with the library. Runtime exception tikhomirov@423: * @throws IllegalArgumentException if inspector argument is null tikhomirov@423: */ tikhomirov@423: public void inspectChangelog(Inspector inspector) throws HgRuntimeException { tikhomirov@169: if (inspector == null) { tikhomirov@169: throw new IllegalArgumentException(); tikhomirov@169: } tikhomirov@512: final Lifecycle lifecycle = lifecycleSetUp(inspector); tikhomirov@295: DataAccess da = null; tikhomirov@169: try { tikhomirov@295: da = getDataStream(); tikhomirov@182: internalInspectChangelog(da, inspector); tikhomirov@295: } catch (IOException ex) { tikhomirov@295: throw new HgInvalidFileException("Bundle.inspectChangelog failed", ex, bundleFile); tikhomirov@37: } finally { tikhomirov@295: if (da != null) { tikhomirov@295: da.done(); tikhomirov@295: } tikhomirov@512: lifecycleTearDown(lifecycle); tikhomirov@37: } tikhomirov@37: } tikhomirov@37: tikhomirov@423: /** tikhomirov@423: * @param inspector callback to visit manifest entries tikhomirov@423: * @throws HgRuntimeException subclass thereof to indicate issues with the library. Runtime exception tikhomirov@423: * @throws IllegalArgumentException if inspector argument is null tikhomirov@423: */ tikhomirov@423: public void inspectManifest(Inspector inspector) throws HgRuntimeException { tikhomirov@169: if (inspector == null) { tikhomirov@169: throw new IllegalArgumentException(); tikhomirov@169: } tikhomirov@512: final Lifecycle lifecycle = lifecycleSetUp(inspector); tikhomirov@295: DataAccess da = null; tikhomirov@36: try { tikhomirov@295: da = getDataStream(); tikhomirov@169: if (da.isEmpty()) { tikhomirov@169: return; tikhomirov@169: } tikhomirov@169: skipGroup(da); // changelog tikhomirov@182: internalInspectManifest(da, inspector); tikhomirov@295: } catch (IOException ex) { tikhomirov@295: throw new HgInvalidFileException("Bundle.inspectManifest failed", ex, bundleFile); tikhomirov@36: } finally { tikhomirov@295: if (da != null) { tikhomirov@295: da.done(); tikhomirov@295: } tikhomirov@512: lifecycleTearDown(lifecycle); tikhomirov@36: } tikhomirov@36: } tikhomirov@36: tikhomirov@423: /** tikhomirov@423: * @param inspector callback to visit file entries tikhomirov@423: * @throws HgRuntimeException subclass thereof to indicate issues with the library. Runtime exception tikhomirov@423: * @throws IllegalArgumentException if inspector argument is null tikhomirov@423: */ tikhomirov@423: public void inspectFiles(Inspector inspector) throws HgRuntimeException { tikhomirov@169: if (inspector == null) { tikhomirov@169: throw new IllegalArgumentException(); tikhomirov@169: } tikhomirov@512: final Lifecycle lifecycle = lifecycleSetUp(inspector); tikhomirov@295: DataAccess da = null; tikhomirov@169: try { tikhomirov@295: da = getDataStream(); tikhomirov@182: if (da.isEmpty()) { tikhomirov@182: return; tikhomirov@169: } tikhomirov@182: skipGroup(da); // changelog tikhomirov@182: if (da.isEmpty()) { tikhomirov@182: return; tikhomirov@169: } tikhomirov@182: skipGroup(da); // manifest tikhomirov@182: internalInspectFiles(da, inspector); tikhomirov@295: } catch (IOException ex) { tikhomirov@295: throw new HgInvalidFileException("Bundle.inspectFiles failed", ex, bundleFile); tikhomirov@169: } finally { tikhomirov@295: if (da != null) { tikhomirov@295: da.done(); tikhomirov@295: } tikhomirov@512: lifecycleTearDown(lifecycle); tikhomirov@169: } tikhomirov@169: } tikhomirov@169: tikhomirov@423: /** tikhomirov@423: * @param inspector visit complete bundle (changelog, manifest and file entries) tikhomirov@423: * @throws HgRuntimeException subclass thereof to indicate issues with the library. Runtime exception tikhomirov@423: * @throws IllegalArgumentException if inspector argument is null tikhomirov@423: */ tikhomirov@423: public void inspectAll(Inspector inspector) throws HgRuntimeException { tikhomirov@169: if (inspector == null) { tikhomirov@169: throw new IllegalArgumentException(); tikhomirov@169: } tikhomirov@512: final Lifecycle lifecycle = lifecycleSetUp(inspector); tikhomirov@295: DataAccess da = null; tikhomirov@169: try { tikhomirov@295: da = getDataStream(); tikhomirov@182: internalInspectChangelog(da, inspector); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: internalInspectManifest(da, inspector); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: internalInspectFiles(da, inspector); tikhomirov@295: } catch (IOException ex) { tikhomirov@295: throw new HgInvalidFileException("Bundle.inspectAll failed", ex, bundleFile); tikhomirov@169: } finally { tikhomirov@295: if (da != null) { tikhomirov@295: da.done(); tikhomirov@295: } tikhomirov@512: lifecycleTearDown(lifecycle); tikhomirov@169: } tikhomirov@169: } tikhomirov@512: tikhomirov@512: // initialize flowControl, check for concurrent usage, starts lifecyle, if any tikhomirov@512: // return non-null only if inspector is interested in lifecycle events tikhomirov@512: private Lifecycle lifecycleSetUp(Inspector inspector) throws ConcurrentModificationException { tikhomirov@512: // Don't need flowControl in case Inspector doesn't implement Lifecycle, tikhomirov@512: // however is handy not to expect it == null inside internalInspect* tikhomirov@512: // XXX Once there's need to make this class thread-safe, tikhomirov@512: // shall move flowControl to thread-local state. tikhomirov@512: if (flowControl != null) { tikhomirov@512: throw new ConcurrentModificationException("HgBundle is in use and not thread-safe yet"); tikhomirov@512: } tikhomirov@512: flowControl = new Lifecycle.BasicCallback(); tikhomirov@512: final Lifecycle lifecycle = Adaptable.Factory.getAdapter(inspector, Lifecycle.class, null); tikhomirov@512: if (lifecycle != null) { tikhomirov@512: lifecycle.start(-1, flowControl, flowControl); tikhomirov@512: } tikhomirov@512: return lifecycle; tikhomirov@512: } tikhomirov@512: tikhomirov@512: private void lifecycleTearDown(Lifecycle lifecycle) { tikhomirov@512: if (lifecycle != null) { tikhomirov@512: lifecycle.finish(flowControl); tikhomirov@512: } tikhomirov@512: flowControl = null; tikhomirov@512: } tikhomirov@169: tikhomirov@628: private void internalInspectChangelog(DataAccess da, Inspector inspector) throws IOException, HgRuntimeException { tikhomirov@182: if (da.isEmpty()) { tikhomirov@182: return; tikhomirov@182: } tikhomirov@182: inspector.changelogStart(); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: readGroup(da, inspector); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: inspector.changelogEnd(); tikhomirov@182: } tikhomirov@182: tikhomirov@628: private void internalInspectManifest(DataAccess da, Inspector inspector) throws IOException, HgRuntimeException { tikhomirov@182: if (da.isEmpty()) { tikhomirov@182: return; tikhomirov@182: } tikhomirov@182: inspector.manifestStart(); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: readGroup(da, inspector); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: inspector.manifestEnd(); tikhomirov@182: } tikhomirov@182: tikhomirov@628: private void internalInspectFiles(DataAccess da, Inspector inspector) throws IOException, HgRuntimeException { tikhomirov@182: while (!da.isEmpty()) { tikhomirov@182: int fnameLen = da.readInt(); tikhomirov@182: if (fnameLen <= 4) { tikhomirov@182: break; // null chunk, the last one. tikhomirov@182: } tikhomirov@182: byte[] fnameBuf = new byte[fnameLen - 4]; tikhomirov@182: da.readBytes(fnameBuf, 0, fnameBuf.length); tikhomirov@182: String name = new String(fnameBuf); tikhomirov@182: inspector.fileStart(name); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: readGroup(da, inspector); tikhomirov@513: if (flowControl.isStopped()) { tikhomirov@513: return; tikhomirov@513: } tikhomirov@182: inspector.fileEnd(name); tikhomirov@182: } tikhomirov@182: } tikhomirov@182: tikhomirov@628: private static void readGroup(DataAccess da, Inspector inspector) throws IOException, HgRuntimeException { tikhomirov@36: int len = da.readInt(); tikhomirov@169: boolean good2go = true; tikhomirov@532: Nodeid prevNodeid = Nodeid.NULL; tikhomirov@169: while (len > 4 && !da.isEmpty() && good2go) { tikhomirov@36: byte[] nb = new byte[80]; tikhomirov@36: da.readBytes(nb, 0, 80); tikhomirov@169: int dataLength = len - 84 /* length field + 4 nodeids */; tikhomirov@169: byte[] data = new byte[dataLength]; tikhomirov@169: da.readBytes(data, 0, dataLength); tikhomirov@169: DataAccess slice = new ByteArrayDataAccess(data); // XXX in fact, may pass a slicing DataAccess. tikhomirov@169: // Just need to make sure that we seek to proper location afterwards (where next GroupElement starts), tikhomirov@169: // regardless whether that slice has read it or not. tikhomirov@532: GroupElement ge = new GroupElement(nb, prevNodeid, slice); tikhomirov@169: good2go = inspector.element(ge); tikhomirov@170: slice.done(); // BADA doesn't implement done(), but it could (e.g. free array) tikhomirov@170: /// and we'd better tell it we are not going to use it any more. However, it's important to ensure Inspector tikhomirov@170: // implementations out there do not retain GroupElement.rawData() tikhomirov@532: prevNodeid = ge.node(); tikhomirov@36: len = da.isEmpty() ? 0 : da.readInt(); tikhomirov@36: } tikhomirov@169: // need to skip up to group end if inspector told he don't want to continue with the group, tikhomirov@169: // because outer code may try to read next group immediately as we return back. tikhomirov@169: while (len > 4 && !da.isEmpty()) { tikhomirov@169: da.skip(len - 4 /* length field */); tikhomirov@169: len = da.isEmpty() ? 0 : da.readInt(); tikhomirov@169: } tikhomirov@36: } tikhomirov@36: tikhomirov@169: private static void skipGroup(DataAccess da) throws IOException { tikhomirov@169: int len = da.readInt(); tikhomirov@169: while (len > 4 && !da.isEmpty()) { tikhomirov@169: da.skip(len - 4); // sizeof(int) tikhomirov@169: len = da.isEmpty() ? 0 : da.readInt(); tikhomirov@169: } tikhomirov@169: } tikhomirov@169: tikhomirov@358: @Experimental(reason="Cumbersome API, rawData and apply with byte[] perhaps need replacement with ByteChannel/ByteBuffer, and better Exceptions. Perhaps, shall split into interface and impl") tikhomirov@169: public static class GroupElement { tikhomirov@169: private final byte[] header; // byte[80] takes 120 bytes, 4 Nodeids - 192 tikhomirov@169: private final DataAccess dataAccess; tikhomirov@329: private Patch patches; tikhomirov@532: private final Nodeid deltaBase; tikhomirov@169: tikhomirov@532: GroupElement(byte[] fourNodeids, Nodeid deltaBaseRev, DataAccess rawDataAccess) { tikhomirov@36: assert fourNodeids != null && fourNodeids.length == 80; tikhomirov@36: header = fourNodeids; tikhomirov@532: deltaBase = deltaBaseRev; tikhomirov@169: dataAccess = rawDataAccess; tikhomirov@36: } tikhomirov@169: tikhomirov@358: /** tikhomirov@358: * node field of the group element tikhomirov@358: * @return node revision, never null tikhomirov@358: */ tikhomirov@36: public Nodeid node() { tikhomirov@36: return Nodeid.fromBinary(header, 0); tikhomirov@36: } tikhomirov@169: tikhomirov@358: /** tikhomirov@358: * p1 (parent 1) field of the group element tikhomirov@358: * @return revision of parent 1, never null tikhomirov@358: */ tikhomirov@36: public Nodeid firstParent() { tikhomirov@36: return Nodeid.fromBinary(header, 20); tikhomirov@36: } tikhomirov@169: tikhomirov@358: /** tikhomirov@358: * p2 (parent 2) field of the group element tikhomirov@358: * @return revision of parent 2, never null tikhomirov@358: */ tikhomirov@36: public Nodeid secondParent() { tikhomirov@36: return Nodeid.fromBinary(header, 40); tikhomirov@36: } tikhomirov@169: tikhomirov@358: /** tikhomirov@358: * cs (changeset link) field of the group element tikhomirov@358: * @return changeset revision, never null tikhomirov@358: */ tikhomirov@358: public Nodeid cset() { tikhomirov@36: return Nodeid.fromBinary(header, 60); tikhomirov@36: } tikhomirov@358: tikhomirov@532: /** tikhomirov@532: * Revision this element keeps patches against. For the patches of the very first revision returns {@link Nodeid#NULL}. tikhomirov@532: * @return revision of delta base, never null tikhomirov@532: */ tikhomirov@532: public Nodeid patchBase() { tikhomirov@532: return deltaBase; tikhomirov@532: } tikhomirov@532: tikhomirov@358: public byte[] rawDataByteArray() throws IOException { // XXX IOException or HgInvalidFileException? tikhomirov@358: return rawData().byteArray(); tikhomirov@358: } tikhomirov@358: tikhomirov@358: public byte[] apply(byte[] baseContent) throws IOException { tikhomirov@358: return apply(new ByteArrayDataAccess(baseContent)); tikhomirov@358: } tikhomirov@169: tikhomirov@358: /*package-local*/ DataAccess rawData() { tikhomirov@169: return dataAccess; tikhomirov@169: } tikhomirov@169: tikhomirov@329: /*package-local*/ Patch patch() throws IOException { tikhomirov@169: if (patches == null) { tikhomirov@169: dataAccess.reset(); tikhomirov@329: patches = new Patch(); tikhomirov@329: patches.read(dataAccess); tikhomirov@169: } tikhomirov@169: return patches; tikhomirov@169: } tikhomirov@169: tikhomirov@358: /*package-local*/ byte[] apply(DataAccess baseContent) throws IOException { tikhomirov@329: return patch().apply(baseContent, -1); tikhomirov@169: } tikhomirov@357: tikhomirov@357: public String toString() { tikhomirov@357: int patchCount; tikhomirov@357: try { tikhomirov@357: patchCount = patch().count(); tikhomirov@357: } catch (IOException ex) { tikhomirov@357: ex.printStackTrace(); tikhomirov@357: patchCount = -1; tikhomirov@357: } tikhomirov@357: return String.format("%s %s %s %s; patches:%d\n", node().shortNotation(), firstParent().shortNotation(), secondParent().shortNotation(), cset().shortNotation(), patchCount); tikhomirov@357: } tikhomirov@36: } tikhomirov@645: tikhomirov@645: @Experimental(reason="Work in progress, not an API") tikhomirov@645: public class BundleSerializer implements DataSerializer.DataSource { tikhomirov@645: tikhomirov@645: public void serialize(DataSerializer out) throws HgIOException, HgRuntimeException { tikhomirov@645: FileInputStream fis = null; tikhomirov@645: try { tikhomirov@645: fis = new FileInputStream(HgBundle.this.bundleFile); tikhomirov@645: byte[] buffer = new byte[8*1024]; tikhomirov@645: int r; tikhomirov@645: while ((r = fis.read(buffer, 0, buffer.length)) > 0) { tikhomirov@645: out.write(buffer, 0, r); tikhomirov@645: } tikhomirov@645: tikhomirov@645: } catch (IOException ex) { tikhomirov@645: throw new HgIOException("Failed to serialize bundle", HgBundle.this.bundleFile); tikhomirov@645: } finally { tikhomirov@654: new FileUtils(HgBundle.this.ctx.getLog(), this).closeQuietly(fis, HgBundle.this.bundleFile); tikhomirov@645: } tikhomirov@645: } tikhomirov@645: tikhomirov@645: public int serializeLength() throws HgRuntimeException { tikhomirov@645: return Internals.ltoi(HgBundle.this.bundleFile.length()); tikhomirov@645: } tikhomirov@645: } tikhomirov@36: }