Author: frm
Date: Wed Sep  5 08:57:56 2018
New Revision: 1840106

URL: http://svn.apache.org/viewvc?rev=1840106&view=rev
Log:
OAK-7721 - Check for too big records when allocating space

Backport r1840024 from trunk.

Modified:
    jackrabbit/oak/branches/1.6/   (props changed)
    
jackrabbit/oak/branches/1.6/oak-segment-tar/src/main/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriter.java
    
jackrabbit/oak/branches/1.6/oak-segment-tar/src/test/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriterTest.java

Propchange: jackrabbit/oak/branches/1.6/
------------------------------------------------------------------------------
--- svn:mergeinfo (original)
+++ svn:mergeinfo Wed Sep  5 08:57:56 2018
@@ -1,3 +1,3 @@
 /jackrabbit/oak/branches/1.0:1665962
-/jackrabbit/oak/trunk:1781068,1781075,1781248,1781386,1781846,1781907,1782000,1782029,1782196,1782447,1782476,1782770,1782945,1782966,1782973,1782990,1783061,1783066,1783089,1783104-1783105,1783110,1783619,1783720,1783731,1783733,1783738,1783742,1783773,1783855,1783891,1784023,1784034,1784130,1784162,1784251,1784401,1784551,1784574,1784689,1785095,1785108,1785161,1785172,1785283,1785652,1785838,1785916-1785917,1785919,1785946,1786122,1787074,1787145,1787151,1787217,1787425,1788056,1788378,1788387-1788389,1788463,1788476,1788850,1789056,1789534,1789925,1789940,1789987,1790006,1790013,1790069,1790077,1790079,1790382,1790502-1790503,1792049,1792463,1792742,1792746,1793013,1793088,1793618,1793627,1793644,1794393,1794417,1794683,1795138,1795314,1795330,1795475,1795488,1795491,1795502,1795594,1795613,1795618,1796144,1796230,1796239,1796274,1796278,1796988,1797378,1798035,1798832,1798834,1799219,1799389,1799393,1799924,1800244,1800269,1800606,1800613,1800974,1801011,1801013,1801118-1801119
 
,1801675,1802260,1802262,1802286,1802548,1802905,1802934,1802938,1802973,1803026,1803247-1803249,1803951,1803953-1803955,1804437,1805851-1805852,1806668,1807308,1807688,1808022,1808125,1808128,1808142,1808240,1808246,1809024,1809026,1809131,1809163,1809178-1809179,1809253,1809255-1809256,1809289,1809745,1811071-1811072,1811155,1811380,1811655,1811952,1811963,1811986,1813192,1813538,1814189,1814332,1814397,1815201,1815438,1815926,1817326,1817919,1817987-1817988,1817990,1818038,1818042,1818056,1818124,1818137,1818554,1818576,1818645,1819048,1819050,1821237,1821325,1821358,1821495,1821516,1821847,1822207,1822850,1823172,1823655,1824896,1825471,1825654,1826237,1826338,1826532,1826640,1826932,1826957,1827472,1827486,1827977,1828502,1829527,1829569,1829587,1829665,1829854,1829864,1829987,1829998,1830019,1830160,1830239,1830748,1831190,1831374,1832379,1832535,1833308,1834648-1834649,1834681,1835060,1837475,1837998,1838637,1839746
+/jackrabbit/oak/trunk:1781068,1781075,1781248,1781386,1781846,1781907,1782000,1782029,1782196,1782447,1782476,1782770,1782945,1782966,1782973,1782990,1783061,1783066,1783089,1783104-1783105,1783110,1783619,1783720,1783731,1783733,1783738,1783742,1783773,1783855,1783891,1784023,1784034,1784130,1784162,1784251,1784401,1784551,1784574,1784689,1785095,1785108,1785161,1785172,1785283,1785652,1785838,1785916-1785917,1785919,1785946,1786122,1787074,1787145,1787151,1787217,1787425,1788056,1788378,1788387-1788389,1788463,1788476,1788850,1789056,1789534,1789925,1789940,1789987,1790006,1790013,1790069,1790077,1790079,1790382,1790502-1790503,1792049,1792463,1792742,1792746,1793013,1793088,1793618,1793627,1793644,1794393,1794417,1794683,1795138,1795314,1795330,1795475,1795488,1795491,1795502,1795594,1795613,1795618,1796144,1796230,1796239,1796274,1796278,1796988,1797378,1798035,1798832,1798834,1799219,1799389,1799393,1799924,1800244,1800269,1800606,1800613,1800974,1801011,1801013,1801118-1801119
 
,1801675,1802260,1802262,1802286,1802548,1802905,1802934,1802938,1802973,1803026,1803247-1803249,1803951,1803953-1803955,1804437,1805851-1805852,1806668,1807308,1807688,1808022,1808125,1808128,1808142,1808240,1808246,1809024,1809026,1809131,1809163,1809178-1809179,1809253,1809255-1809256,1809289,1809745,1811071-1811072,1811155,1811380,1811655,1811952,1811963,1811986,1813192,1813538,1814189,1814332,1814397,1815201,1815438,1815926,1817326,1817919,1817987-1817988,1817990,1818038,1818042,1818056,1818124,1818137,1818554,1818576,1818645,1819048,1819050,1821237,1821325,1821358,1821495,1821516,1821847,1822207,1822850,1823172,1823655,1824896,1825471,1825654,1826237,1826338,1826532,1826640,1826932,1826957,1827472,1827486,1827977,1828502,1829527,1829569,1829587,1829665,1829854,1829864,1829987,1829998,1830019,1830160,1830239,1830748,1831190,1831374,1832379,1832535,1833308,1834648-1834649,1834681,1835060,1837475,1837998,1838637,1839746,1840024
 /jackrabbit/trunk:1345480

Modified: 
jackrabbit/oak/branches/1.6/oak-segment-tar/src/main/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriter.java
URL: 
http://svn.apache.org/viewvc/jackrabbit/oak/branches/1.6/oak-segment-tar/src/main/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriter.java?rev=1840106&r1=1840105&r2=1840106&view=diff
==============================================================================
--- 
jackrabbit/oak/branches/1.6/oak-segment-tar/src/main/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriter.java
 (original)
+++ 
jackrabbit/oak/branches/1.6/oak-segment-tar/src/main/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriter.java
 Wed Sep  5 08:57:56 2018
@@ -470,15 +470,36 @@ public class SegmentBufferWriter impleme
             segmentSize = align(headerSize + recordSize + length, 16);
         }
 
+        // If the resulting segment buffer would be too big we need to allocate
+        // additional space. Allocating additional space is a recursive
+        // operation guarded by the `dirty` flag. The recursion can iterate at
+        // most two times. The base case happens when the `dirty` flag is
+        // `false`: the current buffer is empty, the record is too big to fit 
in
+        // an empty segment, and we fail with an `IllegalArgumentException`. 
The
+        // recursive step happens when the `dirty` flag is `true`:
+        // the current buffer is non-empty, we flush it, allocate a new buffer
+        // for an empty segment, and invoke `prepare()` once more.
+
         if (segmentSize > buffer.length) {
-            flush();
+            if (dirty) {
+                LOG.debug("Flushing full segment {} (headerSize={}, 
recordSize={}, length={}, segmentSize={})",
+                    segment.getSegmentId(), headerSize, recordSize, length, 
segmentSize);
+                flush();
+                return prepare(type, size, ids);
+            }
+            throw new IllegalArgumentException(String.format(
+                "Record too big: type=%s, size=%s, recordIds=%s, total=%s",
+                type,
+                size,
+                ids.size(),
+                recordSize
+            ));
         }
 
         statistics.recordCount++;
 
         length += recordSize;
         position = buffer.length - length;
-        checkState(position >= 0);
 
         int recordNumber = recordNumbers.addRecord(type, position);
         return new RecordId(segment.getSegmentId(), recordNumber);

Modified: 
jackrabbit/oak/branches/1.6/oak-segment-tar/src/test/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriterTest.java
URL: 
http://svn.apache.org/viewvc/jackrabbit/oak/branches/1.6/oak-segment-tar/src/test/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriterTest.java?rev=1840106&r1=1840105&r2=1840106&view=diff
==============================================================================
--- 
jackrabbit/oak/branches/1.6/oak-segment-tar/src/test/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriterTest.java
 (original)
+++ 
jackrabbit/oak/branches/1.6/oak-segment-tar/src/test/java/org/apache/jackrabbit/oak/segment/SegmentBufferWriterTest.java
 Wed Sep  5 08:57:56 2018
@@ -24,6 +24,7 @@ import static org.junit.Assert.assertEqu
 import static org.junit.Assert.assertNotEquals;
 
 import java.io.File;
+import java.util.ArrayList;
 import java.util.List;
 
 import org.apache.jackrabbit.oak.segment.file.FileStore;
@@ -95,4 +96,43 @@ public class SegmentBufferWriterTest {
         assertNotEquals(before, after);
     }
 
+    @Test
+    public void tooBigRecord() throws Exception {
+
+        // See OAK-7721 to understand why this test exists.
+
+        try (FileStore store = openFileStore()) {
+
+            // Please don't change anything from the following statement yet.
+            // Read the next comment to understand why.
+
+            SegmentBufferWriter writer = new SegmentBufferWriter(
+                store,
+                store.getTracker().getSegmentCounter(),
+                store.getReader(),
+                "t",
+                store.getRevisions().getHead().getSegment().getGcGeneration()
+            );
+
+            // The size of the record is chosen with the precise intention to
+            // fool `writer` into having enough space to write the record. In
+            // particular, at the end of `prepare()`, `writer` will have
+            // `this.length = 262144`, which is `MAX_SEGMENT_SIZE`, and
+            // `this.position = 0`. This result is particularly sensitive to 
the
+            // initial content of the segment, which in turn is influenced by
+            // the segment info. Try to change the writer ID in the constructor
+            // of `SegmentBufferWriter` to a longer string, and you will have
+            // `prepare()` throw ISEs because the writer ID is embedded in the
+            // segment info.
+
+            IllegalArgumentException error = null;
+            try {
+                writer.prepare(RecordType.BLOCK, 262101, new 
ArrayList<RecordId>());
+            } catch (IllegalArgumentException e) {
+                error = e;
+            }
+            assertEquals("Record too big: type=BLOCK, size=262101, 
recordIds=0, total=262104", error != null ? error.getMessage() : null);
+        }
+    }
+
 }


Reply via email to