SanityChecker.java
/* ====================================================================
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.poi.hssf.usermodel;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
import java.util.List;
import org.apache.poi.hssf.model.InternalSheet;
import org.apache.poi.hssf.model.InternalWorkbook;
import org.apache.poi.hssf.record.*;
import org.apache.poi.hssf.record.aggregates.PageSettingsBlock;
/**
* Designed to check wither the records written actually make sense.
*/
public class SanityChecker {
static class CheckRecord {
Class<? extends RecordBase> record;
char occurance; // 1 = one time, M = 1..many times, * = 0..many, 0 = optional
private final boolean together;
public CheckRecord(Class<? extends RecordBase> record, char occurance) {
this(record, occurance, true);
}
/**
* @param record The record type to check
* @param occurance The occurance 1 = occurs once, M = occurs many times
* @param together
*/
public CheckRecord(Class<? extends RecordBase> record, char occurance, boolean together) {
this.record = record;
this.occurance = occurance;
this.together = together;
}
public Class<? extends RecordBase> getRecord() {
return record;
}
public boolean isRequired() {
return occurance == '1' || occurance == 'M';
}
public boolean isOptional() {
return occurance == '0' || occurance == '*';
}
public boolean isMany() {
return occurance == '*' || occurance == 'M';
}
public int match(List<? extends RecordBase> records, int recordIdx) {
int firstRecord = findFirstRecord(records, getRecord(), recordIdx);
if (isRequired()) {
return matchRequired(firstRecord, records, recordIdx);
}
return matchOptional(firstRecord, records, recordIdx);
}
private int matchOptional(int firstRecord, List<? extends RecordBase> records, int recordIdx) {
if (firstRecord == -1) {
return recordIdx;
}
return matchOneOrMany(records, firstRecord);
}
private int matchRequired(int firstRecord, List<? extends RecordBase> records, int recordIdx) {
assertNotEquals(-1, firstRecord, "Manditory record missing or out of order: " + record);
return matchOneOrMany(records, firstRecord);
}
private int matchOneOrMany(List<? extends RecordBase> records, int recordIdx) {
if (isZeroOrOne()) {
// check no other records
assertEquals(-1, findFirstRecord(records, getRecord(), recordIdx + 1),
"More than one record matched for " + getRecord().getName());
} else if (isZeroToMany()) {
if (together) {
int nextIdx = findFirstRecord(records, record, recordIdx + 1);
while (nextIdx != -1) {
assertEquals(recordIdx+1, nextIdx, "Records are not together " + record.getName());
recordIdx = nextIdx;
nextIdx = findFirstRecord(records, record, recordIdx + 1);
}
}
}
return recordIdx + 1;
}
private boolean isZeroToMany() {
return occurance == '*' || occurance == 'M';
}
private boolean isZeroOrOne() {
return occurance == '0' || occurance == '1';
}
}
CheckRecord[] workbookRecords = {
new CheckRecord(BOFRecord.class, '1'),
new CheckRecord(InterfaceHdrRecord.class, '1'),
new CheckRecord(MMSRecord.class, '1'),
new CheckRecord(InterfaceEndRecord.class, '1'),
new CheckRecord(WriteAccessRecord.class, '1'),
new CheckRecord(CodepageRecord.class, '1'),
new CheckRecord(DSFRecord.class, '1'),
new CheckRecord(TabIdRecord.class, '1'),
new CheckRecord(FnGroupCountRecord.class, '1'),
new CheckRecord(WindowProtectRecord.class, '1'),
new CheckRecord(ProtectRecord.class, '1'),
new CheckRecord(PasswordRev4Record.class, '1'),
new CheckRecord(WindowOneRecord.class, '1'),
new CheckRecord(BackupRecord.class, '1'),
new CheckRecord(HideObjRecord.class, '1'),
new CheckRecord(DateWindow1904Record.class, '1'),
new CheckRecord(PrecisionRecord.class, '1'),
new CheckRecord(RefreshAllRecord.class, '1'),
new CheckRecord(BookBoolRecord.class, '1'),
new CheckRecord(FontRecord.class, 'M'),
new CheckRecord(FormatRecord.class, 'M'),
new CheckRecord(ExtendedFormatRecord.class, 'M'),
new CheckRecord(StyleRecord.class, 'M'),
new CheckRecord(UseSelFSRecord.class, '1'),
new CheckRecord(BoundSheetRecord.class, 'M'),
new CheckRecord(CountryRecord.class, '1'),
new CheckRecord(SupBookRecord.class, '0'),
new CheckRecord(ExternSheetRecord.class, '0'),
new CheckRecord(NameRecord.class, '*'),
new CheckRecord(SSTRecord.class, '1'),
new CheckRecord(ExtSSTRecord.class, '1'),
new CheckRecord(EOFRecord.class, '1'),
};
CheckRecord[] sheetRecords = {
new CheckRecord(BOFRecord.class, '1'),
new CheckRecord(CalcModeRecord.class, '1'),
new CheckRecord(RefModeRecord.class, '1'),
new CheckRecord(IterationRecord.class, '1'),
new CheckRecord(DeltaRecord.class, '1'),
new CheckRecord(SaveRecalcRecord.class, '1'),
new CheckRecord(PrintHeadersRecord.class, '1'),
new CheckRecord(PrintGridlinesRecord.class, '1'),
new CheckRecord(GridsetRecord.class, '1'),
new CheckRecord(GutsRecord.class, '1'),
new CheckRecord(DefaultRowHeightRecord.class, '1'),
new CheckRecord(WSBoolRecord.class, '1'),
new CheckRecord(PageSettingsBlock.class, '1'),
new CheckRecord(DefaultColWidthRecord.class, '1'),
new CheckRecord(DimensionsRecord.class, '1'),
new CheckRecord(WindowTwoRecord.class, '1'),
new CheckRecord(SelectionRecord.class, '1'),
new CheckRecord(EOFRecord.class, '1')
};
private void checkWorkbookRecords(InternalWorkbook workbook) {
List<org.apache.poi.hssf.record.Record> records = workbook.getRecords();
assertTrue(records.get(0) instanceof BOFRecord);
assertTrue(records.get(records.size() - 1) instanceof EOFRecord);
checkRecordOrder(records, workbookRecords);
}
private void checkSheetRecords(InternalSheet sheet) {
List<RecordBase> records = sheet.getRecords();
assertTrue(records.get(0) instanceof BOFRecord);
assertTrue(records.get(records.size() - 1) instanceof EOFRecord);
checkRecordOrder(records, sheetRecords);
}
void checkHSSFWorkbook(HSSFWorkbook wb) {
checkWorkbookRecords(wb.getWorkbook());
for (int i = 0; i < wb.getNumberOfSheets(); i++)
checkSheetRecords(wb.getSheetAt(i).getSheet());
}
static int findFirstRecord(List<? extends RecordBase> records, Class<? extends RecordBase> record, int startIndex) {
for (int i = startIndex; i < records.size(); i++) {
if (record.getName().equals(records.get(i).getClass().getName()))
return i;
}
return -1;
}
void checkRecordOrder(List<? extends RecordBase> records, CheckRecord[] check) {
int recordIdx = 0;
for (CheckRecord element : check) {
recordIdx = element.match(records, recordIdx);
}
}
}