Skip to content

Commit 079c414

Browse files
leif-scalityleif-scality
committed
CLDSRV-873: return checksum from GetObject
1 parent d9fe250 commit 079c414

File tree

3 files changed

+270
-2
lines changed

3 files changed

+270
-2
lines changed

‎lib/api/objectGet.js‎

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,12 @@ function objectGet(authInfo, request, returnTagCount, log, callback) {
4141
const bucketName = request.bucketName;
4242
const objectKey = request.objectKey;
4343

44+
const checksumMode = request.headers['x-amz-checksum-mode'];
45+
if (checksumMode !== undefined && checksumMode !== 'ENABLED') {
46+
log.debug('invalid x-amz-checksum-mode', { checksumMode });
47+
return callback(errors.InvalidArgument);
48+
}
49+
4450
// returns name of location to get from and key if successful
4551
const locCheckResult =
4652
locationHeaderCheck(request.headers, objectKey, bucketName);
@@ -145,6 +151,7 @@ function objectGet(authInfo, request, returnTagCount, log, callback) {
145151
request.serverAccessLog.objectSize = objLength;
146152
}
147153
let byteRange;
154+
let partNumber = null;
148155
const streamingParams = {};
149156
if (request.headers.range) {
150157
const { range, error } = parseRange(request.headers.range,
@@ -211,8 +218,6 @@ function objectGet(authInfo, request, returnTagCount, log, callback) {
211218
if (dataLocator[0] && dataLocator[0].dataStoreType === 'azure') {
212219
dataLocator[0].azureStreamingOptions = streamingParams;
213220
}
214-
215-
let partNumber = null;
216221
if (request.query && request.query.partNumber !== undefined) {
217222
if (byteRange) {
218223
const error = errorInstances.InvalidRequest
@@ -301,6 +306,16 @@ function objectGet(authInfo, request, returnTagCount, log, callback) {
301306
dataLocator = setPartRanges(dataLocator, byteRange);
302307
}
303308
}
309+
// Checksum is not returned for partNumber requests because per-part
310+
// checksums are not yet stored (S3C-11073).
311+
if (checksumMode === 'ENABLED' && !byteRange && !partNumber) {
312+
const checksum = objMD.checksum;
313+
if (checksum) {
314+
responseMetaHeaders[`x-amz-checksum-${checksum.checksumAlgorithm}`]
315+
= checksum.checksumValue;
316+
responseMetaHeaders['x-amz-checksum-type'] = checksum.checksumType;
317+
}
318+
}
304319
// Check KMS Key access and usability before checking data
305320
// diff with AWS: for empty object (no dataLocator) KMS not checked
306321
return async.each(dataLocator || [],

‎tests/functional/aws-node-sdk/test/object/get.js‎

Lines changed: 98 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,13 @@ const {
2222
const withV4 = require('../support/withV4');
2323
const BucketUtility = require('../../lib/utility/bucket-util');
2424
const changeObjectLock = require('../../../../utilities/objectLock-util');
25+
const { algorithms } = require('../../../../../lib/api/apiUtils/integrity/validateChecksums');
26+
27+
const { crc64NvmeCrtContainer } = require('@aws-sdk/middleware-flexible-checksums');
28+
if (crc64NvmeCrtContainer) {
29+
const { CrtCrc64Nvme } = require('@aws-sdk/crc64-nvme-crt');
30+
crc64NvmeCrtContainer.CrtCrc64Nvme = CrtCrc64Nvme;
31+
}
2532

2633
const changeLockPromise = promisify(changeObjectLock);
2734

@@ -1121,3 +1128,94 @@ describeSkipIfCeph('GET object with object lock', () => {
11211128
});
11221129
});
11231130
});
1131+
1132+
describe('GET object checksum mode', () => {
1133+
withV4(sigCfg => {
1134+
let bucketUtil;
1135+
let s3;
1136+
const checksumBucket = 'checksum-getobject-test';
1137+
const checksumKey = 'checksum-test-object';
1138+
const body = Buffer.from('checksum test body');
1139+
1140+
const expectedDigests = {};
1141+
1142+
const checksumAlgorithms = [
1143+
{ algorithm: 'SHA256', responseField: 'ChecksumSHA256', internalName: 'sha256' },
1144+
{ algorithm: 'SHA1', responseField: 'ChecksumSHA1', internalName: 'sha1' },
1145+
{ algorithm: 'CRC32', responseField: 'ChecksumCRC32', internalName: 'crc32' },
1146+
{ algorithm: 'CRC32C', responseField: 'ChecksumCRC32C', internalName: 'crc32c' },
1147+
{ algorithm: 'CRC64NVME', responseField: 'ChecksumCRC64NVME', internalName: 'crc64nvme' },
1148+
];
1149+
1150+
before(async () => {
1151+
bucketUtil = new BucketUtility('default', sigCfg);
1152+
s3 = bucketUtil.s3;
1153+
await s3.send(new CreateBucketCommand({ Bucket: checksumBucket }));
1154+
1155+
for (const { internalName } of checksumAlgorithms) {
1156+
expectedDigests[internalName] =
1157+
await algorithms[internalName].digest(body);
1158+
}
1159+
});
1160+
1161+
after(async () => {
1162+
await bucketUtil.empty(checksumBucket);
1163+
await s3.send(new DeleteBucketCommand({ Bucket: checksumBucket }));
1164+
});
1165+
1166+
checksumAlgorithms.forEach(({ algorithm, responseField, internalName }) => {
1167+
it(`should return ${responseField} and ChecksumType when ChecksumMode is ENABLED`, async () => {
1168+
const putRes = await s3.send(new PutObjectCommand({
1169+
Bucket: checksumBucket,
1170+
Key: checksumKey,
1171+
Body: body,
1172+
ChecksumAlgorithm: algorithm,
1173+
}));
1174+
const storedChecksum = putRes[responseField];
1175+
assert(storedChecksum, `Expected ${responseField} in PutObject response`);
1176+
1177+
const getRes = await s3.send(new GetObjectCommand({
1178+
Bucket: checksumBucket,
1179+
Key: checksumKey,
1180+
ChecksumMode: 'ENABLED',
1181+
}));
1182+
assert.strictEqual(getRes[responseField], expectedDigests[internalName],
1183+
`${responseField} value mismatch`);
1184+
assert.strictEqual(getRes[responseField], storedChecksum);
1185+
assert.strictEqual(getRes.ChecksumType, 'FULL_OBJECT');
1186+
});
1187+
});
1188+
1189+
it('should not return checksum headers when ChecksumMode is not set', async () => {
1190+
await s3.send(new PutObjectCommand({
1191+
Bucket: checksumBucket,
1192+
Key: checksumKey,
1193+
Body: body,
1194+
ChecksumAlgorithm: 'SHA256',
1195+
}));
1196+
1197+
const getRes = await s3.send(new GetObjectCommand({
1198+
Bucket: checksumBucket,
1199+
Key: checksumKey,
1200+
}));
1201+
assert.strictEqual(getRes.ChecksumSHA256, undefined);
1202+
assert.strictEqual(getRes.ChecksumType, undefined);
1203+
});
1204+
1205+
it('should return an error when ChecksumMode is not ENABLED', async () => {
1206+
await s3.send(new PutObjectCommand({
1207+
Bucket: checksumBucket,
1208+
Key: checksumKey,
1209+
Body: body,
1210+
}));
1211+
1212+
await assert.rejects(
1213+
s3.send(new GetObjectCommand({
1214+
Bucket: checksumBucket,
1215+
Key: checksumKey,
1216+
ChecksumMode: 'DISABLED',
1217+
})),
1218+
);
1219+
});
1220+
});
1221+
});

‎tests/unit/api/objectGet.js‎

Lines changed: 155 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,9 @@ const assert = require('assert');
22
const async = require('async');
33
const crypto = require('crypto');
44
const { parseString } = require('xml2js');
5+
const { models } = require('arsenal');
6+
const { ObjectMD, ObjectMDChecksum } = models;
7+
const { algorithms } = require('../../../lib/api/apiUtils/integrity/validateChecksums');
58

69
const { bucketPut } = require('../../../lib/api/bucketPut');
710
const { cleanup, DummyRequestLogger, makeAuthInfo } = require('../helpers');
@@ -518,3 +521,155 @@ describe('objectGet API', () => {
518521
});
519522
});
520523
});
524+
525+
describe('objectGet API - x-amz-checksum-mode', () => {
526+
const checksumAlgorithms = [
527+
{ name: 'sha256', header: 'x-amz-checksum-sha256' },
528+
{ name: 'sha1', header: 'x-amz-checksum-sha1' },
529+
{ name: 'crc32', header: 'x-amz-checksum-crc32' },
530+
{ name: 'crc32c', header: 'x-amz-checksum-crc32c' },
531+
{ name: 'crc64nvme', header: 'x-amz-checksum-crc64nvme' },
532+
];
533+
534+
const expectedDigests = {};
535+
536+
before(done => {
537+
Promise.all(checksumAlgorithms.map(async ({ name }) => {
538+
expectedDigests[name] = await algorithms[name].digest(postBody);
539+
})).then(() => done(), done);
540+
});
541+
542+
beforeEach(() => cleanup());
543+
544+
checksumAlgorithms.forEach(({ name, header }) => {
545+
it(`should return ${header} and x-amz-checksum-type when mode is ENABLED`, done => {
546+
const md = new ObjectMD(mdColdHelper.baseMd)
547+
.setChecksum(new ObjectMDChecksum(name, expectedDigests[name], 'FULL_OBJECT'));
548+
mdColdHelper.putBucketMock(bucketName, null, () =>
549+
mdColdHelper.putObjectMock(bucketName, objectName, md, () => {
550+
const req = {
551+
bucketName,
552+
namespace,
553+
objectKey: objectName,
554+
headers: { 'x-amz-checksum-mode': 'ENABLED' },
555+
url: `/${bucketName}/${objectName}`,
556+
query: {},
557+
};
558+
objectGet(authInfo, req, false, log, (err, _locator, headers) => {
559+
assert.ifError(err);
560+
assert.strictEqual(headers[header], expectedDigests[name]);
561+
assert.strictEqual(headers['x-amz-checksum-type'], 'FULL_OBJECT');
562+
done();
563+
});
564+
}));
565+
});
566+
});
567+
568+
it('should not return checksum headers when mode is ENABLED but object has no checksum', done => {
569+
mdColdHelper.putBucketMock(bucketName, null, () =>
570+
mdColdHelper.putObjectMock(bucketName, objectName, undefined, () => {
571+
const req = {
572+
bucketName,
573+
namespace,
574+
objectKey: objectName,
575+
headers: { 'x-amz-checksum-mode': 'ENABLED' },
576+
url: `/${bucketName}/${objectName}`,
577+
query: {},
578+
};
579+
objectGet(authInfo, req, false, log, (err, _locator, headers) => {
580+
assert.ifError(err);
581+
checksumAlgorithms.forEach(({ header }) =>
582+
assert.strictEqual(headers[header], undefined));
583+
assert.strictEqual(headers['x-amz-checksum-type'], undefined);
584+
done();
585+
});
586+
}));
587+
});
588+
589+
it('should not return checksum headers when x-amz-checksum-mode is not set', done => {
590+
const md = new ObjectMD(mdColdHelper.baseMd)
591+
.setChecksum(new ObjectMDChecksum('sha256', expectedDigests.sha256, 'FULL_OBJECT'));
592+
mdColdHelper.putBucketMock(bucketName, null, () =>
593+
mdColdHelper.putObjectMock(bucketName, objectName, md, () => {
594+
const req = {
595+
bucketName,
596+
namespace,
597+
objectKey: objectName,
598+
headers: {},
599+
url: `/${bucketName}/${objectName}`,
600+
query: {},
601+
};
602+
objectGet(authInfo, req, false, log, (err, _locator, headers) => {
603+
assert.ifError(err);
604+
checksumAlgorithms.forEach(({ header }) =>
605+
assert.strictEqual(headers[header], undefined));
606+
assert.strictEqual(headers['x-amz-checksum-type'], undefined);
607+
done();
608+
});
609+
}));
610+
});
611+
612+
it('should return InvalidArgument when x-amz-checksum-mode is not ENABLED', done => {
613+
const req = {
614+
bucketName,
615+
namespace,
616+
objectKey: objectName,
617+
headers: { 'x-amz-checksum-mode': 'DISABLED' },
618+
url: `/${bucketName}/${objectName}`,
619+
query: {},
620+
};
621+
objectGet(authInfo, req, false, log, err => {
622+
assert.strictEqual(err.is.InvalidArgument, true);
623+
done();
624+
});
625+
});
626+
627+
it('should not return checksum headers when Range header is set', done => {
628+
const md = new ObjectMD(mdColdHelper.baseMd)
629+
.setChecksum(new ObjectMDChecksum('sha256', expectedDigests.sha256, 'FULL_OBJECT'));
630+
mdColdHelper.putBucketMock(bucketName, null, () =>
631+
mdColdHelper.putObjectMock(bucketName, objectName, md, () => {
632+
const req = {
633+
bucketName,
634+
namespace,
635+
objectKey: objectName,
636+
headers: {
637+
'x-amz-checksum-mode': 'ENABLED',
638+
range: 'bytes=0-3',
639+
},
640+
url: `/${bucketName}/${objectName}`,
641+
query: {},
642+
};
643+
objectGet(authInfo, req, false, log, (err, _locator, headers) => {
644+
assert.ifError(err);
645+
checksumAlgorithms.forEach(({ header }) =>
646+
assert.strictEqual(headers[header], undefined));
647+
assert.strictEqual(headers['x-amz-checksum-type'], undefined);
648+
done();
649+
});
650+
}));
651+
});
652+
653+
it('should not return checksum headers when partNumber is set', done => {
654+
const md = new ObjectMD(mdColdHelper.baseMd)
655+
.setChecksum(new ObjectMDChecksum('sha256', expectedDigests.sha256, 'FULL_OBJECT'));
656+
mdColdHelper.putBucketMock(bucketName, null, () =>
657+
mdColdHelper.putObjectMock(bucketName, objectName, md, () => {
658+
const req = {
659+
bucketName,
660+
namespace,
661+
objectKey: objectName,
662+
headers: { 'x-amz-checksum-mode': 'ENABLED' },
663+
url: `/${bucketName}/${objectName}`,
664+
query: { partNumber: '1' },
665+
};
666+
objectGet(authInfo, req, false, log, (err, _locator, headers) => {
667+
assert.ifError(err);
668+
checksumAlgorithms.forEach(({ header }) =>
669+
assert.strictEqual(headers[header], undefined));
670+
assert.strictEqual(headers['x-amz-checksum-type'], undefined);
671+
done();
672+
});
673+
}));
674+
});
675+
});

0 commit comments

Comments
 (0)