added long commands --memory= and --memlimit-decompress=

dev
Yann Collet 2016-10-14 14:22:32 -07:00
parent 1122349ac2
commit d7b120ab5c
2 changed files with 12 additions and 0 deletions

View File

@ -281,6 +281,16 @@ int main(int argCount, const char* argv[])
memLimit = readU32FromChar(&argument);
continue;
}
if (!strncmp(argument, "--memory=", strlen("--memory="))) {
argument += strlen("--memory=");
memLimit = readU32FromChar(&argument);
continue;
}
if (!strncmp(argument, "--memlimit-decompress=", strlen("--memlimit-decompress="))) {
argument += strlen("--memlimit-decompress=");
memLimit = readU32FromChar(&argument);
continue;
}
/* '-' means stdin/stdout */
if (!strcmp(argument, "-")){

View File

@ -84,6 +84,8 @@ $ZSTD -d - < tmp.zst > $INTOVOID
$ECHO "test : impose memory limitation (must fail)"
$ZSTD -d -f tmp.zst -M2K -c > $INTOVOID && die "decompression needs more memory than allowed"
$ZSTD -d -f tmp.zst --memlimit=2K -c > $INTOVOID && die "decompression needs more memory than allowed" # long command
$ZSTD -d -f tmp.zst --memory=2K -c > $INTOVOID && die "decompression needs more memory than allowed" # long command
$ZSTD -d -f tmp.zst --memlimit-decompress=2K -c > $INTOVOID && die "decompression needs more memory than allowed" # long command
$ECHO "test : overwrite protection"
$ZSTD -q tmp && die "overwrite check failed!"
$ECHO "test : force overwrite"