thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-12-11 17:02:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix segfault in formatRow function

Browse Source
This commit is contained in:
avogar 2021-11-02 16:26:14 +03:00
parent 901ebcede6
commit 62c5951dd7
3 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/Functions/formatRow.cpp
View File

@ -8,6 +8,7 @@
#include <IO/WriteBufferFromVector.h>
#include <IO/WriteHelpers.h>
#include <Processors/Formats/IOutputFormat.h>
#include <Processors/Formats/IRowOutputFormat.h>
#include <base/map.h>
@ -18,6 +19,7 @@ namespace ErrorCodes
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
extern const int NUMBER_OF_ARGUMENTS_DOESNT_MATCH;
extern const int UNKNOWN_FORMAT;
extern const int BAD_ARGUMENTS;
}
namespace
@ -70,6 +72,11 @@ public:
writeChar('\0', buffer);
offsets[row] = buffer.count();
});
/// This function make sense only for row output formats.
if (!dynamic_cast<IRowOutputFormat *>(out.get()))
throw Exception(ErrorCodes::BAD_ARGUMENTS, "Cannot turn rows into a {} format strings. {} function supports only row output formats", format_name, getName());
out->write(arg_columns);
return col_str;
}

0
tests/queries/0_stateless/02113_format_row_bug.reference Normal file
View File

5
tests/queries/0_stateless/02113_format_row_bug.sql Normal file
View File

@ -0,0 +1,5 @@
select formatRow('ORC', number, toDate(number)) from numbers(5); -- { serverError 36 }
select formatRow('Parquet', number, toDate(number)) from numbers(5); -- { serverError 36 }
select formatRow('Arrow', number, toDate(number)) from numbers(5); -- { serverError 36 }
select formatRow('Native', number, toDate(number)) from numbers(5); -- { serverError 36 }