mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-03 13:02:00 +00:00
Eventually merged with master
This commit is contained in:
commit
062034dfd8
3
.gitmodules
vendored
3
.gitmodules
vendored
@ -103,3 +103,6 @@
|
|||||||
[submodule "contrib/fastops"]
|
[submodule "contrib/fastops"]
|
||||||
path = contrib/fastops
|
path = contrib/fastops
|
||||||
url = https://github.com/ClickHouse-Extras/fastops
|
url = https://github.com/ClickHouse-Extras/fastops
|
||||||
|
[submodule "contrib/orc"]
|
||||||
|
path = contrib/orc
|
||||||
|
url = https://github.com/apache/orc
|
||||||
|
20
CHANGELOG.md
20
CHANGELOG.md
@ -1,3 +1,13 @@
|
|||||||
|
## ClickHouse release 19.13.3.26, 2019-08-22
|
||||||
|
|
||||||
|
### Bug Fix
|
||||||
|
* Fix `ALTER TABLE ... UPDATE` query for tables with `enable_mixed_granularity_parts=1`. [#6543](https://github.com/yandex/ClickHouse/pull/6543) ([alesapin](https://github.com/alesapin))
|
||||||
|
* Fix NPE when using IN clause with a subquery with a tuple. [#6125](https://github.com/yandex/ClickHouse/issues/6125) [#6550](https://github.com/yandex/ClickHouse/pull/6550) ([tavplubix](https://github.com/tavplubix))
|
||||||
|
* Fixed an issue that if a stale replica becomes alive, it may still have data parts that were removed by DROP PARTITION. [#6522](https://github.com/yandex/ClickHouse/issues/6522) [#6523](https://github.com/yandex/ClickHouse/pull/6523) ([tavplubix](https://github.com/tavplubix))
|
||||||
|
* Fixed issue with parsing CSV [#6426](https://github.com/yandex/ClickHouse/issues/6426) [#6559](https://github.com/yandex/ClickHouse/pull/6559) ([tavplubix](https://github.com/tavplubix))
|
||||||
|
* Fixed data race in system.parts table and ALTER query. This fixes [#6245](https://github.com/yandex/ClickHouse/issues/6245). [#6513](https://github.com/yandex/ClickHouse/pull/6513) ([alexey-milovidov](https://github.com/alexey-milovidov))
|
||||||
|
* Fixed wrong code in mutations that may lead to memory corruption. Fixed segfault with read of address `0x14c0` that may happed due to concurrent `DROP TABLE` and `SELECT` from `system.parts` or `system.parts_columns`. Fixed race condition in preparation of mutation queries. Fixed deadlock caused by `OPTIMIZE` of Replicated tables and concurrent modification operations like ALTERs. [#6514](https://github.com/yandex/ClickHouse/pull/6514) ([alexey-milovidov](https://github.com/alexey-milovidov))
|
||||||
|
|
||||||
## ClickHouse release 19.13.2.19, 2019-08-14
|
## ClickHouse release 19.13.2.19, 2019-08-14
|
||||||
|
|
||||||
### New Feature
|
### New Feature
|
||||||
@ -31,6 +41,16 @@
|
|||||||
* Fix build with external `libcxx` [#6010](https://github.com/yandex/ClickHouse/pull/6010) ([Ivan](https://github.com/abyss7))
|
* Fix build with external `libcxx` [#6010](https://github.com/yandex/ClickHouse/pull/6010) ([Ivan](https://github.com/abyss7))
|
||||||
* Fix shared build with `rdkafka` library [#6101](https://github.com/yandex/ClickHouse/pull/6101) ([Ivan](https://github.com/abyss7))
|
* Fix shared build with `rdkafka` library [#6101](https://github.com/yandex/ClickHouse/pull/6101) ([Ivan](https://github.com/abyss7))
|
||||||
|
|
||||||
|
## ClickHouse release 19.11.8.46, 2019-08-22
|
||||||
|
|
||||||
|
### Bug Fix
|
||||||
|
* Fix `ALTER TABLE ... UPDATE` query for tables with `enable_mixed_granularity_parts=1`. [#6543](https://github.com/yandex/ClickHouse/pull/6543) ([alesapin](https://github.com/alesapin))
|
||||||
|
* Fix NPE when using IN clause with a subquery with a tuple. [#6125](https://github.com/yandex/ClickHouse/issues/6125) [#6550](https://github.com/yandex/ClickHouse/pull/6550) ([tavplubix](https://github.com/tavplubix))
|
||||||
|
* Fixed an issue that if a stale replica becomes alive, it may still have data parts that were removed by DROP PARTITION. [#6522](https://github.com/yandex/ClickHouse/issues/6522) [#6523](https://github.com/yandex/ClickHouse/pull/6523) ([tavplubix](https://github.com/tavplubix))
|
||||||
|
* Fixed issue with parsing CSV [#6426](https://github.com/yandex/ClickHouse/issues/6426) [#6559](https://github.com/yandex/ClickHouse/pull/6559) ([tavplubix](https://github.com/tavplubix))
|
||||||
|
* Fixed data race in system.parts table and ALTER query. This fixes [#6245](https://github.com/yandex/ClickHouse/issues/6245). [#6513](https://github.com/yandex/ClickHouse/pull/6513) ([alexey-milovidov](https://github.com/alexey-milovidov))
|
||||||
|
* Fixed wrong code in mutations that may lead to memory corruption. Fixed segfault with read of address `0x14c0` that may happed due to concurrent `DROP TABLE` and `SELECT` from `system.parts` or `system.parts_columns`. Fixed race condition in preparation of mutation queries. Fixed deadlock caused by `OPTIMIZE` of Replicated tables and concurrent modification operations like ALTERs. [#6514](https://github.com/yandex/ClickHouse/pull/6514) ([alexey-milovidov](https://github.com/alexey-milovidov))
|
||||||
|
|
||||||
## ClickHouse release 19.11.7.40, 2019-08-14
|
## ClickHouse release 19.11.7.40, 2019-08-14
|
||||||
|
|
||||||
### Bug fix
|
### Bug fix
|
||||||
|
@ -264,9 +264,8 @@ if (USE_STATIC_LIBRARIES AND HAVE_NO_PIE)
|
|||||||
set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${FLAG_NO_PIE}")
|
set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${FLAG_NO_PIE}")
|
||||||
endif ()
|
endif ()
|
||||||
|
|
||||||
# TODO: only make this extra-checks in CI builds, since a lot of contrib libs won't link -
|
# Make this extra-checks for correct library dependencies.
|
||||||
# CI works around this problem by explicitly adding GLIBC_COMPATIBILITY flag.
|
if (NOT SANITIZE)
|
||||||
if (NOT SANITIZE AND YANDEX_OFFICIAL_BUILD)
|
|
||||||
set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,--no-undefined")
|
set (CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -Wl,--no-undefined")
|
||||||
set (CMAKE_SHARED_LINKER_FLAGS "${CMAKE_SHARED_LINKER_FLAGS} -Wl,--no-undefined")
|
set (CMAKE_SHARED_LINKER_FLAGS "${CMAKE_SHARED_LINKER_FLAGS} -Wl,--no-undefined")
|
||||||
endif ()
|
endif ()
|
||||||
@ -481,6 +480,7 @@ include (cmake/find_mimalloc.cmake)
|
|||||||
include (cmake/find_simdjson.cmake)
|
include (cmake/find_simdjson.cmake)
|
||||||
include (cmake/find_rapidjson.cmake)
|
include (cmake/find_rapidjson.cmake)
|
||||||
include (cmake/find_fastops.cmake)
|
include (cmake/find_fastops.cmake)
|
||||||
|
include (cmake/find_orc.cmake)
|
||||||
|
|
||||||
find_contrib_lib(cityhash)
|
find_contrib_lib(cityhash)
|
||||||
find_contrib_lib(farmhash)
|
find_contrib_lib(farmhash)
|
||||||
|
@ -14,6 +14,7 @@ ClickHouse is an open-source column-oriented database management system that all
|
|||||||
|
|
||||||
## Upcoming Events
|
## Upcoming Events
|
||||||
* [ClickHouse Meetup in Moscow](https://yandex.ru/promo/clickhouse/moscow-2019) on September 5.
|
* [ClickHouse Meetup in Moscow](https://yandex.ru/promo/clickhouse/moscow-2019) on September 5.
|
||||||
|
* [ClickHouse Meetup in Munich](https://www.meetup.com/ClickHouse-Meetup-Munich/events/264185199/) on September 17.
|
||||||
* [ClickHouse Meetup in Paris](https://www.eventbrite.com/e/clickhouse-paris-meetup-2019-registration-68493270215) on October 3.
|
* [ClickHouse Meetup in Paris](https://www.eventbrite.com/e/clickhouse-paris-meetup-2019-registration-68493270215) on October 3.
|
||||||
* [ClickHouse Meetup in Hong Kong](https://www.meetup.com/Hong-Kong-Machine-Learning-Meetup/events/263580542/) on October 17.
|
* [ClickHouse Meetup in Hong Kong](https://www.meetup.com/Hong-Kong-Machine-Learning-Meetup/events/263580542/) on October 17.
|
||||||
* [ClickHouse Meetup in Shenzhen](https://www.huodongxing.com/event/3483759917300) on October 20.
|
* [ClickHouse Meetup in Shenzhen](https://www.huodongxing.com/event/3483759917300) on October 20.
|
||||||
|
8
cmake/find_orc.cmake
Normal file
8
cmake/find_orc.cmake
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
##TODO replace hardcode to find procedure
|
||||||
|
|
||||||
|
set(USE_ORC 0)
|
||||||
|
set(USE_INTERNAL_ORC_LIBRARY ON)
|
||||||
|
|
||||||
|
if (ARROW_LIBRARY)
|
||||||
|
set(USE_ORC 1)
|
||||||
|
endif()
|
12
contrib/CMakeLists.txt
vendored
12
contrib/CMakeLists.txt
vendored
@ -10,6 +10,18 @@ endif ()
|
|||||||
|
|
||||||
set_property(DIRECTORY PROPERTY EXCLUDE_FROM_ALL 1)
|
set_property(DIRECTORY PROPERTY EXCLUDE_FROM_ALL 1)
|
||||||
|
|
||||||
|
if (USE_INTERNAL_ORC_LIBRARY)
|
||||||
|
set(BUILD_JAVA OFF)
|
||||||
|
set (ANALYZE_JAVA OFF)
|
||||||
|
set (BUILD_CPP_TESTS OFF)
|
||||||
|
set (BUILD_TOOLS OFF)
|
||||||
|
option(BUILD_JAVA OFF)
|
||||||
|
option (ANALYZE_JAVA OFF)
|
||||||
|
option (BUILD_CPP_TESTS OFF)
|
||||||
|
option (BUILD_TOOLS OFF)
|
||||||
|
set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/contrib/orc/cmake_modules")
|
||||||
|
add_subdirectory(orc)
|
||||||
|
endif()
|
||||||
|
|
||||||
if (USE_INTERNAL_UNWIND_LIBRARY)
|
if (USE_INTERNAL_UNWIND_LIBRARY)
|
||||||
add_subdirectory (libunwind-cmake)
|
add_subdirectory (libunwind-cmake)
|
||||||
|
@ -47,6 +47,71 @@ target_include_directories(${THRIFT_LIBRARY} SYSTEM PUBLIC ${ClickHouse_SOURCE_D
|
|||||||
target_link_libraries(${THRIFT_LIBRARY} PRIVATE Threads::Threads)
|
target_link_libraries(${THRIFT_LIBRARY} PRIVATE Threads::Threads)
|
||||||
|
|
||||||
|
|
||||||
|
# === orc
|
||||||
|
|
||||||
|
set(ORC_SOURCE_DIR ${ClickHouse_SOURCE_DIR}/contrib/orc/c++)
|
||||||
|
set(ORC_INCLUDE_DIR ${ORC_SOURCE_DIR}/include)
|
||||||
|
set(ORC_SOURCE_SRC_DIR ${ORC_SOURCE_DIR}/src)
|
||||||
|
set(ORC_SOURCE_WRAP_DIR ${ORC_SOURCE_DIR}/wrap)
|
||||||
|
|
||||||
|
set(ORC_BUILD_SRC_DIR ${CMAKE_CURRENT_BINARY_DIR}/../orc/c++/src)
|
||||||
|
set(ORC_BUILD_INCLUDE_DIR ${CMAKE_CURRENT_BINARY_DIR}/../orc/c++/include)
|
||||||
|
|
||||||
|
set(GOOGLE_PROTOBUF_DIR ${ClickHouse_SOURCE_DIR}/contrib/protobuf/src/)
|
||||||
|
set(ORC_ADDITION_SOURCE_DIR ${CMAKE_CURRENT_BINARY_DIR})
|
||||||
|
set(ARROW_SRC_DIR ${ClickHouse_SOURCE_DIR}/contrib/arrow/cpp/src)
|
||||||
|
|
||||||
|
set(PROTOBUF_EXECUTABLE ${CMAKE_CURRENT_BINARY_DIR}/../protobuf/cmake/protoc)
|
||||||
|
set(PROTO_DIR ${ORC_SOURCE_DIR}/../proto)
|
||||||
|
|
||||||
|
|
||||||
|
add_custom_command(OUTPUT orc_proto.pb.h orc_proto.pb.cc
|
||||||
|
COMMAND ${PROTOBUF_EXECUTABLE}
|
||||||
|
-I ${PROTO_DIR}
|
||||||
|
--cpp_out="${CMAKE_CURRENT_BINARY_DIR}"
|
||||||
|
"${PROTO_DIR}/orc_proto.proto")
|
||||||
|
|
||||||
|
include_directories(SYSTEM ${ORC_INCLUDE_DIR})
|
||||||
|
include_directories(SYSTEM ${ORC_SOURCE_SRC_DIR})
|
||||||
|
include_directories(SYSTEM ${ORC_SOURCE_WRAP_DIR})
|
||||||
|
include_directories(SYSTEM ${GOOGLE_PROTOBUF_DIR})
|
||||||
|
include_directories(SYSTEM ${ORC_BUILD_SRC_DIR})
|
||||||
|
include_directories(SYSTEM ${ORC_BUILD_INCLUDE_DIR})
|
||||||
|
include_directories(SYSTEM ${ORC_ADDITION_SOURCE_DIR})
|
||||||
|
include_directories(SYSTEM ${ARROW_SRC_DIR})
|
||||||
|
|
||||||
|
|
||||||
|
set(ORC_SRCS
|
||||||
|
${ARROW_SRC_DIR}/arrow/adapters/orc/adapter.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Exceptions.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/OrcFile.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Reader.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/ByteRLE.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/ColumnPrinter.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/ColumnReader.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/ColumnWriter.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Common.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Compression.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Exceptions.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Int128.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/LzoDecompressor.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/MemoryPool.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/OrcFile.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Reader.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/RLE.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/RLEv1.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/RLEv2.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Statistics.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/StripeStream.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Timezone.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/TypeImpl.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Vector.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/Writer.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/io/InputStream.cc
|
||||||
|
${ORC_SOURCE_SRC_DIR}/io/OutputStream.cc
|
||||||
|
${ORC_ADDITION_SOURCE_DIR}/orc_proto.pb.cc
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
# === arrow
|
# === arrow
|
||||||
|
|
||||||
@ -103,6 +168,7 @@ set(ARROW_SRCS
|
|||||||
${LIBRARY_DIR}/util/thread-pool.cc
|
${LIBRARY_DIR}/util/thread-pool.cc
|
||||||
${LIBRARY_DIR}/util/trie.cc
|
${LIBRARY_DIR}/util/trie.cc
|
||||||
${LIBRARY_DIR}/util/utf8.cc
|
${LIBRARY_DIR}/util/utf8.cc
|
||||||
|
${ORC_SRCS}
|
||||||
)
|
)
|
||||||
|
|
||||||
set(ARROW_SRCS ${ARROW_SRCS}
|
set(ARROW_SRCS ${ARROW_SRCS}
|
||||||
@ -151,8 +217,9 @@ endif()
|
|||||||
|
|
||||||
|
|
||||||
add_library(${ARROW_LIBRARY} ${ARROW_SRCS})
|
add_library(${ARROW_LIBRARY} ${ARROW_SRCS})
|
||||||
|
add_dependencies(${ARROW_LIBRARY} protoc)
|
||||||
target_include_directories(${ARROW_LIBRARY} SYSTEM PUBLIC ${ClickHouse_SOURCE_DIR}/contrib/arrow/cpp/src PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/cpp/src ${Boost_INCLUDE_DIRS})
|
target_include_directories(${ARROW_LIBRARY} SYSTEM PUBLIC ${ClickHouse_SOURCE_DIR}/contrib/arrow/cpp/src PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/cpp/src ${Boost_INCLUDE_DIRS})
|
||||||
target_link_libraries(${ARROW_LIBRARY} PRIVATE ${DOUBLE_CONVERSION_LIBRARIES} Threads::Threads)
|
target_link_libraries(${ARROW_LIBRARY} PRIVATE ${DOUBLE_CONVERSION_LIBRARIES} ${Protobuf_LIBRARY} Threads::Threads)
|
||||||
if (ARROW_WITH_LZ4)
|
if (ARROW_WITH_LZ4)
|
||||||
target_link_libraries(${ARROW_LIBRARY} PRIVATE ${LZ4_LIBRARY})
|
target_link_libraries(${ARROW_LIBRARY} PRIVATE ${LZ4_LIBRARY})
|
||||||
endif()
|
endif()
|
||||||
|
1
contrib/orc
vendored
Submodule
1
contrib/orc
vendored
Submodule
@ -0,0 +1 @@
|
|||||||
|
Subproject commit 5981208e39447df84827f6a961d1da76bacb6078
|
@ -114,6 +114,7 @@ add_headers_and_sources(dbms src/Columns)
|
|||||||
add_headers_and_sources(dbms src/Storages)
|
add_headers_and_sources(dbms src/Storages)
|
||||||
add_headers_and_sources(dbms src/Storages/Distributed)
|
add_headers_and_sources(dbms src/Storages/Distributed)
|
||||||
add_headers_and_sources(dbms src/Storages/MergeTree)
|
add_headers_and_sources(dbms src/Storages/MergeTree)
|
||||||
|
add_headers_and_sources(dbms src/Storages/LiveView)
|
||||||
add_headers_and_sources(dbms src/Client)
|
add_headers_and_sources(dbms src/Client)
|
||||||
add_headers_and_sources(dbms src/Formats)
|
add_headers_and_sources(dbms src/Formats)
|
||||||
add_headers_and_sources(dbms src/Processors)
|
add_headers_and_sources(dbms src/Processors)
|
||||||
|
@ -81,7 +81,6 @@ add_subdirectory (extract-from-config)
|
|||||||
add_subdirectory (compressor)
|
add_subdirectory (compressor)
|
||||||
add_subdirectory (copier)
|
add_subdirectory (copier)
|
||||||
add_subdirectory (format)
|
add_subdirectory (format)
|
||||||
add_subdirectory (clang)
|
|
||||||
add_subdirectory (obfuscator)
|
add_subdirectory (obfuscator)
|
||||||
|
|
||||||
if (ENABLE_CLICKHOUSE_ODBC_BRIDGE)
|
if (ENABLE_CLICKHOUSE_ODBC_BRIDGE)
|
||||||
@ -89,25 +88,21 @@ if (ENABLE_CLICKHOUSE_ODBC_BRIDGE)
|
|||||||
endif ()
|
endif ()
|
||||||
|
|
||||||
if (CLICKHOUSE_ONE_SHARED)
|
if (CLICKHOUSE_ONE_SHARED)
|
||||||
add_library(clickhouse-lib SHARED ${CLICKHOUSE_SERVER_SOURCES} ${CLICKHOUSE_CLIENT_SOURCES} ${CLICKHOUSE_LOCAL_SOURCES} ${CLICKHOUSE_BENCHMARK_SOURCES} ${CLICKHOUSE_PERFORMANCE_TEST_SOURCES} ${CLICKHOUSE_COPIER_SOURCES} ${CLICKHOUSE_EXTRACT_FROM_CONFIG_SOURCES} ${CLICKHOUSE_COMPRESSOR_SOURCES} ${CLICKHOUSE_FORMAT_SOURCES} ${CLICKHOUSE_OBFUSCATOR_SOURCES} ${CLICKHOUSE_COMPILER_SOURCES} ${CLICKHOUSE_ODBC_BRIDGE_SOURCES})
|
add_library(clickhouse-lib SHARED ${CLICKHOUSE_SERVER_SOURCES} ${CLICKHOUSE_CLIENT_SOURCES} ${CLICKHOUSE_LOCAL_SOURCES} ${CLICKHOUSE_BENCHMARK_SOURCES} ${CLICKHOUSE_PERFORMANCE_TEST_SOURCES} ${CLICKHOUSE_COPIER_SOURCES} ${CLICKHOUSE_EXTRACT_FROM_CONFIG_SOURCES} ${CLICKHOUSE_COMPRESSOR_SOURCES} ${CLICKHOUSE_FORMAT_SOURCES} ${CLICKHOUSE_OBFUSCATOR_SOURCES} ${CLICKHOUSE_ODBC_BRIDGE_SOURCES})
|
||||||
target_link_libraries(clickhouse-lib ${CLICKHOUSE_SERVER_LINK} ${CLICKHOUSE_CLIENT_LINK} ${CLICKHOUSE_LOCAL_LINK} ${CLICKHOUSE_BENCHMARK_LINK} ${CLICKHOUSE_PERFORMANCE_TEST_LINK} ${CLICKHOUSE_COPIER_LINK} ${CLICKHOUSE_EXTRACT_FROM_CONFIG_LINK} ${CLICKHOUSE_COMPRESSOR_LINK} ${CLICKHOUSE_FORMAT_LINK} ${CLICKHOUSE_OBFUSCATOR_LINK} ${CLICKHOUSE_COMPILER_LINK} ${CLICKHOUSE_ODBC_BRIDGE_LINK})
|
target_link_libraries(clickhouse-lib ${CLICKHOUSE_SERVER_LINK} ${CLICKHOUSE_CLIENT_LINK} ${CLICKHOUSE_LOCAL_LINK} ${CLICKHOUSE_BENCHMARK_LINK} ${CLICKHOUSE_PERFORMANCE_TEST_LINK} ${CLICKHOUSE_COPIER_LINK} ${CLICKHOUSE_EXTRACT_FROM_CONFIG_LINK} ${CLICKHOUSE_COMPRESSOR_LINK} ${CLICKHOUSE_FORMAT_LINK} ${CLICKHOUSE_OBFUSCATOR_LINK} ${CLICKHOUSE_ODBC_BRIDGE_LINK})
|
||||||
target_include_directories(clickhouse-lib ${CLICKHOUSE_SERVER_INCLUDE} ${CLICKHOUSE_CLIENT_INCLUDE} ${CLICKHOUSE_LOCAL_INCLUDE} ${CLICKHOUSE_BENCHMARK_INCLUDE} ${CLICKHOUSE_PERFORMANCE_TEST_INCLUDE} ${CLICKHOUSE_COPIER_INCLUDE} ${CLICKHOUSE_EXTRACT_FROM_CONFIG_INCLUDE} ${CLICKHOUSE_COMPRESSOR_INCLUDE} ${CLICKHOUSE_FORMAT_INCLUDE} ${CLICKHOUSE_OBFUSCATOR_INCLUDE} ${CLICKHOUSE_COMPILER_INCLUDE} ${CLICKHOUSE_ODBC_BRIDGE_INCLUDE})
|
target_include_directories(clickhouse-lib ${CLICKHOUSE_SERVER_INCLUDE} ${CLICKHOUSE_CLIENT_INCLUDE} ${CLICKHOUSE_LOCAL_INCLUDE} ${CLICKHOUSE_BENCHMARK_INCLUDE} ${CLICKHOUSE_PERFORMANCE_TEST_INCLUDE} ${CLICKHOUSE_COPIER_INCLUDE} ${CLICKHOUSE_EXTRACT_FROM_CONFIG_INCLUDE} ${CLICKHOUSE_COMPRESSOR_INCLUDE} ${CLICKHOUSE_FORMAT_INCLUDE} ${CLICKHOUSE_OBFUSCATOR_INCLUDE} ${CLICKHOUSE_ODBC_BRIDGE_INCLUDE})
|
||||||
set_target_properties(clickhouse-lib PROPERTIES SOVERSION ${VERSION_MAJOR}.${VERSION_MINOR} VERSION ${VERSION_SO} OUTPUT_NAME clickhouse DEBUG_POSTFIX "")
|
set_target_properties(clickhouse-lib PROPERTIES SOVERSION ${VERSION_MAJOR}.${VERSION_MINOR} VERSION ${VERSION_SO} OUTPUT_NAME clickhouse DEBUG_POSTFIX "")
|
||||||
install (TARGETS clickhouse-lib LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT clickhouse)
|
install (TARGETS clickhouse-lib LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR} COMPONENT clickhouse)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
if (CLICKHOUSE_SPLIT_BINARY)
|
if (CLICKHOUSE_SPLIT_BINARY)
|
||||||
set (CLICKHOUSE_ALL_TARGETS clickhouse-server clickhouse-client clickhouse-local clickhouse-benchmark clickhouse-performance-test
|
set (CLICKHOUSE_ALL_TARGETS clickhouse-server clickhouse-client clickhouse-local clickhouse-benchmark clickhouse-performance-test
|
||||||
clickhouse-extract-from-config clickhouse-compressor clickhouse-format clickhouse-copier)
|
clickhouse-extract-from-config clickhouse-compressor clickhouse-format clickhouse-obfuscator clickhouse-copier)
|
||||||
|
|
||||||
if (ENABLE_CLICKHOUSE_ODBC_BRIDGE)
|
if (ENABLE_CLICKHOUSE_ODBC_BRIDGE)
|
||||||
list (APPEND CLICKHOUSE_ALL_TARGETS clickhouse-odbc-bridge)
|
list (APPEND CLICKHOUSE_ALL_TARGETS clickhouse-odbc-bridge)
|
||||||
endif ()
|
endif ()
|
||||||
|
|
||||||
if (USE_EMBEDDED_COMPILER)
|
|
||||||
list (APPEND CLICKHOUSE_ALL_TARGETS clickhouse-clang clickhouse-lld)
|
|
||||||
endif ()
|
|
||||||
|
|
||||||
set_target_properties(${CLICKHOUSE_ALL_TARGETS} PROPERTIES RUNTIME_OUTPUT_DIRECTORY ..)
|
set_target_properties(${CLICKHOUSE_ALL_TARGETS} PROPERTIES RUNTIME_OUTPUT_DIRECTORY ..)
|
||||||
|
|
||||||
add_custom_target (clickhouse-bundle ALL DEPENDS ${CLICKHOUSE_ALL_TARGETS})
|
add_custom_target (clickhouse-bundle ALL DEPENDS ${CLICKHOUSE_ALL_TARGETS})
|
||||||
@ -115,10 +110,6 @@ if (CLICKHOUSE_SPLIT_BINARY)
|
|||||||
|
|
||||||
install(PROGRAMS clickhouse-split-helper DESTINATION ${CMAKE_INSTALL_BINDIR} RENAME clickhouse COMPONENT clickhouse)
|
install(PROGRAMS clickhouse-split-helper DESTINATION ${CMAKE_INSTALL_BINDIR} RENAME clickhouse COMPONENT clickhouse)
|
||||||
else ()
|
else ()
|
||||||
if (USE_EMBEDDED_COMPILER)
|
|
||||||
# before add_executable !
|
|
||||||
link_directories (${LLVM_LIBRARY_DIRS})
|
|
||||||
endif ()
|
|
||||||
add_executable (clickhouse main.cpp)
|
add_executable (clickhouse main.cpp)
|
||||||
target_link_libraries (clickhouse PRIVATE clickhouse_common_io string_utils)
|
target_link_libraries (clickhouse PRIVATE clickhouse_common_io string_utils)
|
||||||
target_include_directories (clickhouse BEFORE PRIVATE ${COMMON_INCLUDE_DIR})
|
target_include_directories (clickhouse BEFORE PRIVATE ${COMMON_INCLUDE_DIR})
|
||||||
@ -154,9 +145,6 @@ else ()
|
|||||||
if (ENABLE_CLICKHOUSE_OBFUSCATOR)
|
if (ENABLE_CLICKHOUSE_OBFUSCATOR)
|
||||||
clickhouse_target_link_split_lib(clickhouse obfuscator)
|
clickhouse_target_link_split_lib(clickhouse obfuscator)
|
||||||
endif ()
|
endif ()
|
||||||
if (USE_EMBEDDED_COMPILER)
|
|
||||||
target_link_libraries(clickhouse PRIVATE clickhouse-compiler-lib)
|
|
||||||
endif ()
|
|
||||||
|
|
||||||
set (CLICKHOUSE_BUNDLE)
|
set (CLICKHOUSE_BUNDLE)
|
||||||
if (ENABLE_CLICKHOUSE_SERVER)
|
if (ENABLE_CLICKHOUSE_SERVER)
|
||||||
@ -213,18 +201,8 @@ else ()
|
|||||||
list(APPEND CLICKHOUSE_BUNDLE clickhouse-odbc-bridge)
|
list(APPEND CLICKHOUSE_BUNDLE clickhouse-odbc-bridge)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
# install always because depian package want this files:
|
|
||||||
add_custom_target (clickhouse-clang ALL COMMAND ${CMAKE_COMMAND} -E create_symlink clickhouse clickhouse-clang DEPENDS clickhouse)
|
|
||||||
add_custom_target (clickhouse-lld ALL COMMAND ${CMAKE_COMMAND} -E create_symlink clickhouse clickhouse-lld DEPENDS clickhouse)
|
|
||||||
list(APPEND CLICKHOUSE_BUNDLE clickhouse-clang clickhouse-lld)
|
|
||||||
|
|
||||||
install (TARGETS clickhouse RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR} COMPONENT clickhouse)
|
install (TARGETS clickhouse RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR} COMPONENT clickhouse)
|
||||||
|
|
||||||
install (FILES
|
|
||||||
${CMAKE_CURRENT_BINARY_DIR}/clickhouse-clang
|
|
||||||
${CMAKE_CURRENT_BINARY_DIR}/clickhouse-lld
|
|
||||||
DESTINATION ${CMAKE_INSTALL_BINDIR} COMPONENT clickhouse)
|
|
||||||
|
|
||||||
add_custom_target (clickhouse-bundle ALL DEPENDS ${CLICKHOUSE_BUNDLE})
|
add_custom_target (clickhouse-bundle ALL DEPENDS ${CLICKHOUSE_BUNDLE})
|
||||||
|
|
||||||
endif ()
|
endif ()
|
||||||
|
@ -1,38 +0,0 @@
|
|||||||
if (USE_EMBEDDED_COMPILER)
|
|
||||||
add_subdirectory ("Compiler-${LLVM_VERSION}")
|
|
||||||
endif ()
|
|
||||||
|
|
||||||
if (CLICKHOUSE_SPLIT_BINARY)
|
|
||||||
if (USE_EMBEDDED_COMPILER)
|
|
||||||
link_directories (${LLVM_LIBRARY_DIRS})
|
|
||||||
add_executable (clickhouse-clang clickhouse-clang.cpp)
|
|
||||||
target_link_libraries (clickhouse-clang PRIVATE clickhouse-compiler-lib)
|
|
||||||
add_executable (clickhouse-lld clickhouse-lld.cpp)
|
|
||||||
target_link_libraries (clickhouse-lld PRIVATE clickhouse-compiler-lib)
|
|
||||||
install (TARGETS clickhouse-clang clickhouse-lld RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR} COMPONENT clickhouse)
|
|
||||||
endif ()
|
|
||||||
endif ()
|
|
||||||
|
|
||||||
set (TMP_HEADERS_DIR "${CMAKE_CURRENT_BINARY_DIR}/${INTERNAL_COMPILER_HEADERS_RELATIVE}")
|
|
||||||
# Make and install empty dir for debian package if compiler disabled
|
|
||||||
add_custom_target (make-headers-directory ALL COMMAND ${CMAKE_COMMAND} -E make_directory ${TMP_HEADERS_DIR})
|
|
||||||
install (DIRECTORY ${TMP_HEADERS_DIR} DESTINATION ${CMAKE_INSTALL_DATAROOTDIR}/clickhouse/${INTERNAL_COMPILER_HEADERS_DIR} COMPONENT clickhouse)
|
|
||||||
# TODO: fix on macos copy_headers.sh: sed --posix
|
|
||||||
|
|
||||||
if (USE_EMBEDDED_COMPILER)
|
|
||||||
set (COPY_HEADERS_COMPILER "${CMAKE_CURRENT_BINARY_DIR}/../${INTERNAL_COMPILER_EXECUTABLE}")
|
|
||||||
set (COPY_HEADERS_DEPENDS clickhouse-clang)
|
|
||||||
elseif (EXISTS ${INTERNAL_COMPILER_BIN_ROOT}${INTERNAL_COMPILER_EXECUTABLE})
|
|
||||||
set (COPY_HEADERS_COMPILER "${INTERNAL_COMPILER_BIN_ROOT}${INTERNAL_COMPILER_EXECUTABLE}")
|
|
||||||
endif ()
|
|
||||||
|
|
||||||
if (COPY_HEADERS_COMPILER)
|
|
||||||
add_custom_target (copy-headers [ -f ${TMP_HEADERS_DIR}/dbms/src/Interpreters/SpecializedAggregator.h ] || env CLANG=${COPY_HEADERS_COMPILER} BUILD_PATH=${ClickHouse_BINARY_DIR} DESTDIR=${ClickHouse_SOURCE_DIR} CMAKE_CXX_COMPILER_VERSION=${CMAKE_CXX_COMPILER_VERSION} ${CMAKE_CURRENT_SOURCE_DIR}/copy_headers.sh ${ClickHouse_SOURCE_DIR} ${TMP_HEADERS_DIR} DEPENDS ${COPY_HEADERS_DEPENDS} WORKING_DIRECTORY ${ClickHouse_SOURCE_DIR} SOURCES copy_headers.sh)
|
|
||||||
|
|
||||||
if (USE_INTERNAL_LLVM_LIBRARY)
|
|
||||||
set (CLANG_HEADERS_DIR "${ClickHouse_SOURCE_DIR}/contrib/llvm/clang/lib/Headers")
|
|
||||||
set (CLANG_HEADERS_DEST "${TMP_HEADERS_DIR}/usr/local/lib/clang/${LLVM_VERSION}/include") # original: ${LLVM_LIBRARY_OUTPUT_INTDIR}/clang/${CLANG_VERSION}/include
|
|
||||||
add_custom_target (copy-headers-clang ${CMAKE_COMMAND} -E make_directory ${CLANG_HEADERS_DEST} && ${CMAKE_COMMAND} -E copy_if_different ${CLANG_HEADERS_DIR}/* ${CLANG_HEADERS_DEST} )
|
|
||||||
add_dependencies (copy-headers copy-headers-clang)
|
|
||||||
endif ()
|
|
||||||
endif ()
|
|
@ -1,53 +0,0 @@
|
|||||||
add_definitions(-Wno-error -Wno-unused-parameter -Wno-non-virtual-dtor -U_LIBCPP_DEBUG)
|
|
||||||
|
|
||||||
link_directories(${LLVM_LIBRARY_DIRS})
|
|
||||||
|
|
||||||
add_library(clickhouse-compiler-lib
|
|
||||||
driver.cpp
|
|
||||||
cc1_main.cpp
|
|
||||||
cc1as_main.cpp
|
|
||||||
lld.cpp)
|
|
||||||
|
|
||||||
target_compile_options(clickhouse-compiler-lib PRIVATE -fno-rtti -fno-exceptions -g0)
|
|
||||||
|
|
||||||
string(REPLACE "${INCLUDE_DEBUG_HELPERS}" "" CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS}) # cant compile with -fno-rtti
|
|
||||||
|
|
||||||
llvm_libs_all(REQUIRED_LLVM_LIBRARIES)
|
|
||||||
|
|
||||||
message(STATUS "Using LLVM ${LLVM_VERSION}: ${LLVM_INCLUDE_DIRS} : ${REQUIRED_LLVM_LIBRARIES}")
|
|
||||||
|
|
||||||
target_include_directories(clickhouse-compiler-lib SYSTEM PRIVATE ${LLVM_INCLUDE_DIRS})
|
|
||||||
|
|
||||||
# This is extracted almost directly from CMakeFiles/.../link.txt in LLVM build directory.
|
|
||||||
|
|
||||||
target_link_libraries(clickhouse-compiler-lib PRIVATE
|
|
||||||
|
|
||||||
clangBasic clangCodeGen clangDriver clangFrontend clangFrontendTool
|
|
||||||
clangRewriteFrontend clangARCMigrate clangStaticAnalyzerFrontend
|
|
||||||
clangParse clangSerialization clangSema clangEdit clangStaticAnalyzerCheckers
|
|
||||||
clangASTMatchers clangStaticAnalyzerCore clangAnalysis clangAST clangRewrite clangLex clangBasic
|
|
||||||
|
|
||||||
lldCOFF
|
|
||||||
lldDriver
|
|
||||||
lldELF
|
|
||||||
#lldMinGW
|
|
||||||
lldMachO
|
|
||||||
lldReaderWriter
|
|
||||||
lldYAML
|
|
||||||
#lldCommon
|
|
||||||
lldCore
|
|
||||||
lldConfig
|
|
||||||
|
|
||||||
${REQUIRED_LLVM_LIBRARIES}
|
|
||||||
|
|
||||||
LLVMSupport
|
|
||||||
|
|
||||||
#Polly
|
|
||||||
#PollyISL
|
|
||||||
#PollyPPCG
|
|
||||||
|
|
||||||
PUBLIC ${ZLIB_LIBRARIES} ${EXECINFO_LIBRARIES} Threads::Threads
|
|
||||||
${MALLOC_LIBRARIES}
|
|
||||||
${GLIBC_COMPATIBILITY_LIBRARIES}
|
|
||||||
${MEMCPY_LIBRARIES}
|
|
||||||
)
|
|
@ -1,63 +0,0 @@
|
|||||||
==============================================================================
|
|
||||||
LLVM Release License
|
|
||||||
==============================================================================
|
|
||||||
University of Illinois/NCSA
|
|
||||||
Open Source License
|
|
||||||
|
|
||||||
Copyright (c) 2007-2016 University of Illinois at Urbana-Champaign.
|
|
||||||
All rights reserved.
|
|
||||||
|
|
||||||
Developed by:
|
|
||||||
|
|
||||||
LLVM Team
|
|
||||||
|
|
||||||
University of Illinois at Urbana-Champaign
|
|
||||||
|
|
||||||
http://llvm.org
|
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
||||||
this software and associated documentation files (the "Software"), to deal with
|
|
||||||
the Software without restriction, including without limitation the rights to
|
|
||||||
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
||||||
of the Software, and to permit persons to whom the Software is furnished to do
|
|
||||||
so, subject to the following conditions:
|
|
||||||
|
|
||||||
* Redistributions of source code must retain the above copyright notice,
|
|
||||||
this list of conditions and the following disclaimers.
|
|
||||||
|
|
||||||
* Redistributions in binary form must reproduce the above copyright notice,
|
|
||||||
this list of conditions and the following disclaimers in the
|
|
||||||
documentation and/or other materials provided with the distribution.
|
|
||||||
|
|
||||||
* Neither the names of the LLVM Team, University of Illinois at
|
|
||||||
Urbana-Champaign, nor the names of its contributors may be used to
|
|
||||||
endorse or promote products derived from this Software without specific
|
|
||||||
prior written permission.
|
|
||||||
|
|
||||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
||||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
||||||
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
||||||
CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
||||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
||||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS WITH THE
|
|
||||||
SOFTWARE.
|
|
||||||
|
|
||||||
==============================================================================
|
|
||||||
The LLVM software contains code written by third parties. Such software will
|
|
||||||
have its own individual LICENSE.TXT file in the directory in which it appears.
|
|
||||||
This file will describe the copyrights, license, and restrictions which apply
|
|
||||||
to that code.
|
|
||||||
|
|
||||||
The disclaimer of warranty in the University of Illinois Open Source License
|
|
||||||
applies to all code in the LLVM Distribution, and nothing in any of the
|
|
||||||
other licenses gives permission to use the names of the LLVM Team or the
|
|
||||||
University of Illinois to endorse or promote products derived from this
|
|
||||||
Software.
|
|
||||||
|
|
||||||
The following pieces of software have additional or alternate copyrights,
|
|
||||||
licenses, and/or restrictions:
|
|
||||||
|
|
||||||
Program Directory
|
|
||||||
------- ---------
|
|
||||||
<none yet>
|
|
||||||
|
|
@ -1,242 +0,0 @@
|
|||||||
//===-- cc1_main.cpp - Clang CC1 Compiler Frontend ------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1 functionality, which implements the
|
|
||||||
// core compiler functionality along with a number of additional tools for
|
|
||||||
// demonstration and testing purposes.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "clang/CodeGen/ObjectFilePCHContainerOperations.h"
|
|
||||||
#include "clang/Config/config.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/CompilerInstance.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticBuffer.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "clang/FrontendTool/Utils.h"
|
|
||||||
#include "llvm/ADT/Statistic.h"
|
|
||||||
#include "llvm/LinkAllPasses.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/Compiler.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <cstdio>
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
#include <sys/resource.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// have no .a version in packages
|
|
||||||
#undef LINK_POLLY_INTO_TOOLS
|
|
||||||
|
|
||||||
using namespace clang;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
// Main driver
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool GenCrashDiag) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// Run the interrupt handlers to make sure any special cleanups get done, in
|
|
||||||
// particular that we remove files registered with RemoveFileOnSignal.
|
|
||||||
llvm::sys::RunInterruptHandlers();
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors. When reporting a fatal error, exit
|
|
||||||
// with status 70 to generate crash diagnostics. For BSD systems this is
|
|
||||||
// defined as an internal software error. Otherwise, exit with status 1.
|
|
||||||
exit(GenCrashDiag ? 70 : 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
namespace polly {
|
|
||||||
void initializePollyPasses(llvm::PassRegistry &Registry);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
// The amount of stack we think is "sufficient". If less than this much is
|
|
||||||
// available, we may be unable to reach our template instantiation depth
|
|
||||||
// limit and other similar limits.
|
|
||||||
// FIXME: Unify this with the stack we request when spawning a thread to build
|
|
||||||
// a module.
|
|
||||||
static const int kSufficientStack = 8 << 20;
|
|
||||||
|
|
||||||
#if defined(__linux__) && defined(__PIE__)
|
|
||||||
static size_t getCurrentStackAllocation() {
|
|
||||||
// If we can't compute the current stack usage, allow for 512K of command
|
|
||||||
// line arguments and environment.
|
|
||||||
size_t Usage = 512 * 1024;
|
|
||||||
if (FILE *StatFile = fopen("/proc/self/stat", "r")) {
|
|
||||||
// We assume that the stack extends from its current address to the end of
|
|
||||||
// the environment space. In reality, there is another string literal (the
|
|
||||||
// program name) after the environment, but this is close enough (we only
|
|
||||||
// need to be within 100K or so).
|
|
||||||
unsigned long StackPtr, EnvEnd;
|
|
||||||
// Disable silly GCC -Wformat warning that complains about length
|
|
||||||
// modifiers on ignored format specifiers. We want to retain these
|
|
||||||
// for documentation purposes even though they have no effect.
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic push
|
|
||||||
#pragma GCC diagnostic ignored "-Wformat"
|
|
||||||
#endif
|
|
||||||
if (fscanf(StatFile,
|
|
||||||
"%*d %*s %*c %*d %*d %*d %*d %*d %*u %*lu %*lu %*lu %*lu %*lu "
|
|
||||||
"%*lu %*ld %*ld %*ld %*ld %*ld %*ld %*llu %*lu %*ld %*lu %*lu "
|
|
||||||
"%*lu %*lu %lu %*lu %*lu %*lu %*lu %*lu %*llu %*lu %*lu %*d %*d "
|
|
||||||
"%*u %*u %*llu %*lu %*ld %*lu %*lu %*lu %*lu %*lu %*lu %lu %*d",
|
|
||||||
&StackPtr, &EnvEnd) == 2) {
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic pop
|
|
||||||
#endif
|
|
||||||
Usage = StackPtr < EnvEnd ? EnvEnd - StackPtr : StackPtr - EnvEnd;
|
|
||||||
}
|
|
||||||
fclose(StatFile);
|
|
||||||
}
|
|
||||||
return Usage;
|
|
||||||
}
|
|
||||||
|
|
||||||
#include <alloca.h>
|
|
||||||
|
|
||||||
LLVM_ATTRIBUTE_NOINLINE
|
|
||||||
static void ensureStackAddressSpace(int ExtraChunks = 0) {
|
|
||||||
// Linux kernels prior to 4.1 will sometimes locate the heap of a PIE binary
|
|
||||||
// relatively close to the stack (they are only guaranteed to be 128MiB
|
|
||||||
// apart). This results in crashes if we happen to heap-allocate more than
|
|
||||||
// 128MiB before we reach our stack high-water mark.
|
|
||||||
//
|
|
||||||
// To avoid these crashes, ensure that we have sufficient virtual memory
|
|
||||||
// pages allocated before we start running.
|
|
||||||
size_t Curr = getCurrentStackAllocation();
|
|
||||||
const int kTargetStack = kSufficientStack - 256 * 1024;
|
|
||||||
if (Curr < kTargetStack) {
|
|
||||||
volatile char *volatile Alloc =
|
|
||||||
static_cast<volatile char *>(alloca(kTargetStack - Curr));
|
|
||||||
Alloc[0] = 0;
|
|
||||||
Alloc[kTargetStack - Curr - 1] = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureStackAddressSpace() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/// Attempt to ensure that we have at least 8MiB of usable stack space.
|
|
||||||
static void ensureSufficientStack() {
|
|
||||||
struct rlimit rlim;
|
|
||||||
if (getrlimit(RLIMIT_STACK, &rlim) != 0)
|
|
||||||
return;
|
|
||||||
|
|
||||||
// Increase the soft stack limit to our desired level, if necessary and
|
|
||||||
// possible.
|
|
||||||
if (rlim.rlim_cur != RLIM_INFINITY && rlim.rlim_cur < kSufficientStack) {
|
|
||||||
// Try to allocate sufficient stack.
|
|
||||||
if (rlim.rlim_max == RLIM_INFINITY || rlim.rlim_max >= kSufficientStack)
|
|
||||||
rlim.rlim_cur = kSufficientStack;
|
|
||||||
else if (rlim.rlim_cur == rlim.rlim_max)
|
|
||||||
return;
|
|
||||||
else
|
|
||||||
rlim.rlim_cur = rlim.rlim_max;
|
|
||||||
|
|
||||||
if (setrlimit(RLIMIT_STACK, &rlim) != 0 ||
|
|
||||||
rlim.rlim_cur != kSufficientStack)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// We should now have a stack of size at least kSufficientStack. Ensure
|
|
||||||
// that we can actually use that much, if necessary.
|
|
||||||
ensureStackAddressSpace();
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureSufficientStack() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
int cc1_main(ArrayRef<const char *> Argv, const char *Argv0, void *MainAddr) {
|
|
||||||
ensureSufficientStack();
|
|
||||||
|
|
||||||
std::unique_ptr<CompilerInstance> Clang(new CompilerInstance());
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
// Register the support for object-file-wrapped Clang modules.
|
|
||||||
auto PCHOps = Clang->getPCHContainerOperations();
|
|
||||||
PCHOps->registerWriter(llvm::make_unique<ObjectFilePCHContainerWriter>());
|
|
||||||
PCHOps->registerReader(llvm::make_unique<ObjectFilePCHContainerReader>());
|
|
||||||
|
|
||||||
// Initialize targets first, so that --version shows registered targets.
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
llvm::InitializeAllTargetMCs();
|
|
||||||
llvm::InitializeAllAsmPrinters();
|
|
||||||
llvm::InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
llvm::PassRegistry &Registry = *llvm::PassRegistry::getPassRegistry();
|
|
||||||
polly::initializePollyPasses(Registry);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Buffer diagnostics from argument parsing so that we can output them using a
|
|
||||||
// well formed diagnostic object.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticBuffer *DiagsBuffer = new TextDiagnosticBuffer;
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagsBuffer);
|
|
||||||
bool Success = CompilerInvocation::CreateFromArgs(
|
|
||||||
Clang->getInvocation(), Argv.begin(), Argv.end(), Diags);
|
|
||||||
|
|
||||||
// Infer the builtin include path if unspecified.
|
|
||||||
if (Clang->getHeaderSearchOpts().UseBuiltinIncludes &&
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir.empty())
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir =
|
|
||||||
CompilerInvocation::GetResourcesPath(Argv0, MainAddr);
|
|
||||||
|
|
||||||
// Create the actual diagnostics engine.
|
|
||||||
Clang->createDiagnostics();
|
|
||||||
if (!Clang->hasDiagnostics())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
llvm::install_fatal_error_handler(LLVMErrorHandler,
|
|
||||||
static_cast<void*>(&Clang->getDiagnostics()));
|
|
||||||
|
|
||||||
DiagsBuffer->FlushDiagnostics(Clang->getDiagnostics());
|
|
||||||
if (!Success)
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Execute the frontend actions.
|
|
||||||
Success = ExecuteCompilerInvocation(Clang.get());
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
// Our error handler depends on the Diagnostics object, which we're
|
|
||||||
// potentially about to delete. Uninstall the handler now so that any
|
|
||||||
// later errors use the default handling behavior instead.
|
|
||||||
llvm::remove_fatal_error_handler();
|
|
||||||
|
|
||||||
// When running with -disable-free, don't do any destruction or shutdown.
|
|
||||||
if (Clang->getFrontendOpts().DisableFree) {
|
|
||||||
BuryPointer(std::move(Clang));
|
|
||||||
return !Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
return !Success;
|
|
||||||
}
|
|
@ -1,540 +0,0 @@
|
|||||||
//===-- cc1as_main.cpp - Clang Assembler ---------------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1as functionality, which implements
|
|
||||||
// the direct interface to the LLVM MC based assembler.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Basic/Diagnostic.h"
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/ADT/StringSwitch.h"
|
|
||||||
#include "llvm/ADT/Triple.h"
|
|
||||||
#include "llvm/IR/DataLayout.h"
|
|
||||||
#include "llvm/MC/MCAsmBackend.h"
|
|
||||||
#include "llvm/MC/MCAsmInfo.h"
|
|
||||||
#include "llvm/MC/MCCodeEmitter.h"
|
|
||||||
#include "llvm/MC/MCContext.h"
|
|
||||||
#include "llvm/MC/MCInstrInfo.h"
|
|
||||||
#include "llvm/MC/MCObjectFileInfo.h"
|
|
||||||
#include "llvm/MC/MCParser/MCAsmParser.h"
|
|
||||||
#include "llvm/MC/MCParser/MCTargetAsmParser.h"
|
|
||||||
#include "llvm/MC/MCRegisterInfo.h"
|
|
||||||
#include "llvm/MC/MCStreamer.h"
|
|
||||||
#include "llvm/MC/MCSubtargetInfo.h"
|
|
||||||
#include "llvm/MC/MCTargetOptions.h"
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/FormattedStream.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/MemoryBuffer.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/SourceMgr.h"
|
|
||||||
#include "llvm/Support/TargetRegistry.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace clang::driver::options;
|
|
||||||
using namespace llvm;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
/// \brief Helper class for representing a single invocation of the assembler.
|
|
||||||
struct AssemblerInvocation {
|
|
||||||
/// @name Target Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
/// The name of the target triple to assemble for.
|
|
||||||
std::string Triple;
|
|
||||||
|
|
||||||
/// If given, the name of the target CPU to determine which instructions
|
|
||||||
/// are legal.
|
|
||||||
std::string CPU;
|
|
||||||
|
|
||||||
/// The list of target specific features to enable or disable -- this should
|
|
||||||
/// be a list of strings starting with '+' or '-'.
|
|
||||||
std::vector<std::string> Features;
|
|
||||||
|
|
||||||
/// The list of symbol definitions.
|
|
||||||
std::vector<std::string> SymbolDefs;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Language Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::vector<std::string> IncludePaths;
|
|
||||||
unsigned NoInitialTextSection : 1;
|
|
||||||
unsigned SaveTemporaryLabels : 1;
|
|
||||||
unsigned GenDwarfForAssembly : 1;
|
|
||||||
unsigned RelaxELFRelocations : 1;
|
|
||||||
unsigned DwarfVersion;
|
|
||||||
std::string DwarfDebugFlags;
|
|
||||||
std::string DwarfDebugProducer;
|
|
||||||
std::string DebugCompilationDir;
|
|
||||||
llvm::DebugCompressionType CompressDebugSections =
|
|
||||||
llvm::DebugCompressionType::None;
|
|
||||||
std::string MainFileName;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Frontend Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::string InputFile;
|
|
||||||
std::vector<std::string> LLVMArgs;
|
|
||||||
std::string OutputPath;
|
|
||||||
enum FileType {
|
|
||||||
FT_Asm, ///< Assembly (.s) output, transliterate mode.
|
|
||||||
FT_Null, ///< No output, for timing purposes.
|
|
||||||
FT_Obj ///< Object file output.
|
|
||||||
};
|
|
||||||
FileType OutputType;
|
|
||||||
unsigned ShowHelp : 1;
|
|
||||||
unsigned ShowVersion : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Transliterate Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned OutputAsmVariant;
|
|
||||||
unsigned ShowEncoding : 1;
|
|
||||||
unsigned ShowInst : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Assembler Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned RelaxAll : 1;
|
|
||||||
unsigned NoExecStack : 1;
|
|
||||||
unsigned FatalWarnings : 1;
|
|
||||||
unsigned IncrementalLinkerCompatible : 1;
|
|
||||||
|
|
||||||
/// The name of the relocation model to use.
|
|
||||||
std::string RelocationModel;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
|
|
||||||
public:
|
|
||||||
AssemblerInvocation() {
|
|
||||||
Triple = "";
|
|
||||||
NoInitialTextSection = 0;
|
|
||||||
InputFile = "-";
|
|
||||||
OutputPath = "-";
|
|
||||||
OutputType = FT_Asm;
|
|
||||||
OutputAsmVariant = 0;
|
|
||||||
ShowInst = 0;
|
|
||||||
ShowEncoding = 0;
|
|
||||||
RelaxAll = 0;
|
|
||||||
NoExecStack = 0;
|
|
||||||
FatalWarnings = 0;
|
|
||||||
IncrementalLinkerCompatible = 0;
|
|
||||||
DwarfVersion = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool CreateFromArgs(AssemblerInvocation &Res,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags);
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
bool AssemblerInvocation::CreateFromArgs(AssemblerInvocation &Opts,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
bool Success = true;
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
std::unique_ptr<OptTable> OptTbl(createDriverOptTable());
|
|
||||||
|
|
||||||
const unsigned IncludedFlagsBitmask = options::CC1AsOption;
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args = OptTbl->ParseArgs(Argv, MissingArgIndex, MissingArgCount,
|
|
||||||
IncludedFlagsBitmask);
|
|
||||||
|
|
||||||
// Check for missing argument error.
|
|
||||||
if (MissingArgCount) {
|
|
||||||
Diags.Report(diag::err_drv_missing_argument)
|
|
||||||
<< Args.getArgString(MissingArgIndex) << MissingArgCount;
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Issue errors on unknown arguments.
|
|
||||||
for (const Arg *A : Args.filtered(OPT_UNKNOWN)) {
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << A->getAsString(Args);
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Construct the invocation.
|
|
||||||
|
|
||||||
// Target Options
|
|
||||||
Opts.Triple = llvm::Triple::normalize(Args.getLastArgValue(OPT_triple));
|
|
||||||
Opts.CPU = Args.getLastArgValue(OPT_target_cpu);
|
|
||||||
Opts.Features = Args.getAllArgValues(OPT_target_feature);
|
|
||||||
|
|
||||||
// Use the default target triple if unspecified.
|
|
||||||
if (Opts.Triple.empty())
|
|
||||||
Opts.Triple = llvm::sys::getDefaultTargetTriple();
|
|
||||||
|
|
||||||
// Language Options
|
|
||||||
Opts.IncludePaths = Args.getAllArgValues(OPT_I);
|
|
||||||
Opts.NoInitialTextSection = Args.hasArg(OPT_n);
|
|
||||||
Opts.SaveTemporaryLabels = Args.hasArg(OPT_msave_temp_labels);
|
|
||||||
// Any DebugInfoKind implies GenDwarfForAssembly.
|
|
||||||
Opts.GenDwarfForAssembly = Args.hasArg(OPT_debug_info_kind_EQ);
|
|
||||||
|
|
||||||
if (const Arg *A = Args.getLastArg(OPT_compress_debug_sections,
|
|
||||||
OPT_compress_debug_sections_EQ)) {
|
|
||||||
if (A->getOption().getID() == OPT_compress_debug_sections) {
|
|
||||||
// TODO: be more clever about the compression type auto-detection
|
|
||||||
Opts.CompressDebugSections = llvm::DebugCompressionType::GNU;
|
|
||||||
} else {
|
|
||||||
Opts.CompressDebugSections =
|
|
||||||
llvm::StringSwitch<llvm::DebugCompressionType>(A->getValue())
|
|
||||||
.Case("none", llvm::DebugCompressionType::None)
|
|
||||||
.Case("zlib", llvm::DebugCompressionType::Z)
|
|
||||||
.Case("zlib-gnu", llvm::DebugCompressionType::GNU)
|
|
||||||
.Default(llvm::DebugCompressionType::None);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Opts.RelaxELFRelocations = Args.hasArg(OPT_mrelax_relocations);
|
|
||||||
Opts.DwarfVersion = getLastArgIntValue(Args, OPT_dwarf_version_EQ, 2, Diags);
|
|
||||||
Opts.DwarfDebugFlags = Args.getLastArgValue(OPT_dwarf_debug_flags);
|
|
||||||
Opts.DwarfDebugProducer = Args.getLastArgValue(OPT_dwarf_debug_producer);
|
|
||||||
Opts.DebugCompilationDir = Args.getLastArgValue(OPT_fdebug_compilation_dir);
|
|
||||||
Opts.MainFileName = Args.getLastArgValue(OPT_main_file_name);
|
|
||||||
|
|
||||||
// Frontend Options
|
|
||||||
if (Args.hasArg(OPT_INPUT)) {
|
|
||||||
bool First = true;
|
|
||||||
for (const Arg *A : Args.filtered(OPT_INPUT)) {
|
|
||||||
if (First) {
|
|
||||||
Opts.InputFile = A->getValue();
|
|
||||||
First = false;
|
|
||||||
} else {
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << A->getAsString(Args);
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Opts.LLVMArgs = Args.getAllArgValues(OPT_mllvm);
|
|
||||||
Opts.OutputPath = Args.getLastArgValue(OPT_o);
|
|
||||||
if (Arg *A = Args.getLastArg(OPT_filetype)) {
|
|
||||||
StringRef Name = A->getValue();
|
|
||||||
unsigned OutputType = StringSwitch<unsigned>(Name)
|
|
||||||
.Case("asm", FT_Asm)
|
|
||||||
.Case("null", FT_Null)
|
|
||||||
.Case("obj", FT_Obj)
|
|
||||||
.Default(~0U);
|
|
||||||
if (OutputType == ~0U) {
|
|
||||||
Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Name;
|
|
||||||
Success = false;
|
|
||||||
} else
|
|
||||||
Opts.OutputType = FileType(OutputType);
|
|
||||||
}
|
|
||||||
Opts.ShowHelp = Args.hasArg(OPT_help);
|
|
||||||
Opts.ShowVersion = Args.hasArg(OPT_version);
|
|
||||||
|
|
||||||
// Transliterate Options
|
|
||||||
Opts.OutputAsmVariant =
|
|
||||||
getLastArgIntValue(Args, OPT_output_asm_variant, 0, Diags);
|
|
||||||
Opts.ShowEncoding = Args.hasArg(OPT_show_encoding);
|
|
||||||
Opts.ShowInst = Args.hasArg(OPT_show_inst);
|
|
||||||
|
|
||||||
// Assemble Options
|
|
||||||
Opts.RelaxAll = Args.hasArg(OPT_mrelax_all);
|
|
||||||
Opts.NoExecStack = Args.hasArg(OPT_mno_exec_stack);
|
|
||||||
Opts.FatalWarnings = Args.hasArg(OPT_massembler_fatal_warnings);
|
|
||||||
Opts.RelocationModel = Args.getLastArgValue(OPT_mrelocation_model, "pic");
|
|
||||||
Opts.IncrementalLinkerCompatible =
|
|
||||||
Args.hasArg(OPT_mincremental_linker_compatible);
|
|
||||||
Opts.SymbolDefs = Args.getAllArgValues(OPT_defsym);
|
|
||||||
|
|
||||||
return Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
static std::unique_ptr<raw_fd_ostream>
|
|
||||||
getOutputStream(AssemblerInvocation &Opts, DiagnosticsEngine &Diags,
|
|
||||||
bool Binary) {
|
|
||||||
if (Opts.OutputPath.empty())
|
|
||||||
Opts.OutputPath = "-";
|
|
||||||
|
|
||||||
// Make sure that the Out file gets unlinked from the disk if we get a
|
|
||||||
// SIGINT.
|
|
||||||
if (Opts.OutputPath != "-")
|
|
||||||
sys::RemoveFileOnSignal(Opts.OutputPath);
|
|
||||||
|
|
||||||
std::error_code EC;
|
|
||||||
auto Out = llvm::make_unique<raw_fd_ostream>(
|
|
||||||
Opts.OutputPath, EC, (Binary ? sys::fs::F_None : sys::fs::F_Text));
|
|
||||||
if (EC) {
|
|
||||||
Diags.Report(diag::err_fe_unable_to_open_output) << Opts.OutputPath
|
|
||||||
<< EC.message();
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return Out;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool ExecuteAssembler(AssemblerInvocation &Opts,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
// Get the target specific parser.
|
|
||||||
std::string Error;
|
|
||||||
const Target *TheTarget = TargetRegistry::lookupTarget(Opts.Triple, Error);
|
|
||||||
if (!TheTarget)
|
|
||||||
return Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
ErrorOr<std::unique_ptr<MemoryBuffer>> Buffer =
|
|
||||||
MemoryBuffer::getFileOrSTDIN(Opts.InputFile);
|
|
||||||
|
|
||||||
if (std::error_code EC = Buffer.getError()) {
|
|
||||||
Error = EC.message();
|
|
||||||
return Diags.Report(diag::err_fe_error_reading) << Opts.InputFile;
|
|
||||||
}
|
|
||||||
|
|
||||||
SourceMgr SrcMgr;
|
|
||||||
|
|
||||||
// Tell SrcMgr about this buffer, which is what the parser will pick up.
|
|
||||||
SrcMgr.AddNewSourceBuffer(std::move(*Buffer), SMLoc());
|
|
||||||
|
|
||||||
// Record the location of the include directories so that the lexer can find
|
|
||||||
// it later.
|
|
||||||
SrcMgr.setIncludeDirs(Opts.IncludePaths);
|
|
||||||
|
|
||||||
std::unique_ptr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(Opts.Triple));
|
|
||||||
assert(MRI && "Unable to create target register info!");
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, Opts.Triple));
|
|
||||||
assert(MAI && "Unable to create target asm info!");
|
|
||||||
|
|
||||||
// Ensure MCAsmInfo initialization occurs before any use, otherwise sections
|
|
||||||
// may be created with a combination of default and explicit settings.
|
|
||||||
MAI->setCompressDebugSections(Opts.CompressDebugSections);
|
|
||||||
|
|
||||||
MAI->setRelaxELFRelocations(Opts.RelaxELFRelocations);
|
|
||||||
|
|
||||||
bool IsBinary = Opts.OutputType == AssemblerInvocation::FT_Obj;
|
|
||||||
std::unique_ptr<raw_fd_ostream> FDOS = getOutputStream(Opts, Diags, IsBinary);
|
|
||||||
if (!FDOS)
|
|
||||||
return true;
|
|
||||||
|
|
||||||
// FIXME: This is not pretty. MCContext has a ptr to MCObjectFileInfo and
|
|
||||||
// MCObjectFileInfo needs a MCContext reference in order to initialize itself.
|
|
||||||
std::unique_ptr<MCObjectFileInfo> MOFI(new MCObjectFileInfo());
|
|
||||||
|
|
||||||
MCContext Ctx(MAI.get(), MRI.get(), MOFI.get(), &SrcMgr);
|
|
||||||
|
|
||||||
bool PIC = false;
|
|
||||||
if (Opts.RelocationModel == "static") {
|
|
||||||
PIC = false;
|
|
||||||
} else if (Opts.RelocationModel == "pic") {
|
|
||||||
PIC = true;
|
|
||||||
} else {
|
|
||||||
assert(Opts.RelocationModel == "dynamic-no-pic" &&
|
|
||||||
"Invalid PIC model!");
|
|
||||||
PIC = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
MOFI->InitMCObjectFileInfo(Triple(Opts.Triple), PIC, CodeModel::Default, Ctx);
|
|
||||||
if (Opts.SaveTemporaryLabels)
|
|
||||||
Ctx.setAllowTemporaryLabels(false);
|
|
||||||
if (Opts.GenDwarfForAssembly)
|
|
||||||
Ctx.setGenDwarfForAssembly(true);
|
|
||||||
if (!Opts.DwarfDebugFlags.empty())
|
|
||||||
Ctx.setDwarfDebugFlags(StringRef(Opts.DwarfDebugFlags));
|
|
||||||
if (!Opts.DwarfDebugProducer.empty())
|
|
||||||
Ctx.setDwarfDebugProducer(StringRef(Opts.DwarfDebugProducer));
|
|
||||||
if (!Opts.DebugCompilationDir.empty())
|
|
||||||
Ctx.setCompilationDir(Opts.DebugCompilationDir);
|
|
||||||
if (!Opts.MainFileName.empty())
|
|
||||||
Ctx.setMainFileName(StringRef(Opts.MainFileName));
|
|
||||||
Ctx.setDwarfVersion(Opts.DwarfVersion);
|
|
||||||
|
|
||||||
// Build up the feature string from the target feature list.
|
|
||||||
std::string FS;
|
|
||||||
if (!Opts.Features.empty()) {
|
|
||||||
FS = Opts.Features[0];
|
|
||||||
for (unsigned i = 1, e = Opts.Features.size(); i != e; ++i)
|
|
||||||
FS += "," + Opts.Features[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
std::unique_ptr<MCStreamer> Str;
|
|
||||||
|
|
||||||
std::unique_ptr<MCInstrInfo> MCII(TheTarget->createMCInstrInfo());
|
|
||||||
std::unique_ptr<MCSubtargetInfo> STI(
|
|
||||||
TheTarget->createMCSubtargetInfo(Opts.Triple, Opts.CPU, FS));
|
|
||||||
|
|
||||||
raw_pwrite_stream *Out = FDOS.get();
|
|
||||||
std::unique_ptr<buffer_ostream> BOS;
|
|
||||||
|
|
||||||
// FIXME: There is a bit of code duplication with addPassesToEmitFile.
|
|
||||||
if (Opts.OutputType == AssemblerInvocation::FT_Asm) {
|
|
||||||
MCInstPrinter *IP = TheTarget->createMCInstPrinter(
|
|
||||||
llvm::Triple(Opts.Triple), Opts.OutputAsmVariant, *MAI, *MCII, *MRI);
|
|
||||||
MCCodeEmitter *CE = nullptr;
|
|
||||||
MCAsmBackend *MAB = nullptr;
|
|
||||||
if (Opts.ShowEncoding) {
|
|
||||||
CE = TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx);
|
|
||||||
MCTargetOptions Options;
|
|
||||||
MAB = TheTarget->createMCAsmBackend(*MRI, Opts.Triple, Opts.CPU, Options);
|
|
||||||
}
|
|
||||||
auto FOut = llvm::make_unique<formatted_raw_ostream>(*Out);
|
|
||||||
Str.reset(TheTarget->createAsmStreamer(
|
|
||||||
Ctx, std::move(FOut), /*asmverbose*/ true,
|
|
||||||
/*useDwarfDirectory*/ true, IP, CE, MAB, Opts.ShowInst));
|
|
||||||
} else if (Opts.OutputType == AssemblerInvocation::FT_Null) {
|
|
||||||
Str.reset(createNullStreamer(Ctx));
|
|
||||||
} else {
|
|
||||||
assert(Opts.OutputType == AssemblerInvocation::FT_Obj &&
|
|
||||||
"Invalid file type!");
|
|
||||||
if (!FDOS->supportsSeeking()) {
|
|
||||||
BOS = make_unique<buffer_ostream>(*FDOS);
|
|
||||||
Out = BOS.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
MCCodeEmitter *CE = TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx);
|
|
||||||
MCTargetOptions Options;
|
|
||||||
MCAsmBackend *MAB = TheTarget->createMCAsmBackend(*MRI, Opts.Triple,
|
|
||||||
Opts.CPU, Options);
|
|
||||||
Triple T(Opts.Triple);
|
|
||||||
Str.reset(TheTarget->createMCObjectStreamer(
|
|
||||||
T, Ctx, *MAB, *Out, CE, *STI, Opts.RelaxAll,
|
|
||||||
Opts.IncrementalLinkerCompatible,
|
|
||||||
/*DWARFMustBeAtTheEnd*/ true));
|
|
||||||
Str.get()->InitSections(Opts.NoExecStack);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool Failed = false;
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmParser> Parser(
|
|
||||||
createMCAsmParser(SrcMgr, Ctx, *Str.get(), *MAI));
|
|
||||||
|
|
||||||
// FIXME: init MCTargetOptions from sanitizer flags here.
|
|
||||||
MCTargetOptions Options;
|
|
||||||
std::unique_ptr<MCTargetAsmParser> TAP(
|
|
||||||
TheTarget->createMCAsmParser(*STI, *Parser, *MCII, Options));
|
|
||||||
if (!TAP)
|
|
||||||
Failed = Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
// Set values for symbols, if any.
|
|
||||||
for (auto &S : Opts.SymbolDefs) {
|
|
||||||
auto Pair = StringRef(S).split('=');
|
|
||||||
auto Sym = Pair.first;
|
|
||||||
auto Val = Pair.second;
|
|
||||||
int64_t Value;
|
|
||||||
// We have already error checked this in the driver.
|
|
||||||
Val.getAsInteger(0, Value);
|
|
||||||
Ctx.setSymbolValue(Parser->getStreamer(), Sym, Value);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!Failed) {
|
|
||||||
Parser->setTargetParser(*TAP.get());
|
|
||||||
Failed = Parser->Run(Opts.NoInitialTextSection);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close Streamer first.
|
|
||||||
// It might have a reference to the output stream.
|
|
||||||
Str.reset();
|
|
||||||
// Close the output stream early.
|
|
||||||
BOS.reset();
|
|
||||||
FDOS.reset();
|
|
||||||
|
|
||||||
// Delete output file if there were errors.
|
|
||||||
if (Failed && Opts.OutputPath != "-")
|
|
||||||
sys::fs::remove(Opts.OutputPath);
|
|
||||||
|
|
||||||
return Failed;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool GenCrashDiag) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors.
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
int cc1as_main(ArrayRef<const char *> Argv, const char *Argv0, void *MainAddr) {
|
|
||||||
// Initialize targets and assembly printers/parsers.
|
|
||||||
InitializeAllTargetInfos();
|
|
||||||
InitializeAllTargetMCs();
|
|
||||||
InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
// Construct our diagnostic client.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(errs(), &*DiagOpts);
|
|
||||||
DiagClient->setPrefix("clang -cc1as");
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
ScopedFatalErrorHandler FatalErrorHandler
|
|
||||||
(LLVMErrorHandler, static_cast<void*>(&Diags));
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
AssemblerInvocation Asm;
|
|
||||||
if (!AssemblerInvocation::CreateFromArgs(Asm, Argv, Diags))
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
if (Asm.ShowHelp) {
|
|
||||||
std::unique_ptr<OptTable> Opts(driver::createDriverOptTable());
|
|
||||||
Opts->PrintHelp(llvm::outs(), "clang -cc1as", "Clang Integrated Assembler",
|
|
||||||
/*Include=*/driver::options::CC1AsOption, /*Exclude=*/0);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -version.
|
|
||||||
//
|
|
||||||
// FIXME: Use a better -version message?
|
|
||||||
if (Asm.ShowVersion) {
|
|
||||||
llvm::cl::PrintVersionMessage();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -mllvm.
|
|
||||||
//
|
|
||||||
// FIXME: Remove this, one day.
|
|
||||||
if (!Asm.LLVMArgs.empty()) {
|
|
||||||
unsigned NumArgs = Asm.LLVMArgs.size();
|
|
||||||
auto Args = llvm::make_unique<const char*[]>(NumArgs + 2);
|
|
||||||
Args[0] = "clang (LLVM option parsing)";
|
|
||||||
for (unsigned i = 0; i != NumArgs; ++i)
|
|
||||||
Args[i + 1] = Asm.LLVMArgs[i].c_str();
|
|
||||||
Args[NumArgs + 1] = nullptr;
|
|
||||||
llvm::cl::ParseCommandLineOptions(NumArgs + 1, Args.get());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Execute the invocation, unless there were parsing errors.
|
|
||||||
bool Failed = Diags.hasErrorOccurred() || ExecuteAssembler(Asm, Diags);
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now.
|
|
||||||
TimerGroup::printAll(errs());
|
|
||||||
|
|
||||||
return !!Failed;
|
|
||||||
}
|
|
@ -1,519 +0,0 @@
|
|||||||
//===-- driver.cpp - Clang GCC-Compatible Driver --------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang driver; it is a thin wrapper
|
|
||||||
// for functionality in the Driver clang library.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/Compilation.h"
|
|
||||||
#include "clang/Driver/Driver.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Driver/ToolChain.h"
|
|
||||||
#include "clang/Frontend/ChainedDiagnosticConsumer.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/SerializedDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/ArrayRef.h"
|
|
||||||
#include "llvm/ADT/SmallString.h"
|
|
||||||
#include "llvm/ADT/SmallVector.h"
|
|
||||||
#include "llvm/Config/llvm-config.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Option/Option.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/PrettyStackTrace.h"
|
|
||||||
#include "llvm/Support/Process.h"
|
|
||||||
#include "llvm/Support/Program.h"
|
|
||||||
#include "llvm/Support/Regex.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/StringSaver.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <set>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
std::string GetExecutablePath(const char *Argv0, bool CanonicalPrefixes) {
|
|
||||||
if (!CanonicalPrefixes) {
|
|
||||||
SmallString<128> ExecutablePath(Argv0);
|
|
||||||
// Do a PATH lookup if Argv0 isn't a valid path.
|
|
||||||
if (!llvm::sys::fs::exists(ExecutablePath))
|
|
||||||
if (llvm::ErrorOr<std::string> P =
|
|
||||||
llvm::sys::findProgramByName(ExecutablePath))
|
|
||||||
ExecutablePath = *P;
|
|
||||||
return ExecutablePath.str();
|
|
||||||
}
|
|
||||||
|
|
||||||
// This just needs to be some symbol in the binary; C++ doesn't
|
|
||||||
// allow taking the address of ::main however.
|
|
||||||
void *P = (void*) (intptr_t) GetExecutablePath;
|
|
||||||
return llvm::sys::fs::getMainExecutable(Argv0, P);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const char *GetStableCStr(std::set<std::string> &SavedStrings,
|
|
||||||
StringRef S) {
|
|
||||||
return SavedStrings.insert(S).first->c_str();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a list of edits to the input argument lists.
|
|
||||||
///
|
|
||||||
/// The input string is a space separate list of edits to perform,
|
|
||||||
/// they are applied in order to the input argument lists. Edits
|
|
||||||
/// should be one of the following forms:
|
|
||||||
///
|
|
||||||
/// '#': Silence information about the changes to the command line arguments.
|
|
||||||
///
|
|
||||||
/// '^': Add FOO as a new argument at the beginning of the command line.
|
|
||||||
///
|
|
||||||
/// '+': Add FOO as a new argument at the end of the command line.
|
|
||||||
///
|
|
||||||
/// 's/XXX/YYY/': Substitute the regular expression XXX with YYY in the command
|
|
||||||
/// line.
|
|
||||||
///
|
|
||||||
/// 'xOPTION': Removes all instances of the literal argument OPTION.
|
|
||||||
///
|
|
||||||
/// 'XOPTION': Removes all instances of the literal argument OPTION,
|
|
||||||
/// and the following argument.
|
|
||||||
///
|
|
||||||
/// 'Ox': Removes all flags matching 'O' or 'O[sz0-9]' and adds 'Ox'
|
|
||||||
/// at the end of the command line.
|
|
||||||
///
|
|
||||||
/// \param OS - The stream to write edit information to.
|
|
||||||
/// \param Args - The vector of command line arguments.
|
|
||||||
/// \param Edit - The override command to perform.
|
|
||||||
/// \param SavedStrings - Set to use for storing string representations.
|
|
||||||
static void ApplyOneQAOverride(raw_ostream &OS,
|
|
||||||
SmallVectorImpl<const char*> &Args,
|
|
||||||
StringRef Edit,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
if (Edit[0] == '^') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at beginning\n";
|
|
||||||
Args.insert(Args.begin() + 1, Str);
|
|
||||||
} else if (Edit[0] == '+') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at end\n";
|
|
||||||
Args.push_back(Str);
|
|
||||||
} else if (Edit[0] == 's' && Edit[1] == '/' && Edit.endswith("/") &&
|
|
||||||
Edit.slice(2, Edit.size()-1).find('/') != StringRef::npos) {
|
|
||||||
StringRef MatchPattern = Edit.substr(2).split('/').first;
|
|
||||||
StringRef ReplPattern = Edit.substr(2).split('/').second;
|
|
||||||
ReplPattern = ReplPattern.slice(0, ReplPattern.size()-1);
|
|
||||||
|
|
||||||
for (unsigned i = 1, e = Args.size(); i != e; ++i) {
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (Args[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
std::string Repl = llvm::Regex(MatchPattern).sub(ReplPattern, Args[i]);
|
|
||||||
|
|
||||||
if (Repl != Args[i]) {
|
|
||||||
OS << "### Replacing '" << Args[i] << "' with '" << Repl << "'\n";
|
|
||||||
Args[i] = GetStableCStr(SavedStrings, Repl);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'x' || Edit[0] == 'X') {
|
|
||||||
auto Option = Edit.substr(1);
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
if (Option == Args[i]) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
if (Edit[0] == 'X') {
|
|
||||||
if (i < Args.size()) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
OS << "### Invalid X edit, end of command line!\n";
|
|
||||||
}
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'O') {
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
const char *A = Args[i];
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (A == nullptr)
|
|
||||||
continue;
|
|
||||||
if (A[0] == '-' && A[1] == 'O' &&
|
|
||||||
(A[2] == '\0' ||
|
|
||||||
(A[3] == '\0' && (A[2] == 's' || A[2] == 'z' ||
|
|
||||||
('0' <= A[2] && A[2] <= '9'))))) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
OS << "### Adding argument " << Edit << " at end\n";
|
|
||||||
Args.push_back(GetStableCStr(SavedStrings, '-' + Edit.str()));
|
|
||||||
} else {
|
|
||||||
OS << "### Unrecognized edit: " << Edit << "\n";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a comma separate list of edits to the
|
|
||||||
/// input argument lists. See ApplyOneQAOverride.
|
|
||||||
static void ApplyQAOverride(SmallVectorImpl<const char*> &Args,
|
|
||||||
const char *OverrideStr,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
raw_ostream *OS = &llvm::errs();
|
|
||||||
|
|
||||||
if (OverrideStr[0] == '#') {
|
|
||||||
++OverrideStr;
|
|
||||||
OS = &llvm::nulls();
|
|
||||||
}
|
|
||||||
|
|
||||||
*OS << "### CCC_OVERRIDE_OPTIONS: " << OverrideStr << "\n";
|
|
||||||
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
const char *S = OverrideStr;
|
|
||||||
while (*S) {
|
|
||||||
const char *End = ::strchr(S, ' ');
|
|
||||||
if (!End)
|
|
||||||
End = S + strlen(S);
|
|
||||||
if (End != S)
|
|
||||||
ApplyOneQAOverride(*OS, Args, std::string(S, End), SavedStrings);
|
|
||||||
S = End;
|
|
||||||
if (*S != '\0')
|
|
||||||
++S;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
extern int cc1_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
extern int cc1as_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
|
|
||||||
static void insertTargetAndModeArgs(StringRef Target, StringRef Mode,
|
|
||||||
SmallVectorImpl<const char *> &ArgVector,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
if (!Mode.empty()) {
|
|
||||||
// Add the mode flag to the arguments.
|
|
||||||
auto it = ArgVector.begin();
|
|
||||||
if (it != ArgVector.end())
|
|
||||||
++it;
|
|
||||||
ArgVector.insert(it, GetStableCStr(SavedStrings, Mode));
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!Target.empty()) {
|
|
||||||
auto it = ArgVector.begin();
|
|
||||||
if (it != ArgVector.end())
|
|
||||||
++it;
|
|
||||||
const char *arr[] = {"-target", GetStableCStr(SavedStrings, Target)};
|
|
||||||
ArgVector.insert(it, std::begin(arr), std::end(arr));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void getCLEnvVarOptions(std::string &EnvValue, llvm::StringSaver &Saver,
|
|
||||||
SmallVectorImpl<const char *> &Opts) {
|
|
||||||
llvm::cl::TokenizeWindowsCommandLine(EnvValue, Saver, Opts);
|
|
||||||
// The first instance of '#' should be replaced with '=' in each option.
|
|
||||||
for (const char *Opt : Opts)
|
|
||||||
if (char *NumberSignPtr = const_cast<char *>(::strchr(Opt, '#')))
|
|
||||||
*NumberSignPtr = '=';
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetBackdoorDriverOutputsFromEnvVars(Driver &TheDriver) {
|
|
||||||
// Handle CC_PRINT_OPTIONS and CC_PRINT_OPTIONS_FILE.
|
|
||||||
TheDriver.CCPrintOptions = !!::getenv("CC_PRINT_OPTIONS");
|
|
||||||
if (TheDriver.CCPrintOptions)
|
|
||||||
TheDriver.CCPrintOptionsFilename = ::getenv("CC_PRINT_OPTIONS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_PRINT_HEADERS and CC_PRINT_HEADERS_FILE.
|
|
||||||
TheDriver.CCPrintHeaders = !!::getenv("CC_PRINT_HEADERS");
|
|
||||||
if (TheDriver.CCPrintHeaders)
|
|
||||||
TheDriver.CCPrintHeadersFilename = ::getenv("CC_PRINT_HEADERS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_LOG_DIAGNOSTICS and CC_LOG_DIAGNOSTICS_FILE.
|
|
||||||
TheDriver.CCLogDiagnostics = !!::getenv("CC_LOG_DIAGNOSTICS");
|
|
||||||
if (TheDriver.CCLogDiagnostics)
|
|
||||||
TheDriver.CCLogDiagnosticsFilename = ::getenv("CC_LOG_DIAGNOSTICS_FILE");
|
|
||||||
}
|
|
||||||
|
|
||||||
static void FixupDiagPrefixExeName(TextDiagnosticPrinter *DiagClient,
|
|
||||||
const std::string &Path) {
|
|
||||||
// If the clang binary happens to be named cl.exe for compatibility reasons,
|
|
||||||
// use clang-cl.exe as the prefix to avoid confusion between clang and MSVC.
|
|
||||||
StringRef ExeBasename(llvm::sys::path::filename(Path));
|
|
||||||
if (ExeBasename.equals_lower("cl.exe"))
|
|
||||||
ExeBasename = "clang-cl.exe";
|
|
||||||
DiagClient->setPrefix(ExeBasename);
|
|
||||||
}
|
|
||||||
|
|
||||||
// This lets us create the DiagnosticsEngine with a properly-filled-out
|
|
||||||
// DiagnosticOptions instance.
|
|
||||||
static DiagnosticOptions *
|
|
||||||
CreateAndPopulateDiagOpts(ArrayRef<const char *> argv) {
|
|
||||||
auto *DiagOpts = new DiagnosticOptions;
|
|
||||||
std::unique_ptr<OptTable> Opts(createDriverOptTable());
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args =
|
|
||||||
Opts->ParseArgs(argv.slice(1), MissingArgIndex, MissingArgCount);
|
|
||||||
// We ignore MissingArgCount and the return value of ParseDiagnosticArgs.
|
|
||||||
// Any errors that would be diagnosed here will also be diagnosed later,
|
|
||||||
// when the DiagnosticsEngine actually exists.
|
|
||||||
(void)ParseDiagnosticArgs(*DiagOpts, Args);
|
|
||||||
return DiagOpts;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetInstallDir(SmallVectorImpl<const char *> &argv,
|
|
||||||
Driver &TheDriver, bool CanonicalPrefixes) {
|
|
||||||
// Attempt to find the original path used to invoke the driver, to determine
|
|
||||||
// the installed path. We do this manually, because we want to support that
|
|
||||||
// path being a symlink.
|
|
||||||
SmallString<128> InstalledPath(argv[0]);
|
|
||||||
|
|
||||||
// Do a PATH lookup, if there are no directory components.
|
|
||||||
if (llvm::sys::path::filename(InstalledPath) == InstalledPath)
|
|
||||||
if (llvm::ErrorOr<std::string> Tmp = llvm::sys::findProgramByName(
|
|
||||||
llvm::sys::path::filename(InstalledPath.str())))
|
|
||||||
InstalledPath = *Tmp;
|
|
||||||
|
|
||||||
// FIXME: We don't actually canonicalize this, we just make it absolute.
|
|
||||||
if (CanonicalPrefixes)
|
|
||||||
llvm::sys::fs::make_absolute(InstalledPath);
|
|
||||||
|
|
||||||
StringRef InstalledPathParent(llvm::sys::path::parent_path(InstalledPath));
|
|
||||||
if (llvm::sys::fs::exists(InstalledPathParent))
|
|
||||||
TheDriver.setInstalledDir(InstalledPathParent);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int ExecuteCC1Tool(ArrayRef<const char *> argv, StringRef Tool) {
|
|
||||||
void *GetExecutablePathVP = (void *)(intptr_t) GetExecutablePath;
|
|
||||||
if (Tool == "")
|
|
||||||
return cc1_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
if (Tool == "as")
|
|
||||||
return cc1as_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
|
|
||||||
// Reject unknown tools.
|
|
||||||
llvm::errs() << "error: unknown integrated tool '" << Tool << "'\n";
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
int mainEntryClickHouseClang(int argc_, char **argv_) {
|
|
||||||
llvm::sys::PrintStackTraceOnErrorSignal(argv_[0]);
|
|
||||||
llvm::PrettyStackTraceProgram X(argc_, argv_);
|
|
||||||
llvm::llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
|
|
||||||
|
|
||||||
if (llvm::sys::Process::FixupStandardFileDescriptors())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
SmallVector<const char *, 256> argv;
|
|
||||||
llvm::SpecificBumpPtrAllocator<char> ArgAllocator;
|
|
||||||
std::error_code EC = llvm::sys::Process::GetArgumentVector(
|
|
||||||
argv, llvm::makeArrayRef(argv_, argc_), ArgAllocator);
|
|
||||||
if (EC) {
|
|
||||||
llvm::errs() << "error: couldn't get arguments: " << EC.message() << '\n';
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
std::string ProgName = argv[0];
|
|
||||||
std::pair<std::string, std::string> TargetAndMode =
|
|
||||||
ToolChain::getTargetAndModeFromProgramName(ProgName);
|
|
||||||
|
|
||||||
llvm::BumpPtrAllocator A;
|
|
||||||
llvm::StringSaver Saver(A);
|
|
||||||
|
|
||||||
// Parse response files using the GNU syntax, unless we're in CL mode. There
|
|
||||||
// are two ways to put clang in CL compatibility mode: argv[0] is either
|
|
||||||
// clang-cl or cl, or --driver-mode=cl is on the command line. The normal
|
|
||||||
// command line parsing can't happen until after response file parsing, so we
|
|
||||||
// have to manually search for a --driver-mode=cl argument the hard way.
|
|
||||||
// Finally, our -cc1 tools don't care which tokenization mode we use because
|
|
||||||
// response files written by clang will tokenize the same way in either mode.
|
|
||||||
bool ClangCLMode = false;
|
|
||||||
if (TargetAndMode.second == "--driver-mode=cl" ||
|
|
||||||
std::find_if(argv.begin(), argv.end(), [](const char *F) {
|
|
||||||
return F && strcmp(F, "--driver-mode=cl") == 0;
|
|
||||||
}) != argv.end()) {
|
|
||||||
ClangCLMode = true;
|
|
||||||
}
|
|
||||||
enum { Default, POSIX, Windows } RSPQuoting = Default;
|
|
||||||
for (const char *F : argv) {
|
|
||||||
if (strcmp(F, "--rsp-quoting=posix") == 0)
|
|
||||||
RSPQuoting = POSIX;
|
|
||||||
else if (strcmp(F, "--rsp-quoting=windows") == 0)
|
|
||||||
RSPQuoting = Windows;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Determines whether we want nullptr markers in argv to indicate response
|
|
||||||
// files end-of-lines. We only use this for the /LINK driver argument with
|
|
||||||
// clang-cl.exe on Windows.
|
|
||||||
bool MarkEOLs = ClangCLMode;
|
|
||||||
|
|
||||||
llvm::cl::TokenizerCallback Tokenizer;
|
|
||||||
if (RSPQuoting == Windows || (RSPQuoting == Default && ClangCLMode))
|
|
||||||
Tokenizer = &llvm::cl::TokenizeWindowsCommandLine;
|
|
||||||
else
|
|
||||||
Tokenizer = &llvm::cl::TokenizeGNUCommandLine;
|
|
||||||
|
|
||||||
if (MarkEOLs && argv.size() > 1 && StringRef(argv[1]).startswith("-cc1"))
|
|
||||||
MarkEOLs = false;
|
|
||||||
llvm::cl::ExpandResponseFiles(Saver, Tokenizer, argv, MarkEOLs);
|
|
||||||
|
|
||||||
// Handle -cc1 integrated tools, even if -cc1 was expanded from a response
|
|
||||||
// file.
|
|
||||||
auto FirstArg = std::find_if(argv.begin() + 1, argv.end(),
|
|
||||||
[](const char *A) { return A != nullptr; });
|
|
||||||
if (FirstArg != argv.end() && StringRef(*FirstArg).startswith("-cc1")) {
|
|
||||||
// If -cc1 came from a response file, remove the EOL sentinels.
|
|
||||||
if (MarkEOLs) {
|
|
||||||
auto newEnd = std::remove(argv.begin(), argv.end(), nullptr);
|
|
||||||
argv.resize(newEnd - argv.begin());
|
|
||||||
}
|
|
||||||
return ExecuteCC1Tool(argv, argv[1] + 4);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool CanonicalPrefixes = true;
|
|
||||||
for (int i = 1, size = argv.size(); i < size; ++i) {
|
|
||||||
// Skip end-of-line response file markers
|
|
||||||
if (argv[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
if (StringRef(argv[i]) == "-no-canonical-prefixes") {
|
|
||||||
CanonicalPrefixes = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle CL and _CL_ which permits additional command line options to be
|
|
||||||
// prepended or appended.
|
|
||||||
if (ClangCLMode) {
|
|
||||||
// Arguments in "CL" are prepended.
|
|
||||||
llvm::Optional<std::string> OptCL = llvm::sys::Process::GetEnv("CL");
|
|
||||||
if (OptCL.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> PrependedOpts;
|
|
||||||
getCLEnvVarOptions(OptCL.getValue(), Saver, PrependedOpts);
|
|
||||||
|
|
||||||
// Insert right after the program name to prepend to the argument list.
|
|
||||||
argv.insert(argv.begin() + 1, PrependedOpts.begin(), PrependedOpts.end());
|
|
||||||
}
|
|
||||||
// Arguments in "_CL_" are appended.
|
|
||||||
llvm::Optional<std::string> Opt_CL_ = llvm::sys::Process::GetEnv("_CL_");
|
|
||||||
if (Opt_CL_.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> AppendedOpts;
|
|
||||||
getCLEnvVarOptions(Opt_CL_.getValue(), Saver, AppendedOpts);
|
|
||||||
|
|
||||||
// Insert at the end of the argument list to append.
|
|
||||||
argv.append(AppendedOpts.begin(), AppendedOpts.end());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::set<std::string> SavedStrings;
|
|
||||||
// Handle CCC_OVERRIDE_OPTIONS, used for editing a command line behind the
|
|
||||||
// scenes.
|
|
||||||
if (const char *OverrideStr = ::getenv("CCC_OVERRIDE_OPTIONS")) {
|
|
||||||
// FIXME: Driver shouldn't take extra initial argument.
|
|
||||||
ApplyQAOverride(argv, OverrideStr, SavedStrings);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string Path = GetExecutablePath(argv[0], CanonicalPrefixes);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts =
|
|
||||||
CreateAndPopulateDiagOpts(argv);
|
|
||||||
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(llvm::errs(), &*DiagOpts);
|
|
||||||
FixupDiagPrefixExeName(DiagClient, Path);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
if (!DiagOpts->DiagnosticSerializationFile.empty()) {
|
|
||||||
auto SerializedConsumer =
|
|
||||||
clang::serialized_diags::create(DiagOpts->DiagnosticSerializationFile,
|
|
||||||
&*DiagOpts, /*MergeChildRecords=*/true);
|
|
||||||
Diags.setClient(new ChainedDiagnosticConsumer(
|
|
||||||
Diags.takeClient(), std::move(SerializedConsumer)));
|
|
||||||
}
|
|
||||||
|
|
||||||
ProcessWarningOptions(Diags, *DiagOpts, /*ReportDiags=*/false);
|
|
||||||
|
|
||||||
Driver TheDriver(Path, llvm::sys::getDefaultTargetTriple(), Diags);
|
|
||||||
SetInstallDir(argv, TheDriver, CanonicalPrefixes);
|
|
||||||
|
|
||||||
insertTargetAndModeArgs(TargetAndMode.first, TargetAndMode.second, argv,
|
|
||||||
SavedStrings);
|
|
||||||
|
|
||||||
SetBackdoorDriverOutputsFromEnvVars(TheDriver);
|
|
||||||
|
|
||||||
std::unique_ptr<Compilation> C(TheDriver.BuildCompilation(argv));
|
|
||||||
int Res = 1;
|
|
||||||
if (C && !C->containsError()) {
|
|
||||||
SmallVector<std::pair<int, const Command *>, 4> FailingCommands;
|
|
||||||
Res = TheDriver.ExecuteCompilation(*C, FailingCommands);
|
|
||||||
|
|
||||||
// Force a crash to test the diagnostics.
|
|
||||||
if (TheDriver.GenReproducer) {
|
|
||||||
Diags.Report(diag::err_drv_force_crash)
|
|
||||||
<< !::getenv("FORCE_CLANG_DIAGNOSTICS_CRASH");
|
|
||||||
|
|
||||||
// Pretend that every command failed.
|
|
||||||
FailingCommands.clear();
|
|
||||||
for (const auto &J : C->getJobs())
|
|
||||||
if (const Command *C = dyn_cast<Command>(&J))
|
|
||||||
FailingCommands.push_back(std::make_pair(-1, C));
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const auto &P : FailingCommands) {
|
|
||||||
int CommandRes = P.first;
|
|
||||||
const Command *FailingCommand = P.second;
|
|
||||||
if (!Res)
|
|
||||||
Res = CommandRes;
|
|
||||||
|
|
||||||
// If result status is < 0, then the driver command signalled an error.
|
|
||||||
// If result status is 70, then the driver command reported a fatal error.
|
|
||||||
// On Windows, abort will return an exit code of 3. In these cases,
|
|
||||||
// generate additional diagnostic information if possible.
|
|
||||||
bool DiagnoseCrash = CommandRes < 0 || CommandRes == 70;
|
|
||||||
#ifdef LLVM_ON_WIN32
|
|
||||||
DiagnoseCrash |= CommandRes == 3;
|
|
||||||
#endif
|
|
||||||
if (DiagnoseCrash) {
|
|
||||||
TheDriver.generateCompilationDiagnostics(*C, *FailingCommand);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Diags.getClient()->finish();
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
#ifdef LLVM_ON_WIN32
|
|
||||||
// Exit status should not be negative on Win32, unless abnormal termination.
|
|
||||||
// Once abnormal termiation was caught, negative status should not be
|
|
||||||
// propagated.
|
|
||||||
if (Res < 0)
|
|
||||||
Res = 1;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// If we have multiple failing commands, we return the result of the first
|
|
||||||
// failing command.
|
|
||||||
return Res;
|
|
||||||
}
|
|
@ -1,23 +0,0 @@
|
|||||||
#include "lld/Driver/Driver.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/ADT/StringSwitch.h"
|
|
||||||
#include "llvm/ADT/Twine.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/PrettyStackTrace.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
|
|
||||||
using namespace lld;
|
|
||||||
using namespace llvm;
|
|
||||||
using namespace llvm::sys;
|
|
||||||
|
|
||||||
int mainEntryClickHouseLLD(int Argc, char **Argv)
|
|
||||||
{
|
|
||||||
// Standard set up, so program fails gracefully.
|
|
||||||
sys::PrintStackTraceOnErrorSignal(Argv[0]);
|
|
||||||
PrettyStackTraceProgram StackPrinter(Argc, Argv);
|
|
||||||
llvm_shutdown_obj Shutdown;
|
|
||||||
|
|
||||||
std::vector<const char *> Args(Argv, Argv + Argc);
|
|
||||||
return !elf::link(Args, true);
|
|
||||||
}
|
|
@ -1 +0,0 @@
|
|||||||
Compiler-5.0.0
|
|
@ -1 +0,0 @@
|
|||||||
Compiler-5.0.0
|
|
@ -1,54 +0,0 @@
|
|||||||
|
|
||||||
add_definitions(-Wno-error -Wno-unused-parameter -Wno-non-virtual-dtor -U_LIBCPP_DEBUG)
|
|
||||||
|
|
||||||
link_directories(${LLVM_LIBRARY_DIRS})
|
|
||||||
|
|
||||||
add_library(clickhouse-compiler-lib
|
|
||||||
driver.cpp
|
|
||||||
cc1_main.cpp
|
|
||||||
cc1as_main.cpp
|
|
||||||
lld.cpp)
|
|
||||||
|
|
||||||
target_compile_options(clickhouse-compiler-lib PRIVATE -fno-rtti -fno-exceptions -g0)
|
|
||||||
|
|
||||||
string(REPLACE "${INCLUDE_DEBUG_HELPERS}" "" CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS}) # cant compile with -fno-rtti
|
|
||||||
|
|
||||||
llvm_libs_all(REQUIRED_LLVM_LIBRARIES)
|
|
||||||
|
|
||||||
message(STATUS "Using LLVM ${LLVM_VERSION}: ${LLVM_INCLUDE_DIRS} : ${REQUIRED_LLVM_LIBRARIES}")
|
|
||||||
|
|
||||||
target_include_directories(clickhouse-compiler-lib SYSTEM PRIVATE ${LLVM_INCLUDE_DIRS})
|
|
||||||
|
|
||||||
# This is extracted almost directly from CMakeFiles/.../link.txt in LLVM build directory.
|
|
||||||
|
|
||||||
target_link_libraries(clickhouse-compiler-lib PRIVATE
|
|
||||||
|
|
||||||
clangBasic clangCodeGen clangDriver
|
|
||||||
clangFrontend
|
|
||||||
clangFrontendTool
|
|
||||||
clangRewriteFrontend clangARCMigrate clangStaticAnalyzerFrontend
|
|
||||||
clangParse clangSerialization clangSema clangEdit clangStaticAnalyzerCheckers
|
|
||||||
clangASTMatchers clangStaticAnalyzerCore clangAnalysis clangAST clangRewrite clangLex clangBasic
|
|
||||||
|
|
||||||
lldCOFF
|
|
||||||
lldDriver
|
|
||||||
lldELF
|
|
||||||
lldMinGW
|
|
||||||
lldMachO
|
|
||||||
lldReaderWriter
|
|
||||||
lldYAML
|
|
||||||
lldCommon
|
|
||||||
lldCore
|
|
||||||
#lldWasm
|
|
||||||
|
|
||||||
${REQUIRED_LLVM_LIBRARIES}
|
|
||||||
|
|
||||||
#Polly
|
|
||||||
#PollyISL
|
|
||||||
#PollyPPCG
|
|
||||||
|
|
||||||
PUBLIC ${ZLIB_LIBRARIES} ${EXECINFO_LIBRARIES} Threads::Threads
|
|
||||||
${MALLOC_LIBRARIES}
|
|
||||||
${GLIBC_COMPATIBILITY_LIBRARIES}
|
|
||||||
${MEMCPY_LIBRARIES}
|
|
||||||
)
|
|
@ -1,63 +0,0 @@
|
|||||||
==============================================================================
|
|
||||||
LLVM Release License
|
|
||||||
==============================================================================
|
|
||||||
University of Illinois/NCSA
|
|
||||||
Open Source License
|
|
||||||
|
|
||||||
Copyright (c) 2007-2016 University of Illinois at Urbana-Champaign.
|
|
||||||
All rights reserved.
|
|
||||||
|
|
||||||
Developed by:
|
|
||||||
|
|
||||||
LLVM Team
|
|
||||||
|
|
||||||
University of Illinois at Urbana-Champaign
|
|
||||||
|
|
||||||
http://llvm.org
|
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
||||||
this software and associated documentation files (the "Software"), to deal with
|
|
||||||
the Software without restriction, including without limitation the rights to
|
|
||||||
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
||||||
of the Software, and to permit persons to whom the Software is furnished to do
|
|
||||||
so, subject to the following conditions:
|
|
||||||
|
|
||||||
* Redistributions of source code must retain the above copyright notice,
|
|
||||||
this list of conditions and the following disclaimers.
|
|
||||||
|
|
||||||
* Redistributions in binary form must reproduce the above copyright notice,
|
|
||||||
this list of conditions and the following disclaimers in the
|
|
||||||
documentation and/or other materials provided with the distribution.
|
|
||||||
|
|
||||||
* Neither the names of the LLVM Team, University of Illinois at
|
|
||||||
Urbana-Champaign, nor the names of its contributors may be used to
|
|
||||||
endorse or promote products derived from this Software without specific
|
|
||||||
prior written permission.
|
|
||||||
|
|
||||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
||||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
||||||
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
||||||
CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
||||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
||||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS WITH THE
|
|
||||||
SOFTWARE.
|
|
||||||
|
|
||||||
==============================================================================
|
|
||||||
The LLVM software contains code written by third parties. Such software will
|
|
||||||
have its own individual LICENSE.TXT file in the directory in which it appears.
|
|
||||||
This file will describe the copyrights, license, and restrictions which apply
|
|
||||||
to that code.
|
|
||||||
|
|
||||||
The disclaimer of warranty in the University of Illinois Open Source License
|
|
||||||
applies to all code in the LLVM Distribution, and nothing in any of the
|
|
||||||
other licenses gives permission to use the names of the LLVM Team or the
|
|
||||||
University of Illinois to endorse or promote products derived from this
|
|
||||||
Software.
|
|
||||||
|
|
||||||
The following pieces of software have additional or alternate copyrights,
|
|
||||||
licenses, and/or restrictions:
|
|
||||||
|
|
||||||
Program Directory
|
|
||||||
------- ---------
|
|
||||||
<none yet>
|
|
||||||
|
|
@ -1,242 +0,0 @@
|
|||||||
//===-- cc1_main.cpp - Clang CC1 Compiler Frontend ------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1 functionality, which implements the
|
|
||||||
// core compiler functionality along with a number of additional tools for
|
|
||||||
// demonstration and testing purposes.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "clang/CodeGen/ObjectFilePCHContainerOperations.h"
|
|
||||||
#include "clang/Config/config.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/CompilerInstance.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticBuffer.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "clang/FrontendTool/Utils.h"
|
|
||||||
#include "llvm/ADT/Statistic.h"
|
|
||||||
#include "llvm/LinkAllPasses.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/Compiler.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <cstdio>
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
#include <sys/resource.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// have no .a version in packages
|
|
||||||
#undef LINK_POLLY_INTO_TOOLS
|
|
||||||
|
|
||||||
using namespace clang;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
// Main driver
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool GenCrashDiag) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// Run the interrupt handlers to make sure any special cleanups get done, in
|
|
||||||
// particular that we remove files registered with RemoveFileOnSignal.
|
|
||||||
llvm::sys::RunInterruptHandlers();
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors. When reporting a fatal error, exit
|
|
||||||
// with status 70 to generate crash diagnostics. For BSD systems this is
|
|
||||||
// defined as an internal software error. Otherwise, exit with status 1.
|
|
||||||
exit(GenCrashDiag ? 70 : 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
namespace polly {
|
|
||||||
void initializePollyPasses(llvm::PassRegistry &Registry);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
// The amount of stack we think is "sufficient". If less than this much is
|
|
||||||
// available, we may be unable to reach our template instantiation depth
|
|
||||||
// limit and other similar limits.
|
|
||||||
// FIXME: Unify this with the stack we request when spawning a thread to build
|
|
||||||
// a module.
|
|
||||||
static const int kSufficientStack = 8 << 20;
|
|
||||||
|
|
||||||
#if defined(__linux__) && defined(__PIE__)
|
|
||||||
static size_t getCurrentStackAllocation() {
|
|
||||||
// If we can't compute the current stack usage, allow for 512K of command
|
|
||||||
// line arguments and environment.
|
|
||||||
size_t Usage = 512 * 1024;
|
|
||||||
if (FILE *StatFile = fopen("/proc/self/stat", "r")) {
|
|
||||||
// We assume that the stack extends from its current address to the end of
|
|
||||||
// the environment space. In reality, there is another string literal (the
|
|
||||||
// program name) after the environment, but this is close enough (we only
|
|
||||||
// need to be within 100K or so).
|
|
||||||
unsigned long StackPtr, EnvEnd;
|
|
||||||
// Disable silly GCC -Wformat warning that complains about length
|
|
||||||
// modifiers on ignored format specifiers. We want to retain these
|
|
||||||
// for documentation purposes even though they have no effect.
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic push
|
|
||||||
#pragma GCC diagnostic ignored "-Wformat"
|
|
||||||
#endif
|
|
||||||
if (fscanf(StatFile,
|
|
||||||
"%*d %*s %*c %*d %*d %*d %*d %*d %*u %*lu %*lu %*lu %*lu %*lu "
|
|
||||||
"%*lu %*ld %*ld %*ld %*ld %*ld %*ld %*llu %*lu %*ld %*lu %*lu "
|
|
||||||
"%*lu %*lu %lu %*lu %*lu %*lu %*lu %*lu %*llu %*lu %*lu %*d %*d "
|
|
||||||
"%*u %*u %*llu %*lu %*ld %*lu %*lu %*lu %*lu %*lu %*lu %lu %*d",
|
|
||||||
&StackPtr, &EnvEnd) == 2) {
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic pop
|
|
||||||
#endif
|
|
||||||
Usage = StackPtr < EnvEnd ? EnvEnd - StackPtr : StackPtr - EnvEnd;
|
|
||||||
}
|
|
||||||
fclose(StatFile);
|
|
||||||
}
|
|
||||||
return Usage;
|
|
||||||
}
|
|
||||||
|
|
||||||
#include <alloca.h>
|
|
||||||
|
|
||||||
LLVM_ATTRIBUTE_NOINLINE
|
|
||||||
static void ensureStackAddressSpace(int ExtraChunks = 0) {
|
|
||||||
// Linux kernels prior to 4.1 will sometimes locate the heap of a PIE binary
|
|
||||||
// relatively close to the stack (they are only guaranteed to be 128MiB
|
|
||||||
// apart). This results in crashes if we happen to heap-allocate more than
|
|
||||||
// 128MiB before we reach our stack high-water mark.
|
|
||||||
//
|
|
||||||
// To avoid these crashes, ensure that we have sufficient virtual memory
|
|
||||||
// pages allocated before we start running.
|
|
||||||
size_t Curr = getCurrentStackAllocation();
|
|
||||||
const int kTargetStack = kSufficientStack - 256 * 1024;
|
|
||||||
if (Curr < kTargetStack) {
|
|
||||||
volatile char *volatile Alloc =
|
|
||||||
static_cast<volatile char *>(alloca(kTargetStack - Curr));
|
|
||||||
Alloc[0] = 0;
|
|
||||||
Alloc[kTargetStack - Curr - 1] = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureStackAddressSpace() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/// Attempt to ensure that we have at least 8MiB of usable stack space.
|
|
||||||
static void ensureSufficientStack() {
|
|
||||||
struct rlimit rlim;
|
|
||||||
if (getrlimit(RLIMIT_STACK, &rlim) != 0)
|
|
||||||
return;
|
|
||||||
|
|
||||||
// Increase the soft stack limit to our desired level, if necessary and
|
|
||||||
// possible.
|
|
||||||
if (rlim.rlim_cur != RLIM_INFINITY && rlim.rlim_cur < kSufficientStack) {
|
|
||||||
// Try to allocate sufficient stack.
|
|
||||||
if (rlim.rlim_max == RLIM_INFINITY || rlim.rlim_max >= kSufficientStack)
|
|
||||||
rlim.rlim_cur = kSufficientStack;
|
|
||||||
else if (rlim.rlim_cur == rlim.rlim_max)
|
|
||||||
return;
|
|
||||||
else
|
|
||||||
rlim.rlim_cur = rlim.rlim_max;
|
|
||||||
|
|
||||||
if (setrlimit(RLIMIT_STACK, &rlim) != 0 ||
|
|
||||||
rlim.rlim_cur != kSufficientStack)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// We should now have a stack of size at least kSufficientStack. Ensure
|
|
||||||
// that we can actually use that much, if necessary.
|
|
||||||
ensureStackAddressSpace();
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureSufficientStack() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
int cc1_main(ArrayRef<const char *> Argv, const char *Argv0, void *MainAddr) {
|
|
||||||
ensureSufficientStack();
|
|
||||||
|
|
||||||
std::unique_ptr<CompilerInstance> Clang(new CompilerInstance());
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
// Register the support for object-file-wrapped Clang modules.
|
|
||||||
auto PCHOps = Clang->getPCHContainerOperations();
|
|
||||||
PCHOps->registerWriter(llvm::make_unique<ObjectFilePCHContainerWriter>());
|
|
||||||
PCHOps->registerReader(llvm::make_unique<ObjectFilePCHContainerReader>());
|
|
||||||
|
|
||||||
// Initialize targets first, so that --version shows registered targets.
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
llvm::InitializeAllTargetMCs();
|
|
||||||
llvm::InitializeAllAsmPrinters();
|
|
||||||
llvm::InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
llvm::PassRegistry &Registry = *llvm::PassRegistry::getPassRegistry();
|
|
||||||
polly::initializePollyPasses(Registry);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Buffer diagnostics from argument parsing so that we can output them using a
|
|
||||||
// well formed diagnostic object.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticBuffer *DiagsBuffer = new TextDiagnosticBuffer;
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagsBuffer);
|
|
||||||
bool Success = CompilerInvocation::CreateFromArgs(
|
|
||||||
Clang->getInvocation(), Argv.begin(), Argv.end(), Diags);
|
|
||||||
|
|
||||||
// Infer the builtin include path if unspecified.
|
|
||||||
if (Clang->getHeaderSearchOpts().UseBuiltinIncludes &&
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir.empty())
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir =
|
|
||||||
CompilerInvocation::GetResourcesPath(Argv0, MainAddr);
|
|
||||||
|
|
||||||
// Create the actual diagnostics engine.
|
|
||||||
Clang->createDiagnostics();
|
|
||||||
if (!Clang->hasDiagnostics())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
llvm::install_fatal_error_handler(LLVMErrorHandler,
|
|
||||||
static_cast<void*>(&Clang->getDiagnostics()));
|
|
||||||
|
|
||||||
DiagsBuffer->FlushDiagnostics(Clang->getDiagnostics());
|
|
||||||
if (!Success)
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Execute the frontend actions.
|
|
||||||
Success = ExecuteCompilerInvocation(Clang.get());
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
// Our error handler depends on the Diagnostics object, which we're
|
|
||||||
// potentially about to delete. Uninstall the handler now so that any
|
|
||||||
// later errors use the default handling behavior instead.
|
|
||||||
llvm::remove_fatal_error_handler();
|
|
||||||
|
|
||||||
// When running with -disable-free, don't do any destruction or shutdown.
|
|
||||||
if (Clang->getFrontendOpts().DisableFree) {
|
|
||||||
BuryPointer(std::move(Clang));
|
|
||||||
return !Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
return !Success;
|
|
||||||
}
|
|
@ -1,540 +0,0 @@
|
|||||||
//===-- cc1as_main.cpp - Clang Assembler ---------------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1as functionality, which implements
|
|
||||||
// the direct interface to the LLVM MC based assembler.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Basic/Diagnostic.h"
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/ADT/StringSwitch.h"
|
|
||||||
#include "llvm/ADT/Triple.h"
|
|
||||||
#include "llvm/IR/DataLayout.h"
|
|
||||||
#include "llvm/MC/MCAsmBackend.h"
|
|
||||||
#include "llvm/MC/MCAsmInfo.h"
|
|
||||||
#include "llvm/MC/MCCodeEmitter.h"
|
|
||||||
#include "llvm/MC/MCContext.h"
|
|
||||||
#include "llvm/MC/MCInstrInfo.h"
|
|
||||||
#include "llvm/MC/MCObjectFileInfo.h"
|
|
||||||
#include "llvm/MC/MCParser/MCAsmParser.h"
|
|
||||||
#include "llvm/MC/MCParser/MCTargetAsmParser.h"
|
|
||||||
#include "llvm/MC/MCRegisterInfo.h"
|
|
||||||
#include "llvm/MC/MCStreamer.h"
|
|
||||||
#include "llvm/MC/MCSubtargetInfo.h"
|
|
||||||
#include "llvm/MC/MCTargetOptions.h"
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/FormattedStream.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/MemoryBuffer.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/SourceMgr.h"
|
|
||||||
#include "llvm/Support/TargetRegistry.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace clang::driver::options;
|
|
||||||
using namespace llvm;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
/// \brief Helper class for representing a single invocation of the assembler.
|
|
||||||
struct AssemblerInvocation {
|
|
||||||
/// @name Target Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
/// The name of the target triple to assemble for.
|
|
||||||
std::string Triple;
|
|
||||||
|
|
||||||
/// If given, the name of the target CPU to determine which instructions
|
|
||||||
/// are legal.
|
|
||||||
std::string CPU;
|
|
||||||
|
|
||||||
/// The list of target specific features to enable or disable -- this should
|
|
||||||
/// be a list of strings starting with '+' or '-'.
|
|
||||||
std::vector<std::string> Features;
|
|
||||||
|
|
||||||
/// The list of symbol definitions.
|
|
||||||
std::vector<std::string> SymbolDefs;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Language Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::vector<std::string> IncludePaths;
|
|
||||||
unsigned NoInitialTextSection : 1;
|
|
||||||
unsigned SaveTemporaryLabels : 1;
|
|
||||||
unsigned GenDwarfForAssembly : 1;
|
|
||||||
unsigned RelaxELFRelocations : 1;
|
|
||||||
unsigned DwarfVersion;
|
|
||||||
std::string DwarfDebugFlags;
|
|
||||||
std::string DwarfDebugProducer;
|
|
||||||
std::string DebugCompilationDir;
|
|
||||||
llvm::DebugCompressionType CompressDebugSections =
|
|
||||||
llvm::DebugCompressionType::None;
|
|
||||||
std::string MainFileName;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Frontend Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::string InputFile;
|
|
||||||
std::vector<std::string> LLVMArgs;
|
|
||||||
std::string OutputPath;
|
|
||||||
enum FileType {
|
|
||||||
FT_Asm, ///< Assembly (.s) output, transliterate mode.
|
|
||||||
FT_Null, ///< No output, for timing purposes.
|
|
||||||
FT_Obj ///< Object file output.
|
|
||||||
};
|
|
||||||
FileType OutputType;
|
|
||||||
unsigned ShowHelp : 1;
|
|
||||||
unsigned ShowVersion : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Transliterate Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned OutputAsmVariant;
|
|
||||||
unsigned ShowEncoding : 1;
|
|
||||||
unsigned ShowInst : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Assembler Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned RelaxAll : 1;
|
|
||||||
unsigned NoExecStack : 1;
|
|
||||||
unsigned FatalWarnings : 1;
|
|
||||||
unsigned IncrementalLinkerCompatible : 1;
|
|
||||||
|
|
||||||
/// The name of the relocation model to use.
|
|
||||||
std::string RelocationModel;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
|
|
||||||
public:
|
|
||||||
AssemblerInvocation() {
|
|
||||||
Triple = "";
|
|
||||||
NoInitialTextSection = 0;
|
|
||||||
InputFile = "-";
|
|
||||||
OutputPath = "-";
|
|
||||||
OutputType = FT_Asm;
|
|
||||||
OutputAsmVariant = 0;
|
|
||||||
ShowInst = 0;
|
|
||||||
ShowEncoding = 0;
|
|
||||||
RelaxAll = 0;
|
|
||||||
NoExecStack = 0;
|
|
||||||
FatalWarnings = 0;
|
|
||||||
IncrementalLinkerCompatible = 0;
|
|
||||||
DwarfVersion = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool CreateFromArgs(AssemblerInvocation &Res,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags);
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
bool AssemblerInvocation::CreateFromArgs(AssemblerInvocation &Opts,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
bool Success = true;
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
std::unique_ptr<OptTable> OptTbl(createDriverOptTable());
|
|
||||||
|
|
||||||
const unsigned IncludedFlagsBitmask = options::CC1AsOption;
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args = OptTbl->ParseArgs(Argv, MissingArgIndex, MissingArgCount,
|
|
||||||
IncludedFlagsBitmask);
|
|
||||||
|
|
||||||
// Check for missing argument error.
|
|
||||||
if (MissingArgCount) {
|
|
||||||
Diags.Report(diag::err_drv_missing_argument)
|
|
||||||
<< Args.getArgString(MissingArgIndex) << MissingArgCount;
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Issue errors on unknown arguments.
|
|
||||||
for (const Arg *A : Args.filtered(OPT_UNKNOWN)) {
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << A->getAsString(Args);
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Construct the invocation.
|
|
||||||
|
|
||||||
// Target Options
|
|
||||||
Opts.Triple = llvm::Triple::normalize(Args.getLastArgValue(OPT_triple));
|
|
||||||
Opts.CPU = Args.getLastArgValue(OPT_target_cpu);
|
|
||||||
Opts.Features = Args.getAllArgValues(OPT_target_feature);
|
|
||||||
|
|
||||||
// Use the default target triple if unspecified.
|
|
||||||
if (Opts.Triple.empty())
|
|
||||||
Opts.Triple = llvm::sys::getDefaultTargetTriple();
|
|
||||||
|
|
||||||
// Language Options
|
|
||||||
Opts.IncludePaths = Args.getAllArgValues(OPT_I);
|
|
||||||
Opts.NoInitialTextSection = Args.hasArg(OPT_n);
|
|
||||||
Opts.SaveTemporaryLabels = Args.hasArg(OPT_msave_temp_labels);
|
|
||||||
// Any DebugInfoKind implies GenDwarfForAssembly.
|
|
||||||
Opts.GenDwarfForAssembly = Args.hasArg(OPT_debug_info_kind_EQ);
|
|
||||||
|
|
||||||
if (const Arg *A = Args.getLastArg(OPT_compress_debug_sections,
|
|
||||||
OPT_compress_debug_sections_EQ)) {
|
|
||||||
if (A->getOption().getID() == OPT_compress_debug_sections) {
|
|
||||||
// TODO: be more clever about the compression type auto-detection
|
|
||||||
Opts.CompressDebugSections = llvm::DebugCompressionType::GNU;
|
|
||||||
} else {
|
|
||||||
Opts.CompressDebugSections =
|
|
||||||
llvm::StringSwitch<llvm::DebugCompressionType>(A->getValue())
|
|
||||||
.Case("none", llvm::DebugCompressionType::None)
|
|
||||||
.Case("zlib", llvm::DebugCompressionType::Z)
|
|
||||||
.Case("zlib-gnu", llvm::DebugCompressionType::GNU)
|
|
||||||
.Default(llvm::DebugCompressionType::None);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Opts.RelaxELFRelocations = Args.hasArg(OPT_mrelax_relocations);
|
|
||||||
Opts.DwarfVersion = getLastArgIntValue(Args, OPT_dwarf_version_EQ, 2, Diags);
|
|
||||||
Opts.DwarfDebugFlags = Args.getLastArgValue(OPT_dwarf_debug_flags);
|
|
||||||
Opts.DwarfDebugProducer = Args.getLastArgValue(OPT_dwarf_debug_producer);
|
|
||||||
Opts.DebugCompilationDir = Args.getLastArgValue(OPT_fdebug_compilation_dir);
|
|
||||||
Opts.MainFileName = Args.getLastArgValue(OPT_main_file_name);
|
|
||||||
|
|
||||||
// Frontend Options
|
|
||||||
if (Args.hasArg(OPT_INPUT)) {
|
|
||||||
bool First = true;
|
|
||||||
for (const Arg *A : Args.filtered(OPT_INPUT)) {
|
|
||||||
if (First) {
|
|
||||||
Opts.InputFile = A->getValue();
|
|
||||||
First = false;
|
|
||||||
} else {
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << A->getAsString(Args);
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Opts.LLVMArgs = Args.getAllArgValues(OPT_mllvm);
|
|
||||||
Opts.OutputPath = Args.getLastArgValue(OPT_o);
|
|
||||||
if (Arg *A = Args.getLastArg(OPT_filetype)) {
|
|
||||||
StringRef Name = A->getValue();
|
|
||||||
unsigned OutputType = StringSwitch<unsigned>(Name)
|
|
||||||
.Case("asm", FT_Asm)
|
|
||||||
.Case("null", FT_Null)
|
|
||||||
.Case("obj", FT_Obj)
|
|
||||||
.Default(~0U);
|
|
||||||
if (OutputType == ~0U) {
|
|
||||||
Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Name;
|
|
||||||
Success = false;
|
|
||||||
} else
|
|
||||||
Opts.OutputType = FileType(OutputType);
|
|
||||||
}
|
|
||||||
Opts.ShowHelp = Args.hasArg(OPT_help);
|
|
||||||
Opts.ShowVersion = Args.hasArg(OPT_version);
|
|
||||||
|
|
||||||
// Transliterate Options
|
|
||||||
Opts.OutputAsmVariant =
|
|
||||||
getLastArgIntValue(Args, OPT_output_asm_variant, 0, Diags);
|
|
||||||
Opts.ShowEncoding = Args.hasArg(OPT_show_encoding);
|
|
||||||
Opts.ShowInst = Args.hasArg(OPT_show_inst);
|
|
||||||
|
|
||||||
// Assemble Options
|
|
||||||
Opts.RelaxAll = Args.hasArg(OPT_mrelax_all);
|
|
||||||
Opts.NoExecStack = Args.hasArg(OPT_mno_exec_stack);
|
|
||||||
Opts.FatalWarnings = Args.hasArg(OPT_massembler_fatal_warnings);
|
|
||||||
Opts.RelocationModel = Args.getLastArgValue(OPT_mrelocation_model, "pic");
|
|
||||||
Opts.IncrementalLinkerCompatible =
|
|
||||||
Args.hasArg(OPT_mincremental_linker_compatible);
|
|
||||||
Opts.SymbolDefs = Args.getAllArgValues(OPT_defsym);
|
|
||||||
|
|
||||||
return Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
static std::unique_ptr<raw_fd_ostream>
|
|
||||||
getOutputStream(AssemblerInvocation &Opts, DiagnosticsEngine &Diags,
|
|
||||||
bool Binary) {
|
|
||||||
if (Opts.OutputPath.empty())
|
|
||||||
Opts.OutputPath = "-";
|
|
||||||
|
|
||||||
// Make sure that the Out file gets unlinked from the disk if we get a
|
|
||||||
// SIGINT.
|
|
||||||
if (Opts.OutputPath != "-")
|
|
||||||
sys::RemoveFileOnSignal(Opts.OutputPath);
|
|
||||||
|
|
||||||
std::error_code EC;
|
|
||||||
auto Out = llvm::make_unique<raw_fd_ostream>(
|
|
||||||
Opts.OutputPath, EC, (Binary ? sys::fs::F_None : sys::fs::F_Text));
|
|
||||||
if (EC) {
|
|
||||||
Diags.Report(diag::err_fe_unable_to_open_output) << Opts.OutputPath
|
|
||||||
<< EC.message();
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return Out;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool ExecuteAssembler(AssemblerInvocation &Opts,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
// Get the target specific parser.
|
|
||||||
std::string Error;
|
|
||||||
const Target *TheTarget = TargetRegistry::lookupTarget(Opts.Triple, Error);
|
|
||||||
if (!TheTarget)
|
|
||||||
return Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
ErrorOr<std::unique_ptr<MemoryBuffer>> Buffer =
|
|
||||||
MemoryBuffer::getFileOrSTDIN(Opts.InputFile);
|
|
||||||
|
|
||||||
if (std::error_code EC = Buffer.getError()) {
|
|
||||||
Error = EC.message();
|
|
||||||
return Diags.Report(diag::err_fe_error_reading) << Opts.InputFile;
|
|
||||||
}
|
|
||||||
|
|
||||||
SourceMgr SrcMgr;
|
|
||||||
|
|
||||||
// Tell SrcMgr about this buffer, which is what the parser will pick up.
|
|
||||||
SrcMgr.AddNewSourceBuffer(std::move(*Buffer), SMLoc());
|
|
||||||
|
|
||||||
// Record the location of the include directories so that the lexer can find
|
|
||||||
// it later.
|
|
||||||
SrcMgr.setIncludeDirs(Opts.IncludePaths);
|
|
||||||
|
|
||||||
std::unique_ptr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(Opts.Triple));
|
|
||||||
assert(MRI && "Unable to create target register info!");
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, Opts.Triple));
|
|
||||||
assert(MAI && "Unable to create target asm info!");
|
|
||||||
|
|
||||||
// Ensure MCAsmInfo initialization occurs before any use, otherwise sections
|
|
||||||
// may be created with a combination of default and explicit settings.
|
|
||||||
MAI->setCompressDebugSections(Opts.CompressDebugSections);
|
|
||||||
|
|
||||||
MAI->setRelaxELFRelocations(Opts.RelaxELFRelocations);
|
|
||||||
|
|
||||||
bool IsBinary = Opts.OutputType == AssemblerInvocation::FT_Obj;
|
|
||||||
std::unique_ptr<raw_fd_ostream> FDOS = getOutputStream(Opts, Diags, IsBinary);
|
|
||||||
if (!FDOS)
|
|
||||||
return true;
|
|
||||||
|
|
||||||
// FIXME: This is not pretty. MCContext has a ptr to MCObjectFileInfo and
|
|
||||||
// MCObjectFileInfo needs a MCContext reference in order to initialize itself.
|
|
||||||
std::unique_ptr<MCObjectFileInfo> MOFI(new MCObjectFileInfo());
|
|
||||||
|
|
||||||
MCContext Ctx(MAI.get(), MRI.get(), MOFI.get(), &SrcMgr);
|
|
||||||
|
|
||||||
bool PIC = false;
|
|
||||||
if (Opts.RelocationModel == "static") {
|
|
||||||
PIC = false;
|
|
||||||
} else if (Opts.RelocationModel == "pic") {
|
|
||||||
PIC = true;
|
|
||||||
} else {
|
|
||||||
assert(Opts.RelocationModel == "dynamic-no-pic" &&
|
|
||||||
"Invalid PIC model!");
|
|
||||||
PIC = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
MOFI->InitMCObjectFileInfo(Triple(Opts.Triple), PIC, Ctx);
|
|
||||||
if (Opts.SaveTemporaryLabels)
|
|
||||||
Ctx.setAllowTemporaryLabels(false);
|
|
||||||
if (Opts.GenDwarfForAssembly)
|
|
||||||
Ctx.setGenDwarfForAssembly(true);
|
|
||||||
if (!Opts.DwarfDebugFlags.empty())
|
|
||||||
Ctx.setDwarfDebugFlags(StringRef(Opts.DwarfDebugFlags));
|
|
||||||
if (!Opts.DwarfDebugProducer.empty())
|
|
||||||
Ctx.setDwarfDebugProducer(StringRef(Opts.DwarfDebugProducer));
|
|
||||||
if (!Opts.DebugCompilationDir.empty())
|
|
||||||
Ctx.setCompilationDir(Opts.DebugCompilationDir);
|
|
||||||
if (!Opts.MainFileName.empty())
|
|
||||||
Ctx.setMainFileName(StringRef(Opts.MainFileName));
|
|
||||||
Ctx.setDwarfVersion(Opts.DwarfVersion);
|
|
||||||
|
|
||||||
// Build up the feature string from the target feature list.
|
|
||||||
std::string FS;
|
|
||||||
if (!Opts.Features.empty()) {
|
|
||||||
FS = Opts.Features[0];
|
|
||||||
for (unsigned i = 1, e = Opts.Features.size(); i != e; ++i)
|
|
||||||
FS += "," + Opts.Features[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
std::unique_ptr<MCStreamer> Str;
|
|
||||||
|
|
||||||
std::unique_ptr<MCInstrInfo> MCII(TheTarget->createMCInstrInfo());
|
|
||||||
std::unique_ptr<MCSubtargetInfo> STI(
|
|
||||||
TheTarget->createMCSubtargetInfo(Opts.Triple, Opts.CPU, FS));
|
|
||||||
|
|
||||||
raw_pwrite_stream *Out = FDOS.get();
|
|
||||||
std::unique_ptr<buffer_ostream> BOS;
|
|
||||||
|
|
||||||
// FIXME: There is a bit of code duplication with addPassesToEmitFile.
|
|
||||||
if (Opts.OutputType == AssemblerInvocation::FT_Asm) {
|
|
||||||
MCInstPrinter *IP = TheTarget->createMCInstPrinter(
|
|
||||||
llvm::Triple(Opts.Triple), Opts.OutputAsmVariant, *MAI, *MCII, *MRI);
|
|
||||||
MCCodeEmitter *CE = nullptr;
|
|
||||||
MCAsmBackend *MAB = nullptr;
|
|
||||||
if (Opts.ShowEncoding) {
|
|
||||||
CE = TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx);
|
|
||||||
MCTargetOptions Options;
|
|
||||||
MAB = TheTarget->createMCAsmBackend(*STI, *MRI, Options);
|
|
||||||
}
|
|
||||||
auto FOut = llvm::make_unique<formatted_raw_ostream>(*Out);
|
|
||||||
Str.reset(TheTarget->createAsmStreamer(
|
|
||||||
Ctx, std::move(FOut), /*asmverbose*/ true,
|
|
||||||
/*useDwarfDirectory*/ true, IP, CE, MAB, Opts.ShowInst));
|
|
||||||
} else if (Opts.OutputType == AssemblerInvocation::FT_Null) {
|
|
||||||
Str.reset(createNullStreamer(Ctx));
|
|
||||||
} else {
|
|
||||||
assert(Opts.OutputType == AssemblerInvocation::FT_Obj &&
|
|
||||||
"Invalid file type!");
|
|
||||||
if (!FDOS->supportsSeeking()) {
|
|
||||||
BOS = make_unique<buffer_ostream>(*FDOS);
|
|
||||||
Out = BOS.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
MCCodeEmitter *CE = TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx);
|
|
||||||
MCTargetOptions Options;
|
|
||||||
MCAsmBackend *MAB = TheTarget->createMCAsmBackend(*STI, *MRI, Options);
|
|
||||||
Triple T(Opts.Triple);
|
|
||||||
Str.reset(TheTarget->createMCObjectStreamer(
|
|
||||||
T, Ctx, std::unique_ptr<MCAsmBackend>(MAB), *Out, std::unique_ptr<MCCodeEmitter>(CE), *STI,
|
|
||||||
Opts.RelaxAll, Opts.IncrementalLinkerCompatible,
|
|
||||||
/*DWARFMustBeAtTheEnd*/ true));
|
|
||||||
Str.get()->InitSections(Opts.NoExecStack);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool Failed = false;
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmParser> Parser(
|
|
||||||
createMCAsmParser(SrcMgr, Ctx, *Str.get(), *MAI));
|
|
||||||
|
|
||||||
// FIXME: init MCTargetOptions from sanitizer flags here.
|
|
||||||
MCTargetOptions Options;
|
|
||||||
std::unique_ptr<MCTargetAsmParser> TAP(
|
|
||||||
TheTarget->createMCAsmParser(*STI, *Parser, *MCII, Options));
|
|
||||||
if (!TAP)
|
|
||||||
Failed = Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
// Set values for symbols, if any.
|
|
||||||
for (auto &S : Opts.SymbolDefs) {
|
|
||||||
auto Pair = StringRef(S).split('=');
|
|
||||||
auto Sym = Pair.first;
|
|
||||||
auto Val = Pair.second;
|
|
||||||
int64_t Value = 0;
|
|
||||||
// We have already error checked this in the driver.
|
|
||||||
Val.getAsInteger(0, Value);
|
|
||||||
Ctx.setSymbolValue(Parser->getStreamer(), Sym, Value);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!Failed) {
|
|
||||||
Parser->setTargetParser(*TAP.get());
|
|
||||||
Failed = Parser->Run(Opts.NoInitialTextSection);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close Streamer first.
|
|
||||||
// It might have a reference to the output stream.
|
|
||||||
Str.reset();
|
|
||||||
// Close the output stream early.
|
|
||||||
BOS.reset();
|
|
||||||
FDOS.reset();
|
|
||||||
|
|
||||||
// Delete output file if there were errors.
|
|
||||||
if (Failed && Opts.OutputPath != "-")
|
|
||||||
sys::fs::remove(Opts.OutputPath);
|
|
||||||
|
|
||||||
return Failed;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool GenCrashDiag) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors.
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
int cc1as_main(ArrayRef<const char *> Argv, const char *Argv0, void *MainAddr) {
|
|
||||||
// Initialize targets and assembly printers/parsers.
|
|
||||||
InitializeAllTargetInfos();
|
|
||||||
InitializeAllTargetMCs();
|
|
||||||
InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
// Construct our diagnostic client.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(errs(), &*DiagOpts);
|
|
||||||
DiagClient->setPrefix("clang -cc1as");
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
ScopedFatalErrorHandler FatalErrorHandler
|
|
||||||
(LLVMErrorHandler, static_cast<void*>(&Diags));
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
AssemblerInvocation Asm;
|
|
||||||
if (!AssemblerInvocation::CreateFromArgs(Asm, Argv, Diags))
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
if (Asm.ShowHelp) {
|
|
||||||
std::unique_ptr<OptTable> Opts(driver::createDriverOptTable());
|
|
||||||
Opts->PrintHelp(llvm::outs(), "clang -cc1as", "Clang Integrated Assembler",
|
|
||||||
/*Include=*/driver::options::CC1AsOption, /*Exclude=*/0,
|
|
||||||
/*ShowAllAliases=*/false);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -version.
|
|
||||||
//
|
|
||||||
// FIXME: Use a better -version message?
|
|
||||||
if (Asm.ShowVersion) {
|
|
||||||
llvm::cl::PrintVersionMessage();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -mllvm.
|
|
||||||
//
|
|
||||||
// FIXME: Remove this, one day.
|
|
||||||
if (!Asm.LLVMArgs.empty()) {
|
|
||||||
unsigned NumArgs = Asm.LLVMArgs.size();
|
|
||||||
auto Args = llvm::make_unique<const char*[]>(NumArgs + 2);
|
|
||||||
Args[0] = "clang (LLVM option parsing)";
|
|
||||||
for (unsigned i = 0; i != NumArgs; ++i)
|
|
||||||
Args[i + 1] = Asm.LLVMArgs[i].c_str();
|
|
||||||
Args[NumArgs + 1] = nullptr;
|
|
||||||
llvm::cl::ParseCommandLineOptions(NumArgs + 1, Args.get());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Execute the invocation, unless there were parsing errors.
|
|
||||||
bool Failed = Diags.hasErrorOccurred() || ExecuteAssembler(Asm, Diags);
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now.
|
|
||||||
TimerGroup::printAll(errs());
|
|
||||||
|
|
||||||
return !!Failed;
|
|
||||||
}
|
|
@ -1,520 +0,0 @@
|
|||||||
//===-- driver.cpp - Clang GCC-Compatible Driver --------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang driver; it is a thin wrapper
|
|
||||||
// for functionality in the Driver clang library.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/Compilation.h"
|
|
||||||
#include "clang/Driver/Driver.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Driver/ToolChain.h"
|
|
||||||
#include "clang/Frontend/ChainedDiagnosticConsumer.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/SerializedDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/ArrayRef.h"
|
|
||||||
#include "llvm/ADT/SmallString.h"
|
|
||||||
#include "llvm/ADT/SmallVector.h"
|
|
||||||
#include "llvm/Config/llvm-config.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Option/Option.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/PrettyStackTrace.h"
|
|
||||||
#include "llvm/Support/Process.h"
|
|
||||||
#include "llvm/Support/Program.h"
|
|
||||||
#include "llvm/Support/Regex.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/StringSaver.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <set>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
std::string GetExecutablePath(const char *Argv0, bool CanonicalPrefixes) {
|
|
||||||
if (!CanonicalPrefixes) {
|
|
||||||
SmallString<128> ExecutablePath(Argv0);
|
|
||||||
// Do a PATH lookup if Argv0 isn't a valid path.
|
|
||||||
if (!llvm::sys::fs::exists(ExecutablePath))
|
|
||||||
if (llvm::ErrorOr<std::string> P =
|
|
||||||
llvm::sys::findProgramByName(ExecutablePath))
|
|
||||||
ExecutablePath = *P;
|
|
||||||
return ExecutablePath.str();
|
|
||||||
}
|
|
||||||
|
|
||||||
// This just needs to be some symbol in the binary; C++ doesn't
|
|
||||||
// allow taking the address of ::main however.
|
|
||||||
void *P = (void*) (intptr_t) GetExecutablePath;
|
|
||||||
return llvm::sys::fs::getMainExecutable(Argv0, P);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const char *GetStableCStr(std::set<std::string> &SavedStrings,
|
|
||||||
StringRef S) {
|
|
||||||
return SavedStrings.insert(S).first->c_str();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a list of edits to the input argument lists.
|
|
||||||
///
|
|
||||||
/// The input string is a space separate list of edits to perform,
|
|
||||||
/// they are applied in order to the input argument lists. Edits
|
|
||||||
/// should be one of the following forms:
|
|
||||||
///
|
|
||||||
/// '#': Silence information about the changes to the command line arguments.
|
|
||||||
///
|
|
||||||
/// '^': Add FOO as a new argument at the beginning of the command line.
|
|
||||||
///
|
|
||||||
/// '+': Add FOO as a new argument at the end of the command line.
|
|
||||||
///
|
|
||||||
/// 's/XXX/YYY/': Substitute the regular expression XXX with YYY in the command
|
|
||||||
/// line.
|
|
||||||
///
|
|
||||||
/// 'xOPTION': Removes all instances of the literal argument OPTION.
|
|
||||||
///
|
|
||||||
/// 'XOPTION': Removes all instances of the literal argument OPTION,
|
|
||||||
/// and the following argument.
|
|
||||||
///
|
|
||||||
/// 'Ox': Removes all flags matching 'O' or 'O[sz0-9]' and adds 'Ox'
|
|
||||||
/// at the end of the command line.
|
|
||||||
///
|
|
||||||
/// \param OS - The stream to write edit information to.
|
|
||||||
/// \param Args - The vector of command line arguments.
|
|
||||||
/// \param Edit - The override command to perform.
|
|
||||||
/// \param SavedStrings - Set to use for storing string representations.
|
|
||||||
static void ApplyOneQAOverride(raw_ostream &OS,
|
|
||||||
SmallVectorImpl<const char*> &Args,
|
|
||||||
StringRef Edit,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
if (Edit[0] == '^') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at beginning\n";
|
|
||||||
Args.insert(Args.begin() + 1, Str);
|
|
||||||
} else if (Edit[0] == '+') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at end\n";
|
|
||||||
Args.push_back(Str);
|
|
||||||
} else if (Edit[0] == 's' && Edit[1] == '/' && Edit.endswith("/") &&
|
|
||||||
Edit.slice(2, Edit.size()-1).find('/') != StringRef::npos) {
|
|
||||||
StringRef MatchPattern = Edit.substr(2).split('/').first;
|
|
||||||
StringRef ReplPattern = Edit.substr(2).split('/').second;
|
|
||||||
ReplPattern = ReplPattern.slice(0, ReplPattern.size()-1);
|
|
||||||
|
|
||||||
for (unsigned i = 1, e = Args.size(); i != e; ++i) {
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (Args[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
std::string Repl = llvm::Regex(MatchPattern).sub(ReplPattern, Args[i]);
|
|
||||||
|
|
||||||
if (Repl != Args[i]) {
|
|
||||||
OS << "### Replacing '" << Args[i] << "' with '" << Repl << "'\n";
|
|
||||||
Args[i] = GetStableCStr(SavedStrings, Repl);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'x' || Edit[0] == 'X') {
|
|
||||||
auto Option = Edit.substr(1);
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
if (Option == Args[i]) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
if (Edit[0] == 'X') {
|
|
||||||
if (i < Args.size()) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
OS << "### Invalid X edit, end of command line!\n";
|
|
||||||
}
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'O') {
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
const char *A = Args[i];
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (A == nullptr)
|
|
||||||
continue;
|
|
||||||
if (A[0] == '-' && A[1] == 'O' &&
|
|
||||||
(A[2] == '\0' ||
|
|
||||||
(A[3] == '\0' && (A[2] == 's' || A[2] == 'z' ||
|
|
||||||
('0' <= A[2] && A[2] <= '9'))))) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
OS << "### Adding argument " << Edit << " at end\n";
|
|
||||||
Args.push_back(GetStableCStr(SavedStrings, '-' + Edit.str()));
|
|
||||||
} else {
|
|
||||||
OS << "### Unrecognized edit: " << Edit << "\n";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a comma separate list of edits to the
|
|
||||||
/// input argument lists. See ApplyOneQAOverride.
|
|
||||||
static void ApplyQAOverride(SmallVectorImpl<const char*> &Args,
|
|
||||||
const char *OverrideStr,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
raw_ostream *OS = &llvm::errs();
|
|
||||||
|
|
||||||
if (OverrideStr[0] == '#') {
|
|
||||||
++OverrideStr;
|
|
||||||
OS = &llvm::nulls();
|
|
||||||
}
|
|
||||||
|
|
||||||
*OS << "### CCC_OVERRIDE_OPTIONS: " << OverrideStr << "\n";
|
|
||||||
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
const char *S = OverrideStr;
|
|
||||||
while (*S) {
|
|
||||||
const char *End = ::strchr(S, ' ');
|
|
||||||
if (!End)
|
|
||||||
End = S + strlen(S);
|
|
||||||
if (End != S)
|
|
||||||
ApplyOneQAOverride(*OS, Args, std::string(S, End), SavedStrings);
|
|
||||||
S = End;
|
|
||||||
if (*S != '\0')
|
|
||||||
++S;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
extern int cc1_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
extern int cc1as_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
|
|
||||||
static void insertTargetAndModeArgs(const ParsedClangName &NameParts,
|
|
||||||
SmallVectorImpl<const char *> &ArgVector,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
// Put target and mode arguments at the start of argument list so that
|
|
||||||
// arguments specified in command line could override them. Avoid putting
|
|
||||||
// them at index 0, as an option like '-cc1' must remain the first.
|
|
||||||
auto InsertionPoint = ArgVector.begin();
|
|
||||||
if (InsertionPoint != ArgVector.end())
|
|
||||||
++InsertionPoint;
|
|
||||||
|
|
||||||
if (NameParts.DriverMode) {
|
|
||||||
// Add the mode flag to the arguments.
|
|
||||||
ArgVector.insert(InsertionPoint,
|
|
||||||
GetStableCStr(SavedStrings, NameParts.DriverMode));
|
|
||||||
}
|
|
||||||
|
|
||||||
if (NameParts.TargetIsValid) {
|
|
||||||
const char *arr[] = {"-target", GetStableCStr(SavedStrings,
|
|
||||||
NameParts.TargetPrefix)};
|
|
||||||
ArgVector.insert(InsertionPoint, std::begin(arr), std::end(arr));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void getCLEnvVarOptions(std::string &EnvValue, llvm::StringSaver &Saver,
|
|
||||||
SmallVectorImpl<const char *> &Opts) {
|
|
||||||
llvm::cl::TokenizeWindowsCommandLine(EnvValue, Saver, Opts);
|
|
||||||
// The first instance of '#' should be replaced with '=' in each option.
|
|
||||||
for (const char *Opt : Opts)
|
|
||||||
if (char *NumberSignPtr = const_cast<char *>(::strchr(Opt, '#')))
|
|
||||||
*NumberSignPtr = '=';
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetBackdoorDriverOutputsFromEnvVars(Driver &TheDriver) {
|
|
||||||
// Handle CC_PRINT_OPTIONS and CC_PRINT_OPTIONS_FILE.
|
|
||||||
TheDriver.CCPrintOptions = !!::getenv("CC_PRINT_OPTIONS");
|
|
||||||
if (TheDriver.CCPrintOptions)
|
|
||||||
TheDriver.CCPrintOptionsFilename = ::getenv("CC_PRINT_OPTIONS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_PRINT_HEADERS and CC_PRINT_HEADERS_FILE.
|
|
||||||
TheDriver.CCPrintHeaders = !!::getenv("CC_PRINT_HEADERS");
|
|
||||||
if (TheDriver.CCPrintHeaders)
|
|
||||||
TheDriver.CCPrintHeadersFilename = ::getenv("CC_PRINT_HEADERS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_LOG_DIAGNOSTICS and CC_LOG_DIAGNOSTICS_FILE.
|
|
||||||
TheDriver.CCLogDiagnostics = !!::getenv("CC_LOG_DIAGNOSTICS");
|
|
||||||
if (TheDriver.CCLogDiagnostics)
|
|
||||||
TheDriver.CCLogDiagnosticsFilename = ::getenv("CC_LOG_DIAGNOSTICS_FILE");
|
|
||||||
}
|
|
||||||
|
|
||||||
static void FixupDiagPrefixExeName(TextDiagnosticPrinter *DiagClient,
|
|
||||||
const std::string &Path) {
|
|
||||||
// If the clang binary happens to be named cl.exe for compatibility reasons,
|
|
||||||
// use clang-cl.exe as the prefix to avoid confusion between clang and MSVC.
|
|
||||||
StringRef ExeBasename(llvm::sys::path::filename(Path));
|
|
||||||
if (ExeBasename.equals_lower("cl.exe"))
|
|
||||||
ExeBasename = "clang-cl.exe";
|
|
||||||
DiagClient->setPrefix(ExeBasename);
|
|
||||||
}
|
|
||||||
|
|
||||||
// This lets us create the DiagnosticsEngine with a properly-filled-out
|
|
||||||
// DiagnosticOptions instance.
|
|
||||||
static DiagnosticOptions *
|
|
||||||
CreateAndPopulateDiagOpts(ArrayRef<const char *> argv) {
|
|
||||||
auto *DiagOpts = new DiagnosticOptions;
|
|
||||||
std::unique_ptr<OptTable> Opts(createDriverOptTable());
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args =
|
|
||||||
Opts->ParseArgs(argv.slice(1), MissingArgIndex, MissingArgCount);
|
|
||||||
// We ignore MissingArgCount and the return value of ParseDiagnosticArgs.
|
|
||||||
// Any errors that would be diagnosed here will also be diagnosed later,
|
|
||||||
// when the DiagnosticsEngine actually exists.
|
|
||||||
(void)ParseDiagnosticArgs(*DiagOpts, Args);
|
|
||||||
return DiagOpts;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetInstallDir(SmallVectorImpl<const char *> &argv,
|
|
||||||
Driver &TheDriver, bool CanonicalPrefixes) {
|
|
||||||
// Attempt to find the original path used to invoke the driver, to determine
|
|
||||||
// the installed path. We do this manually, because we want to support that
|
|
||||||
// path being a symlink.
|
|
||||||
SmallString<128> InstalledPath(argv[0]);
|
|
||||||
|
|
||||||
// Do a PATH lookup, if there are no directory components.
|
|
||||||
if (llvm::sys::path::filename(InstalledPath) == InstalledPath)
|
|
||||||
if (llvm::ErrorOr<std::string> Tmp = llvm::sys::findProgramByName(
|
|
||||||
llvm::sys::path::filename(InstalledPath.str())))
|
|
||||||
InstalledPath = *Tmp;
|
|
||||||
|
|
||||||
// FIXME: We don't actually canonicalize this, we just make it absolute.
|
|
||||||
if (CanonicalPrefixes)
|
|
||||||
llvm::sys::fs::make_absolute(InstalledPath);
|
|
||||||
|
|
||||||
StringRef InstalledPathParent(llvm::sys::path::parent_path(InstalledPath));
|
|
||||||
if (llvm::sys::fs::exists(InstalledPathParent))
|
|
||||||
TheDriver.setInstalledDir(InstalledPathParent);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int ExecuteCC1Tool(ArrayRef<const char *> argv, StringRef Tool) {
|
|
||||||
void *GetExecutablePathVP = (void *)(intptr_t) GetExecutablePath;
|
|
||||||
if (Tool == "")
|
|
||||||
return cc1_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
if (Tool == "as")
|
|
||||||
return cc1as_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
|
|
||||||
// Reject unknown tools.
|
|
||||||
llvm::errs() << "error: unknown integrated tool '" << Tool << "'\n";
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
int mainEntryClickHouseClang(int argc_, char **argv_) {
|
|
||||||
llvm::sys::PrintStackTraceOnErrorSignal(argv_[0]);
|
|
||||||
llvm::PrettyStackTraceProgram X(argc_, argv_);
|
|
||||||
llvm::llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
|
|
||||||
|
|
||||||
if (llvm::sys::Process::FixupStandardFileDescriptors())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
SmallVector<const char *, 256> argv;
|
|
||||||
llvm::SpecificBumpPtrAllocator<char> ArgAllocator;
|
|
||||||
std::error_code EC = llvm::sys::Process::GetArgumentVector(
|
|
||||||
argv, llvm::makeArrayRef(argv_, argc_), ArgAllocator);
|
|
||||||
if (EC) {
|
|
||||||
llvm::errs() << "error: couldn't get arguments: " << EC.message() << '\n';
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
auto TargetAndMode = ToolChain::getTargetAndModeFromProgramName(argv[0]);
|
|
||||||
|
|
||||||
llvm::BumpPtrAllocator A;
|
|
||||||
llvm::StringSaver Saver(A);
|
|
||||||
|
|
||||||
// Parse response files using the GNU syntax, unless we're in CL mode. There
|
|
||||||
// are two ways to put clang in CL compatibility mode: argv[0] is either
|
|
||||||
// clang-cl or cl, or --driver-mode=cl is on the command line. The normal
|
|
||||||
// command line parsing can't happen until after response file parsing, so we
|
|
||||||
// have to manually search for a --driver-mode=cl argument the hard way.
|
|
||||||
// Finally, our -cc1 tools don't care which tokenization mode we use because
|
|
||||||
// response files written by clang will tokenize the same way in either mode.
|
|
||||||
bool ClangCLMode = false;
|
|
||||||
if (StringRef(TargetAndMode.DriverMode).equals("--driver-mode=cl") ||
|
|
||||||
std::find_if(argv.begin(), argv.end(), [](const char *F) {
|
|
||||||
return F && strcmp(F, "--driver-mode=cl") == 0;
|
|
||||||
}) != argv.end()) {
|
|
||||||
ClangCLMode = true;
|
|
||||||
}
|
|
||||||
enum { Default, POSIX, Windows } RSPQuoting = Default;
|
|
||||||
for (const char *F : argv) {
|
|
||||||
if (strcmp(F, "--rsp-quoting=posix") == 0)
|
|
||||||
RSPQuoting = POSIX;
|
|
||||||
else if (strcmp(F, "--rsp-quoting=windows") == 0)
|
|
||||||
RSPQuoting = Windows;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Determines whether we want nullptr markers in argv to indicate response
|
|
||||||
// files end-of-lines. We only use this for the /LINK driver argument with
|
|
||||||
// clang-cl.exe on Windows.
|
|
||||||
bool MarkEOLs = ClangCLMode;
|
|
||||||
|
|
||||||
llvm::cl::TokenizerCallback Tokenizer;
|
|
||||||
if (RSPQuoting == Windows || (RSPQuoting == Default && ClangCLMode))
|
|
||||||
Tokenizer = &llvm::cl::TokenizeWindowsCommandLine;
|
|
||||||
else
|
|
||||||
Tokenizer = &llvm::cl::TokenizeGNUCommandLine;
|
|
||||||
|
|
||||||
if (MarkEOLs && argv.size() > 1 && StringRef(argv[1]).startswith("-cc1"))
|
|
||||||
MarkEOLs = false;
|
|
||||||
llvm::cl::ExpandResponseFiles(Saver, Tokenizer, argv, MarkEOLs);
|
|
||||||
|
|
||||||
// Handle -cc1 integrated tools, even if -cc1 was expanded from a response
|
|
||||||
// file.
|
|
||||||
auto FirstArg = std::find_if(argv.begin() + 1, argv.end(),
|
|
||||||
[](const char *A) { return A != nullptr; });
|
|
||||||
if (FirstArg != argv.end() && StringRef(*FirstArg).startswith("-cc1")) {
|
|
||||||
// If -cc1 came from a response file, remove the EOL sentinels.
|
|
||||||
if (MarkEOLs) {
|
|
||||||
auto newEnd = std::remove(argv.begin(), argv.end(), nullptr);
|
|
||||||
argv.resize(newEnd - argv.begin());
|
|
||||||
}
|
|
||||||
return ExecuteCC1Tool(argv, argv[1] + 4);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool CanonicalPrefixes = true;
|
|
||||||
for (int i = 1, size = argv.size(); i < size; ++i) {
|
|
||||||
// Skip end-of-line response file markers
|
|
||||||
if (argv[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
if (StringRef(argv[i]) == "-no-canonical-prefixes") {
|
|
||||||
CanonicalPrefixes = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle CL and _CL_ which permits additional command line options to be
|
|
||||||
// prepended or appended.
|
|
||||||
if (ClangCLMode) {
|
|
||||||
// Arguments in "CL" are prepended.
|
|
||||||
llvm::Optional<std::string> OptCL = llvm::sys::Process::GetEnv("CL");
|
|
||||||
if (OptCL.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> PrependedOpts;
|
|
||||||
getCLEnvVarOptions(OptCL.getValue(), Saver, PrependedOpts);
|
|
||||||
|
|
||||||
// Insert right after the program name to prepend to the argument list.
|
|
||||||
argv.insert(argv.begin() + 1, PrependedOpts.begin(), PrependedOpts.end());
|
|
||||||
}
|
|
||||||
// Arguments in "_CL_" are appended.
|
|
||||||
llvm::Optional<std::string> Opt_CL_ = llvm::sys::Process::GetEnv("_CL_");
|
|
||||||
if (Opt_CL_.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> AppendedOpts;
|
|
||||||
getCLEnvVarOptions(Opt_CL_.getValue(), Saver, AppendedOpts);
|
|
||||||
|
|
||||||
// Insert at the end of the argument list to append.
|
|
||||||
argv.append(AppendedOpts.begin(), AppendedOpts.end());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::set<std::string> SavedStrings;
|
|
||||||
// Handle CCC_OVERRIDE_OPTIONS, used for editing a command line behind the
|
|
||||||
// scenes.
|
|
||||||
if (const char *OverrideStr = ::getenv("CCC_OVERRIDE_OPTIONS")) {
|
|
||||||
// FIXME: Driver shouldn't take extra initial argument.
|
|
||||||
ApplyQAOverride(argv, OverrideStr, SavedStrings);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string Path = GetExecutablePath(argv[0], CanonicalPrefixes);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts =
|
|
||||||
CreateAndPopulateDiagOpts(argv);
|
|
||||||
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(llvm::errs(), &*DiagOpts);
|
|
||||||
FixupDiagPrefixExeName(DiagClient, Path);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
if (!DiagOpts->DiagnosticSerializationFile.empty()) {
|
|
||||||
auto SerializedConsumer =
|
|
||||||
clang::serialized_diags::create(DiagOpts->DiagnosticSerializationFile,
|
|
||||||
&*DiagOpts, /*MergeChildRecords=*/true);
|
|
||||||
Diags.setClient(new ChainedDiagnosticConsumer(
|
|
||||||
Diags.takeClient(), std::move(SerializedConsumer)));
|
|
||||||
}
|
|
||||||
|
|
||||||
ProcessWarningOptions(Diags, *DiagOpts, /*ReportDiags=*/false);
|
|
||||||
|
|
||||||
Driver TheDriver(Path, llvm::sys::getDefaultTargetTriple(), Diags);
|
|
||||||
SetInstallDir(argv, TheDriver, CanonicalPrefixes);
|
|
||||||
TheDriver.setTargetAndMode(TargetAndMode);
|
|
||||||
|
|
||||||
insertTargetAndModeArgs(TargetAndMode, argv, SavedStrings);
|
|
||||||
|
|
||||||
SetBackdoorDriverOutputsFromEnvVars(TheDriver);
|
|
||||||
|
|
||||||
std::unique_ptr<Compilation> C(TheDriver.BuildCompilation(argv));
|
|
||||||
int Res = 1;
|
|
||||||
if (C && !C->containsError()) {
|
|
||||||
SmallVector<std::pair<int, const Command *>, 4> FailingCommands;
|
|
||||||
Res = TheDriver.ExecuteCompilation(*C, FailingCommands);
|
|
||||||
|
|
||||||
// Force a crash to test the diagnostics.
|
|
||||||
if (TheDriver.GenReproducer) {
|
|
||||||
Diags.Report(diag::err_drv_force_crash)
|
|
||||||
<< !::getenv("FORCE_CLANG_DIAGNOSTICS_CRASH");
|
|
||||||
|
|
||||||
// Pretend that every command failed.
|
|
||||||
FailingCommands.clear();
|
|
||||||
for (const auto &J : C->getJobs())
|
|
||||||
if (const Command *C = dyn_cast<Command>(&J))
|
|
||||||
FailingCommands.push_back(std::make_pair(-1, C));
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const auto &P : FailingCommands) {
|
|
||||||
int CommandRes = P.first;
|
|
||||||
const Command *FailingCommand = P.second;
|
|
||||||
if (!Res)
|
|
||||||
Res = CommandRes;
|
|
||||||
|
|
||||||
// If result status is < 0, then the driver command signalled an error.
|
|
||||||
// If result status is 70, then the driver command reported a fatal error.
|
|
||||||
// On Windows, abort will return an exit code of 3. In these cases,
|
|
||||||
// generate additional diagnostic information if possible.
|
|
||||||
bool DiagnoseCrash = CommandRes < 0 || CommandRes == 70;
|
|
||||||
#ifdef LLVM_ON_WIN32
|
|
||||||
DiagnoseCrash |= CommandRes == 3;
|
|
||||||
#endif
|
|
||||||
if (DiagnoseCrash) {
|
|
||||||
TheDriver.generateCompilationDiagnostics(*C, *FailingCommand);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Diags.getClient()->finish();
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
#ifdef LLVM_ON_WIN32
|
|
||||||
// Exit status should not be negative on Win32, unless abnormal termination.
|
|
||||||
// Once abnormal termiation was caught, negative status should not be
|
|
||||||
// propagated.
|
|
||||||
if (Res < 0)
|
|
||||||
Res = 1;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// If we have multiple failing commands, we return the result of the first
|
|
||||||
// failing command.
|
|
||||||
return Res;
|
|
||||||
}
|
|
@ -1,23 +0,0 @@
|
|||||||
#include "lld/Common/Driver.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/ADT/StringSwitch.h"
|
|
||||||
#include "llvm/ADT/Twine.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/PrettyStackTrace.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
|
|
||||||
using namespace lld;
|
|
||||||
using namespace llvm;
|
|
||||||
using namespace llvm::sys;
|
|
||||||
|
|
||||||
int mainEntryClickHouseLLD(int Argc, char **Argv)
|
|
||||||
{
|
|
||||||
// Standard set up, so program fails gracefully.
|
|
||||||
sys::PrintStackTraceOnErrorSignal(Argv[0]);
|
|
||||||
PrettyStackTraceProgram StackPrinter(Argc, Argv);
|
|
||||||
llvm_shutdown_obj Shutdown;
|
|
||||||
|
|
||||||
std::vector<const char *> Args(Argv, Argv + Argc);
|
|
||||||
return !elf::link(Args, true);
|
|
||||||
}
|
|
@ -1 +0,0 @@
|
|||||||
Compiler-6.0.0
|
|
@ -1 +0,0 @@
|
|||||||
Compiler-6.0.0
|
|
@ -1,49 +0,0 @@
|
|||||||
add_definitions(-Wno-error -Wno-unused-parameter -Wno-non-virtual-dtor -U_LIBCPP_DEBUG)
|
|
||||||
|
|
||||||
link_directories(${LLVM_LIBRARY_DIRS})
|
|
||||||
|
|
||||||
add_library(clickhouse-compiler-lib
|
|
||||||
driver.cpp
|
|
||||||
cc1_main.cpp
|
|
||||||
cc1gen_reproducer_main.cpp
|
|
||||||
cc1as_main.cpp
|
|
||||||
lld.cpp)
|
|
||||||
|
|
||||||
target_compile_options(clickhouse-compiler-lib PRIVATE -fno-rtti -fno-exceptions -g0)
|
|
||||||
|
|
||||||
string(REPLACE "${INCLUDE_DEBUG_HELPERS}" "" CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS}) # cant compile with -fno-rtti
|
|
||||||
|
|
||||||
llvm_libs_all(REQUIRED_LLVM_LIBRARIES)
|
|
||||||
|
|
||||||
message(STATUS "Using LLVM ${LLVM_VERSION}: ${LLVM_INCLUDE_DIRS} : ${REQUIRED_LLVM_LIBRARIES}")
|
|
||||||
|
|
||||||
target_include_directories(clickhouse-compiler-lib SYSTEM PRIVATE ${LLVM_INCLUDE_DIRS})
|
|
||||||
|
|
||||||
# This is extracted almost directly from CMakeFiles/.../link.txt in LLVM build directory.
|
|
||||||
|
|
||||||
target_link_libraries(clickhouse-compiler-lib PRIVATE
|
|
||||||
clangBasic clangCodeGen clangDriver
|
|
||||||
clangFrontend
|
|
||||||
clangFrontendTool
|
|
||||||
clangRewriteFrontend clangARCMigrate clangStaticAnalyzerFrontend
|
|
||||||
clangParse clangSerialization clangSema clangEdit clangStaticAnalyzerCheckers
|
|
||||||
clangASTMatchers clangStaticAnalyzerCore clangAnalysis clangAST clangRewrite clangLex clangBasic
|
|
||||||
clangCrossTU clangIndex
|
|
||||||
|
|
||||||
lldCOFF
|
|
||||||
lldDriver
|
|
||||||
lldELF
|
|
||||||
lldMinGW
|
|
||||||
lldMachO
|
|
||||||
lldReaderWriter
|
|
||||||
lldYAML
|
|
||||||
lldCommon
|
|
||||||
lldCore
|
|
||||||
|
|
||||||
${REQUIRED_LLVM_LIBRARIES}
|
|
||||||
|
|
||||||
PUBLIC ${ZLIB_LIBRARIES} ${EXECINFO_LIBRARIES} Threads::Threads
|
|
||||||
${MALLOC_LIBRARIES}
|
|
||||||
${GLIBC_COMPATIBILITY_LIBRARIES}
|
|
||||||
${MEMCPY_LIBRARIES}
|
|
||||||
)
|
|
@ -1,239 +0,0 @@
|
|||||||
//===-- cc1_main.cpp - Clang CC1 Compiler Frontend ------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1 functionality, which implements the
|
|
||||||
// core compiler functionality along with a number of additional tools for
|
|
||||||
// demonstration and testing purposes.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "clang/CodeGen/ObjectFilePCHContainerOperations.h"
|
|
||||||
#include "clang/Config/config.h"
|
|
||||||
#include "clang/Basic/Stack.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/CompilerInstance.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticBuffer.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "clang/FrontendTool/Utils.h"
|
|
||||||
#include "llvm/ADT/Statistic.h"
|
|
||||||
#include "llvm/Config/llvm-config.h"
|
|
||||||
#include "llvm/LinkAllPasses.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/Compiler.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <cstdio>
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
#include <sys/resource.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// have no .a version in packages
|
|
||||||
#undef LINK_POLLY_INTO_TOOLS
|
|
||||||
|
|
||||||
using namespace clang;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
// Main driver
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool GenCrashDiag) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// Run the interrupt handlers to make sure any special cleanups get done, in
|
|
||||||
// particular that we remove files registered with RemoveFileOnSignal.
|
|
||||||
llvm::sys::RunInterruptHandlers();
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors. When reporting a fatal error, exit
|
|
||||||
// with status 70 to generate crash diagnostics. For BSD systems this is
|
|
||||||
// defined as an internal software error. Otherwise, exit with status 1.
|
|
||||||
exit(GenCrashDiag ? 70 : 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
namespace polly {
|
|
||||||
void initializePollyPasses(llvm::PassRegistry &Registry);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
#if defined(__linux__) && defined(__PIE__)
|
|
||||||
static size_t getCurrentStackAllocation() {
|
|
||||||
// If we can't compute the current stack usage, allow for 512K of command
|
|
||||||
// line arguments and environment.
|
|
||||||
size_t Usage = 512 * 1024;
|
|
||||||
if (FILE *StatFile = fopen("/proc/self/stat", "r")) {
|
|
||||||
// We assume that the stack extends from its current address to the end of
|
|
||||||
// the environment space. In reality, there is another string literal (the
|
|
||||||
// program name) after the environment, but this is close enough (we only
|
|
||||||
// need to be within 100K or so).
|
|
||||||
unsigned long StackPtr, EnvEnd;
|
|
||||||
// Disable silly GCC -Wformat warning that complains about length
|
|
||||||
// modifiers on ignored format specifiers. We want to retain these
|
|
||||||
// for documentation purposes even though they have no effect.
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic push
|
|
||||||
#pragma GCC diagnostic ignored "-Wformat"
|
|
||||||
#endif
|
|
||||||
if (fscanf(StatFile,
|
|
||||||
"%*d %*s %*c %*d %*d %*d %*d %*d %*u %*lu %*lu %*lu %*lu %*lu "
|
|
||||||
"%*lu %*ld %*ld %*ld %*ld %*ld %*ld %*llu %*lu %*ld %*lu %*lu "
|
|
||||||
"%*lu %*lu %lu %*lu %*lu %*lu %*lu %*lu %*llu %*lu %*lu %*d %*d "
|
|
||||||
"%*u %*u %*llu %*lu %*ld %*lu %*lu %*lu %*lu %*lu %*lu %lu %*d",
|
|
||||||
&StackPtr, &EnvEnd) == 2) {
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic pop
|
|
||||||
#endif
|
|
||||||
Usage = StackPtr < EnvEnd ? EnvEnd - StackPtr : StackPtr - EnvEnd;
|
|
||||||
}
|
|
||||||
fclose(StatFile);
|
|
||||||
}
|
|
||||||
return Usage;
|
|
||||||
}
|
|
||||||
|
|
||||||
#include <alloca.h>
|
|
||||||
|
|
||||||
LLVM_ATTRIBUTE_NOINLINE
|
|
||||||
static void ensureStackAddressSpace() {
|
|
||||||
// Linux kernels prior to 4.1 will sometimes locate the heap of a PIE binary
|
|
||||||
// relatively close to the stack (they are only guaranteed to be 128MiB
|
|
||||||
// apart). This results in crashes if we happen to heap-allocate more than
|
|
||||||
// 128MiB before we reach our stack high-water mark.
|
|
||||||
//
|
|
||||||
// To avoid these crashes, ensure that we have sufficient virtual memory
|
|
||||||
// pages allocated before we start running.
|
|
||||||
size_t Curr = getCurrentStackAllocation();
|
|
||||||
const int kTargetStack = DesiredStackSize - 256 * 1024;
|
|
||||||
if (Curr < kTargetStack) {
|
|
||||||
volatile char *volatile Alloc =
|
|
||||||
static_cast<volatile char *>(alloca(kTargetStack - Curr));
|
|
||||||
Alloc[0] = 0;
|
|
||||||
Alloc[kTargetStack - Curr - 1] = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureStackAddressSpace() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/// Attempt to ensure that we have at least 8MiB of usable stack space.
|
|
||||||
static void ensureSufficientStack() {
|
|
||||||
struct rlimit rlim;
|
|
||||||
if (getrlimit(RLIMIT_STACK, &rlim) != 0)
|
|
||||||
return;
|
|
||||||
|
|
||||||
// Increase the soft stack limit to our desired level, if necessary and
|
|
||||||
// possible.
|
|
||||||
if (rlim.rlim_cur != RLIM_INFINITY &&
|
|
||||||
rlim.rlim_cur < rlim_t(DesiredStackSize)) {
|
|
||||||
// Try to allocate sufficient stack.
|
|
||||||
if (rlim.rlim_max == RLIM_INFINITY ||
|
|
||||||
rlim.rlim_max >= rlim_t(DesiredStackSize))
|
|
||||||
rlim.rlim_cur = DesiredStackSize;
|
|
||||||
else if (rlim.rlim_cur == rlim.rlim_max)
|
|
||||||
return;
|
|
||||||
else
|
|
||||||
rlim.rlim_cur = rlim.rlim_max;
|
|
||||||
|
|
||||||
if (setrlimit(RLIMIT_STACK, &rlim) != 0 ||
|
|
||||||
rlim.rlim_cur != DesiredStackSize)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// We should now have a stack of size at least DesiredStackSize. Ensure
|
|
||||||
// that we can actually use that much, if necessary.
|
|
||||||
ensureStackAddressSpace();
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureSufficientStack() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
int cc1_main(ArrayRef<const char *> Argv, const char *Argv0, void *MainAddr) {
|
|
||||||
ensureSufficientStack();
|
|
||||||
|
|
||||||
std::unique_ptr<CompilerInstance> Clang(new CompilerInstance());
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
// Register the support for object-file-wrapped Clang modules.
|
|
||||||
auto PCHOps = Clang->getPCHContainerOperations();
|
|
||||||
PCHOps->registerWriter(llvm::make_unique<ObjectFilePCHContainerWriter>());
|
|
||||||
PCHOps->registerReader(llvm::make_unique<ObjectFilePCHContainerReader>());
|
|
||||||
|
|
||||||
// Initialize targets first, so that --version shows registered targets.
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
llvm::InitializeAllTargetMCs();
|
|
||||||
llvm::InitializeAllAsmPrinters();
|
|
||||||
llvm::InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
llvm::PassRegistry &Registry = *llvm::PassRegistry::getPassRegistry();
|
|
||||||
polly::initializePollyPasses(Registry);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Buffer diagnostics from argument parsing so that we can output them using a
|
|
||||||
// well formed diagnostic object.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticBuffer *DiagsBuffer = new TextDiagnosticBuffer;
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagsBuffer);
|
|
||||||
bool Success = CompilerInvocation::CreateFromArgs(
|
|
||||||
Clang->getInvocation(), Argv.begin(), Argv.end(), Diags);
|
|
||||||
|
|
||||||
// Infer the builtin include path if unspecified.
|
|
||||||
if (Clang->getHeaderSearchOpts().UseBuiltinIncludes &&
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir.empty())
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir =
|
|
||||||
CompilerInvocation::GetResourcesPath(Argv0, MainAddr);
|
|
||||||
|
|
||||||
// Create the actual diagnostics engine.
|
|
||||||
Clang->createDiagnostics();
|
|
||||||
if (!Clang->hasDiagnostics())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
llvm::install_fatal_error_handler(LLVMErrorHandler,
|
|
||||||
static_cast<void*>(&Clang->getDiagnostics()));
|
|
||||||
|
|
||||||
DiagsBuffer->FlushDiagnostics(Clang->getDiagnostics());
|
|
||||||
if (!Success)
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Execute the frontend actions.
|
|
||||||
Success = ExecuteCompilerInvocation(Clang.get());
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
// Our error handler depends on the Diagnostics object, which we're
|
|
||||||
// potentially about to delete. Uninstall the handler now so that any
|
|
||||||
// later errors use the default handling behavior instead.
|
|
||||||
llvm::remove_fatal_error_handler();
|
|
||||||
|
|
||||||
// When running with -disable-free, don't do any destruction or shutdown.
|
|
||||||
if (Clang->getFrontendOpts().DisableFree) {
|
|
||||||
BuryPointer(std::move(Clang));
|
|
||||||
return !Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
return !Success;
|
|
||||||
}
|
|
@ -1,572 +0,0 @@
|
|||||||
//===-- cc1as_main.cpp - Clang Assembler ---------------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1as functionality, which implements
|
|
||||||
// the direct interface to the LLVM MC based assembler.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Basic/Diagnostic.h"
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/ADT/StringSwitch.h"
|
|
||||||
#include "llvm/ADT/Triple.h"
|
|
||||||
#include "llvm/IR/DataLayout.h"
|
|
||||||
#include "llvm/MC/MCAsmBackend.h"
|
|
||||||
#include "llvm/MC/MCAsmInfo.h"
|
|
||||||
#include "llvm/MC/MCCodeEmitter.h"
|
|
||||||
#include "llvm/MC/MCContext.h"
|
|
||||||
#include "llvm/MC/MCInstrInfo.h"
|
|
||||||
#include "llvm/MC/MCObjectFileInfo.h"
|
|
||||||
#include "llvm/MC/MCObjectWriter.h"
|
|
||||||
#include "llvm/MC/MCParser/MCAsmParser.h"
|
|
||||||
#include "llvm/MC/MCParser/MCTargetAsmParser.h"
|
|
||||||
#include "llvm/MC/MCRegisterInfo.h"
|
|
||||||
#include "llvm/MC/MCStreamer.h"
|
|
||||||
#include "llvm/MC/MCSubtargetInfo.h"
|
|
||||||
#include "llvm/MC/MCTargetOptions.h"
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/FormattedStream.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/MemoryBuffer.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/SourceMgr.h"
|
|
||||||
#include "llvm/Support/TargetRegistry.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace clang::driver::options;
|
|
||||||
using namespace llvm;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
/// Helper class for representing a single invocation of the assembler.
|
|
||||||
struct AssemblerInvocation {
|
|
||||||
/// @name Target Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
/// The name of the target triple to assemble for.
|
|
||||||
std::string Triple;
|
|
||||||
|
|
||||||
/// If given, the name of the target CPU to determine which instructions
|
|
||||||
/// are legal.
|
|
||||||
std::string CPU;
|
|
||||||
|
|
||||||
/// The list of target specific features to enable or disable -- this should
|
|
||||||
/// be a list of strings starting with '+' or '-'.
|
|
||||||
std::vector<std::string> Features;
|
|
||||||
|
|
||||||
/// The list of symbol definitions.
|
|
||||||
std::vector<std::string> SymbolDefs;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Language Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::vector<std::string> IncludePaths;
|
|
||||||
unsigned NoInitialTextSection : 1;
|
|
||||||
unsigned SaveTemporaryLabels : 1;
|
|
||||||
unsigned GenDwarfForAssembly : 1;
|
|
||||||
unsigned RelaxELFRelocations : 1;
|
|
||||||
unsigned DwarfVersion;
|
|
||||||
std::string DwarfDebugFlags;
|
|
||||||
std::string DwarfDebugProducer;
|
|
||||||
std::string DebugCompilationDir;
|
|
||||||
std::map<const std::string, const std::string> DebugPrefixMap;
|
|
||||||
llvm::DebugCompressionType CompressDebugSections =
|
|
||||||
llvm::DebugCompressionType::None;
|
|
||||||
std::string MainFileName;
|
|
||||||
std::string SplitDwarfFile;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Frontend Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::string InputFile;
|
|
||||||
std::vector<std::string> LLVMArgs;
|
|
||||||
std::string OutputPath;
|
|
||||||
enum FileType {
|
|
||||||
FT_Asm, ///< Assembly (.s) output, transliterate mode.
|
|
||||||
FT_Null, ///< No output, for timing purposes.
|
|
||||||
FT_Obj ///< Object file output.
|
|
||||||
};
|
|
||||||
FileType OutputType;
|
|
||||||
unsigned ShowHelp : 1;
|
|
||||||
unsigned ShowVersion : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Transliterate Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned OutputAsmVariant;
|
|
||||||
unsigned ShowEncoding : 1;
|
|
||||||
unsigned ShowInst : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Assembler Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned RelaxAll : 1;
|
|
||||||
unsigned NoExecStack : 1;
|
|
||||||
unsigned FatalWarnings : 1;
|
|
||||||
unsigned IncrementalLinkerCompatible : 1;
|
|
||||||
|
|
||||||
/// The name of the relocation model to use.
|
|
||||||
std::string RelocationModel;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
|
|
||||||
public:
|
|
||||||
AssemblerInvocation() {
|
|
||||||
Triple = "";
|
|
||||||
NoInitialTextSection = 0;
|
|
||||||
InputFile = "-";
|
|
||||||
OutputPath = "-";
|
|
||||||
OutputType = FT_Asm;
|
|
||||||
OutputAsmVariant = 0;
|
|
||||||
ShowInst = 0;
|
|
||||||
ShowEncoding = 0;
|
|
||||||
RelaxAll = 0;
|
|
||||||
NoExecStack = 0;
|
|
||||||
FatalWarnings = 0;
|
|
||||||
IncrementalLinkerCompatible = 0;
|
|
||||||
DwarfVersion = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool CreateFromArgs(AssemblerInvocation &Res,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags);
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
bool AssemblerInvocation::CreateFromArgs(AssemblerInvocation &Opts,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
bool Success = true;
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
std::unique_ptr<OptTable> OptTbl(createDriverOptTable());
|
|
||||||
|
|
||||||
const unsigned IncludedFlagsBitmask = options::CC1AsOption;
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args = OptTbl->ParseArgs(Argv, MissingArgIndex, MissingArgCount,
|
|
||||||
IncludedFlagsBitmask);
|
|
||||||
|
|
||||||
// Check for missing argument error.
|
|
||||||
if (MissingArgCount) {
|
|
||||||
Diags.Report(diag::err_drv_missing_argument)
|
|
||||||
<< Args.getArgString(MissingArgIndex) << MissingArgCount;
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Issue errors on unknown arguments.
|
|
||||||
for (const Arg *A : Args.filtered(OPT_UNKNOWN)) {
|
|
||||||
auto ArgString = A->getAsString(Args);
|
|
||||||
std::string Nearest;
|
|
||||||
if (OptTbl->findNearest(ArgString, Nearest, IncludedFlagsBitmask) > 1)
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << ArgString;
|
|
||||||
else
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument_with_suggestion)
|
|
||||||
<< ArgString << Nearest;
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Construct the invocation.
|
|
||||||
|
|
||||||
// Target Options
|
|
||||||
Opts.Triple = llvm::Triple::normalize(Args.getLastArgValue(OPT_triple));
|
|
||||||
Opts.CPU = Args.getLastArgValue(OPT_target_cpu);
|
|
||||||
Opts.Features = Args.getAllArgValues(OPT_target_feature);
|
|
||||||
|
|
||||||
// Use the default target triple if unspecified.
|
|
||||||
if (Opts.Triple.empty())
|
|
||||||
Opts.Triple = llvm::sys::getDefaultTargetTriple();
|
|
||||||
|
|
||||||
// Language Options
|
|
||||||
Opts.IncludePaths = Args.getAllArgValues(OPT_I);
|
|
||||||
Opts.NoInitialTextSection = Args.hasArg(OPT_n);
|
|
||||||
Opts.SaveTemporaryLabels = Args.hasArg(OPT_msave_temp_labels);
|
|
||||||
// Any DebugInfoKind implies GenDwarfForAssembly.
|
|
||||||
Opts.GenDwarfForAssembly = Args.hasArg(OPT_debug_info_kind_EQ);
|
|
||||||
|
|
||||||
if (const Arg *A = Args.getLastArg(OPT_compress_debug_sections,
|
|
||||||
OPT_compress_debug_sections_EQ)) {
|
|
||||||
if (A->getOption().getID() == OPT_compress_debug_sections) {
|
|
||||||
// TODO: be more clever about the compression type auto-detection
|
|
||||||
Opts.CompressDebugSections = llvm::DebugCompressionType::GNU;
|
|
||||||
} else {
|
|
||||||
Opts.CompressDebugSections =
|
|
||||||
llvm::StringSwitch<llvm::DebugCompressionType>(A->getValue())
|
|
||||||
.Case("none", llvm::DebugCompressionType::None)
|
|
||||||
.Case("zlib", llvm::DebugCompressionType::Z)
|
|
||||||
.Case("zlib-gnu", llvm::DebugCompressionType::GNU)
|
|
||||||
.Default(llvm::DebugCompressionType::None);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Opts.RelaxELFRelocations = Args.hasArg(OPT_mrelax_relocations);
|
|
||||||
Opts.DwarfVersion = getLastArgIntValue(Args, OPT_dwarf_version_EQ, 2, Diags);
|
|
||||||
Opts.DwarfDebugFlags = Args.getLastArgValue(OPT_dwarf_debug_flags);
|
|
||||||
Opts.DwarfDebugProducer = Args.getLastArgValue(OPT_dwarf_debug_producer);
|
|
||||||
Opts.DebugCompilationDir = Args.getLastArgValue(OPT_fdebug_compilation_dir);
|
|
||||||
Opts.MainFileName = Args.getLastArgValue(OPT_main_file_name);
|
|
||||||
|
|
||||||
for (const auto &Arg : Args.getAllArgValues(OPT_fdebug_prefix_map_EQ))
|
|
||||||
Opts.DebugPrefixMap.insert(StringRef(Arg).split('='));
|
|
||||||
|
|
||||||
// Frontend Options
|
|
||||||
if (Args.hasArg(OPT_INPUT)) {
|
|
||||||
bool First = true;
|
|
||||||
for (const Arg *A : Args.filtered(OPT_INPUT)) {
|
|
||||||
if (First) {
|
|
||||||
Opts.InputFile = A->getValue();
|
|
||||||
First = false;
|
|
||||||
} else {
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << A->getAsString(Args);
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Opts.LLVMArgs = Args.getAllArgValues(OPT_mllvm);
|
|
||||||
Opts.OutputPath = Args.getLastArgValue(OPT_o);
|
|
||||||
Opts.SplitDwarfFile = Args.getLastArgValue(OPT_split_dwarf_file);
|
|
||||||
if (Arg *A = Args.getLastArg(OPT_filetype)) {
|
|
||||||
StringRef Name = A->getValue();
|
|
||||||
unsigned OutputType = StringSwitch<unsigned>(Name)
|
|
||||||
.Case("asm", FT_Asm)
|
|
||||||
.Case("null", FT_Null)
|
|
||||||
.Case("obj", FT_Obj)
|
|
||||||
.Default(~0U);
|
|
||||||
if (OutputType == ~0U) {
|
|
||||||
Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Name;
|
|
||||||
Success = false;
|
|
||||||
} else
|
|
||||||
Opts.OutputType = FileType(OutputType);
|
|
||||||
}
|
|
||||||
Opts.ShowHelp = Args.hasArg(OPT_help);
|
|
||||||
Opts.ShowVersion = Args.hasArg(OPT_version);
|
|
||||||
|
|
||||||
// Transliterate Options
|
|
||||||
Opts.OutputAsmVariant =
|
|
||||||
getLastArgIntValue(Args, OPT_output_asm_variant, 0, Diags);
|
|
||||||
Opts.ShowEncoding = Args.hasArg(OPT_show_encoding);
|
|
||||||
Opts.ShowInst = Args.hasArg(OPT_show_inst);
|
|
||||||
|
|
||||||
// Assemble Options
|
|
||||||
Opts.RelaxAll = Args.hasArg(OPT_mrelax_all);
|
|
||||||
Opts.NoExecStack = Args.hasArg(OPT_mno_exec_stack);
|
|
||||||
Opts.FatalWarnings = Args.hasArg(OPT_massembler_fatal_warnings);
|
|
||||||
Opts.RelocationModel = Args.getLastArgValue(OPT_mrelocation_model, "pic");
|
|
||||||
Opts.IncrementalLinkerCompatible =
|
|
||||||
Args.hasArg(OPT_mincremental_linker_compatible);
|
|
||||||
Opts.SymbolDefs = Args.getAllArgValues(OPT_defsym);
|
|
||||||
|
|
||||||
return Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
static std::unique_ptr<raw_fd_ostream>
|
|
||||||
getOutputStream(StringRef Path, DiagnosticsEngine &Diags, bool Binary) {
|
|
||||||
// Make sure that the Out file gets unlinked from the disk if we get a
|
|
||||||
// SIGINT.
|
|
||||||
if (Path != "-")
|
|
||||||
sys::RemoveFileOnSignal(Path);
|
|
||||||
|
|
||||||
std::error_code EC;
|
|
||||||
auto Out = llvm::make_unique<raw_fd_ostream>(
|
|
||||||
Path, EC, (Binary ? sys::fs::F_None : sys::fs::F_Text));
|
|
||||||
if (EC) {
|
|
||||||
Diags.Report(diag::err_fe_unable_to_open_output) << Path << EC.message();
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return Out;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool ExecuteAssembler(AssemblerInvocation &Opts,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
// Get the target specific parser.
|
|
||||||
std::string Error;
|
|
||||||
const Target *TheTarget = TargetRegistry::lookupTarget(Opts.Triple, Error);
|
|
||||||
if (!TheTarget)
|
|
||||||
return Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
ErrorOr<std::unique_ptr<MemoryBuffer>> Buffer =
|
|
||||||
MemoryBuffer::getFileOrSTDIN(Opts.InputFile);
|
|
||||||
|
|
||||||
if (std::error_code EC = Buffer.getError()) {
|
|
||||||
Error = EC.message();
|
|
||||||
return Diags.Report(diag::err_fe_error_reading) << Opts.InputFile;
|
|
||||||
}
|
|
||||||
|
|
||||||
SourceMgr SrcMgr;
|
|
||||||
|
|
||||||
// Tell SrcMgr about this buffer, which is what the parser will pick up.
|
|
||||||
SrcMgr.AddNewSourceBuffer(std::move(*Buffer), SMLoc());
|
|
||||||
|
|
||||||
// Record the location of the include directories so that the lexer can find
|
|
||||||
// it later.
|
|
||||||
SrcMgr.setIncludeDirs(Opts.IncludePaths);
|
|
||||||
|
|
||||||
std::unique_ptr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(Opts.Triple));
|
|
||||||
assert(MRI && "Unable to create target register info!");
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, Opts.Triple));
|
|
||||||
assert(MAI && "Unable to create target asm info!");
|
|
||||||
|
|
||||||
// Ensure MCAsmInfo initialization occurs before any use, otherwise sections
|
|
||||||
// may be created with a combination of default and explicit settings.
|
|
||||||
MAI->setCompressDebugSections(Opts.CompressDebugSections);
|
|
||||||
|
|
||||||
MAI->setRelaxELFRelocations(Opts.RelaxELFRelocations);
|
|
||||||
|
|
||||||
bool IsBinary = Opts.OutputType == AssemblerInvocation::FT_Obj;
|
|
||||||
if (Opts.OutputPath.empty())
|
|
||||||
Opts.OutputPath = "-";
|
|
||||||
std::unique_ptr<raw_fd_ostream> FDOS =
|
|
||||||
getOutputStream(Opts.OutputPath, Diags, IsBinary);
|
|
||||||
if (!FDOS)
|
|
||||||
return true;
|
|
||||||
std::unique_ptr<raw_fd_ostream> DwoOS;
|
|
||||||
if (!Opts.SplitDwarfFile.empty())
|
|
||||||
DwoOS = getOutputStream(Opts.SplitDwarfFile, Diags, IsBinary);
|
|
||||||
|
|
||||||
// FIXME: This is not pretty. MCContext has a ptr to MCObjectFileInfo and
|
|
||||||
// MCObjectFileInfo needs a MCContext reference in order to initialize itself.
|
|
||||||
std::unique_ptr<MCObjectFileInfo> MOFI(new MCObjectFileInfo());
|
|
||||||
|
|
||||||
MCContext Ctx(MAI.get(), MRI.get(), MOFI.get(), &SrcMgr);
|
|
||||||
|
|
||||||
bool PIC = false;
|
|
||||||
if (Opts.RelocationModel == "static") {
|
|
||||||
PIC = false;
|
|
||||||
} else if (Opts.RelocationModel == "pic") {
|
|
||||||
PIC = true;
|
|
||||||
} else {
|
|
||||||
assert(Opts.RelocationModel == "dynamic-no-pic" &&
|
|
||||||
"Invalid PIC model!");
|
|
||||||
PIC = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
MOFI->InitMCObjectFileInfo(Triple(Opts.Triple), PIC, Ctx);
|
|
||||||
if (Opts.SaveTemporaryLabels)
|
|
||||||
Ctx.setAllowTemporaryLabels(false);
|
|
||||||
if (Opts.GenDwarfForAssembly)
|
|
||||||
Ctx.setGenDwarfForAssembly(true);
|
|
||||||
if (!Opts.DwarfDebugFlags.empty())
|
|
||||||
Ctx.setDwarfDebugFlags(StringRef(Opts.DwarfDebugFlags));
|
|
||||||
if (!Opts.DwarfDebugProducer.empty())
|
|
||||||
Ctx.setDwarfDebugProducer(StringRef(Opts.DwarfDebugProducer));
|
|
||||||
if (!Opts.DebugCompilationDir.empty())
|
|
||||||
Ctx.setCompilationDir(Opts.DebugCompilationDir);
|
|
||||||
if (!Opts.DebugPrefixMap.empty())
|
|
||||||
for (const auto &KV : Opts.DebugPrefixMap)
|
|
||||||
Ctx.addDebugPrefixMapEntry(KV.first, KV.second);
|
|
||||||
if (!Opts.MainFileName.empty())
|
|
||||||
Ctx.setMainFileName(StringRef(Opts.MainFileName));
|
|
||||||
Ctx.setDwarfVersion(Opts.DwarfVersion);
|
|
||||||
|
|
||||||
// Build up the feature string from the target feature list.
|
|
||||||
std::string FS;
|
|
||||||
if (!Opts.Features.empty()) {
|
|
||||||
FS = Opts.Features[0];
|
|
||||||
for (unsigned i = 1, e = Opts.Features.size(); i != e; ++i)
|
|
||||||
FS += "," + Opts.Features[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
std::unique_ptr<MCStreamer> Str;
|
|
||||||
|
|
||||||
std::unique_ptr<MCInstrInfo> MCII(TheTarget->createMCInstrInfo());
|
|
||||||
std::unique_ptr<MCSubtargetInfo> STI(
|
|
||||||
TheTarget->createMCSubtargetInfo(Opts.Triple, Opts.CPU, FS));
|
|
||||||
|
|
||||||
raw_pwrite_stream *Out = FDOS.get();
|
|
||||||
std::unique_ptr<buffer_ostream> BOS;
|
|
||||||
|
|
||||||
// FIXME: There is a bit of code duplication with addPassesToEmitFile.
|
|
||||||
if (Opts.OutputType == AssemblerInvocation::FT_Asm) {
|
|
||||||
MCInstPrinter *IP = TheTarget->createMCInstPrinter(
|
|
||||||
llvm::Triple(Opts.Triple), Opts.OutputAsmVariant, *MAI, *MCII, *MRI);
|
|
||||||
|
|
||||||
std::unique_ptr<MCCodeEmitter> CE;
|
|
||||||
if (Opts.ShowEncoding)
|
|
||||||
CE.reset(TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx));
|
|
||||||
MCTargetOptions MCOptions;
|
|
||||||
std::unique_ptr<MCAsmBackend> MAB(
|
|
||||||
TheTarget->createMCAsmBackend(*STI, *MRI, MCOptions));
|
|
||||||
|
|
||||||
auto FOut = llvm::make_unique<formatted_raw_ostream>(*Out);
|
|
||||||
Str.reset(TheTarget->createAsmStreamer(
|
|
||||||
Ctx, std::move(FOut), /*asmverbose*/ true,
|
|
||||||
/*useDwarfDirectory*/ true, IP, std::move(CE), std::move(MAB),
|
|
||||||
Opts.ShowInst));
|
|
||||||
} else if (Opts.OutputType == AssemblerInvocation::FT_Null) {
|
|
||||||
Str.reset(createNullStreamer(Ctx));
|
|
||||||
} else {
|
|
||||||
assert(Opts.OutputType == AssemblerInvocation::FT_Obj &&
|
|
||||||
"Invalid file type!");
|
|
||||||
if (!FDOS->supportsSeeking()) {
|
|
||||||
BOS = make_unique<buffer_ostream>(*FDOS);
|
|
||||||
Out = BOS.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
std::unique_ptr<MCCodeEmitter> CE(
|
|
||||||
TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx));
|
|
||||||
MCTargetOptions MCOptions;
|
|
||||||
std::unique_ptr<MCAsmBackend> MAB(
|
|
||||||
TheTarget->createMCAsmBackend(*STI, *MRI, MCOptions));
|
|
||||||
std::unique_ptr<MCObjectWriter> OW =
|
|
||||||
DwoOS ? MAB->createDwoObjectWriter(*Out, *DwoOS)
|
|
||||||
: MAB->createObjectWriter(*Out);
|
|
||||||
|
|
||||||
Triple T(Opts.Triple);
|
|
||||||
Str.reset(TheTarget->createMCObjectStreamer(
|
|
||||||
T, Ctx, std::move(MAB), std::move(OW), std::move(CE), *STI,
|
|
||||||
Opts.RelaxAll, Opts.IncrementalLinkerCompatible,
|
|
||||||
/*DWARFMustBeAtTheEnd*/ true));
|
|
||||||
Str.get()->InitSections(Opts.NoExecStack);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Assembly to object compilation should leverage assembly info.
|
|
||||||
Str->setUseAssemblerInfoForParsing(true);
|
|
||||||
|
|
||||||
bool Failed = false;
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmParser> Parser(
|
|
||||||
createMCAsmParser(SrcMgr, Ctx, *Str.get(), *MAI));
|
|
||||||
|
|
||||||
// FIXME: init MCTargetOptions from sanitizer flags here.
|
|
||||||
MCTargetOptions Options;
|
|
||||||
std::unique_ptr<MCTargetAsmParser> TAP(
|
|
||||||
TheTarget->createMCAsmParser(*STI, *Parser, *MCII, Options));
|
|
||||||
if (!TAP)
|
|
||||||
Failed = Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
// Set values for symbols, if any.
|
|
||||||
for (auto &S : Opts.SymbolDefs) {
|
|
||||||
auto Pair = StringRef(S).split('=');
|
|
||||||
auto Sym = Pair.first;
|
|
||||||
auto Val = Pair.second;
|
|
||||||
int64_t Value = 1;
|
|
||||||
// We have already error checked this in the driver.
|
|
||||||
Val.getAsInteger(0, Value);
|
|
||||||
Ctx.setSymbolValue(Parser->getStreamer(), Sym, Value);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!Failed) {
|
|
||||||
Parser->setTargetParser(*TAP.get());
|
|
||||||
Failed = Parser->Run(Opts.NoInitialTextSection);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close Streamer first.
|
|
||||||
// It might have a reference to the output stream.
|
|
||||||
Str.reset();
|
|
||||||
// Close the output stream early.
|
|
||||||
BOS.reset();
|
|
||||||
FDOS.reset();
|
|
||||||
|
|
||||||
// Delete output file if there were errors.
|
|
||||||
if (Failed) {
|
|
||||||
if (Opts.OutputPath != "-")
|
|
||||||
sys::fs::remove(Opts.OutputPath);
|
|
||||||
if (!Opts.SplitDwarfFile.empty() && Opts.SplitDwarfFile != "-")
|
|
||||||
sys::fs::remove(Opts.SplitDwarfFile);
|
|
||||||
}
|
|
||||||
|
|
||||||
return Failed;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool GenCrashDiag) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors.
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
int cc1as_main(ArrayRef<const char *> Argv, const char *Argv0, void *MainAddr) {
|
|
||||||
// Initialize targets and assembly printers/parsers.
|
|
||||||
InitializeAllTargetInfos();
|
|
||||||
InitializeAllTargetMCs();
|
|
||||||
InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
// Construct our diagnostic client.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(errs(), &*DiagOpts);
|
|
||||||
DiagClient->setPrefix("clang -cc1as");
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
ScopedFatalErrorHandler FatalErrorHandler
|
|
||||||
(LLVMErrorHandler, static_cast<void*>(&Diags));
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
AssemblerInvocation Asm;
|
|
||||||
if (!AssemblerInvocation::CreateFromArgs(Asm, Argv, Diags))
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
if (Asm.ShowHelp) {
|
|
||||||
std::unique_ptr<OptTable> Opts(driver::createDriverOptTable());
|
|
||||||
Opts->PrintHelp(llvm::outs(), "clang -cc1as", "Clang Integrated Assembler",
|
|
||||||
/*Include=*/driver::options::CC1AsOption, /*Exclude=*/0,
|
|
||||||
/*ShowAllAliases=*/false);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -version.
|
|
||||||
//
|
|
||||||
// FIXME: Use a better -version message?
|
|
||||||
if (Asm.ShowVersion) {
|
|
||||||
llvm::cl::PrintVersionMessage();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -mllvm.
|
|
||||||
//
|
|
||||||
// FIXME: Remove this, one day.
|
|
||||||
if (!Asm.LLVMArgs.empty()) {
|
|
||||||
unsigned NumArgs = Asm.LLVMArgs.size();
|
|
||||||
auto Args = llvm::make_unique<const char*[]>(NumArgs + 2);
|
|
||||||
Args[0] = "clang (LLVM option parsing)";
|
|
||||||
for (unsigned i = 0; i != NumArgs; ++i)
|
|
||||||
Args[i + 1] = Asm.LLVMArgs[i].c_str();
|
|
||||||
Args[NumArgs + 1] = nullptr;
|
|
||||||
llvm::cl::ParseCommandLineOptions(NumArgs + 1, Args.get());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Execute the invocation, unless there were parsing errors.
|
|
||||||
bool Failed = Diags.hasErrorOccurred() || ExecuteAssembler(Asm, Diags);
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now.
|
|
||||||
TimerGroup::printAll(errs());
|
|
||||||
|
|
||||||
return !!Failed;
|
|
||||||
}
|
|
@ -1,196 +0,0 @@
|
|||||||
//===-- cc1gen_reproducer_main.cpp - Clang reproducer generator ----------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1gen-reproducer functionality, which
|
|
||||||
// generates reproducers for invocations for clang-based tools.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Basic/Diagnostic.h"
|
|
||||||
#include "clang/Basic/LLVM.h"
|
|
||||||
#include "clang/Basic/VirtualFileSystem.h"
|
|
||||||
#include "clang/Driver/Compilation.h"
|
|
||||||
#include "clang/Driver/Driver.h"
|
|
||||||
#include "llvm/ADT/ArrayRef.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/YAMLTraits.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
|
|
||||||
using namespace clang;
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
struct UnsavedFileHash {
|
|
||||||
std::string Name;
|
|
||||||
std::string MD5;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct ClangInvocationInfo {
|
|
||||||
std::string Toolchain;
|
|
||||||
std::string LibclangOperation;
|
|
||||||
std::string LibclangOptions;
|
|
||||||
std::vector<std::string> Arguments;
|
|
||||||
std::vector<std::string> InvocationArguments;
|
|
||||||
std::vector<UnsavedFileHash> UnsavedFileHashes;
|
|
||||||
bool Dump = false;
|
|
||||||
};
|
|
||||||
|
|
||||||
} // end anonymous namespace
|
|
||||||
|
|
||||||
LLVM_YAML_IS_SEQUENCE_VECTOR(UnsavedFileHash)
|
|
||||||
|
|
||||||
namespace llvm {
|
|
||||||
namespace yaml {
|
|
||||||
|
|
||||||
template <> struct MappingTraits<UnsavedFileHash> {
|
|
||||||
static void mapping(IO &IO, UnsavedFileHash &Info) {
|
|
||||||
IO.mapRequired("name", Info.Name);
|
|
||||||
IO.mapRequired("md5", Info.MD5);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <> struct MappingTraits<ClangInvocationInfo> {
|
|
||||||
static void mapping(IO &IO, ClangInvocationInfo &Info) {
|
|
||||||
IO.mapRequired("toolchain", Info.Toolchain);
|
|
||||||
IO.mapOptional("libclang.operation", Info.LibclangOperation);
|
|
||||||
IO.mapOptional("libclang.opts", Info.LibclangOptions);
|
|
||||||
IO.mapRequired("args", Info.Arguments);
|
|
||||||
IO.mapOptional("invocation-args", Info.InvocationArguments);
|
|
||||||
IO.mapOptional("unsaved_file_hashes", Info.UnsavedFileHashes);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
} // end namespace yaml
|
|
||||||
} // end namespace llvm
|
|
||||||
|
|
||||||
static std::string generateReproducerMetaInfo(const ClangInvocationInfo &Info) {
|
|
||||||
std::string Result;
|
|
||||||
llvm::raw_string_ostream OS(Result);
|
|
||||||
OS << '{';
|
|
||||||
bool NeedComma = false;
|
|
||||||
auto EmitKey = [&](StringRef Key) {
|
|
||||||
if (NeedComma)
|
|
||||||
OS << ", ";
|
|
||||||
NeedComma = true;
|
|
||||||
OS << '"' << Key << "\": ";
|
|
||||||
};
|
|
||||||
auto EmitStringKey = [&](StringRef Key, StringRef Value) {
|
|
||||||
if (Value.empty())
|
|
||||||
return;
|
|
||||||
EmitKey(Key);
|
|
||||||
OS << '"' << Value << '"';
|
|
||||||
};
|
|
||||||
EmitStringKey("libclang.operation", Info.LibclangOperation);
|
|
||||||
EmitStringKey("libclang.opts", Info.LibclangOptions);
|
|
||||||
if (!Info.InvocationArguments.empty()) {
|
|
||||||
EmitKey("invocation-args");
|
|
||||||
OS << '[';
|
|
||||||
for (const auto &Arg : llvm::enumerate(Info.InvocationArguments)) {
|
|
||||||
if (Arg.index())
|
|
||||||
OS << ',';
|
|
||||||
OS << '"' << Arg.value() << '"';
|
|
||||||
}
|
|
||||||
OS << ']';
|
|
||||||
}
|
|
||||||
OS << '}';
|
|
||||||
// FIXME: Compare unsaved file hashes and report mismatch in the reproducer.
|
|
||||||
if (Info.Dump)
|
|
||||||
llvm::outs() << "REPRODUCER METAINFO: " << OS.str() << "\n";
|
|
||||||
return std::move(OS.str());
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Generates a reproducer for a set of arguments from a specific invocation.
|
|
||||||
static llvm::Optional<driver::Driver::CompilationDiagnosticReport>
|
|
||||||
generateReproducerForInvocationArguments(ArrayRef<const char *> Argv,
|
|
||||||
const ClangInvocationInfo &Info) {
|
|
||||||
using namespace driver;
|
|
||||||
auto TargetAndMode = ToolChain::getTargetAndModeFromProgramName(Argv[0]);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions;
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, new IgnoringDiagConsumer());
|
|
||||||
ProcessWarningOptions(Diags, *DiagOpts, /*ReportDiags=*/false);
|
|
||||||
Driver TheDriver(Argv[0], llvm::sys::getDefaultTargetTriple(), Diags);
|
|
||||||
TheDriver.setTargetAndMode(TargetAndMode);
|
|
||||||
|
|
||||||
std::unique_ptr<Compilation> C(TheDriver.BuildCompilation(Argv));
|
|
||||||
if (C && !C->containsError()) {
|
|
||||||
for (const auto &J : C->getJobs()) {
|
|
||||||
if (const Command *Cmd = dyn_cast<Command>(&J)) {
|
|
||||||
Driver::CompilationDiagnosticReport Report;
|
|
||||||
TheDriver.generateCompilationDiagnostics(
|
|
||||||
*C, *Cmd, generateReproducerMetaInfo(Info), &Report);
|
|
||||||
return Report;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string GetExecutablePath(const char *Argv0, bool CanonicalPrefixes);
|
|
||||||
|
|
||||||
static void printReproducerInformation(
|
|
||||||
llvm::raw_ostream &OS, const ClangInvocationInfo &Info,
|
|
||||||
const driver::Driver::CompilationDiagnosticReport &Report) {
|
|
||||||
OS << "REPRODUCER:\n";
|
|
||||||
OS << "{\n";
|
|
||||||
OS << R"("files":[)";
|
|
||||||
for (const auto &File : llvm::enumerate(Report.TemporaryFiles)) {
|
|
||||||
if (File.index())
|
|
||||||
OS << ',';
|
|
||||||
OS << '"' << File.value() << '"';
|
|
||||||
}
|
|
||||||
OS << "]\n}\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
int cc1gen_reproducer_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr) {
|
|
||||||
if (Argv.size() < 1) {
|
|
||||||
llvm::errs() << "error: missing invocation file\n";
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
// Parse the invocation descriptor.
|
|
||||||
StringRef Input = Argv[0];
|
|
||||||
llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> Buffer =
|
|
||||||
llvm::MemoryBuffer::getFile(Input);
|
|
||||||
if (!Buffer) {
|
|
||||||
llvm::errs() << "error: failed to read " << Input << ": "
|
|
||||||
<< Buffer.getError().message() << "\n";
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
llvm::yaml::Input YAML(Buffer.get()->getBuffer());
|
|
||||||
ClangInvocationInfo InvocationInfo;
|
|
||||||
YAML >> InvocationInfo;
|
|
||||||
if (Argv.size() > 1 && Argv[1] == StringRef("-v"))
|
|
||||||
InvocationInfo.Dump = true;
|
|
||||||
|
|
||||||
// Create an invocation that will produce the reproducer.
|
|
||||||
std::vector<const char *> DriverArgs;
|
|
||||||
for (const auto &Arg : InvocationInfo.Arguments)
|
|
||||||
DriverArgs.push_back(Arg.c_str());
|
|
||||||
std::string Path = GetExecutablePath(Argv0, /*CanonicalPrefixes=*/true);
|
|
||||||
DriverArgs[0] = Path.c_str();
|
|
||||||
llvm::Optional<driver::Driver::CompilationDiagnosticReport> Report =
|
|
||||||
generateReproducerForInvocationArguments(DriverArgs, InvocationInfo);
|
|
||||||
|
|
||||||
// Emit the information about the reproduce files to stdout.
|
|
||||||
int Result = 1;
|
|
||||||
if (Report) {
|
|
||||||
printReproducerInformation(llvm::outs(), InvocationInfo, *Report);
|
|
||||||
Result = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Remove the input file.
|
|
||||||
llvm::sys::fs::remove(Input);
|
|
||||||
return Result;
|
|
||||||
}
|
|
@ -1,514 +0,0 @@
|
|||||||
//===-- driver.cpp - Clang GCC-Compatible Driver --------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang driver; it is a thin wrapper
|
|
||||||
// for functionality in the Driver clang library.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Driver/Driver.h"
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/Compilation.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Driver/ToolChain.h"
|
|
||||||
#include "clang/Frontend/ChainedDiagnosticConsumer.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/SerializedDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/ArrayRef.h"
|
|
||||||
#include "llvm/ADT/SmallString.h"
|
|
||||||
#include "llvm/ADT/SmallVector.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Option/Option.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/InitLLVM.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/Process.h"
|
|
||||||
#include "llvm/Support/Program.h"
|
|
||||||
#include "llvm/Support/Regex.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/StringSaver.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <set>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
std::string GetExecutablePath(const char *Argv0, bool CanonicalPrefixes) {
|
|
||||||
if (!CanonicalPrefixes) {
|
|
||||||
SmallString<128> ExecutablePath(Argv0);
|
|
||||||
// Do a PATH lookup if Argv0 isn't a valid path.
|
|
||||||
if (!llvm::sys::fs::exists(ExecutablePath))
|
|
||||||
if (llvm::ErrorOr<std::string> P =
|
|
||||||
llvm::sys::findProgramByName(ExecutablePath))
|
|
||||||
ExecutablePath = *P;
|
|
||||||
return ExecutablePath.str();
|
|
||||||
}
|
|
||||||
|
|
||||||
// This just needs to be some symbol in the binary; C++ doesn't
|
|
||||||
// allow taking the address of ::main however.
|
|
||||||
void *P = (void*) (intptr_t) GetExecutablePath;
|
|
||||||
return llvm::sys::fs::getMainExecutable(Argv0, P);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const char *GetStableCStr(std::set<std::string> &SavedStrings,
|
|
||||||
StringRef S) {
|
|
||||||
return SavedStrings.insert(S).first->c_str();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a list of edits to the input argument lists.
|
|
||||||
///
|
|
||||||
/// The input string is a space separate list of edits to perform,
|
|
||||||
/// they are applied in order to the input argument lists. Edits
|
|
||||||
/// should be one of the following forms:
|
|
||||||
///
|
|
||||||
/// '#': Silence information about the changes to the command line arguments.
|
|
||||||
///
|
|
||||||
/// '^': Add FOO as a new argument at the beginning of the command line.
|
|
||||||
///
|
|
||||||
/// '+': Add FOO as a new argument at the end of the command line.
|
|
||||||
///
|
|
||||||
/// 's/XXX/YYY/': Substitute the regular expression XXX with YYY in the command
|
|
||||||
/// line.
|
|
||||||
///
|
|
||||||
/// 'xOPTION': Removes all instances of the literal argument OPTION.
|
|
||||||
///
|
|
||||||
/// 'XOPTION': Removes all instances of the literal argument OPTION,
|
|
||||||
/// and the following argument.
|
|
||||||
///
|
|
||||||
/// 'Ox': Removes all flags matching 'O' or 'O[sz0-9]' and adds 'Ox'
|
|
||||||
/// at the end of the command line.
|
|
||||||
///
|
|
||||||
/// \param OS - The stream to write edit information to.
|
|
||||||
/// \param Args - The vector of command line arguments.
|
|
||||||
/// \param Edit - The override command to perform.
|
|
||||||
/// \param SavedStrings - Set to use for storing string representations.
|
|
||||||
static void ApplyOneQAOverride(raw_ostream &OS,
|
|
||||||
SmallVectorImpl<const char*> &Args,
|
|
||||||
StringRef Edit,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
if (Edit[0] == '^') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at beginning\n";
|
|
||||||
Args.insert(Args.begin() + 1, Str);
|
|
||||||
} else if (Edit[0] == '+') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at end\n";
|
|
||||||
Args.push_back(Str);
|
|
||||||
} else if (Edit[0] == 's' && Edit[1] == '/' && Edit.endswith("/") &&
|
|
||||||
Edit.slice(2, Edit.size()-1).find('/') != StringRef::npos) {
|
|
||||||
StringRef MatchPattern = Edit.substr(2).split('/').first;
|
|
||||||
StringRef ReplPattern = Edit.substr(2).split('/').second;
|
|
||||||
ReplPattern = ReplPattern.slice(0, ReplPattern.size()-1);
|
|
||||||
|
|
||||||
for (unsigned i = 1, e = Args.size(); i != e; ++i) {
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (Args[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
std::string Repl = llvm::Regex(MatchPattern).sub(ReplPattern, Args[i]);
|
|
||||||
|
|
||||||
if (Repl != Args[i]) {
|
|
||||||
OS << "### Replacing '" << Args[i] << "' with '" << Repl << "'\n";
|
|
||||||
Args[i] = GetStableCStr(SavedStrings, Repl);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'x' || Edit[0] == 'X') {
|
|
||||||
auto Option = Edit.substr(1);
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
if (Option == Args[i]) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
if (Edit[0] == 'X') {
|
|
||||||
if (i < Args.size()) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
OS << "### Invalid X edit, end of command line!\n";
|
|
||||||
}
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'O') {
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
const char *A = Args[i];
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (A == nullptr)
|
|
||||||
continue;
|
|
||||||
if (A[0] == '-' && A[1] == 'O' &&
|
|
||||||
(A[2] == '\0' ||
|
|
||||||
(A[3] == '\0' && (A[2] == 's' || A[2] == 'z' ||
|
|
||||||
('0' <= A[2] && A[2] <= '9'))))) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
OS << "### Adding argument " << Edit << " at end\n";
|
|
||||||
Args.push_back(GetStableCStr(SavedStrings, '-' + Edit.str()));
|
|
||||||
} else {
|
|
||||||
OS << "### Unrecognized edit: " << Edit << "\n";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a comma separate list of edits to the
|
|
||||||
/// input argument lists. See ApplyOneQAOverride.
|
|
||||||
static void ApplyQAOverride(SmallVectorImpl<const char*> &Args,
|
|
||||||
const char *OverrideStr,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
raw_ostream *OS = &llvm::errs();
|
|
||||||
|
|
||||||
if (OverrideStr[0] == '#') {
|
|
||||||
++OverrideStr;
|
|
||||||
OS = &llvm::nulls();
|
|
||||||
}
|
|
||||||
|
|
||||||
*OS << "### CCC_OVERRIDE_OPTIONS: " << OverrideStr << "\n";
|
|
||||||
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
const char *S = OverrideStr;
|
|
||||||
while (*S) {
|
|
||||||
const char *End = ::strchr(S, ' ');
|
|
||||||
if (!End)
|
|
||||||
End = S + strlen(S);
|
|
||||||
if (End != S)
|
|
||||||
ApplyOneQAOverride(*OS, Args, std::string(S, End), SavedStrings);
|
|
||||||
S = End;
|
|
||||||
if (*S != '\0')
|
|
||||||
++S;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
extern int cc1_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
extern int cc1as_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
extern int cc1gen_reproducer_main(ArrayRef<const char *> Argv,
|
|
||||||
const char *Argv0, void *MainAddr);
|
|
||||||
|
|
||||||
static void insertTargetAndModeArgs(const ParsedClangName &NameParts,
|
|
||||||
SmallVectorImpl<const char *> &ArgVector,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
// Put target and mode arguments at the start of argument list so that
|
|
||||||
// arguments specified in command line could override them. Avoid putting
|
|
||||||
// them at index 0, as an option like '-cc1' must remain the first.
|
|
||||||
int InsertionPoint = 0;
|
|
||||||
if (ArgVector.size() > 0)
|
|
||||||
++InsertionPoint;
|
|
||||||
|
|
||||||
if (NameParts.DriverMode) {
|
|
||||||
// Add the mode flag to the arguments.
|
|
||||||
ArgVector.insert(ArgVector.begin() + InsertionPoint,
|
|
||||||
GetStableCStr(SavedStrings, NameParts.DriverMode));
|
|
||||||
}
|
|
||||||
|
|
||||||
if (NameParts.TargetIsValid) {
|
|
||||||
const char *arr[] = {"-target", GetStableCStr(SavedStrings,
|
|
||||||
NameParts.TargetPrefix)};
|
|
||||||
ArgVector.insert(ArgVector.begin() + InsertionPoint,
|
|
||||||
std::begin(arr), std::end(arr));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void getCLEnvVarOptions(std::string &EnvValue, llvm::StringSaver &Saver,
|
|
||||||
SmallVectorImpl<const char *> &Opts) {
|
|
||||||
llvm::cl::TokenizeWindowsCommandLine(EnvValue, Saver, Opts);
|
|
||||||
// The first instance of '#' should be replaced with '=' in each option.
|
|
||||||
for (const char *Opt : Opts)
|
|
||||||
if (char *NumberSignPtr = const_cast<char *>(::strchr(Opt, '#')))
|
|
||||||
*NumberSignPtr = '=';
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetBackdoorDriverOutputsFromEnvVars(Driver &TheDriver) {
|
|
||||||
// Handle CC_PRINT_OPTIONS and CC_PRINT_OPTIONS_FILE.
|
|
||||||
TheDriver.CCPrintOptions = !!::getenv("CC_PRINT_OPTIONS");
|
|
||||||
if (TheDriver.CCPrintOptions)
|
|
||||||
TheDriver.CCPrintOptionsFilename = ::getenv("CC_PRINT_OPTIONS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_PRINT_HEADERS and CC_PRINT_HEADERS_FILE.
|
|
||||||
TheDriver.CCPrintHeaders = !!::getenv("CC_PRINT_HEADERS");
|
|
||||||
if (TheDriver.CCPrintHeaders)
|
|
||||||
TheDriver.CCPrintHeadersFilename = ::getenv("CC_PRINT_HEADERS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_LOG_DIAGNOSTICS and CC_LOG_DIAGNOSTICS_FILE.
|
|
||||||
TheDriver.CCLogDiagnostics = !!::getenv("CC_LOG_DIAGNOSTICS");
|
|
||||||
if (TheDriver.CCLogDiagnostics)
|
|
||||||
TheDriver.CCLogDiagnosticsFilename = ::getenv("CC_LOG_DIAGNOSTICS_FILE");
|
|
||||||
}
|
|
||||||
|
|
||||||
static void FixupDiagPrefixExeName(TextDiagnosticPrinter *DiagClient,
|
|
||||||
const std::string &Path) {
|
|
||||||
// If the clang binary happens to be named cl.exe for compatibility reasons,
|
|
||||||
// use clang-cl.exe as the prefix to avoid confusion between clang and MSVC.
|
|
||||||
StringRef ExeBasename(llvm::sys::path::filename(Path));
|
|
||||||
if (ExeBasename.equals_lower("cl.exe"))
|
|
||||||
ExeBasename = "clang-cl.exe";
|
|
||||||
DiagClient->setPrefix(ExeBasename);
|
|
||||||
}
|
|
||||||
|
|
||||||
// This lets us create the DiagnosticsEngine with a properly-filled-out
|
|
||||||
// DiagnosticOptions instance.
|
|
||||||
static DiagnosticOptions *
|
|
||||||
CreateAndPopulateDiagOpts(ArrayRef<const char *> argv) {
|
|
||||||
auto *DiagOpts = new DiagnosticOptions;
|
|
||||||
std::unique_ptr<OptTable> Opts(createDriverOptTable());
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args =
|
|
||||||
Opts->ParseArgs(argv.slice(1), MissingArgIndex, MissingArgCount);
|
|
||||||
// We ignore MissingArgCount and the return value of ParseDiagnosticArgs.
|
|
||||||
// Any errors that would be diagnosed here will also be diagnosed later,
|
|
||||||
// when the DiagnosticsEngine actually exists.
|
|
||||||
(void)ParseDiagnosticArgs(*DiagOpts, Args);
|
|
||||||
return DiagOpts;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetInstallDir(SmallVectorImpl<const char *> &argv,
|
|
||||||
Driver &TheDriver, bool CanonicalPrefixes) {
|
|
||||||
// Attempt to find the original path used to invoke the driver, to determine
|
|
||||||
// the installed path. We do this manually, because we want to support that
|
|
||||||
// path being a symlink.
|
|
||||||
SmallString<128> InstalledPath(argv[0]);
|
|
||||||
|
|
||||||
// Do a PATH lookup, if there are no directory components.
|
|
||||||
if (llvm::sys::path::filename(InstalledPath) == InstalledPath)
|
|
||||||
if (llvm::ErrorOr<std::string> Tmp = llvm::sys::findProgramByName(
|
|
||||||
llvm::sys::path::filename(InstalledPath.str())))
|
|
||||||
InstalledPath = *Tmp;
|
|
||||||
|
|
||||||
// FIXME: We don't actually canonicalize this, we just make it absolute.
|
|
||||||
if (CanonicalPrefixes)
|
|
||||||
llvm::sys::fs::make_absolute(InstalledPath);
|
|
||||||
|
|
||||||
StringRef InstalledPathParent(llvm::sys::path::parent_path(InstalledPath));
|
|
||||||
if (llvm::sys::fs::exists(InstalledPathParent))
|
|
||||||
TheDriver.setInstalledDir(InstalledPathParent);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int ExecuteCC1Tool(ArrayRef<const char *> argv, StringRef Tool) {
|
|
||||||
void *GetExecutablePathVP = (void *)(intptr_t) GetExecutablePath;
|
|
||||||
if (Tool == "")
|
|
||||||
return cc1_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
if (Tool == "as")
|
|
||||||
return cc1as_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
if (Tool == "gen-reproducer")
|
|
||||||
return cc1gen_reproducer_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
|
|
||||||
// Reject unknown tools.
|
|
||||||
llvm::errs() << "error: unknown integrated tool '" << Tool << "'. "
|
|
||||||
<< "Valid tools include '-cc1' and '-cc1as'.\n";
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
int mainEntryClickHouseClang(int argc_, /* const */ char **argv_) {
|
|
||||||
llvm::InitLLVM X(argc_, argv_);
|
|
||||||
SmallVector<const char *, 256> argv(argv_, argv_ + argc_);
|
|
||||||
|
|
||||||
if (llvm::sys::Process::FixupStandardFileDescriptors())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
auto TargetAndMode = ToolChain::getTargetAndModeFromProgramName(argv[0]);
|
|
||||||
|
|
||||||
llvm::BumpPtrAllocator A;
|
|
||||||
llvm::StringSaver Saver(A);
|
|
||||||
|
|
||||||
// Parse response files using the GNU syntax, unless we're in CL mode. There
|
|
||||||
// are two ways to put clang in CL compatibility mode: argv[0] is either
|
|
||||||
// clang-cl or cl, or --driver-mode=cl is on the command line. The normal
|
|
||||||
// command line parsing can't happen until after response file parsing, so we
|
|
||||||
// have to manually search for a --driver-mode=cl argument the hard way.
|
|
||||||
// Finally, our -cc1 tools don't care which tokenization mode we use because
|
|
||||||
// response files written by clang will tokenize the same way in either mode.
|
|
||||||
bool ClangCLMode = false;
|
|
||||||
if (StringRef(TargetAndMode.DriverMode).equals("--driver-mode=cl") ||
|
|
||||||
std::find_if(argv.begin(), argv.end(), [](const char *F) {
|
|
||||||
return F && strcmp(F, "--driver-mode=cl") == 0;
|
|
||||||
}) != argv.end()) {
|
|
||||||
ClangCLMode = true;
|
|
||||||
}
|
|
||||||
enum { Default, POSIX, Windows } RSPQuoting = Default;
|
|
||||||
for (const char *F : argv) {
|
|
||||||
if (strcmp(F, "--rsp-quoting=posix") == 0)
|
|
||||||
RSPQuoting = POSIX;
|
|
||||||
else if (strcmp(F, "--rsp-quoting=windows") == 0)
|
|
||||||
RSPQuoting = Windows;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Determines whether we want nullptr markers in argv to indicate response
|
|
||||||
// files end-of-lines. We only use this for the /LINK driver argument with
|
|
||||||
// clang-cl.exe on Windows.
|
|
||||||
bool MarkEOLs = ClangCLMode;
|
|
||||||
|
|
||||||
llvm::cl::TokenizerCallback Tokenizer;
|
|
||||||
if (RSPQuoting == Windows || (RSPQuoting == Default && ClangCLMode))
|
|
||||||
Tokenizer = &llvm::cl::TokenizeWindowsCommandLine;
|
|
||||||
else
|
|
||||||
Tokenizer = &llvm::cl::TokenizeGNUCommandLine;
|
|
||||||
|
|
||||||
if (MarkEOLs && argv.size() > 1 && StringRef(argv[1]).startswith("-cc1"))
|
|
||||||
MarkEOLs = false;
|
|
||||||
llvm::cl::ExpandResponseFiles(Saver, Tokenizer, argv, MarkEOLs);
|
|
||||||
|
|
||||||
// Handle -cc1 integrated tools, even if -cc1 was expanded from a response
|
|
||||||
// file.
|
|
||||||
auto FirstArg = std::find_if(argv.begin() + 1, argv.end(),
|
|
||||||
[](const char *A) { return A != nullptr; });
|
|
||||||
if (FirstArg != argv.end() && StringRef(*FirstArg).startswith("-cc1")) {
|
|
||||||
// If -cc1 came from a response file, remove the EOL sentinels.
|
|
||||||
if (MarkEOLs) {
|
|
||||||
auto newEnd = std::remove(argv.begin(), argv.end(), nullptr);
|
|
||||||
argv.resize(newEnd - argv.begin());
|
|
||||||
}
|
|
||||||
return ExecuteCC1Tool(argv, argv[1] + 4);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool CanonicalPrefixes = true;
|
|
||||||
for (int i = 1, size = argv.size(); i < size; ++i) {
|
|
||||||
// Skip end-of-line response file markers
|
|
||||||
if (argv[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
if (StringRef(argv[i]) == "-no-canonical-prefixes") {
|
|
||||||
CanonicalPrefixes = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle CL and _CL_ which permits additional command line options to be
|
|
||||||
// prepended or appended.
|
|
||||||
if (ClangCLMode) {
|
|
||||||
// Arguments in "CL" are prepended.
|
|
||||||
llvm::Optional<std::string> OptCL = llvm::sys::Process::GetEnv("CL");
|
|
||||||
if (OptCL.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> PrependedOpts;
|
|
||||||
getCLEnvVarOptions(OptCL.getValue(), Saver, PrependedOpts);
|
|
||||||
|
|
||||||
// Insert right after the program name to prepend to the argument list.
|
|
||||||
argv.insert(argv.begin() + 1, PrependedOpts.begin(), PrependedOpts.end());
|
|
||||||
}
|
|
||||||
// Arguments in "_CL_" are appended.
|
|
||||||
llvm::Optional<std::string> Opt_CL_ = llvm::sys::Process::GetEnv("_CL_");
|
|
||||||
if (Opt_CL_.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> AppendedOpts;
|
|
||||||
getCLEnvVarOptions(Opt_CL_.getValue(), Saver, AppendedOpts);
|
|
||||||
|
|
||||||
// Insert at the end of the argument list to append.
|
|
||||||
argv.append(AppendedOpts.begin(), AppendedOpts.end());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::set<std::string> SavedStrings;
|
|
||||||
// Handle CCC_OVERRIDE_OPTIONS, used for editing a command line behind the
|
|
||||||
// scenes.
|
|
||||||
if (const char *OverrideStr = ::getenv("CCC_OVERRIDE_OPTIONS")) {
|
|
||||||
// FIXME: Driver shouldn't take extra initial argument.
|
|
||||||
ApplyQAOverride(argv, OverrideStr, SavedStrings);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string Path = GetExecutablePath(argv[0], CanonicalPrefixes);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts =
|
|
||||||
CreateAndPopulateDiagOpts(argv);
|
|
||||||
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(llvm::errs(), &*DiagOpts);
|
|
||||||
FixupDiagPrefixExeName(DiagClient, Path);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
if (!DiagOpts->DiagnosticSerializationFile.empty()) {
|
|
||||||
auto SerializedConsumer =
|
|
||||||
clang::serialized_diags::create(DiagOpts->DiagnosticSerializationFile,
|
|
||||||
&*DiagOpts, /*MergeChildRecords=*/true);
|
|
||||||
Diags.setClient(new ChainedDiagnosticConsumer(
|
|
||||||
Diags.takeClient(), std::move(SerializedConsumer)));
|
|
||||||
}
|
|
||||||
|
|
||||||
ProcessWarningOptions(Diags, *DiagOpts, /*ReportDiags=*/false);
|
|
||||||
|
|
||||||
Driver TheDriver(Path, llvm::sys::getDefaultTargetTriple(), Diags);
|
|
||||||
SetInstallDir(argv, TheDriver, CanonicalPrefixes);
|
|
||||||
TheDriver.setTargetAndMode(TargetAndMode);
|
|
||||||
|
|
||||||
insertTargetAndModeArgs(TargetAndMode, argv, SavedStrings);
|
|
||||||
|
|
||||||
SetBackdoorDriverOutputsFromEnvVars(TheDriver);
|
|
||||||
|
|
||||||
std::unique_ptr<Compilation> C(TheDriver.BuildCompilation(argv));
|
|
||||||
int Res = 1;
|
|
||||||
if (C && !C->containsError()) {
|
|
||||||
SmallVector<std::pair<int, const Command *>, 4> FailingCommands;
|
|
||||||
Res = TheDriver.ExecuteCompilation(*C, FailingCommands);
|
|
||||||
|
|
||||||
// Force a crash to test the diagnostics.
|
|
||||||
if (TheDriver.GenReproducer) {
|
|
||||||
Diags.Report(diag::err_drv_force_crash)
|
|
||||||
<< !::getenv("FORCE_CLANG_DIAGNOSTICS_CRASH");
|
|
||||||
|
|
||||||
// Pretend that every command failed.
|
|
||||||
FailingCommands.clear();
|
|
||||||
for (const auto &J : C->getJobs())
|
|
||||||
if (const Command *C = dyn_cast<Command>(&J))
|
|
||||||
FailingCommands.push_back(std::make_pair(-1, C));
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const auto &P : FailingCommands) {
|
|
||||||
int CommandRes = P.first;
|
|
||||||
const Command *FailingCommand = P.second;
|
|
||||||
if (!Res)
|
|
||||||
Res = CommandRes;
|
|
||||||
|
|
||||||
// If result status is < 0, then the driver command signalled an error.
|
|
||||||
// If result status is 70, then the driver command reported a fatal error.
|
|
||||||
// On Windows, abort will return an exit code of 3. In these cases,
|
|
||||||
// generate additional diagnostic information if possible.
|
|
||||||
bool DiagnoseCrash = CommandRes < 0 || CommandRes == 70;
|
|
||||||
#ifdef _WIN32
|
|
||||||
DiagnoseCrash |= CommandRes == 3;
|
|
||||||
#endif
|
|
||||||
if (DiagnoseCrash) {
|
|
||||||
TheDriver.generateCompilationDiagnostics(*C, *FailingCommand);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Diags.getClient()->finish();
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
// Exit status should not be negative on Win32, unless abnormal termination.
|
|
||||||
// Once abnormal termiation was caught, negative status should not be
|
|
||||||
// propagated.
|
|
||||||
if (Res < 0)
|
|
||||||
Res = 1;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// If we have multiple failing commands, we return the result of the first
|
|
||||||
// failing command.
|
|
||||||
return Res;
|
|
||||||
}
|
|
@ -1,150 +0,0 @@
|
|||||||
//===- tools/lld/lld.cpp - Linker Driver Dispatcher -----------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Linker
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This file contains the main function of the lld executable. The main
|
|
||||||
// function is a thin wrapper which dispatches to the platform specific
|
|
||||||
// driver.
|
|
||||||
//
|
|
||||||
// lld is a single executable that contains four different linkers for ELF,
|
|
||||||
// COFF, WebAssembly and Mach-O. The main function dispatches according to
|
|
||||||
// argv[0] (i.e. command name). The most common name for each target is shown
|
|
||||||
// below:
|
|
||||||
//
|
|
||||||
// - ld.lld: ELF (Unix)
|
|
||||||
// - ld64: Mach-O (macOS)
|
|
||||||
// - lld-link: COFF (Windows)
|
|
||||||
// - ld-wasm: WebAssembly
|
|
||||||
//
|
|
||||||
// lld can be invoked as "lld" along with "-flavor" option. This is for
|
|
||||||
// backward compatibility and not recommended.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "lld/Common/Driver.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/ADT/StringSwitch.h"
|
|
||||||
#include "llvm/ADT/Twine.h"
|
|
||||||
#include "llvm/Support/InitLLVM.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include <cstdlib>
|
|
||||||
using namespace lld;
|
|
||||||
using namespace llvm;
|
|
||||||
using namespace llvm::sys;
|
|
||||||
|
|
||||||
/*
|
|
||||||
|
|
||||||
enum Flavor {
|
|
||||||
Invalid,
|
|
||||||
Gnu, // -flavor gnu
|
|
||||||
WinLink, // -flavor link
|
|
||||||
Darwin, // -flavor darwin
|
|
||||||
Wasm, // -flavor wasm
|
|
||||||
};
|
|
||||||
|
|
||||||
LLVM_ATTRIBUTE_NORETURN static void die(const Twine &S) {
|
|
||||||
errs() << S << "\n";
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
static Flavor getFlavor(StringRef S) {
|
|
||||||
return StringSwitch<Flavor>(S)
|
|
||||||
.CasesLower("ld", "ld.lld", "gnu", Gnu)
|
|
||||||
.CasesLower("wasm", "ld-wasm", Wasm)
|
|
||||||
.CaseLower("link", WinLink)
|
|
||||||
.CasesLower("ld64", "ld64.lld", "darwin", Darwin)
|
|
||||||
.Default(Invalid);
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool isPETarget(const std::vector<const char *> &V) {
|
|
||||||
for (auto It = V.begin(); It + 1 != V.end(); ++It) {
|
|
||||||
if (StringRef(*It) != "-m")
|
|
||||||
continue;
|
|
||||||
StringRef S = *(It + 1);
|
|
||||||
return S == "i386pe" || S == "i386pep" || S == "thumb2pe" || S == "arm64pe";
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
static Flavor parseProgname(StringRef Progname) {
|
|
||||||
#if __APPLE__
|
|
||||||
// Use Darwin driver for "ld" on Darwin.
|
|
||||||
if (Progname == "ld")
|
|
||||||
return Darwin;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if LLVM_ON_UNIX
|
|
||||||
// Use GNU driver for "ld" on other Unix-like system.
|
|
||||||
if (Progname == "ld")
|
|
||||||
return Gnu;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Progname may be something like "lld-gnu". Parse it.
|
|
||||||
SmallVector<StringRef, 3> V;
|
|
||||||
Progname.split(V, "-");
|
|
||||||
for (StringRef S : V)
|
|
||||||
if (Flavor F = getFlavor(S))
|
|
||||||
return F;
|
|
||||||
return Invalid;
|
|
||||||
}
|
|
||||||
|
|
||||||
static Flavor parseFlavor(std::vector<const char *> &V) {
|
|
||||||
// Parse -flavor option.
|
|
||||||
if (V.size() > 1 && V[1] == StringRef("-flavor")) {
|
|
||||||
if (V.size() <= 2)
|
|
||||||
die("missing arg value for '-flavor'");
|
|
||||||
Flavor F = getFlavor(V[2]);
|
|
||||||
if (F == Invalid)
|
|
||||||
die("Unknown flavor: " + StringRef(V[2]));
|
|
||||||
V.erase(V.begin() + 1, V.begin() + 3);
|
|
||||||
return F;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deduct the flavor from argv[0].
|
|
||||||
StringRef Arg0 = path::filename(V[0]);
|
|
||||||
if (Arg0.endswith_lower(".exe"))
|
|
||||||
Arg0 = Arg0.drop_back(4);
|
|
||||||
return parseProgname(Arg0);
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
// If this function returns true, lld calls _exit() so that it quickly
|
|
||||||
// exits without invoking destructors of globally allocated objects.
|
|
||||||
//
|
|
||||||
// We don't want to do that if we are running tests though, because
|
|
||||||
// doing that breaks leak sanitizer. So, lit sets this environment variable,
|
|
||||||
// and we use it to detect whether we are running tests or not.
|
|
||||||
static bool canExitEarly() { return StringRef(getenv("LLD_IN_TEST")) != "1"; }
|
|
||||||
|
|
||||||
/// Universal linker main(). This linker emulates the gnu, darwin, or
|
|
||||||
/// windows linker based on the argv[0] or -flavor option.
|
|
||||||
int mainEntryClickHouseLLD(int Argc, /* const */ char **Argv) {
|
|
||||||
InitLLVM X(Argc, Argv);
|
|
||||||
|
|
||||||
std::vector<const char *> Args(Argv, Argv + Argc);
|
|
||||||
/*
|
|
||||||
switch (parseFlavor(Args)) {
|
|
||||||
case Gnu:
|
|
||||||
if (isPETarget(Args))
|
|
||||||
return !mingw::link(Args);
|
|
||||||
*/
|
|
||||||
return !elf::link(Args, canExitEarly());
|
|
||||||
/*
|
|
||||||
case WinLink:
|
|
||||||
return !coff::link(Args, canExitEarly());
|
|
||||||
case Darwin:
|
|
||||||
return !mach_o::link(Args, canExitEarly());
|
|
||||||
case Wasm:
|
|
||||||
return !wasm::link(Args, canExitEarly());
|
|
||||||
default:
|
|
||||||
die("lld is a generic driver.\n"
|
|
||||||
"Invoke ld.lld (Unix), ld64.lld (macOS), lld-link (Windows), wasm-lld"
|
|
||||||
" (WebAssembly) instead");
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
}
|
|
@ -1,49 +0,0 @@
|
|||||||
add_definitions(-Wno-error -Wno-unused-parameter -Wno-non-virtual-dtor -U_LIBCPP_DEBUG)
|
|
||||||
|
|
||||||
link_directories(${LLVM_LIBRARY_DIRS})
|
|
||||||
|
|
||||||
add_library(clickhouse-compiler-lib
|
|
||||||
driver.cpp
|
|
||||||
cc1_main.cpp
|
|
||||||
cc1as_main.cpp
|
|
||||||
lld.cpp)
|
|
||||||
|
|
||||||
target_compile_options(clickhouse-compiler-lib PRIVATE -fno-rtti -fno-exceptions -g0)
|
|
||||||
|
|
||||||
string(REPLACE "${INCLUDE_DEBUG_HELPERS}" "" CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS}) # cant compile with -fno-rtti
|
|
||||||
|
|
||||||
llvm_libs_all(REQUIRED_LLVM_LIBRARIES)
|
|
||||||
|
|
||||||
message(STATUS "Using LLVM ${LLVM_VERSION}: ${LLVM_INCLUDE_DIRS} : ${REQUIRED_LLVM_LIBRARIES}")
|
|
||||||
|
|
||||||
target_include_directories(clickhouse-compiler-lib SYSTEM PRIVATE ${LLVM_INCLUDE_DIRS})
|
|
||||||
|
|
||||||
# This is extracted almost directly from CMakeFiles/.../link.txt in LLVM build directory.
|
|
||||||
|
|
||||||
target_link_libraries(clickhouse-compiler-lib PRIVATE
|
|
||||||
|
|
||||||
clangBasic clangCodeGen clangDriver
|
|
||||||
clangFrontend
|
|
||||||
clangFrontendTool
|
|
||||||
clangRewriteFrontend clangARCMigrate clangStaticAnalyzerFrontend
|
|
||||||
clangParse clangSerialization clangSema clangEdit clangStaticAnalyzerCheckers
|
|
||||||
clangASTMatchers clangStaticAnalyzerCore clangAnalysis clangAST clangRewrite clangLex clangBasic
|
|
||||||
clangCrossTU clangIndex
|
|
||||||
|
|
||||||
lldCOFF
|
|
||||||
lldDriver
|
|
||||||
lldELF
|
|
||||||
lldMinGW
|
|
||||||
lldMachO
|
|
||||||
lldReaderWriter
|
|
||||||
lldYAML
|
|
||||||
lldCommon
|
|
||||||
lldCore
|
|
||||||
|
|
||||||
${REQUIRED_LLVM_LIBRARIES}
|
|
||||||
|
|
||||||
PUBLIC ${ZLIB_LIBRARIES} ${EXECINFO_LIBRARIES} Threads::Threads
|
|
||||||
${MALLOC_LIBRARIES}
|
|
||||||
${GLIBC_COMPATIBILITY_LIBRARIES}
|
|
||||||
${MEMCPY_LIBRARIES}
|
|
||||||
)
|
|
@ -1,243 +0,0 @@
|
|||||||
//===-- cc1_main.cpp - Clang CC1 Compiler Frontend ------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1 functionality, which implements the
|
|
||||||
// core compiler functionality along with a number of additional tools for
|
|
||||||
// demonstration and testing purposes.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "clang/CodeGen/ObjectFilePCHContainerOperations.h"
|
|
||||||
#include "clang/Config/config.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/CompilerInstance.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticBuffer.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "clang/FrontendTool/Utils.h"
|
|
||||||
#include "llvm/ADT/Statistic.h"
|
|
||||||
#include "llvm/Config/llvm-config.h"
|
|
||||||
#include "llvm/LinkAllPasses.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/Compiler.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/ManagedStatic.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <cstdio>
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
#include <sys/resource.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// have no .a version in packages
|
|
||||||
#undef LINK_POLLY_INTO_TOOLS
|
|
||||||
|
|
||||||
using namespace clang;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
// Main driver
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool GenCrashDiag) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// Run the interrupt handlers to make sure any special cleanups get done, in
|
|
||||||
// particular that we remove files registered with RemoveFileOnSignal.
|
|
||||||
llvm::sys::RunInterruptHandlers();
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors. When reporting a fatal error, exit
|
|
||||||
// with status 70 to generate crash diagnostics. For BSD systems this is
|
|
||||||
// defined as an internal software error. Otherwise, exit with status 1.
|
|
||||||
exit(GenCrashDiag ? 70 : 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
namespace polly {
|
|
||||||
void initializePollyPasses(llvm::PassRegistry &Registry);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef CLANG_HAVE_RLIMITS
|
|
||||||
// The amount of stack we think is "sufficient". If less than this much is
|
|
||||||
// available, we may be unable to reach our template instantiation depth
|
|
||||||
// limit and other similar limits.
|
|
||||||
// FIXME: Unify this with the stack we request when spawning a thread to build
|
|
||||||
// a module.
|
|
||||||
static const int kSufficientStack = 8 << 20;
|
|
||||||
|
|
||||||
#if defined(__linux__) && defined(__PIE__)
|
|
||||||
static size_t getCurrentStackAllocation() {
|
|
||||||
// If we can't compute the current stack usage, allow for 512K of command
|
|
||||||
// line arguments and environment.
|
|
||||||
size_t Usage = 512 * 1024;
|
|
||||||
if (FILE *StatFile = fopen("/proc/self/stat", "r")) {
|
|
||||||
// We assume that the stack extends from its current address to the end of
|
|
||||||
// the environment space. In reality, there is another string literal (the
|
|
||||||
// program name) after the environment, but this is close enough (we only
|
|
||||||
// need to be within 100K or so).
|
|
||||||
unsigned long StackPtr, EnvEnd;
|
|
||||||
// Disable silly GCC -Wformat warning that complains about length
|
|
||||||
// modifiers on ignored format specifiers. We want to retain these
|
|
||||||
// for documentation purposes even though they have no effect.
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic push
|
|
||||||
#pragma GCC diagnostic ignored "-Wformat"
|
|
||||||
#endif
|
|
||||||
if (fscanf(StatFile,
|
|
||||||
"%*d %*s %*c %*d %*d %*d %*d %*d %*u %*lu %*lu %*lu %*lu %*lu "
|
|
||||||
"%*lu %*ld %*ld %*ld %*ld %*ld %*ld %*llu %*lu %*ld %*lu %*lu "
|
|
||||||
"%*lu %*lu %lu %*lu %*lu %*lu %*lu %*lu %*llu %*lu %*lu %*d %*d "
|
|
||||||
"%*u %*u %*llu %*lu %*ld %*lu %*lu %*lu %*lu %*lu %*lu %lu %*d",
|
|
||||||
&StackPtr, &EnvEnd) == 2) {
|
|
||||||
#if defined(__GNUC__) && !defined(__clang__)
|
|
||||||
#pragma GCC diagnostic pop
|
|
||||||
#endif
|
|
||||||
Usage = StackPtr < EnvEnd ? EnvEnd - StackPtr : StackPtr - EnvEnd;
|
|
||||||
}
|
|
||||||
fclose(StatFile);
|
|
||||||
}
|
|
||||||
return Usage;
|
|
||||||
}
|
|
||||||
|
|
||||||
#include <alloca.h>
|
|
||||||
|
|
||||||
LLVM_ATTRIBUTE_NOINLINE
|
|
||||||
static void ensureStackAddressSpace(int ExtraChunks = 0) {
|
|
||||||
// Linux kernels prior to 4.1 will sometimes locate the heap of a PIE binary
|
|
||||||
// relatively close to the stack (they are only guaranteed to be 128MiB
|
|
||||||
// apart). This results in crashes if we happen to heap-allocate more than
|
|
||||||
// 128MiB before we reach our stack high-water mark.
|
|
||||||
//
|
|
||||||
// To avoid these crashes, ensure that we have sufficient virtual memory
|
|
||||||
// pages allocated before we start running.
|
|
||||||
size_t Curr = getCurrentStackAllocation();
|
|
||||||
const int kTargetStack = kSufficientStack - 256 * 1024;
|
|
||||||
if (Curr < kTargetStack) {
|
|
||||||
volatile char *volatile Alloc =
|
|
||||||
static_cast<volatile char *>(alloca(kTargetStack - Curr));
|
|
||||||
Alloc[0] = 0;
|
|
||||||
Alloc[kTargetStack - Curr - 1] = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureStackAddressSpace() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/// Attempt to ensure that we have at least 8MiB of usable stack space.
|
|
||||||
static void ensureSufficientStack() {
|
|
||||||
struct rlimit rlim;
|
|
||||||
if (getrlimit(RLIMIT_STACK, &rlim) != 0)
|
|
||||||
return;
|
|
||||||
|
|
||||||
// Increase the soft stack limit to our desired level, if necessary and
|
|
||||||
// possible.
|
|
||||||
if (rlim.rlim_cur != RLIM_INFINITY && rlim.rlim_cur < kSufficientStack) {
|
|
||||||
// Try to allocate sufficient stack.
|
|
||||||
if (rlim.rlim_max == RLIM_INFINITY || rlim.rlim_max >= kSufficientStack)
|
|
||||||
rlim.rlim_cur = kSufficientStack;
|
|
||||||
else if (rlim.rlim_cur == rlim.rlim_max)
|
|
||||||
return;
|
|
||||||
else
|
|
||||||
rlim.rlim_cur = rlim.rlim_max;
|
|
||||||
|
|
||||||
if (setrlimit(RLIMIT_STACK, &rlim) != 0 ||
|
|
||||||
rlim.rlim_cur != kSufficientStack)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// We should now have a stack of size at least kSufficientStack. Ensure
|
|
||||||
// that we can actually use that much, if necessary.
|
|
||||||
ensureStackAddressSpace();
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
static void ensureSufficientStack() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
int cc1_main(ArrayRef<const char *> Argv, const char *Argv0, void *MainAddr) {
|
|
||||||
ensureSufficientStack();
|
|
||||||
|
|
||||||
std::unique_ptr<CompilerInstance> Clang(new CompilerInstance());
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
// Register the support for object-file-wrapped Clang modules.
|
|
||||||
auto PCHOps = Clang->getPCHContainerOperations();
|
|
||||||
PCHOps->registerWriter(llvm::make_unique<ObjectFilePCHContainerWriter>());
|
|
||||||
PCHOps->registerReader(llvm::make_unique<ObjectFilePCHContainerReader>());
|
|
||||||
|
|
||||||
// Initialize targets first, so that --version shows registered targets.
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
llvm::InitializeAllTargetMCs();
|
|
||||||
llvm::InitializeAllAsmPrinters();
|
|
||||||
llvm::InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
#ifdef LINK_POLLY_INTO_TOOLS
|
|
||||||
llvm::PassRegistry &Registry = *llvm::PassRegistry::getPassRegistry();
|
|
||||||
polly::initializePollyPasses(Registry);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Buffer diagnostics from argument parsing so that we can output them using a
|
|
||||||
// well formed diagnostic object.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticBuffer *DiagsBuffer = new TextDiagnosticBuffer;
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagsBuffer);
|
|
||||||
bool Success = CompilerInvocation::CreateFromArgs(
|
|
||||||
Clang->getInvocation(), Argv.begin(), Argv.end(), Diags);
|
|
||||||
|
|
||||||
// Infer the builtin include path if unspecified.
|
|
||||||
if (Clang->getHeaderSearchOpts().UseBuiltinIncludes &&
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir.empty())
|
|
||||||
Clang->getHeaderSearchOpts().ResourceDir =
|
|
||||||
CompilerInvocation::GetResourcesPath(Argv0, MainAddr);
|
|
||||||
|
|
||||||
// Create the actual diagnostics engine.
|
|
||||||
Clang->createDiagnostics();
|
|
||||||
if (!Clang->hasDiagnostics())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
llvm::install_fatal_error_handler(LLVMErrorHandler,
|
|
||||||
static_cast<void*>(&Clang->getDiagnostics()));
|
|
||||||
|
|
||||||
DiagsBuffer->FlushDiagnostics(Clang->getDiagnostics());
|
|
||||||
if (!Success)
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
// Execute the frontend actions.
|
|
||||||
Success = ExecuteCompilerInvocation(Clang.get());
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
// Our error handler depends on the Diagnostics object, which we're
|
|
||||||
// potentially about to delete. Uninstall the handler now so that any
|
|
||||||
// later errors use the default handling behavior instead.
|
|
||||||
llvm::remove_fatal_error_handler();
|
|
||||||
|
|
||||||
// When running with -disable-free, don't do any destruction or shutdown.
|
|
||||||
if (Clang->getFrontendOpts().DisableFree) {
|
|
||||||
BuryPointer(std::move(Clang));
|
|
||||||
return !Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
return !Success;
|
|
||||||
}
|
|
@ -1,555 +0,0 @@
|
|||||||
//===-- cc1as_main.cpp - Clang Assembler ---------------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang -cc1as functionality, which implements
|
|
||||||
// the direct interface to the LLVM MC based assembler.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Basic/Diagnostic.h"
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Frontend/FrontendDiagnostic.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/STLExtras.h"
|
|
||||||
#include "llvm/ADT/StringSwitch.h"
|
|
||||||
#include "llvm/ADT/Triple.h"
|
|
||||||
#include "llvm/IR/DataLayout.h"
|
|
||||||
#include "llvm/MC/MCAsmBackend.h"
|
|
||||||
#include "llvm/MC/MCAsmInfo.h"
|
|
||||||
#include "llvm/MC/MCCodeEmitter.h"
|
|
||||||
#include "llvm/MC/MCContext.h"
|
|
||||||
#include "llvm/MC/MCInstrInfo.h"
|
|
||||||
#include "llvm/MC/MCObjectFileInfo.h"
|
|
||||||
#include "llvm/MC/MCParser/MCAsmParser.h"
|
|
||||||
#include "llvm/MC/MCParser/MCTargetAsmParser.h"
|
|
||||||
#include "llvm/MC/MCRegisterInfo.h"
|
|
||||||
#include "llvm/MC/MCStreamer.h"
|
|
||||||
#include "llvm/MC/MCSubtargetInfo.h"
|
|
||||||
#include "llvm/MC/MCTargetOptions.h"
|
|
||||||
#include "llvm/Option/Arg.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/FormattedStream.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/MemoryBuffer.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/SourceMgr.h"
|
|
||||||
#include "llvm/Support/TargetRegistry.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace clang::driver::options;
|
|
||||||
using namespace llvm;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
/// \brief Helper class for representing a single invocation of the assembler.
|
|
||||||
struct AssemblerInvocation {
|
|
||||||
/// @name Target Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
/// The name of the target triple to assemble for.
|
|
||||||
std::string Triple;
|
|
||||||
|
|
||||||
/// If given, the name of the target CPU to determine which instructions
|
|
||||||
/// are legal.
|
|
||||||
std::string CPU;
|
|
||||||
|
|
||||||
/// The list of target specific features to enable or disable -- this should
|
|
||||||
/// be a list of strings starting with '+' or '-'.
|
|
||||||
std::vector<std::string> Features;
|
|
||||||
|
|
||||||
/// The list of symbol definitions.
|
|
||||||
std::vector<std::string> SymbolDefs;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Language Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::vector<std::string> IncludePaths;
|
|
||||||
unsigned NoInitialTextSection : 1;
|
|
||||||
unsigned SaveTemporaryLabels : 1;
|
|
||||||
unsigned GenDwarfForAssembly : 1;
|
|
||||||
unsigned RelaxELFRelocations : 1;
|
|
||||||
unsigned DwarfVersion;
|
|
||||||
std::string DwarfDebugFlags;
|
|
||||||
std::string DwarfDebugProducer;
|
|
||||||
std::string DebugCompilationDir;
|
|
||||||
llvm::DebugCompressionType CompressDebugSections =
|
|
||||||
llvm::DebugCompressionType::None;
|
|
||||||
std::string MainFileName;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Frontend Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
std::string InputFile;
|
|
||||||
std::vector<std::string> LLVMArgs;
|
|
||||||
std::string OutputPath;
|
|
||||||
enum FileType {
|
|
||||||
FT_Asm, ///< Assembly (.s) output, transliterate mode.
|
|
||||||
FT_Null, ///< No output, for timing purposes.
|
|
||||||
FT_Obj ///< Object file output.
|
|
||||||
};
|
|
||||||
FileType OutputType;
|
|
||||||
unsigned ShowHelp : 1;
|
|
||||||
unsigned ShowVersion : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Transliterate Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned OutputAsmVariant;
|
|
||||||
unsigned ShowEncoding : 1;
|
|
||||||
unsigned ShowInst : 1;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
/// @name Assembler Options
|
|
||||||
/// @{
|
|
||||||
|
|
||||||
unsigned RelaxAll : 1;
|
|
||||||
unsigned NoExecStack : 1;
|
|
||||||
unsigned FatalWarnings : 1;
|
|
||||||
unsigned IncrementalLinkerCompatible : 1;
|
|
||||||
|
|
||||||
/// The name of the relocation model to use.
|
|
||||||
std::string RelocationModel;
|
|
||||||
|
|
||||||
/// @}
|
|
||||||
|
|
||||||
public:
|
|
||||||
AssemblerInvocation() {
|
|
||||||
Triple = "";
|
|
||||||
NoInitialTextSection = 0;
|
|
||||||
InputFile = "-";
|
|
||||||
OutputPath = "-";
|
|
||||||
OutputType = FT_Asm;
|
|
||||||
OutputAsmVariant = 0;
|
|
||||||
ShowInst = 0;
|
|
||||||
ShowEncoding = 0;
|
|
||||||
RelaxAll = 0;
|
|
||||||
NoExecStack = 0;
|
|
||||||
FatalWarnings = 0;
|
|
||||||
IncrementalLinkerCompatible = 0;
|
|
||||||
DwarfVersion = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool CreateFromArgs(AssemblerInvocation &Res,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags);
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
bool AssemblerInvocation::CreateFromArgs(AssemblerInvocation &Opts,
|
|
||||||
ArrayRef<const char *> Argv,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
bool Success = true;
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
std::unique_ptr<OptTable> OptTbl(createDriverOptTable());
|
|
||||||
|
|
||||||
const unsigned IncludedFlagsBitmask = options::CC1AsOption;
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args = OptTbl->ParseArgs(Argv, MissingArgIndex, MissingArgCount,
|
|
||||||
IncludedFlagsBitmask);
|
|
||||||
|
|
||||||
// Check for missing argument error.
|
|
||||||
if (MissingArgCount) {
|
|
||||||
Diags.Report(diag::err_drv_missing_argument)
|
|
||||||
<< Args.getArgString(MissingArgIndex) << MissingArgCount;
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Issue errors on unknown arguments.
|
|
||||||
for (const Arg *A : Args.filtered(OPT_UNKNOWN)) {
|
|
||||||
auto ArgString = A->getAsString(Args);
|
|
||||||
std::string Nearest;
|
|
||||||
if (OptTbl->findNearest(ArgString, Nearest, IncludedFlagsBitmask) > 1)
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << ArgString;
|
|
||||||
else
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument_with_suggestion)
|
|
||||||
<< ArgString << Nearest;
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Construct the invocation.
|
|
||||||
|
|
||||||
// Target Options
|
|
||||||
Opts.Triple = llvm::Triple::normalize(Args.getLastArgValue(OPT_triple));
|
|
||||||
Opts.CPU = Args.getLastArgValue(OPT_target_cpu);
|
|
||||||
Opts.Features = Args.getAllArgValues(OPT_target_feature);
|
|
||||||
|
|
||||||
// Use the default target triple if unspecified.
|
|
||||||
if (Opts.Triple.empty())
|
|
||||||
Opts.Triple = llvm::sys::getDefaultTargetTriple();
|
|
||||||
|
|
||||||
// Language Options
|
|
||||||
Opts.IncludePaths = Args.getAllArgValues(OPT_I);
|
|
||||||
Opts.NoInitialTextSection = Args.hasArg(OPT_n);
|
|
||||||
Opts.SaveTemporaryLabels = Args.hasArg(OPT_msave_temp_labels);
|
|
||||||
// Any DebugInfoKind implies GenDwarfForAssembly.
|
|
||||||
Opts.GenDwarfForAssembly = Args.hasArg(OPT_debug_info_kind_EQ);
|
|
||||||
|
|
||||||
if (const Arg *A = Args.getLastArg(OPT_compress_debug_sections,
|
|
||||||
OPT_compress_debug_sections_EQ)) {
|
|
||||||
if (A->getOption().getID() == OPT_compress_debug_sections) {
|
|
||||||
// TODO: be more clever about the compression type auto-detection
|
|
||||||
Opts.CompressDebugSections = llvm::DebugCompressionType::GNU;
|
|
||||||
} else {
|
|
||||||
Opts.CompressDebugSections =
|
|
||||||
llvm::StringSwitch<llvm::DebugCompressionType>(A->getValue())
|
|
||||||
.Case("none", llvm::DebugCompressionType::None)
|
|
||||||
.Case("zlib", llvm::DebugCompressionType::Z)
|
|
||||||
.Case("zlib-gnu", llvm::DebugCompressionType::GNU)
|
|
||||||
.Default(llvm::DebugCompressionType::None);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Opts.RelaxELFRelocations = Args.hasArg(OPT_mrelax_relocations);
|
|
||||||
Opts.DwarfVersion = getLastArgIntValue(Args, OPT_dwarf_version_EQ, 2, Diags);
|
|
||||||
Opts.DwarfDebugFlags = Args.getLastArgValue(OPT_dwarf_debug_flags);
|
|
||||||
Opts.DwarfDebugProducer = Args.getLastArgValue(OPT_dwarf_debug_producer);
|
|
||||||
Opts.DebugCompilationDir = Args.getLastArgValue(OPT_fdebug_compilation_dir);
|
|
||||||
Opts.MainFileName = Args.getLastArgValue(OPT_main_file_name);
|
|
||||||
|
|
||||||
// Frontend Options
|
|
||||||
if (Args.hasArg(OPT_INPUT)) {
|
|
||||||
bool First = true;
|
|
||||||
for (const Arg *A : Args.filtered(OPT_INPUT)) {
|
|
||||||
if (First) {
|
|
||||||
Opts.InputFile = A->getValue();
|
|
||||||
First = false;
|
|
||||||
} else {
|
|
||||||
Diags.Report(diag::err_drv_unknown_argument) << A->getAsString(Args);
|
|
||||||
Success = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Opts.LLVMArgs = Args.getAllArgValues(OPT_mllvm);
|
|
||||||
Opts.OutputPath = Args.getLastArgValue(OPT_o);
|
|
||||||
if (Arg *A = Args.getLastArg(OPT_filetype)) {
|
|
||||||
StringRef Name = A->getValue();
|
|
||||||
unsigned OutputType = StringSwitch<unsigned>(Name)
|
|
||||||
.Case("asm", FT_Asm)
|
|
||||||
.Case("null", FT_Null)
|
|
||||||
.Case("obj", FT_Obj)
|
|
||||||
.Default(~0U);
|
|
||||||
if (OutputType == ~0U) {
|
|
||||||
Diags.Report(diag::err_drv_invalid_value) << A->getAsString(Args) << Name;
|
|
||||||
Success = false;
|
|
||||||
} else
|
|
||||||
Opts.OutputType = FileType(OutputType);
|
|
||||||
}
|
|
||||||
Opts.ShowHelp = Args.hasArg(OPT_help);
|
|
||||||
Opts.ShowVersion = Args.hasArg(OPT_version);
|
|
||||||
|
|
||||||
// Transliterate Options
|
|
||||||
Opts.OutputAsmVariant =
|
|
||||||
getLastArgIntValue(Args, OPT_output_asm_variant, 0, Diags);
|
|
||||||
Opts.ShowEncoding = Args.hasArg(OPT_show_encoding);
|
|
||||||
Opts.ShowInst = Args.hasArg(OPT_show_inst);
|
|
||||||
|
|
||||||
// Assemble Options
|
|
||||||
Opts.RelaxAll = Args.hasArg(OPT_mrelax_all);
|
|
||||||
Opts.NoExecStack = Args.hasArg(OPT_mno_exec_stack);
|
|
||||||
Opts.FatalWarnings = Args.hasArg(OPT_massembler_fatal_warnings);
|
|
||||||
Opts.RelocationModel = Args.getLastArgValue(OPT_mrelocation_model, "pic");
|
|
||||||
Opts.IncrementalLinkerCompatible =
|
|
||||||
Args.hasArg(OPT_mincremental_linker_compatible);
|
|
||||||
Opts.SymbolDefs = Args.getAllArgValues(OPT_defsym);
|
|
||||||
|
|
||||||
return Success;
|
|
||||||
}
|
|
||||||
|
|
||||||
static std::unique_ptr<raw_fd_ostream>
|
|
||||||
getOutputStream(AssemblerInvocation &Opts, DiagnosticsEngine &Diags,
|
|
||||||
bool Binary) {
|
|
||||||
if (Opts.OutputPath.empty())
|
|
||||||
Opts.OutputPath = "-";
|
|
||||||
|
|
||||||
// Make sure that the Out file gets unlinked from the disk if we get a
|
|
||||||
// SIGINT.
|
|
||||||
if (Opts.OutputPath != "-")
|
|
||||||
sys::RemoveFileOnSignal(Opts.OutputPath);
|
|
||||||
|
|
||||||
std::error_code EC;
|
|
||||||
auto Out = llvm::make_unique<raw_fd_ostream>(
|
|
||||||
Opts.OutputPath, EC, (Binary ? sys::fs::F_None : sys::fs::F_Text));
|
|
||||||
if (EC) {
|
|
||||||
Diags.Report(diag::err_fe_unable_to_open_output) << Opts.OutputPath
|
|
||||||
<< EC.message();
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return Out;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool ExecuteAssembler(AssemblerInvocation &Opts,
|
|
||||||
DiagnosticsEngine &Diags) {
|
|
||||||
// Get the target specific parser.
|
|
||||||
std::string Error;
|
|
||||||
const Target *TheTarget = TargetRegistry::lookupTarget(Opts.Triple, Error);
|
|
||||||
if (!TheTarget)
|
|
||||||
return Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
ErrorOr<std::unique_ptr<MemoryBuffer>> Buffer =
|
|
||||||
MemoryBuffer::getFileOrSTDIN(Opts.InputFile);
|
|
||||||
|
|
||||||
if (std::error_code EC = Buffer.getError()) {
|
|
||||||
Error = EC.message();
|
|
||||||
return Diags.Report(diag::err_fe_error_reading) << Opts.InputFile;
|
|
||||||
}
|
|
||||||
|
|
||||||
SourceMgr SrcMgr;
|
|
||||||
|
|
||||||
// Tell SrcMgr about this buffer, which is what the parser will pick up.
|
|
||||||
SrcMgr.AddNewSourceBuffer(std::move(*Buffer), SMLoc());
|
|
||||||
|
|
||||||
// Record the location of the include directories so that the lexer can find
|
|
||||||
// it later.
|
|
||||||
SrcMgr.setIncludeDirs(Opts.IncludePaths);
|
|
||||||
|
|
||||||
std::unique_ptr<MCRegisterInfo> MRI(TheTarget->createMCRegInfo(Opts.Triple));
|
|
||||||
assert(MRI && "Unable to create target register info!");
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, Opts.Triple));
|
|
||||||
assert(MAI && "Unable to create target asm info!");
|
|
||||||
|
|
||||||
// Ensure MCAsmInfo initialization occurs before any use, otherwise sections
|
|
||||||
// may be created with a combination of default and explicit settings.
|
|
||||||
MAI->setCompressDebugSections(Opts.CompressDebugSections);
|
|
||||||
|
|
||||||
MAI->setRelaxELFRelocations(Opts.RelaxELFRelocations);
|
|
||||||
|
|
||||||
bool IsBinary = Opts.OutputType == AssemblerInvocation::FT_Obj;
|
|
||||||
std::unique_ptr<raw_fd_ostream> FDOS = getOutputStream(Opts, Diags, IsBinary);
|
|
||||||
if (!FDOS)
|
|
||||||
return true;
|
|
||||||
|
|
||||||
// FIXME: This is not pretty. MCContext has a ptr to MCObjectFileInfo and
|
|
||||||
// MCObjectFileInfo needs a MCContext reference in order to initialize itself.
|
|
||||||
std::unique_ptr<MCObjectFileInfo> MOFI(new MCObjectFileInfo());
|
|
||||||
|
|
||||||
MCContext Ctx(MAI.get(), MRI.get(), MOFI.get(), &SrcMgr);
|
|
||||||
|
|
||||||
bool PIC = false;
|
|
||||||
if (Opts.RelocationModel == "static") {
|
|
||||||
PIC = false;
|
|
||||||
} else if (Opts.RelocationModel == "pic") {
|
|
||||||
PIC = true;
|
|
||||||
} else {
|
|
||||||
assert(Opts.RelocationModel == "dynamic-no-pic" &&
|
|
||||||
"Invalid PIC model!");
|
|
||||||
PIC = false;
|
|
||||||
}
|
|
||||||
|
|
||||||
MOFI->InitMCObjectFileInfo(Triple(Opts.Triple), PIC, Ctx);
|
|
||||||
if (Opts.SaveTemporaryLabels)
|
|
||||||
Ctx.setAllowTemporaryLabels(false);
|
|
||||||
if (Opts.GenDwarfForAssembly)
|
|
||||||
Ctx.setGenDwarfForAssembly(true);
|
|
||||||
if (!Opts.DwarfDebugFlags.empty())
|
|
||||||
Ctx.setDwarfDebugFlags(StringRef(Opts.DwarfDebugFlags));
|
|
||||||
if (!Opts.DwarfDebugProducer.empty())
|
|
||||||
Ctx.setDwarfDebugProducer(StringRef(Opts.DwarfDebugProducer));
|
|
||||||
if (!Opts.DebugCompilationDir.empty())
|
|
||||||
Ctx.setCompilationDir(Opts.DebugCompilationDir);
|
|
||||||
if (!Opts.MainFileName.empty())
|
|
||||||
Ctx.setMainFileName(StringRef(Opts.MainFileName));
|
|
||||||
Ctx.setDwarfVersion(Opts.DwarfVersion);
|
|
||||||
|
|
||||||
// Build up the feature string from the target feature list.
|
|
||||||
std::string FS;
|
|
||||||
if (!Opts.Features.empty()) {
|
|
||||||
FS = Opts.Features[0];
|
|
||||||
for (unsigned i = 1, e = Opts.Features.size(); i != e; ++i)
|
|
||||||
FS += "," + Opts.Features[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
std::unique_ptr<MCStreamer> Str;
|
|
||||||
|
|
||||||
std::unique_ptr<MCInstrInfo> MCII(TheTarget->createMCInstrInfo());
|
|
||||||
std::unique_ptr<MCSubtargetInfo> STI(
|
|
||||||
TheTarget->createMCSubtargetInfo(Opts.Triple, Opts.CPU, FS));
|
|
||||||
|
|
||||||
raw_pwrite_stream *Out = FDOS.get();
|
|
||||||
std::unique_ptr<buffer_ostream> BOS;
|
|
||||||
|
|
||||||
// FIXME: There is a bit of code duplication with addPassesToEmitFile.
|
|
||||||
if (Opts.OutputType == AssemblerInvocation::FT_Asm) {
|
|
||||||
MCInstPrinter *IP = TheTarget->createMCInstPrinter(
|
|
||||||
llvm::Triple(Opts.Triple), Opts.OutputAsmVariant, *MAI, *MCII, *MRI);
|
|
||||||
|
|
||||||
std::unique_ptr<MCCodeEmitter> CE;
|
|
||||||
if (Opts.ShowEncoding)
|
|
||||||
CE.reset(TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx));
|
|
||||||
MCTargetOptions MCOptions;
|
|
||||||
std::unique_ptr<MCAsmBackend> MAB(
|
|
||||||
TheTarget->createMCAsmBackend(*STI, *MRI, MCOptions));
|
|
||||||
|
|
||||||
auto FOut = llvm::make_unique<formatted_raw_ostream>(*Out);
|
|
||||||
Str.reset(TheTarget->createAsmStreamer(
|
|
||||||
Ctx, std::move(FOut), /*asmverbose*/ true,
|
|
||||||
/*useDwarfDirectory*/ true, IP, std::move(CE), std::move(MAB),
|
|
||||||
Opts.ShowInst));
|
|
||||||
} else if (Opts.OutputType == AssemblerInvocation::FT_Null) {
|
|
||||||
Str.reset(createNullStreamer(Ctx));
|
|
||||||
} else {
|
|
||||||
assert(Opts.OutputType == AssemblerInvocation::FT_Obj &&
|
|
||||||
"Invalid file type!");
|
|
||||||
if (!FDOS->supportsSeeking()) {
|
|
||||||
BOS = make_unique<buffer_ostream>(*FDOS);
|
|
||||||
Out = BOS.get();
|
|
||||||
}
|
|
||||||
|
|
||||||
std::unique_ptr<MCCodeEmitter> CE(
|
|
||||||
TheTarget->createMCCodeEmitter(*MCII, *MRI, Ctx));
|
|
||||||
MCTargetOptions MCOptions;
|
|
||||||
std::unique_ptr<MCAsmBackend> MAB(
|
|
||||||
TheTarget->createMCAsmBackend(*STI, *MRI, MCOptions));
|
|
||||||
|
|
||||||
Triple T(Opts.Triple);
|
|
||||||
Str.reset(TheTarget->createMCObjectStreamer(
|
|
||||||
T, Ctx, std::move(MAB), *Out, std::move(CE), *STI, Opts.RelaxAll,
|
|
||||||
Opts.IncrementalLinkerCompatible,
|
|
||||||
/*DWARFMustBeAtTheEnd*/ true));
|
|
||||||
Str.get()->InitSections(Opts.NoExecStack);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Assembly to object compilation should leverage assembly info.
|
|
||||||
Str->setUseAssemblerInfoForParsing(true);
|
|
||||||
|
|
||||||
bool Failed = false;
|
|
||||||
|
|
||||||
std::unique_ptr<MCAsmParser> Parser(
|
|
||||||
createMCAsmParser(SrcMgr, Ctx, *Str.get(), *MAI));
|
|
||||||
|
|
||||||
// FIXME: init MCTargetOptions from sanitizer flags here.
|
|
||||||
MCTargetOptions Options;
|
|
||||||
std::unique_ptr<MCTargetAsmParser> TAP(
|
|
||||||
TheTarget->createMCAsmParser(*STI, *Parser, *MCII, Options));
|
|
||||||
if (!TAP)
|
|
||||||
Failed = Diags.Report(diag::err_target_unknown_triple) << Opts.Triple;
|
|
||||||
|
|
||||||
// Set values for symbols, if any.
|
|
||||||
for (auto &S : Opts.SymbolDefs) {
|
|
||||||
auto Pair = StringRef(S).split('=');
|
|
||||||
auto Sym = Pair.first;
|
|
||||||
auto Val = Pair.second;
|
|
||||||
int64_t Value;
|
|
||||||
// We have already error checked this in the driver.
|
|
||||||
Val.getAsInteger(0, Value);
|
|
||||||
Ctx.setSymbolValue(Parser->getStreamer(), Sym, Value);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!Failed) {
|
|
||||||
Parser->setTargetParser(*TAP.get());
|
|
||||||
Failed = Parser->Run(Opts.NoInitialTextSection);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close Streamer first.
|
|
||||||
// It might have a reference to the output stream.
|
|
||||||
Str.reset();
|
|
||||||
// Close the output stream early.
|
|
||||||
BOS.reset();
|
|
||||||
FDOS.reset();
|
|
||||||
|
|
||||||
// Delete output file if there were errors.
|
|
||||||
if (Failed && Opts.OutputPath != "-")
|
|
||||||
sys::fs::remove(Opts.OutputPath);
|
|
||||||
|
|
||||||
return Failed;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void LLVMErrorHandler(void *UserData, const std::string &Message,
|
|
||||||
bool /*GenCrashDiag*/) {
|
|
||||||
DiagnosticsEngine &Diags = *static_cast<DiagnosticsEngine*>(UserData);
|
|
||||||
|
|
||||||
Diags.Report(diag::err_fe_error_backend) << Message;
|
|
||||||
|
|
||||||
// We cannot recover from llvm errors.
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
int cc1as_main(ArrayRef<const char *> Argv, const char */*Argv0*/, void */*MainAddr*/) {
|
|
||||||
// Initialize targets and assembly printers/parsers.
|
|
||||||
InitializeAllTargetInfos();
|
|
||||||
InitializeAllTargetMCs();
|
|
||||||
InitializeAllAsmParsers();
|
|
||||||
|
|
||||||
// Construct our diagnostic client.
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions();
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(errs(), &*DiagOpts);
|
|
||||||
DiagClient->setPrefix("clang -cc1as");
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
// Set an error handler, so that any LLVM backend diagnostics go through our
|
|
||||||
// error handler.
|
|
||||||
ScopedFatalErrorHandler FatalErrorHandler
|
|
||||||
(LLVMErrorHandler, static_cast<void*>(&Diags));
|
|
||||||
|
|
||||||
// Parse the arguments.
|
|
||||||
AssemblerInvocation Asm;
|
|
||||||
if (!AssemblerInvocation::CreateFromArgs(Asm, Argv, Diags))
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
if (Asm.ShowHelp) {
|
|
||||||
std::unique_ptr<OptTable> Opts(driver::createDriverOptTable());
|
|
||||||
Opts->PrintHelp(llvm::outs(), "clang -cc1as", "Clang Integrated Assembler",
|
|
||||||
/*Include=*/driver::options::CC1AsOption, /*Exclude=*/0,
|
|
||||||
/*ShowAllAliases=*/false);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -version.
|
|
||||||
//
|
|
||||||
// FIXME: Use a better -version message?
|
|
||||||
if (Asm.ShowVersion) {
|
|
||||||
llvm::cl::PrintVersionMessage();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Honor -mllvm.
|
|
||||||
//
|
|
||||||
// FIXME: Remove this, one day.
|
|
||||||
if (!Asm.LLVMArgs.empty()) {
|
|
||||||
unsigned NumArgs = Asm.LLVMArgs.size();
|
|
||||||
auto Args = llvm::make_unique<const char*[]>(NumArgs + 2);
|
|
||||||
Args[0] = "clang (LLVM option parsing)";
|
|
||||||
for (unsigned i = 0; i != NumArgs; ++i)
|
|
||||||
Args[i + 1] = Asm.LLVMArgs[i].c_str();
|
|
||||||
Args[NumArgs + 1] = nullptr;
|
|
||||||
llvm::cl::ParseCommandLineOptions(NumArgs + 1, Args.get());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Execute the invocation, unless there were parsing errors.
|
|
||||||
bool Failed = Diags.hasErrorOccurred() || ExecuteAssembler(Asm, Diags);
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now.
|
|
||||||
TimerGroup::printAll(errs());
|
|
||||||
|
|
||||||
return !!Failed;
|
|
||||||
}
|
|
@ -1,512 +0,0 @@
|
|||||||
//===-- driver.cpp - Clang GCC-Compatible Driver --------------------------===//
|
|
||||||
//
|
|
||||||
// The LLVM Compiler Infrastructure
|
|
||||||
//
|
|
||||||
// This file is distributed under the University of Illinois Open Source
|
|
||||||
// License. See LICENSE.TXT for details.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
//
|
|
||||||
// This is the entry point to the clang driver; it is a thin wrapper
|
|
||||||
// for functionality in the Driver clang library.
|
|
||||||
//
|
|
||||||
//===----------------------------------------------------------------------===//
|
|
||||||
|
|
||||||
#include "clang/Driver/Driver.h"
|
|
||||||
#include "clang/Basic/DiagnosticOptions.h"
|
|
||||||
#include "clang/Driver/Compilation.h"
|
|
||||||
#include "clang/Driver/DriverDiagnostic.h"
|
|
||||||
#include "clang/Driver/Options.h"
|
|
||||||
#include "clang/Driver/ToolChain.h"
|
|
||||||
#include "clang/Frontend/ChainedDiagnosticConsumer.h"
|
|
||||||
#include "clang/Frontend/CompilerInvocation.h"
|
|
||||||
#include "clang/Frontend/SerializedDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
||||||
#include "clang/Frontend/Utils.h"
|
|
||||||
#include "llvm/ADT/ArrayRef.h"
|
|
||||||
#include "llvm/ADT/SmallString.h"
|
|
||||||
#include "llvm/ADT/SmallVector.h"
|
|
||||||
#include "llvm/Option/ArgList.h"
|
|
||||||
#include "llvm/Option/OptTable.h"
|
|
||||||
#include "llvm/Option/Option.h"
|
|
||||||
#include "llvm/Support/CommandLine.h"
|
|
||||||
#include "llvm/Support/ErrorHandling.h"
|
|
||||||
#include "llvm/Support/FileSystem.h"
|
|
||||||
#include "llvm/Support/Host.h"
|
|
||||||
#include "llvm/Support/InitLLVM.h"
|
|
||||||
#include "llvm/Support/Path.h"
|
|
||||||
#include "llvm/Support/Process.h"
|
|
||||||
#include "llvm/Support/Program.h"
|
|
||||||
#include "llvm/Support/Regex.h"
|
|
||||||
#include "llvm/Support/Signals.h"
|
|
||||||
#include "llvm/Support/StringSaver.h"
|
|
||||||
#include "llvm/Support/TargetSelect.h"
|
|
||||||
#include "llvm/Support/Timer.h"
|
|
||||||
#include "llvm/Support/raw_ostream.h"
|
|
||||||
#include <memory>
|
|
||||||
#include <set>
|
|
||||||
#include <system_error>
|
|
||||||
using namespace clang;
|
|
||||||
using namespace clang::driver;
|
|
||||||
using namespace llvm::opt;
|
|
||||||
|
|
||||||
std::string GetExecutablePath(const char *Argv0, bool CanonicalPrefixes) {
|
|
||||||
if (!CanonicalPrefixes) {
|
|
||||||
SmallString<128> ExecutablePath(Argv0);
|
|
||||||
// Do a PATH lookup if Argv0 isn't a valid path.
|
|
||||||
if (!llvm::sys::fs::exists(ExecutablePath))
|
|
||||||
if (llvm::ErrorOr<std::string> P =
|
|
||||||
llvm::sys::findProgramByName(ExecutablePath))
|
|
||||||
ExecutablePath = *P;
|
|
||||||
return ExecutablePath.str();
|
|
||||||
}
|
|
||||||
|
|
||||||
// This just needs to be some symbol in the binary; C++ doesn't
|
|
||||||
// allow taking the address of ::main however.
|
|
||||||
void *P = (void*) (intptr_t) GetExecutablePath;
|
|
||||||
return llvm::sys::fs::getMainExecutable(Argv0, P);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const char *GetStableCStr(std::set<std::string> &SavedStrings,
|
|
||||||
StringRef S) {
|
|
||||||
return SavedStrings.insert(S).first->c_str();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a list of edits to the input argument lists.
|
|
||||||
///
|
|
||||||
/// The input string is a space separate list of edits to perform,
|
|
||||||
/// they are applied in order to the input argument lists. Edits
|
|
||||||
/// should be one of the following forms:
|
|
||||||
///
|
|
||||||
/// '#': Silence information about the changes to the command line arguments.
|
|
||||||
///
|
|
||||||
/// '^': Add FOO as a new argument at the beginning of the command line.
|
|
||||||
///
|
|
||||||
/// '+': Add FOO as a new argument at the end of the command line.
|
|
||||||
///
|
|
||||||
/// 's/XXX/YYY/': Substitute the regular expression XXX with YYY in the command
|
|
||||||
/// line.
|
|
||||||
///
|
|
||||||
/// 'xOPTION': Removes all instances of the literal argument OPTION.
|
|
||||||
///
|
|
||||||
/// 'XOPTION': Removes all instances of the literal argument OPTION,
|
|
||||||
/// and the following argument.
|
|
||||||
///
|
|
||||||
/// 'Ox': Removes all flags matching 'O' or 'O[sz0-9]' and adds 'Ox'
|
|
||||||
/// at the end of the command line.
|
|
||||||
///
|
|
||||||
/// \param OS - The stream to write edit information to.
|
|
||||||
/// \param Args - The vector of command line arguments.
|
|
||||||
/// \param Edit - The override command to perform.
|
|
||||||
/// \param SavedStrings - Set to use for storing string representations.
|
|
||||||
static void ApplyOneQAOverride(raw_ostream &OS,
|
|
||||||
SmallVectorImpl<const char*> &Args,
|
|
||||||
StringRef Edit,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
if (Edit[0] == '^') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at beginning\n";
|
|
||||||
Args.insert(Args.begin() + 1, Str);
|
|
||||||
} else if (Edit[0] == '+') {
|
|
||||||
const char *Str =
|
|
||||||
GetStableCStr(SavedStrings, Edit.substr(1));
|
|
||||||
OS << "### Adding argument " << Str << " at end\n";
|
|
||||||
Args.push_back(Str);
|
|
||||||
} else if (Edit[0] == 's' && Edit[1] == '/' && Edit.endswith("/") &&
|
|
||||||
Edit.slice(2, Edit.size()-1).find('/') != StringRef::npos) {
|
|
||||||
StringRef MatchPattern = Edit.substr(2).split('/').first;
|
|
||||||
StringRef ReplPattern = Edit.substr(2).split('/').second;
|
|
||||||
ReplPattern = ReplPattern.slice(0, ReplPattern.size()-1);
|
|
||||||
|
|
||||||
for (unsigned i = 1, e = Args.size(); i != e; ++i) {
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (Args[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
std::string Repl = llvm::Regex(MatchPattern).sub(ReplPattern, Args[i]);
|
|
||||||
|
|
||||||
if (Repl != Args[i]) {
|
|
||||||
OS << "### Replacing '" << Args[i] << "' with '" << Repl << "'\n";
|
|
||||||
Args[i] = GetStableCStr(SavedStrings, Repl);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'x' || Edit[0] == 'X') {
|
|
||||||
auto Option = Edit.substr(1);
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
if (Option == Args[i]) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
if (Edit[0] == 'X') {
|
|
||||||
if (i < Args.size()) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
OS << "### Invalid X edit, end of command line!\n";
|
|
||||||
}
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
} else if (Edit[0] == 'O') {
|
|
||||||
for (unsigned i = 1; i < Args.size();) {
|
|
||||||
const char *A = Args[i];
|
|
||||||
// Ignore end-of-line response file markers
|
|
||||||
if (A == nullptr)
|
|
||||||
continue;
|
|
||||||
if (A[0] == '-' && A[1] == 'O' &&
|
|
||||||
(A[2] == '\0' ||
|
|
||||||
(A[3] == '\0' && (A[2] == 's' || A[2] == 'z' ||
|
|
||||||
('0' <= A[2] && A[2] <= '9'))))) {
|
|
||||||
OS << "### Deleting argument " << Args[i] << '\n';
|
|
||||||
Args.erase(Args.begin() + i);
|
|
||||||
} else
|
|
||||||
++i;
|
|
||||||
}
|
|
||||||
OS << "### Adding argument " << Edit << " at end\n";
|
|
||||||
Args.push_back(GetStableCStr(SavedStrings, '-' + Edit.str()));
|
|
||||||
} else {
|
|
||||||
OS << "### Unrecognized edit: " << Edit << "\n";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// ApplyQAOverride - Apply a comma separate list of edits to the
|
|
||||||
/// input argument lists. See ApplyOneQAOverride.
|
|
||||||
static void ApplyQAOverride(SmallVectorImpl<const char*> &Args,
|
|
||||||
const char *OverrideStr,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
raw_ostream *OS = &llvm::errs();
|
|
||||||
|
|
||||||
if (OverrideStr[0] == '#') {
|
|
||||||
++OverrideStr;
|
|
||||||
OS = &llvm::nulls();
|
|
||||||
}
|
|
||||||
|
|
||||||
*OS << "### CCC_OVERRIDE_OPTIONS: " << OverrideStr << "\n";
|
|
||||||
|
|
||||||
// This does not need to be efficient.
|
|
||||||
|
|
||||||
const char *S = OverrideStr;
|
|
||||||
while (*S) {
|
|
||||||
const char *End = ::strchr(S, ' ');
|
|
||||||
if (!End)
|
|
||||||
End = S + strlen(S);
|
|
||||||
if (End != S)
|
|
||||||
ApplyOneQAOverride(*OS, Args, std::string(S, End), SavedStrings);
|
|
||||||
S = End;
|
|
||||||
if (*S != '\0')
|
|
||||||
++S;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
extern int cc1_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
extern int cc1as_main(ArrayRef<const char *> Argv, const char *Argv0,
|
|
||||||
void *MainAddr);
|
|
||||||
extern int cc1gen_reproducer_main(ArrayRef<const char *> Argv,
|
|
||||||
const char *Argv0, void *MainAddr);
|
|
||||||
|
|
||||||
static void insertTargetAndModeArgs(const ParsedClangName &NameParts,
|
|
||||||
SmallVectorImpl<const char *> &ArgVector,
|
|
||||||
std::set<std::string> &SavedStrings) {
|
|
||||||
// Put target and mode arguments at the start of argument list so that
|
|
||||||
// arguments specified in command line could override them. Avoid putting
|
|
||||||
// them at index 0, as an option like '-cc1' must remain the first.
|
|
||||||
int InsertionPoint = 0;
|
|
||||||
if (ArgVector.size() > 0)
|
|
||||||
++InsertionPoint;
|
|
||||||
|
|
||||||
if (NameParts.DriverMode) {
|
|
||||||
// Add the mode flag to the arguments.
|
|
||||||
ArgVector.insert(ArgVector.begin() + InsertionPoint,
|
|
||||||
GetStableCStr(SavedStrings, NameParts.DriverMode));
|
|
||||||
}
|
|
||||||
|
|
||||||
if (NameParts.TargetIsValid) {
|
|
||||||
const char *arr[] = {"-target", GetStableCStr(SavedStrings,
|
|
||||||
NameParts.TargetPrefix)};
|
|
||||||
ArgVector.insert(ArgVector.begin() + InsertionPoint,
|
|
||||||
std::begin(arr), std::end(arr));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void getCLEnvVarOptions(std::string &EnvValue, llvm::StringSaver &Saver,
|
|
||||||
SmallVectorImpl<const char *> &Opts) {
|
|
||||||
llvm::cl::TokenizeWindowsCommandLine(EnvValue, Saver, Opts);
|
|
||||||
// The first instance of '#' should be replaced with '=' in each option.
|
|
||||||
for (const char *Opt : Opts)
|
|
||||||
if (char *NumberSignPtr = const_cast<char *>(::strchr(Opt, '#')))
|
|
||||||
*NumberSignPtr = '=';
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetBackdoorDriverOutputsFromEnvVars(Driver &TheDriver) {
|
|
||||||
// Handle CC_PRINT_OPTIONS and CC_PRINT_OPTIONS_FILE.
|
|
||||||
TheDriver.CCPrintOptions = !!::getenv("CC_PRINT_OPTIONS");
|
|
||||||
if (TheDriver.CCPrintOptions)
|
|
||||||
TheDriver.CCPrintOptionsFilename = ::getenv("CC_PRINT_OPTIONS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_PRINT_HEADERS and CC_PRINT_HEADERS_FILE.
|
|
||||||
TheDriver.CCPrintHeaders = !!::getenv("CC_PRINT_HEADERS");
|
|
||||||
if (TheDriver.CCPrintHeaders)
|
|
||||||
TheDriver.CCPrintHeadersFilename = ::getenv("CC_PRINT_HEADERS_FILE");
|
|
||||||
|
|
||||||
// Handle CC_LOG_DIAGNOSTICS and CC_LOG_DIAGNOSTICS_FILE.
|
|
||||||
TheDriver.CCLogDiagnostics = !!::getenv("CC_LOG_DIAGNOSTICS");
|
|
||||||
if (TheDriver.CCLogDiagnostics)
|
|
||||||
TheDriver.CCLogDiagnosticsFilename = ::getenv("CC_LOG_DIAGNOSTICS_FILE");
|
|
||||||
}
|
|
||||||
|
|
||||||
static void FixupDiagPrefixExeName(TextDiagnosticPrinter *DiagClient,
|
|
||||||
const std::string &Path) {
|
|
||||||
// If the clang binary happens to be named cl.exe for compatibility reasons,
|
|
||||||
// use clang-cl.exe as the prefix to avoid confusion between clang and MSVC.
|
|
||||||
StringRef ExeBasename(llvm::sys::path::filename(Path));
|
|
||||||
if (ExeBasename.equals_lower("cl.exe"))
|
|
||||||
ExeBasename = "clang-cl.exe";
|
|
||||||
DiagClient->setPrefix(ExeBasename);
|
|
||||||
}
|
|
||||||
|
|
||||||
// This lets us create the DiagnosticsEngine with a properly-filled-out
|
|
||||||
// DiagnosticOptions instance.
|
|
||||||
static DiagnosticOptions *
|
|
||||||
CreateAndPopulateDiagOpts(ArrayRef<const char *> argv) {
|
|
||||||
auto *DiagOpts = new DiagnosticOptions;
|
|
||||||
std::unique_ptr<OptTable> Opts(createDriverOptTable());
|
|
||||||
unsigned MissingArgIndex, MissingArgCount;
|
|
||||||
InputArgList Args =
|
|
||||||
Opts->ParseArgs(argv.slice(1), MissingArgIndex, MissingArgCount);
|
|
||||||
// We ignore MissingArgCount and the return value of ParseDiagnosticArgs.
|
|
||||||
// Any errors that would be diagnosed here will also be diagnosed later,
|
|
||||||
// when the DiagnosticsEngine actually exists.
|
|
||||||
(void)ParseDiagnosticArgs(*DiagOpts, Args);
|
|
||||||
return DiagOpts;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void SetInstallDir(SmallVectorImpl<const char *> &argv,
|
|
||||||
Driver &TheDriver, bool CanonicalPrefixes) {
|
|
||||||
// Attempt to find the original path used to invoke the driver, to determine
|
|
||||||
// the installed path. We do this manually, because we want to support that
|
|
||||||
// path being a symlink.
|
|
||||||
SmallString<128> InstalledPath(argv[0]);
|
|
||||||
|
|
||||||
// Do a PATH lookup, if there are no directory components.
|
|
||||||
if (llvm::sys::path::filename(InstalledPath) == InstalledPath)
|
|
||||||
if (llvm::ErrorOr<std::string> Tmp = llvm::sys::findProgramByName(
|
|
||||||
llvm::sys::path::filename(InstalledPath.str())))
|
|
||||||
InstalledPath = *Tmp;
|
|
||||||
|
|
||||||
// FIXME: We don't actually canonicalize this, we just make it absolute.
|
|
||||||
if (CanonicalPrefixes)
|
|
||||||
llvm::sys::fs::make_absolute(InstalledPath);
|
|
||||||
|
|
||||||
StringRef InstalledPathParent(llvm::sys::path::parent_path(InstalledPath));
|
|
||||||
if (llvm::sys::fs::exists(InstalledPathParent))
|
|
||||||
TheDriver.setInstalledDir(InstalledPathParent);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int ExecuteCC1Tool(ArrayRef<const char *> argv, StringRef Tool) {
|
|
||||||
void *GetExecutablePathVP = (void *)(intptr_t) GetExecutablePath;
|
|
||||||
if (Tool == "")
|
|
||||||
return cc1_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
if (Tool == "as")
|
|
||||||
return cc1as_main(argv.slice(2), argv[0], GetExecutablePathVP);
|
|
||||||
|
|
||||||
// Reject unknown tools.
|
|
||||||
llvm::errs() << "error: unknown integrated tool '" << Tool << "'. "
|
|
||||||
<< "Valid tools include '-cc1' and '-cc1as'.\n";
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
int mainEntryClickHouseClang(int argc_, char **argv_) {
|
|
||||||
llvm::InitLLVM X(argc_, argv_);
|
|
||||||
SmallVector<const char *, 256> argv(argv_, argv_ + argc_);
|
|
||||||
|
|
||||||
if (llvm::sys::Process::FixupStandardFileDescriptors())
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
llvm::InitializeAllTargets();
|
|
||||||
auto TargetAndMode = ToolChain::getTargetAndModeFromProgramName(argv[0]);
|
|
||||||
|
|
||||||
llvm::BumpPtrAllocator A;
|
|
||||||
llvm::StringSaver Saver(A);
|
|
||||||
|
|
||||||
// Parse response files using the GNU syntax, unless we're in CL mode. There
|
|
||||||
// are two ways to put clang in CL compatibility mode: argv[0] is either
|
|
||||||
// clang-cl or cl, or --driver-mode=cl is on the command line. The normal
|
|
||||||
// command line parsing can't happen until after response file parsing, so we
|
|
||||||
// have to manually search for a --driver-mode=cl argument the hard way.
|
|
||||||
// Finally, our -cc1 tools don't care which tokenization mode we use because
|
|
||||||
// response files written by clang will tokenize the same way in either mode.
|
|
||||||
bool ClangCLMode = false;
|
|
||||||
if (StringRef(TargetAndMode.DriverMode).equals("--driver-mode=cl") ||
|
|
||||||
std::find_if(argv.begin(), argv.end(), [](const char *F) {
|
|
||||||
return F && strcmp(F, "--driver-mode=cl") == 0;
|
|
||||||
}) != argv.end()) {
|
|
||||||
ClangCLMode = true;
|
|
||||||
}
|
|
||||||
enum { Default, POSIX, Windows } RSPQuoting = Default;
|
|
||||||
for (const char *F : argv) {
|
|
||||||
if (strcmp(F, "--rsp-quoting=posix") == 0)
|
|
||||||
RSPQuoting = POSIX;
|
|
||||||
else if (strcmp(F, "--rsp-quoting=windows") == 0)
|
|
||||||
RSPQuoting = Windows;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Determines whether we want nullptr markers in argv to indicate response
|
|
||||||
// files end-of-lines. We only use this for the /LINK driver argument with
|
|
||||||
// clang-cl.exe on Windows.
|
|
||||||
bool MarkEOLs = ClangCLMode;
|
|
||||||
|
|
||||||
llvm::cl::TokenizerCallback Tokenizer;
|
|
||||||
if (RSPQuoting == Windows || (RSPQuoting == Default && ClangCLMode))
|
|
||||||
Tokenizer = &llvm::cl::TokenizeWindowsCommandLine;
|
|
||||||
else
|
|
||||||
Tokenizer = &llvm::cl::TokenizeGNUCommandLine;
|
|
||||||
|
|
||||||
if (MarkEOLs && argv.size() > 1 && StringRef(argv[1]).startswith("-cc1"))
|
|
||||||
MarkEOLs = false;
|
|
||||||
llvm::cl::ExpandResponseFiles(Saver, Tokenizer, argv, MarkEOLs);
|
|
||||||
|
|
||||||
// Handle -cc1 integrated tools, even if -cc1 was expanded from a response
|
|
||||||
// file.
|
|
||||||
auto FirstArg = std::find_if(argv.begin() + 1, argv.end(),
|
|
||||||
[](const char *A) { return A != nullptr; });
|
|
||||||
if (FirstArg != argv.end() && StringRef(*FirstArg).startswith("-cc1")) {
|
|
||||||
// If -cc1 came from a response file, remove the EOL sentinels.
|
|
||||||
if (MarkEOLs) {
|
|
||||||
auto newEnd = std::remove(argv.begin(), argv.end(), nullptr);
|
|
||||||
argv.resize(newEnd - argv.begin());
|
|
||||||
}
|
|
||||||
return ExecuteCC1Tool(argv, argv[1] + 4);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool CanonicalPrefixes = true;
|
|
||||||
for (int i = 1, size = argv.size(); i < size; ++i) {
|
|
||||||
// Skip end-of-line response file markers
|
|
||||||
if (argv[i] == nullptr)
|
|
||||||
continue;
|
|
||||||
if (StringRef(argv[i]) == "-no-canonical-prefixes") {
|
|
||||||
CanonicalPrefixes = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle CL and _CL_ which permits additional command line options to be
|
|
||||||
// prepended or appended.
|
|
||||||
if (ClangCLMode) {
|
|
||||||
// Arguments in "CL" are prepended.
|
|
||||||
llvm::Optional<std::string> OptCL = llvm::sys::Process::GetEnv("CL");
|
|
||||||
if (OptCL.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> PrependedOpts;
|
|
||||||
getCLEnvVarOptions(OptCL.getValue(), Saver, PrependedOpts);
|
|
||||||
|
|
||||||
// Insert right after the program name to prepend to the argument list.
|
|
||||||
argv.insert(argv.begin() + 1, PrependedOpts.begin(), PrependedOpts.end());
|
|
||||||
}
|
|
||||||
// Arguments in "_CL_" are appended.
|
|
||||||
llvm::Optional<std::string> Opt_CL_ = llvm::sys::Process::GetEnv("_CL_");
|
|
||||||
if (Opt_CL_.hasValue()) {
|
|
||||||
SmallVector<const char *, 8> AppendedOpts;
|
|
||||||
getCLEnvVarOptions(Opt_CL_.getValue(), Saver, AppendedOpts);
|
|
||||||
|
|
||||||
// Insert at the end of the argument list to append.
|
|
||||||
argv.append(AppendedOpts.begin(), AppendedOpts.end());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
std::set<std::string> SavedStrings;
|
|
||||||
// Handle CCC_OVERRIDE_OPTIONS, used for editing a command line behind the
|
|
||||||
// scenes.
|
|
||||||
if (const char *OverrideStr = ::getenv("CCC_OVERRIDE_OPTIONS")) {
|
|
||||||
// FIXME: Driver shouldn't take extra initial argument.
|
|
||||||
ApplyQAOverride(argv, OverrideStr, SavedStrings);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string Path = GetExecutablePath(argv[0], CanonicalPrefixes);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts =
|
|
||||||
CreateAndPopulateDiagOpts(argv);
|
|
||||||
|
|
||||||
TextDiagnosticPrinter *DiagClient
|
|
||||||
= new TextDiagnosticPrinter(llvm::errs(), &*DiagOpts);
|
|
||||||
FixupDiagPrefixExeName(DiagClient, Path);
|
|
||||||
|
|
||||||
IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs());
|
|
||||||
|
|
||||||
DiagnosticsEngine Diags(DiagID, &*DiagOpts, DiagClient);
|
|
||||||
|
|
||||||
if (!DiagOpts->DiagnosticSerializationFile.empty()) {
|
|
||||||
auto SerializedConsumer =
|
|
||||||
clang::serialized_diags::create(DiagOpts->DiagnosticSerializationFile,
|
|
||||||
&*DiagOpts, /*MergeChildRecords=*/true);
|
|
||||||
Diags.setClient(new ChainedDiagnosticConsumer(
|
|
||||||
Diags.takeClient(), std::move(SerializedConsumer)));
|
|
||||||
}
|
|
||||||
|
|
||||||
ProcessWarningOptions(Diags, *DiagOpts, /*ReportDiags=*/false);
|
|
||||||
|
|
||||||
Driver TheDriver(Path, llvm::sys::getDefaultTargetTriple(), Diags);
|
|
||||||
SetInstallDir(argv, TheDriver, CanonicalPrefixes);
|
|
||||||
TheDriver.setTargetAndMode(TargetAndMode);
|
|
||||||
|
|
||||||
insertTargetAndModeArgs(TargetAndMode, argv, SavedStrings);
|
|
||||||
|
|
||||||
SetBackdoorDriverOutputsFromEnvVars(TheDriver);
|
|
||||||
|
|
||||||
std::unique_ptr<Compilation> C(TheDriver.BuildCompilation(argv));
|
|
||||||
int Res = 1;
|
|
||||||
if (C && !C->containsError()) {
|
|
||||||
SmallVector<std::pair<int, const Command *>, 4> FailingCommands;
|
|
||||||
Res = TheDriver.ExecuteCompilation(*C, FailingCommands);
|
|
||||||
|
|
||||||
// Force a crash to test the diagnostics.
|
|
||||||
if (TheDriver.GenReproducer) {
|
|
||||||
Diags.Report(diag::err_drv_force_crash)
|
|
||||||
<< !::getenv("FORCE_CLANG_DIAGNOSTICS_CRASH");
|
|
||||||
|
|
||||||
// Pretend that every command failed.
|
|
||||||
FailingCommands.clear();
|
|
||||||
for (const auto &J : C->getJobs())
|
|
||||||
if (const Command *C = dyn_cast<Command>(&J))
|
|
||||||
FailingCommands.push_back(std::make_pair(-1, C));
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const auto &P : FailingCommands) {
|
|
||||||
int CommandRes = P.first;
|
|
||||||
const Command *FailingCommand = P.second;
|
|
||||||
if (!Res)
|
|
||||||
Res = CommandRes;
|
|
||||||
|
|
||||||
// If result status is < 0, then the driver command signalled an error.
|
|
||||||
// If result status is 70, then the driver command reported a fatal error.
|
|
||||||
// On Windows, abort will return an exit code of 3. In these cases,
|
|
||||||
// generate additional diagnostic information if possible.
|
|
||||||
bool DiagnoseCrash = CommandRes < 0 || CommandRes == 70;
|
|
||||||
#ifdef _WIN32
|
|
||||||
DiagnoseCrash |= CommandRes == 3;
|
|
||||||
#endif
|
|
||||||
if (DiagnoseCrash) {
|
|
||||||
TheDriver.generateCompilationDiagnostics(*C, *FailingCommand);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Diags.getClient()->finish();
|
|
||||||
|
|
||||||
// If any timers were active but haven't been destroyed yet, print their
|
|
||||||
// results now. This happens in -disable-free mode.
|
|
||||||
llvm::TimerGroup::printAll(llvm::errs());
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
// Exit status should not be negative on Win32, unless abnormal termination.
|
|
||||||
// Once abnormal termiation was caught, negative status should not be
|
|
||||||
// propagated.
|
|
||||||
if (Res < 0)
|
|
||||||
Res = 1;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// If we have multiple failing commands, we return the result of the first
|
|
||||||
// failing command.
|
|
||||||
return Res;
|
|
||||||
}
|
|
@ -1,10 +0,0 @@
|
|||||||
#include "lld/Common/Driver.h"
|
|
||||||
#include "llvm/Support/InitLLVM.h"
|
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
int mainEntryClickHouseLLD(int argc, char ** argv)
|
|
||||||
{
|
|
||||||
llvm::InitLLVM X(argc, argv);
|
|
||||||
std::vector<const char *> args(argv, argv + argc);
|
|
||||||
return !lld::elf::link(args, false);
|
|
||||||
}
|
|
@ -1 +0,0 @@
|
|||||||
Compiler-7.0.0
|
|
@ -1 +0,0 @@
|
|||||||
Compiler-7.0.0
|
|
@ -1,2 +0,0 @@
|
|||||||
int mainEntryClickHouseClang(int argc, char ** argv);
|
|
||||||
int main(int argc_, char ** argv_) { return mainEntryClickHouseClang(argc_, argv_); }
|
|
@ -1,2 +0,0 @@
|
|||||||
int mainEntryClickHouseLLD(int argc, char ** argv);
|
|
||||||
int main(int argc_, char ** argv_) { return mainEntryClickHouseLLD(argc_, argv_); }
|
|
@ -1,100 +0,0 @@
|
|||||||
#!/usr/bin/env bash
|
|
||||||
|
|
||||||
set -e
|
|
||||||
#set -x
|
|
||||||
#echo "Args: $*"; env | sort
|
|
||||||
|
|
||||||
# Этот скрипт собирает все заголовочные файлы, нужные для компиляции некоторого translation unit-а
|
|
||||||
# и копирует их с сохранением путей в директорию DST.
|
|
||||||
# Это затем может быть использовано, чтобы скомпилировать translation unit на другом сервере,
|
|
||||||
# используя ровно такой же набор заголовочных файлов.
|
|
||||||
#
|
|
||||||
# Требуется clang, желательно наиболее свежий (trunk).
|
|
||||||
#
|
|
||||||
# Используется при сборке пакетов.
|
|
||||||
# Заголовочные файлы записываются в пакет clickhouse-common, в директорию /usr/share/clickhouse/headers.
|
|
||||||
#
|
|
||||||
# Если вы хотите установить их самостоятельно, без сборки пакета,
|
|
||||||
# чтобы clickhouse-server видел их там, где ожидается, выполните:
|
|
||||||
#
|
|
||||||
# sudo ./copy_headers.sh . /usr/share/clickhouse/headers/
|
|
||||||
|
|
||||||
SOURCE_PATH=${1:-../../..}
|
|
||||||
DST=${2:-$SOURCE_PATH/../headers}
|
|
||||||
BUILD_PATH=${BUILD_PATH=${3:-$SOURCE_PATH/build}}
|
|
||||||
|
|
||||||
PATH="/usr/local/bin:/usr/local/sbin:/usr/bin:$PATH"
|
|
||||||
|
|
||||||
if [[ -z $CLANG ]]; then
|
|
||||||
CLANG="clang"
|
|
||||||
fi
|
|
||||||
|
|
||||||
START_HEADERS=$(echo \
|
|
||||||
$BUILD_PATH/dbms/src/Common/config_version.h \
|
|
||||||
$SOURCE_PATH/dbms/src/Interpreters/SpecializedAggregator.h \
|
|
||||||
$SOURCE_PATH/dbms/src/AggregateFunctions/AggregateFunction*.h)
|
|
||||||
|
|
||||||
for header in $START_HEADERS; do
|
|
||||||
START_HEADERS_INCLUDE+="-include $header "
|
|
||||||
done
|
|
||||||
|
|
||||||
|
|
||||||
GCC_ROOT=`$CLANG -v 2>&1 | grep "Selected GCC installation"| sed -n -e 's/^.*: //p'`
|
|
||||||
|
|
||||||
# TODO: Does not work on macos?
|
|
||||||
GCC_ROOT=${GCC_ROOT:=/usr/lib/clang/${CMAKE_CXX_COMPILER_VERSION}}
|
|
||||||
|
|
||||||
# Опция -mcx16 для того, чтобы выбиралось больше заголовочных файлов (с запасом).
|
|
||||||
# The latter options are the same that are added while building packages.
|
|
||||||
for src_file in $(echo | $CLANG -M -xc++ -std=c++1z -Wall -Werror -msse2 -msse4 -mcx16 -mpopcnt -O3 -g -fPIC -fstack-protector -D_FORTIFY_SOURCE=2 \
|
|
||||||
-I $GCC_ROOT/include \
|
|
||||||
-I $GCC_ROOT/include-fixed \
|
|
||||||
$(cat "$BUILD_PATH/include_directories.txt") \
|
|
||||||
$START_HEADERS_INCLUDE \
|
|
||||||
- |
|
|
||||||
tr -d '\\' |
|
|
||||||
sed -E -e 's/^-\.o://');
|
|
||||||
do
|
|
||||||
dst_file=$src_file;
|
|
||||||
[ -n $BUILD_PATH ] && dst_file=$(echo $dst_file | sed -E -e "s!^$BUILD_PATH!!")
|
|
||||||
[ -n $DESTDIR ] && dst_file=$(echo $dst_file | sed -E -e "s!^$DESTDIR!!")
|
|
||||||
dst_file=$(echo $dst_file | sed -E -e 's/build\///') # for simplicity reasons, will put generated headers near the rest.
|
|
||||||
mkdir -p "$DST/$(echo $dst_file | sed -E -e 's/\/[^/]*$/\//')";
|
|
||||||
cp "$src_file" "$DST/$dst_file";
|
|
||||||
done
|
|
||||||
|
|
||||||
|
|
||||||
# Копируем больше заголовочных файлов с интринсиками, так как на серверах, куда будут устанавливаться
|
|
||||||
# заголовочные файлы, будет использоваться опция -march=native.
|
|
||||||
|
|
||||||
for src_file in $(ls -1 $($CLANG -v -xc++ - <<<'' 2>&1 | grep '^ /' | grep 'include' | grep -E '/lib/clang/|/include/clang/')/*.h | grep -vE 'arm|altivec|Intrin');
|
|
||||||
do
|
|
||||||
dst_file=$src_file;
|
|
||||||
[ -n $BUILD_PATH ] && dst_file=$(echo $dst_file | sed -E -e "s!^$BUILD_PATH!!")
|
|
||||||
[ -n $DESTDIR ] && dst_file=$(echo $dst_file | sed -E -e "s!^$DESTDIR!!")
|
|
||||||
mkdir -p "$DST/$(echo $dst_file | sed -E -e 's/\/[^/]*$/\//')";
|
|
||||||
cp "$src_file" "$DST/$dst_file";
|
|
||||||
done
|
|
||||||
|
|
||||||
if [ -d "$SOURCE_PATH/contrib/boost/libs/smart_ptr/include/boost/smart_ptr/detail" ]; then
|
|
||||||
# Even more platform-specific headers
|
|
||||||
for src_file in $(ls -1 $SOURCE_PATH/contrib/boost/libs/smart_ptr/include/boost/smart_ptr/detail/*);
|
|
||||||
do
|
|
||||||
dst_file=$src_file;
|
|
||||||
[ -n $BUILD_PATH ] && dst_file=$(echo $dst_file | sed -E -e "s!^$BUILD_PATH!!")
|
|
||||||
[ -n $DESTDIR ] && dst_file=$(echo $dst_file | sed -E -e "s!^$DESTDIR!!")
|
|
||||||
mkdir -p "$DST/$(echo $dst_file | sed -E -e 's/\/[^/]*$/\//')";
|
|
||||||
cp "$src_file" "$DST/$dst_file";
|
|
||||||
done
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ -d "$SOURCE_PATH/contrib/boost/boost/smart_ptr/detail" ]; then
|
|
||||||
for src_file in $(ls -1 $SOURCE_PATH/contrib/boost/boost/smart_ptr/detail/*);
|
|
||||||
do
|
|
||||||
dst_file=$src_file;
|
|
||||||
[ -n $BUILD_PATH ] && dst_file=$(echo $dst_file | sed -E -e "s!^$BUILD_PATH!!")
|
|
||||||
[ -n $DESTDIR ] && dst_file=$(echo $dst_file | sed -E -e "s!^$DESTDIR!!")
|
|
||||||
mkdir -p "$DST/$(echo $dst_file | sed -E -e 's/\/[^/]*$/\//')";
|
|
||||||
cp "$src_file" "$DST/$dst_file";
|
|
||||||
done
|
|
||||||
fi
|
|
@ -56,11 +56,6 @@ int mainEntryClickHouseObfuscator(int argc, char ** argv);
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
#if USE_EMBEDDED_COMPILER
|
|
||||||
int mainEntryClickHouseClang(int argc, char ** argv);
|
|
||||||
int mainEntryClickHouseLLD(int argc, char ** argv);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
namespace
|
namespace
|
||||||
{
|
{
|
||||||
|
|
||||||
@ -100,12 +95,6 @@ std::pair<const char *, MainFunc> clickhouse_applications[] =
|
|||||||
#if ENABLE_CLICKHOUSE_OBFUSCATOR || !defined(ENABLE_CLICKHOUSE_OBFUSCATOR)
|
#if ENABLE_CLICKHOUSE_OBFUSCATOR || !defined(ENABLE_CLICKHOUSE_OBFUSCATOR)
|
||||||
{"obfuscator", mainEntryClickHouseObfuscator},
|
{"obfuscator", mainEntryClickHouseObfuscator},
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if USE_EMBEDDED_COMPILER
|
|
||||||
{"clang", mainEntryClickHouseClang},
|
|
||||||
{"clang++", mainEntryClickHouseClang},
|
|
||||||
{"lld", mainEntryClickHouseLLD},
|
|
||||||
#endif
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@ -152,11 +141,6 @@ int main(int argc_, char ** argv_)
|
|||||||
/// will work only after additional call of this function.
|
/// will work only after additional call of this function.
|
||||||
updatePHDRCache();
|
updatePHDRCache();
|
||||||
|
|
||||||
#if USE_EMBEDDED_COMPILER
|
|
||||||
if (argc_ >= 2 && 0 == strcmp(argv_[1], "-cc1"))
|
|
||||||
return mainEntryClickHouseClang(argc_, argv_);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if USE_TCMALLOC
|
#if USE_TCMALLOC
|
||||||
/** Without this option, tcmalloc returns memory to OS too frequently for medium-sized memory allocations
|
/** Without this option, tcmalloc returns memory to OS too frequently for medium-sized memory allocations
|
||||||
* (like IO buffers, column vectors, hash tables, etc.),
|
* (like IO buffers, column vectors, hash tables, etc.),
|
||||||
|
@ -22,6 +22,7 @@
|
|||||||
#include <Common/StringUtils/StringUtils.h>
|
#include <Common/StringUtils/StringUtils.h>
|
||||||
#include <Common/HashTable/HashMap.h>
|
#include <Common/HashTable/HashMap.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <Core/Block.h>
|
#include <Core/Block.h>
|
||||||
#include <common/StringRef.h>
|
#include <common/StringRef.h>
|
||||||
#include <common/DateLUT.h>
|
#include <common/DateLUT.h>
|
||||||
@ -263,11 +264,11 @@ public:
|
|||||||
|
|
||||||
ColumnPtr generate(const IColumn & column) override
|
ColumnPtr generate(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const auto & src_data = static_cast<const ColumnVector<Float> &>(column).getData();
|
const auto & src_data = assert_cast<const ColumnVector<Float> &>(column).getData();
|
||||||
size_t size = src_data.size();
|
size_t size = src_data.size();
|
||||||
|
|
||||||
auto res_column = ColumnVector<Float>::create(size);
|
auto res_column = ColumnVector<Float>::create(size);
|
||||||
auto & res_data = static_cast<ColumnVector<Float> &>(*res_column).getData();
|
auto & res_data = assert_cast<ColumnVector<Float> &>(*res_column).getData();
|
||||||
|
|
||||||
for (size_t i = 0; i < size; ++i)
|
for (size_t i = 0; i < size; ++i)
|
||||||
{
|
{
|
||||||
@ -355,7 +356,7 @@ public:
|
|||||||
|
|
||||||
ColumnPtr generate(const IColumn & column) override
|
ColumnPtr generate(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const ColumnFixedString & column_fixed_string = static_cast<const ColumnFixedString &>(column);
|
const ColumnFixedString & column_fixed_string = assert_cast<const ColumnFixedString &>(column);
|
||||||
const size_t string_size = column_fixed_string.getN();
|
const size_t string_size = column_fixed_string.getN();
|
||||||
|
|
||||||
const auto & src_data = column_fixed_string.getChars();
|
const auto & src_data = column_fixed_string.getChars();
|
||||||
@ -392,11 +393,11 @@ public:
|
|||||||
|
|
||||||
ColumnPtr generate(const IColumn & column) override
|
ColumnPtr generate(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const auto & src_data = static_cast<const ColumnVector<UInt32> &>(column).getData();
|
const auto & src_data = assert_cast<const ColumnVector<UInt32> &>(column).getData();
|
||||||
size_t size = src_data.size();
|
size_t size = src_data.size();
|
||||||
|
|
||||||
auto res_column = ColumnVector<UInt32>::create(size);
|
auto res_column = ColumnVector<UInt32>::create(size);
|
||||||
auto & res_data = static_cast<ColumnVector<UInt32> &>(*res_column).getData();
|
auto & res_data = assert_cast<ColumnVector<UInt32> &>(*res_column).getData();
|
||||||
|
|
||||||
for (size_t i = 0; i < size; ++i)
|
for (size_t i = 0; i < size; ++i)
|
||||||
{
|
{
|
||||||
@ -749,7 +750,7 @@ public:
|
|||||||
|
|
||||||
void train(const IColumn & column) override
|
void train(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const ColumnString & column_string = static_cast<const ColumnString &>(column);
|
const ColumnString & column_string = assert_cast<const ColumnString &>(column);
|
||||||
size_t size = column_string.size();
|
size_t size = column_string.size();
|
||||||
|
|
||||||
for (size_t i = 0; i < size; ++i)
|
for (size_t i = 0; i < size; ++i)
|
||||||
@ -766,7 +767,7 @@ public:
|
|||||||
|
|
||||||
ColumnPtr generate(const IColumn & column) override
|
ColumnPtr generate(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const ColumnString & column_string = static_cast<const ColumnString &>(column);
|
const ColumnString & column_string = assert_cast<const ColumnString &>(column);
|
||||||
size_t size = column_string.size();
|
size_t size = column_string.size();
|
||||||
|
|
||||||
auto res_column = ColumnString::create();
|
auto res_column = ColumnString::create();
|
||||||
@ -801,7 +802,7 @@ public:
|
|||||||
|
|
||||||
void train(const IColumn & column) override
|
void train(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const ColumnArray & column_array = static_cast<const ColumnArray &>(column);
|
const ColumnArray & column_array = assert_cast<const ColumnArray &>(column);
|
||||||
const IColumn & nested_column = column_array.getData();
|
const IColumn & nested_column = column_array.getData();
|
||||||
|
|
||||||
nested_model->train(nested_column);
|
nested_model->train(nested_column);
|
||||||
@ -814,7 +815,7 @@ public:
|
|||||||
|
|
||||||
ColumnPtr generate(const IColumn & column) override
|
ColumnPtr generate(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const ColumnArray & column_array = static_cast<const ColumnArray &>(column);
|
const ColumnArray & column_array = assert_cast<const ColumnArray &>(column);
|
||||||
const IColumn & nested_column = column_array.getData();
|
const IColumn & nested_column = column_array.getData();
|
||||||
|
|
||||||
ColumnPtr new_nested_column = nested_model->generate(nested_column);
|
ColumnPtr new_nested_column = nested_model->generate(nested_column);
|
||||||
@ -834,7 +835,7 @@ public:
|
|||||||
|
|
||||||
void train(const IColumn & column) override
|
void train(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const ColumnNullable & column_nullable = static_cast<const ColumnNullable &>(column);
|
const ColumnNullable & column_nullable = assert_cast<const ColumnNullable &>(column);
|
||||||
const IColumn & nested_column = column_nullable.getNestedColumn();
|
const IColumn & nested_column = column_nullable.getNestedColumn();
|
||||||
|
|
||||||
nested_model->train(nested_column);
|
nested_model->train(nested_column);
|
||||||
@ -847,7 +848,7 @@ public:
|
|||||||
|
|
||||||
ColumnPtr generate(const IColumn & column) override
|
ColumnPtr generate(const IColumn & column) override
|
||||||
{
|
{
|
||||||
const ColumnNullable & column_nullable = static_cast<const ColumnNullable &>(column);
|
const ColumnNullable & column_nullable = assert_cast<const ColumnNullable &>(column);
|
||||||
const IColumn & nested_column = column_nullable.getNestedColumn();
|
const IColumn & nested_column = column_nullable.getNestedColumn();
|
||||||
|
|
||||||
ColumnPtr new_nested_column = nested_model->generate(nested_column);
|
ColumnPtr new_nested_column = nested_model->generate(nested_column);
|
||||||
|
@ -3,6 +3,7 @@
|
|||||||
#include <Columns/ColumnNullable.h>
|
#include <Columns/ColumnNullable.h>
|
||||||
#include <Columns/ColumnString.h>
|
#include <Columns/ColumnString.h>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <IO/ReadHelpers.h>
|
#include <IO/ReadHelpers.h>
|
||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
#include <common/logger_useful.h>
|
#include <common/logger_useful.h>
|
||||||
@ -44,46 +45,46 @@ namespace
|
|||||||
switch (type)
|
switch (type)
|
||||||
{
|
{
|
||||||
case ValueType::vtUInt8:
|
case ValueType::vtUInt8:
|
||||||
static_cast<ColumnUInt8 &>(column).insertValue(value.convert<UInt64>());
|
assert_cast<ColumnUInt8 &>(column).insertValue(value.convert<UInt64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtUInt16:
|
case ValueType::vtUInt16:
|
||||||
static_cast<ColumnUInt16 &>(column).insertValue(value.convert<UInt64>());
|
assert_cast<ColumnUInt16 &>(column).insertValue(value.convert<UInt64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtUInt32:
|
case ValueType::vtUInt32:
|
||||||
static_cast<ColumnUInt32 &>(column).insertValue(value.convert<UInt64>());
|
assert_cast<ColumnUInt32 &>(column).insertValue(value.convert<UInt64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtUInt64:
|
case ValueType::vtUInt64:
|
||||||
static_cast<ColumnUInt64 &>(column).insertValue(value.convert<UInt64>());
|
assert_cast<ColumnUInt64 &>(column).insertValue(value.convert<UInt64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtInt8:
|
case ValueType::vtInt8:
|
||||||
static_cast<ColumnInt8 &>(column).insertValue(value.convert<Int64>());
|
assert_cast<ColumnInt8 &>(column).insertValue(value.convert<Int64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtInt16:
|
case ValueType::vtInt16:
|
||||||
static_cast<ColumnInt16 &>(column).insertValue(value.convert<Int64>());
|
assert_cast<ColumnInt16 &>(column).insertValue(value.convert<Int64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtInt32:
|
case ValueType::vtInt32:
|
||||||
static_cast<ColumnInt32 &>(column).insertValue(value.convert<Int64>());
|
assert_cast<ColumnInt32 &>(column).insertValue(value.convert<Int64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtInt64:
|
case ValueType::vtInt64:
|
||||||
static_cast<ColumnInt64 &>(column).insertValue(value.convert<Int64>());
|
assert_cast<ColumnInt64 &>(column).insertValue(value.convert<Int64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtFloat32:
|
case ValueType::vtFloat32:
|
||||||
static_cast<ColumnFloat32 &>(column).insertValue(value.convert<Float64>());
|
assert_cast<ColumnFloat32 &>(column).insertValue(value.convert<Float64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtFloat64:
|
case ValueType::vtFloat64:
|
||||||
static_cast<ColumnFloat64 &>(column).insertValue(value.convert<Float64>());
|
assert_cast<ColumnFloat64 &>(column).insertValue(value.convert<Float64>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtString:
|
case ValueType::vtString:
|
||||||
static_cast<ColumnString &>(column).insert(value.convert<String>());
|
assert_cast<ColumnString &>(column).insert(value.convert<String>());
|
||||||
break;
|
break;
|
||||||
case ValueType::vtDate:
|
case ValueType::vtDate:
|
||||||
static_cast<ColumnUInt16 &>(column).insertValue(UInt16{LocalDate{value.convert<String>()}.getDayNum()});
|
assert_cast<ColumnUInt16 &>(column).insertValue(UInt16{LocalDate{value.convert<String>()}.getDayNum()});
|
||||||
break;
|
break;
|
||||||
case ValueType::vtDateTime:
|
case ValueType::vtDateTime:
|
||||||
static_cast<ColumnUInt32 &>(column).insertValue(time_t{LocalDateTime{value.convert<String>()}});
|
assert_cast<ColumnUInt32 &>(column).insertValue(time_t{LocalDateTime{value.convert<String>()}});
|
||||||
break;
|
break;
|
||||||
case ValueType::vtUUID:
|
case ValueType::vtUUID:
|
||||||
static_cast<ColumnUInt128 &>(column).insert(parse<UUID>(value.convert<std::string>()));
|
assert_cast<ColumnUInt128 &>(column).insert(parse<UUID>(value.convert<std::string>()));
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -114,7 +115,7 @@ Block ODBCBlockInputStream::readImpl()
|
|||||||
{
|
{
|
||||||
if (description.types[idx].second)
|
if (description.types[idx].second)
|
||||||
{
|
{
|
||||||
ColumnNullable & column_nullable = static_cast<ColumnNullable &>(*columns[idx]);
|
ColumnNullable & column_nullable = assert_cast<ColumnNullable &>(*columns[idx]);
|
||||||
insertValue(column_nullable.getNestedColumn(), description.types[idx].first, value);
|
insertValue(column_nullable.getNestedColumn(), description.types[idx].first, value);
|
||||||
column_nullable.getNullMapData().emplace_back(0);
|
column_nullable.getNullMapData().emplace_back(0);
|
||||||
}
|
}
|
||||||
|
@ -217,6 +217,7 @@
|
|||||||
|
|
||||||
See https://clickhouse.yandex/docs/en/table_engines/replication/
|
See https://clickhouse.yandex/docs/en/table_engines/replication/
|
||||||
-->
|
-->
|
||||||
|
|
||||||
<zookeeper incl="zookeeper-servers" optional="true" />
|
<zookeeper incl="zookeeper-servers" optional="true" />
|
||||||
|
|
||||||
<!-- Substitutions for parameters of replicated tables.
|
<!-- Substitutions for parameters of replicated tables.
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <DataTypes/DataTypeArray.h>
|
#include <DataTypes/DataTypeArray.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
@ -81,9 +82,9 @@ public:
|
|||||||
const IColumn * nested[num_arguments];
|
const IColumn * nested[num_arguments];
|
||||||
|
|
||||||
for (size_t i = 0; i < num_arguments; ++i)
|
for (size_t i = 0; i < num_arguments; ++i)
|
||||||
nested[i] = &static_cast<const ColumnArray &>(*columns[i]).getData();
|
nested[i] = &assert_cast<const ColumnArray &>(*columns[i]).getData();
|
||||||
|
|
||||||
const ColumnArray & first_array_column = static_cast<const ColumnArray &>(*columns[0]);
|
const ColumnArray & first_array_column = assert_cast<const ColumnArray &>(*columns[0]);
|
||||||
const IColumn::Offsets & offsets = first_array_column.getOffsets();
|
const IColumn::Offsets & offsets = first_array_column.getOffsets();
|
||||||
|
|
||||||
size_t begin = offsets[row_num - 1];
|
size_t begin = offsets[row_num - 1];
|
||||||
@ -92,7 +93,7 @@ public:
|
|||||||
/// Sanity check. NOTE We can implement specialization for a case with single argument, if the check will hurt performance.
|
/// Sanity check. NOTE We can implement specialization for a case with single argument, if the check will hurt performance.
|
||||||
for (size_t i = 1; i < num_arguments; ++i)
|
for (size_t i = 1; i < num_arguments; ++i)
|
||||||
{
|
{
|
||||||
const ColumnArray & ith_column = static_cast<const ColumnArray &>(*columns[i]);
|
const ColumnArray & ith_column = assert_cast<const ColumnArray &>(*columns[i]);
|
||||||
const IColumn::Offsets & ith_offsets = ith_column.getOffsets();
|
const IColumn::Offsets & ith_offsets = ith_column.getOffsets();
|
||||||
|
|
||||||
if (ith_offsets[row_num] != end || (row_num != 0 && ith_offsets[row_num - 1] != begin))
|
if (ith_offsets[row_num] != end || (row_num != 0 && ith_offsets[row_num - 1] != begin))
|
||||||
|
@ -5,6 +5,7 @@
|
|||||||
|
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <Columns/ColumnVector.h>
|
#include <Columns/ColumnVector.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -55,7 +56,7 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
this->data(place).update(static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
this->data(place).update(assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
||||||
}
|
}
|
||||||
|
|
||||||
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
||||||
@ -75,7 +76,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnVector<T> &>(to).getData().push_back(this->data(place).value);
|
assert_cast<ColumnVector<T> &>(to).getData().push_back(this->data(place).value);
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
#include <AggregateFunctions/Helpers.h>
|
#include <AggregateFunctions/Helpers.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -151,7 +152,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnFloat64 &>(to).getData().push_back(getBoundingRatio(data(place)));
|
assert_cast<ColumnFloat64 &>(to).getData().push_back(getBoundingRatio(data(place)));
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override
|
const char * getHeaderFilePath() const override
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <Columns/ColumnNullable.h>
|
#include <Columns/ColumnNullable.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -59,7 +60,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(data(place).count);
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(data(place).count);
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
@ -86,7 +87,7 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
data(place).count += !static_cast<const ColumnNullable &>(*columns[0]).isNullAt(row_num);
|
data(place).count += !assert_cast<const ColumnNullable &>(*columns[0]).isNullAt(row_num);
|
||||||
}
|
}
|
||||||
|
|
||||||
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
||||||
@ -106,7 +107,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(data(place).count);
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(data(place).count);
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
@ -143,7 +144,7 @@ public:
|
|||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
for (size_t i = 0; i < number_of_arguments; ++i)
|
for (size_t i = 0; i < number_of_arguments; ++i)
|
||||||
if (is_nullable[i] && static_cast<const ColumnNullable &>(*columns[i]).isNullAt(row_num))
|
if (is_nullable[i] && assert_cast<const ColumnNullable &>(*columns[i]).isNullAt(row_num))
|
||||||
return;
|
return;
|
||||||
|
|
||||||
++data(place).count;
|
++data(place).count;
|
||||||
@ -166,7 +167,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(data(place).count);
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(data(place).count);
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
#include <AggregateFunctions/UniqVariadicHash.h>
|
#include <AggregateFunctions/UniqVariadicHash.h>
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <Columns/ColumnVector.h>
|
#include <Columns/ColumnVector.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <cmath>
|
#include <cmath>
|
||||||
|
|
||||||
@ -115,7 +116,7 @@ public:
|
|||||||
if constexpr (!std::is_same_v<UInt128, Value>)
|
if constexpr (!std::is_same_v<UInt128, Value>)
|
||||||
{
|
{
|
||||||
/// Here we manage only with numerical types
|
/// Here we manage only with numerical types
|
||||||
const auto & column = static_cast<const ColumnVector <Value> &>(*columns[0]);
|
const auto & column = assert_cast<const ColumnVector <Value> &>(*columns[0]);
|
||||||
this->data(place).add(column.getData()[row_num]);
|
this->data(place).add(column.getData()[row_num]);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
@ -141,7 +142,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
auto & column = static_cast<ColumnVector<Float64> &>(to);
|
auto & column = assert_cast<ColumnVector<Float64> &>(to);
|
||||||
column.getData().push_back(this->data(place).get());
|
column.getData().push_back(this->data(place).get());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <DataTypes/DataTypeArray.h>
|
#include <DataTypes/DataTypeArray.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -149,9 +150,9 @@ public:
|
|||||||
const IColumn * nested[num_arguments];
|
const IColumn * nested[num_arguments];
|
||||||
|
|
||||||
for (size_t i = 0; i < num_arguments; ++i)
|
for (size_t i = 0; i < num_arguments; ++i)
|
||||||
nested[i] = &static_cast<const ColumnArray &>(*columns[i]).getData();
|
nested[i] = &assert_cast<const ColumnArray &>(*columns[i]).getData();
|
||||||
|
|
||||||
const ColumnArray & first_array_column = static_cast<const ColumnArray &>(*columns[0]);
|
const ColumnArray & first_array_column = assert_cast<const ColumnArray &>(*columns[0]);
|
||||||
const IColumn::Offsets & offsets = first_array_column.getOffsets();
|
const IColumn::Offsets & offsets = first_array_column.getOffsets();
|
||||||
|
|
||||||
size_t begin = offsets[row_num - 1];
|
size_t begin = offsets[row_num - 1];
|
||||||
@ -160,7 +161,7 @@ public:
|
|||||||
/// Sanity check. NOTE We can implement specialization for a case with single argument, if the check will hurt performance.
|
/// Sanity check. NOTE We can implement specialization for a case with single argument, if the check will hurt performance.
|
||||||
for (size_t i = 1; i < num_arguments; ++i)
|
for (size_t i = 1; i < num_arguments; ++i)
|
||||||
{
|
{
|
||||||
const ColumnArray & ith_column = static_cast<const ColumnArray &>(*columns[i]);
|
const ColumnArray & ith_column = assert_cast<const ColumnArray &>(*columns[i]);
|
||||||
const IColumn::Offsets & ith_offsets = ith_column.getOffsets();
|
const IColumn::Offsets & ith_offsets = ith_column.getOffsets();
|
||||||
|
|
||||||
if (ith_offsets[row_num] != end || (row_num != 0 && ith_offsets[row_num - 1] != begin))
|
if (ith_offsets[row_num] != end || (row_num != 0 && ith_offsets[row_num - 1] != begin))
|
||||||
@ -228,7 +229,7 @@ public:
|
|||||||
{
|
{
|
||||||
const AggregateFunctionForEachData & state = data(place);
|
const AggregateFunctionForEachData & state = data(place);
|
||||||
|
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
IColumn & elems_to = arr_to.getData();
|
IColumn & elems_to = arr_to.getData();
|
||||||
|
|
||||||
|
@ -12,6 +12,7 @@
|
|||||||
#include <Columns/ColumnString.h>
|
#include <Columns/ColumnString.h>
|
||||||
|
|
||||||
#include <Common/ArenaAllocator.h>
|
#include <Common/ArenaAllocator.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -68,7 +69,7 @@ public:
|
|||||||
if (limit_num_elems && this->data(place).value.size() >= max_elems)
|
if (limit_num_elems && this->data(place).value.size() >= max_elems)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
this->data(place).value.push_back(static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num], arena);
|
this->data(place).value.push_back(assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num], arena);
|
||||||
}
|
}
|
||||||
|
|
||||||
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena * arena) const override
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena * arena) const override
|
||||||
@ -119,14 +120,14 @@ public:
|
|||||||
const auto & value = this->data(place).value;
|
const auto & value = this->data(place).value;
|
||||||
size_t size = value.size();
|
size_t size = value.size();
|
||||||
|
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
|
|
||||||
offsets_to.push_back(offsets_to.back() + size);
|
offsets_to.push_back(offsets_to.back() + size);
|
||||||
|
|
||||||
if (size)
|
if (size)
|
||||||
{
|
{
|
||||||
typename ColumnVector<T>::Container & data_to = static_cast<ColumnVector<T> &>(arr_to.getData()).getData();
|
typename ColumnVector<T>::Container & data_to = assert_cast<ColumnVector<T> &>(arr_to.getData()).getData();
|
||||||
data_to.insert(this->data(place).value.begin(), this->data(place).value.end());
|
data_to.insert(this->data(place).value.begin(), this->data(place).value.end());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -191,7 +192,7 @@ struct GroupArrayListNodeString : public GroupArrayListNodeBase<GroupArrayListNo
|
|||||||
/// Create node from string
|
/// Create node from string
|
||||||
static Node * allocate(const IColumn & column, size_t row_num, Arena * arena)
|
static Node * allocate(const IColumn & column, size_t row_num, Arena * arena)
|
||||||
{
|
{
|
||||||
StringRef string = static_cast<const ColumnString &>(column).getDataAt(row_num);
|
StringRef string = assert_cast<const ColumnString &>(column).getDataAt(row_num);
|
||||||
|
|
||||||
Node * node = reinterpret_cast<Node *>(arena->alignedAlloc(sizeof(Node) + string.size, alignof(Node)));
|
Node * node = reinterpret_cast<Node *>(arena->alignedAlloc(sizeof(Node) + string.size, alignof(Node)));
|
||||||
node->next = nullptr;
|
node->next = nullptr;
|
||||||
@ -203,7 +204,7 @@ struct GroupArrayListNodeString : public GroupArrayListNodeBase<GroupArrayListNo
|
|||||||
|
|
||||||
void insertInto(IColumn & column)
|
void insertInto(IColumn & column)
|
||||||
{
|
{
|
||||||
static_cast<ColumnString &>(column).insertData(data(), size);
|
assert_cast<ColumnString &>(column).insertData(data(), size);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -374,7 +375,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
auto & column_array = static_cast<ColumnArray &>(to);
|
auto & column_array = assert_cast<ColumnArray &>(to);
|
||||||
|
|
||||||
auto & offsets = column_array.getOffsets();
|
auto & offsets = column_array.getOffsets();
|
||||||
offsets.push_back(offsets.back() + data(place).elems);
|
offsets.push_back(offsets.back() + data(place).elems);
|
||||||
@ -383,7 +384,7 @@ public:
|
|||||||
|
|
||||||
if (std::is_same_v<Node, GroupArrayListNodeString>)
|
if (std::is_same_v<Node, GroupArrayListNodeString>)
|
||||||
{
|
{
|
||||||
auto & string_offsets = static_cast<ColumnString &>(column_data).getOffsets();
|
auto & string_offsets = assert_cast<ColumnString &>(column_data).getOffsets();
|
||||||
string_offsets.reserve(string_offsets.size() + data(place).elems);
|
string_offsets.reserve(string_offsets.size() + data(place).elems);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -10,6 +10,7 @@
|
|||||||
#include <Columns/ColumnVector.h>
|
#include <Columns/ColumnVector.h>
|
||||||
|
|
||||||
#include <Common/FieldVisitors.h>
|
#include <Common/FieldVisitors.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <Interpreters/convertFieldToType.h>
|
#include <Interpreters/convertFieldToType.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
@ -180,7 +181,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
ColumnArray & to_array = static_cast<ColumnArray &>(to);
|
ColumnArray & to_array = assert_cast<ColumnArray &>(to);
|
||||||
IColumn & to_data = to_array.getData();
|
IColumn & to_data = to_array.getData();
|
||||||
ColumnArray::Offsets & to_offsets = to_array.getOffsets();
|
ColumnArray::Offsets & to_offsets = to_array.getOffsets();
|
||||||
|
|
||||||
|
@ -11,6 +11,7 @@
|
|||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
|
|
||||||
#include <Common/ArenaAllocator.h>
|
#include <Common/ArenaAllocator.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -164,7 +165,7 @@ public:
|
|||||||
const auto & data = this->data(place);
|
const auto & data = this->data(place);
|
||||||
size_t size = data.value.size();
|
size_t size = data.value.size();
|
||||||
|
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
|
|
||||||
offsets_to.push_back(offsets_to.back() + size);
|
offsets_to.push_back(offsets_to.back() + size);
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <Columns/ColumnVector.h>
|
#include <Columns/ColumnVector.h>
|
||||||
#include <boost/noncopyable.hpp>
|
#include <Common/assert_cast.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <AggregateFunctions/AggregateFunctionGroupBitmapData.h>
|
#include <AggregateFunctions/AggregateFunctionGroupBitmapData.h>
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
@ -26,7 +26,7 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
this->data(place).rbs.add(static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
this->data(place).rbs.add(assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
||||||
}
|
}
|
||||||
|
|
||||||
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
||||||
@ -46,7 +46,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnVector<T> &>(to).getData().push_back(this->data(place).rbs.size());
|
assert_cast<ColumnVector<T> &>(to).getData().push_back(this->data(place).rbs.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
|
@ -10,6 +10,7 @@
|
|||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
|
|
||||||
#include <Common/HashTable/HashSet.h>
|
#include <Common/HashTable/HashSet.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -63,7 +64,7 @@ public:
|
|||||||
{
|
{
|
||||||
if (limit_num_elems && this->data(place).value.size() >= max_elems)
|
if (limit_num_elems && this->data(place).value.size() >= max_elems)
|
||||||
return;
|
return;
|
||||||
this->data(place).value.insert(static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
this->data(place).value.insert(assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
||||||
}
|
}
|
||||||
|
|
||||||
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
||||||
@ -100,7 +101,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
|
|
||||||
const typename State::Set & set = this->data(place).value;
|
const typename State::Set & set = this->data(place).value;
|
||||||
@ -108,7 +109,7 @@ public:
|
|||||||
|
|
||||||
offsets_to.push_back(offsets_to.back() + size);
|
offsets_to.push_back(offsets_to.back() + size);
|
||||||
|
|
||||||
typename ColumnVector<T>::Container & data_to = static_cast<ColumnVector<T> &>(arr_to.getData()).getData();
|
typename ColumnVector<T>::Container & data_to = assert_cast<ColumnVector<T> &>(arr_to.getData()).getData();
|
||||||
size_t old_size = data_to.size();
|
size_t old_size = data_to.size();
|
||||||
data_to.resize(old_size + size);
|
data_to.resize(old_size + size);
|
||||||
|
|
||||||
@ -254,7 +255,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
IColumn & data_to = arr_to.getData();
|
IColumn & data_to = arr_to.getData();
|
||||||
|
|
||||||
|
@ -6,6 +6,7 @@
|
|||||||
#include <Columns/ColumnVector.h>
|
#include <Columns/ColumnVector.h>
|
||||||
#include <Columns/ColumnTuple.h>
|
#include <Columns/ColumnTuple.h>
|
||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <DataTypes/DataTypeArray.h>
|
#include <DataTypes/DataTypeArray.h>
|
||||||
@ -333,7 +334,7 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
auto val = static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num];
|
auto val = assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num];
|
||||||
this->data(place).add(static_cast<Data::Mean>(val), 1, max_bins);
|
this->data(place).add(static_cast<Data::Mean>(val), 1, max_bins);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -356,13 +357,13 @@ public:
|
|||||||
{
|
{
|
||||||
auto & data = this->data(const_cast<AggregateDataPtr>(place));
|
auto & data = this->data(const_cast<AggregateDataPtr>(place));
|
||||||
|
|
||||||
auto & to_array = static_cast<ColumnArray &>(to);
|
auto & to_array = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = to_array.getOffsets();
|
ColumnArray::Offsets & offsets_to = to_array.getOffsets();
|
||||||
auto & to_tuple = static_cast<ColumnTuple &>(to_array.getData());
|
auto & to_tuple = assert_cast<ColumnTuple &>(to_array.getData());
|
||||||
|
|
||||||
auto & to_lower = static_cast<ColumnVector<Data::Mean> &>(to_tuple.getColumn(0));
|
auto & to_lower = assert_cast<ColumnVector<Data::Mean> &>(to_tuple.getColumn(0));
|
||||||
auto & to_upper = static_cast<ColumnVector<Data::Mean> &>(to_tuple.getColumn(1));
|
auto & to_upper = assert_cast<ColumnVector<Data::Mean> &>(to_tuple.getColumn(1));
|
||||||
auto & to_weights = static_cast<ColumnVector<Data::Weight> &>(to_tuple.getColumn(2));
|
auto & to_weights = assert_cast<ColumnVector<Data::Weight> &>(to_tuple.getColumn(2));
|
||||||
data.insertResultInto(to_lower, to_upper, to_weights, max_bins);
|
data.insertResultInto(to_lower, to_upper, to_weights, max_bins);
|
||||||
|
|
||||||
offsets_to.push_back(to_tuple.size());
|
offsets_to.push_back(to_tuple.size());
|
||||||
|
@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
|
|
||||||
@ -75,7 +76,7 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
||||||
{
|
{
|
||||||
if (static_cast<const ColumnUInt8 &>(*columns[num_arguments - 1]).getData()[row_num])
|
if (assert_cast<const ColumnUInt8 &>(*columns[num_arguments - 1]).getData()[row_num])
|
||||||
nested_func->add(place, columns, row_num, arena);
|
nested_func->add(place, columns, row_num, arena);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
#include <Columns/ColumnTuple.h>
|
#include <Columns/ColumnTuple.h>
|
||||||
#include <Common/FieldVisitors.h>
|
#include <Common/FieldVisitors.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include "AggregateFunctionFactory.h"
|
#include "AggregateFunctionFactory.h"
|
||||||
#include "FactoryHelpers.h"
|
#include "FactoryHelpers.h"
|
||||||
#include "Helpers.h"
|
#include "Helpers.h"
|
||||||
@ -147,14 +148,14 @@ void LinearModelData::returnWeights(IColumn & to) const
|
|||||||
{
|
{
|
||||||
size_t size = weights.size() + 1;
|
size_t size = weights.size() + 1;
|
||||||
|
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
|
|
||||||
size_t old_size = offsets_to.back();
|
size_t old_size = offsets_to.back();
|
||||||
offsets_to.push_back(old_size + size);
|
offsets_to.push_back(old_size + size);
|
||||||
|
|
||||||
typename ColumnFloat64::Container & val_to
|
typename ColumnFloat64::Container & val_to
|
||||||
= static_cast<ColumnFloat64 &>(arr_to.getData()).getData();
|
= assert_cast<ColumnFloat64 &>(arr_to.getData()).getData();
|
||||||
|
|
||||||
val_to.reserve(old_size + size);
|
val_to.reserve(old_size + size);
|
||||||
for (size_t i = 0; i + 1 < size; ++i)
|
for (size_t i = 0; i + 1 < size; ++i)
|
||||||
|
@ -10,6 +10,7 @@
|
|||||||
|
|
||||||
#include <Common/ArenaAllocator.h>
|
#include <Common/ArenaAllocator.h>
|
||||||
#include <Common/NaNUtils.h>
|
#include <Common/NaNUtils.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -88,8 +89,8 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
||||||
{
|
{
|
||||||
PointType left = static_cast<const ColumnVector<PointType> &>(*columns[0]).getData()[row_num];
|
PointType left = assert_cast<const ColumnVector<PointType> &>(*columns[0]).getData()[row_num];
|
||||||
PointType right = static_cast<const ColumnVector<PointType> &>(*columns[1]).getData()[row_num];
|
PointType right = assert_cast<const ColumnVector<PointType> &>(*columns[1]).getData()[row_num];
|
||||||
|
|
||||||
if (!isNaN(left))
|
if (!isNaN(left))
|
||||||
this->data(place).value.push_back(std::make_pair(left, Int64(1)), arena);
|
this->data(place).value.push_back(std::make_pair(left, Int64(1)), arena);
|
||||||
@ -152,12 +153,12 @@ public:
|
|||||||
|
|
||||||
if (kind == AggregateFunctionIntersectionsKind::Count)
|
if (kind == AggregateFunctionIntersectionsKind::Count)
|
||||||
{
|
{
|
||||||
auto & result_column = static_cast<ColumnUInt64 &>(to).getData();
|
auto & result_column = assert_cast<ColumnUInt64 &>(to).getData();
|
||||||
result_column.push_back(max_intersections);
|
result_column.push_back(max_intersections);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
auto & result_column = static_cast<ColumnVector<PointType> &>(to).getData();
|
auto & result_column = assert_cast<ColumnVector<PointType> &>(to).getData();
|
||||||
result_column.push_back(position_of_max_intersections);
|
result_column.push_back(position_of_max_intersections);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -4,6 +4,7 @@
|
|||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <Columns/ColumnAggregateFunction.h>
|
#include <Columns/ColumnAggregateFunction.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -70,7 +71,7 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
||||||
{
|
{
|
||||||
nested_func->merge(place, static_cast<const ColumnAggregateFunction &>(*columns[0]).getData()[row_num], arena);
|
nested_func->merge(place, assert_cast<const ColumnAggregateFunction &>(*columns[0]).getData()[row_num], arena);
|
||||||
}
|
}
|
||||||
|
|
||||||
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena * arena) const override
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena * arena) const override
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
#include <Columns/ColumnString.h>
|
#include <Columns/ColumnString.h>
|
||||||
#include <DataTypes/IDataType.h>
|
#include <DataTypes/IDataType.h>
|
||||||
#include <common/StringRef.h>
|
#include <common/StringRef.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -38,9 +39,9 @@ public:
|
|||||||
void insertResultInto(IColumn & to) const
|
void insertResultInto(IColumn & to) const
|
||||||
{
|
{
|
||||||
if (has())
|
if (has())
|
||||||
static_cast<ColumnVector<T> &>(to).getData().push_back(value);
|
assert_cast<ColumnVector<T> &>(to).getData().push_back(value);
|
||||||
else
|
else
|
||||||
static_cast<ColumnVector<T> &>(to).insertDefault();
|
assert_cast<ColumnVector<T> &>(to).insertDefault();
|
||||||
}
|
}
|
||||||
|
|
||||||
void write(WriteBuffer & buf, const IDataType & /*data_type*/) const
|
void write(WriteBuffer & buf, const IDataType & /*data_type*/) const
|
||||||
@ -61,7 +62,7 @@ public:
|
|||||||
void change(const IColumn & column, size_t row_num, Arena *)
|
void change(const IColumn & column, size_t row_num, Arena *)
|
||||||
{
|
{
|
||||||
has_value = true;
|
has_value = true;
|
||||||
value = static_cast<const ColumnVector<T> &>(column).getData()[row_num];
|
value = assert_cast<const ColumnVector<T> &>(column).getData()[row_num];
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Assuming to.has()
|
/// Assuming to.has()
|
||||||
@ -112,7 +113,7 @@ public:
|
|||||||
|
|
||||||
bool changeIfLess(const IColumn & column, size_t row_num, Arena * arena)
|
bool changeIfLess(const IColumn & column, size_t row_num, Arena * arena)
|
||||||
{
|
{
|
||||||
if (!has() || static_cast<const ColumnVector<T> &>(column).getData()[row_num] < value)
|
if (!has() || assert_cast<const ColumnVector<T> &>(column).getData()[row_num] < value)
|
||||||
{
|
{
|
||||||
change(column, row_num, arena);
|
change(column, row_num, arena);
|
||||||
return true;
|
return true;
|
||||||
@ -134,7 +135,7 @@ public:
|
|||||||
|
|
||||||
bool changeIfGreater(const IColumn & column, size_t row_num, Arena * arena)
|
bool changeIfGreater(const IColumn & column, size_t row_num, Arena * arena)
|
||||||
{
|
{
|
||||||
if (!has() || static_cast<const ColumnVector<T> &>(column).getData()[row_num] > value)
|
if (!has() || assert_cast<const ColumnVector<T> &>(column).getData()[row_num] > value)
|
||||||
{
|
{
|
||||||
change(column, row_num, arena);
|
change(column, row_num, arena);
|
||||||
return true;
|
return true;
|
||||||
@ -161,7 +162,7 @@ public:
|
|||||||
|
|
||||||
bool isEqualTo(const IColumn & column, size_t row_num) const
|
bool isEqualTo(const IColumn & column, size_t row_num) const
|
||||||
{
|
{
|
||||||
return has() && static_cast<const ColumnVector<T> &>(column).getData()[row_num] == value;
|
return has() && assert_cast<const ColumnVector<T> &>(column).getData()[row_num] == value;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -204,9 +205,9 @@ public:
|
|||||||
void insertResultInto(IColumn & to) const
|
void insertResultInto(IColumn & to) const
|
||||||
{
|
{
|
||||||
if (has())
|
if (has())
|
||||||
static_cast<ColumnString &>(to).insertDataWithTerminatingZero(getData(), size);
|
assert_cast<ColumnString &>(to).insertDataWithTerminatingZero(getData(), size);
|
||||||
else
|
else
|
||||||
static_cast<ColumnString &>(to).insertDefault();
|
assert_cast<ColumnString &>(to).insertDefault();
|
||||||
}
|
}
|
||||||
|
|
||||||
void write(WriteBuffer & buf, const IDataType & /*data_type*/) const
|
void write(WriteBuffer & buf, const IDataType & /*data_type*/) const
|
||||||
@ -281,7 +282,7 @@ public:
|
|||||||
|
|
||||||
void change(const IColumn & column, size_t row_num, Arena * arena)
|
void change(const IColumn & column, size_t row_num, Arena * arena)
|
||||||
{
|
{
|
||||||
changeImpl(static_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num), arena);
|
changeImpl(assert_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num), arena);
|
||||||
}
|
}
|
||||||
|
|
||||||
void change(const Self & to, Arena * arena)
|
void change(const Self & to, Arena * arena)
|
||||||
@ -330,7 +331,7 @@ public:
|
|||||||
|
|
||||||
bool changeIfLess(const IColumn & column, size_t row_num, Arena * arena)
|
bool changeIfLess(const IColumn & column, size_t row_num, Arena * arena)
|
||||||
{
|
{
|
||||||
if (!has() || static_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num) < getStringRef())
|
if (!has() || assert_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num) < getStringRef())
|
||||||
{
|
{
|
||||||
change(column, row_num, arena);
|
change(column, row_num, arena);
|
||||||
return true;
|
return true;
|
||||||
@ -352,7 +353,7 @@ public:
|
|||||||
|
|
||||||
bool changeIfGreater(const IColumn & column, size_t row_num, Arena * arena)
|
bool changeIfGreater(const IColumn & column, size_t row_num, Arena * arena)
|
||||||
{
|
{
|
||||||
if (!has() || static_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num) > getStringRef())
|
if (!has() || assert_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num) > getStringRef())
|
||||||
{
|
{
|
||||||
change(column, row_num, arena);
|
change(column, row_num, arena);
|
||||||
return true;
|
return true;
|
||||||
@ -379,7 +380,7 @@ public:
|
|||||||
|
|
||||||
bool isEqualTo(const IColumn & column, size_t row_num) const
|
bool isEqualTo(const IColumn & column, size_t row_num) const
|
||||||
{
|
{
|
||||||
return has() && static_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num) == getStringRef();
|
return has() && assert_cast<const ColumnString &>(column).getDataAtWithTerminatingZero(row_num) == getStringRef();
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -3,6 +3,7 @@
|
|||||||
#include <array>
|
#include <array>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <Columns/ColumnNullable.h>
|
#include <Columns/ColumnNullable.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <DataTypes/DataTypeNullable.h>
|
#include <DataTypes/DataTypeNullable.h>
|
||||||
#include <IO/ReadHelpers.h>
|
#include <IO/ReadHelpers.h>
|
||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
@ -149,7 +150,7 @@ public:
|
|||||||
{
|
{
|
||||||
if (result_is_nullable)
|
if (result_is_nullable)
|
||||||
{
|
{
|
||||||
ColumnNullable & to_concrete = static_cast<ColumnNullable &>(to);
|
ColumnNullable & to_concrete = assert_cast<ColumnNullable &>(to);
|
||||||
if (getFlag(place))
|
if (getFlag(place))
|
||||||
{
|
{
|
||||||
nested_function->insertResultInto(nestedPlace(place), to_concrete.getNestedColumn());
|
nested_function->insertResultInto(nestedPlace(place), to_concrete.getNestedColumn());
|
||||||
@ -194,7 +195,7 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, size_t row_num, Arena * arena) const override
|
||||||
{
|
{
|
||||||
const ColumnNullable * column = static_cast<const ColumnNullable *>(columns[0]);
|
const ColumnNullable * column = assert_cast<const ColumnNullable *>(columns[0]);
|
||||||
if (!column->isNullAt(row_num))
|
if (!column->isNullAt(row_num))
|
||||||
{
|
{
|
||||||
this->setFlag(place);
|
this->setFlag(place);
|
||||||
@ -233,7 +234,7 @@ public:
|
|||||||
{
|
{
|
||||||
if (is_nullable[i])
|
if (is_nullable[i])
|
||||||
{
|
{
|
||||||
const ColumnNullable & nullable_col = static_cast<const ColumnNullable &>(*columns[i]);
|
const ColumnNullable & nullable_col = assert_cast<const ColumnNullable &>(*columns[i]);
|
||||||
if (nullable_col.isNullAt(row_num))
|
if (nullable_col.isNullAt(row_num))
|
||||||
{
|
{
|
||||||
/// If at least one column has a null value in the current row,
|
/// If at least one column has a null value in the current row,
|
||||||
|
@ -21,6 +21,7 @@
|
|||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <IO/ReadHelpers.h>
|
#include <IO/ReadHelpers.h>
|
||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <type_traits>
|
#include <type_traits>
|
||||||
|
|
||||||
@ -143,7 +144,7 @@ public:
|
|||||||
|
|
||||||
if constexpr (returns_many)
|
if constexpr (returns_many)
|
||||||
{
|
{
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
|
|
||||||
size_t size = levels.size();
|
size_t size = levels.size();
|
||||||
@ -154,7 +155,7 @@ public:
|
|||||||
|
|
||||||
if constexpr (returns_float)
|
if constexpr (returns_float)
|
||||||
{
|
{
|
||||||
auto & data_to = static_cast<ColumnVector<FloatReturnType> &>(arr_to.getData()).getData();
|
auto & data_to = assert_cast<ColumnVector<FloatReturnType> &>(arr_to.getData()).getData();
|
||||||
size_t old_size = data_to.size();
|
size_t old_size = data_to.size();
|
||||||
data_to.resize(data_to.size() + size);
|
data_to.resize(data_to.size() + size);
|
||||||
|
|
||||||
@ -172,7 +173,7 @@ public:
|
|||||||
else
|
else
|
||||||
{
|
{
|
||||||
if constexpr (returns_float)
|
if constexpr (returns_float)
|
||||||
static_cast<ColumnVector<FloatReturnType> &>(to).getData().push_back(data.getFloat(level));
|
assert_cast<ColumnVector<FloatReturnType> &>(to).getData().push_back(data.getFloat(level));
|
||||||
else
|
else
|
||||||
static_cast<ColVecType &>(to).getData().push_back(data.get(level));
|
static_cast<ColVecType &>(to).getData().push_back(data.get(level));
|
||||||
}
|
}
|
||||||
|
@ -3,6 +3,7 @@
|
|||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
#include <DataTypes/DataTypeArray.h>
|
#include <DataTypes/DataTypeArray.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -169,8 +170,8 @@ public:
|
|||||||
ConstAggregateDataPtr place,
|
ConstAggregateDataPtr place,
|
||||||
IColumn & to) const override
|
IColumn & to) const override
|
||||||
{
|
{
|
||||||
auto & col = static_cast<ColumnArray &>(to);
|
auto & col = assert_cast<ColumnArray &>(to);
|
||||||
auto & col_offsets = static_cast<ColumnArray::ColumnOffsets &>(col.getOffsetsColumn());
|
auto & col_offsets = assert_cast<ColumnArray::ColumnOffsets &>(col.getOffsetsColumn());
|
||||||
|
|
||||||
for (size_t i = 0; i < total; ++i)
|
for (size_t i = 0; i < total; ++i)
|
||||||
nested_function->insertResultInto(place + i * sod, col.getData());
|
nested_function->insertResultInto(place + i * sod, col.getData());
|
||||||
|
@ -5,6 +5,7 @@
|
|||||||
#include <unordered_set>
|
#include <unordered_set>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <DataTypes/DataTypeArray.h>
|
#include <DataTypes/DataTypeArray.h>
|
||||||
#include <IO/ReadHelpers.h>
|
#include <IO/ReadHelpers.h>
|
||||||
@ -100,7 +101,7 @@ public:
|
|||||||
{
|
{
|
||||||
for (const auto i : ext::range(0, events_size))
|
for (const auto i : ext::range(0, events_size))
|
||||||
{
|
{
|
||||||
auto event = static_cast<const ColumnVector<UInt8> *>(columns[i])->getData()[row_num];
|
auto event = assert_cast<const ColumnVector<UInt8> *>(columns[i])->getData()[row_num];
|
||||||
if (event)
|
if (event)
|
||||||
{
|
{
|
||||||
this->data(place).add(i);
|
this->data(place).add(i);
|
||||||
@ -125,8 +126,8 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
auto & data_to = static_cast<ColumnUInt8 &>(static_cast<ColumnArray &>(to).getData()).getData();
|
auto & data_to = assert_cast<ColumnUInt8 &>(assert_cast<ColumnArray &>(to).getData()).getData();
|
||||||
auto & offsets_to = static_cast<ColumnArray &>(to).getOffsets();
|
auto & offsets_to = assert_cast<ColumnArray &>(to).getOffsets();
|
||||||
|
|
||||||
ColumnArray::Offset current_offset = data_to.size();
|
ColumnArray::Offset current_offset = data_to.size();
|
||||||
data_to.resize(current_offset + events_size);
|
data_to.resize(current_offset + events_size);
|
||||||
|
@ -4,6 +4,7 @@
|
|||||||
#include <DataTypes/DataTypeDateTime.h>
|
#include <DataTypes/DataTypeDateTime.h>
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <ext/range.h>
|
#include <ext/range.h>
|
||||||
#include <Common/PODArray.h>
|
#include <Common/PODArray.h>
|
||||||
#include <IO/ReadHelpers.h>
|
#include <IO/ReadHelpers.h>
|
||||||
@ -152,12 +153,12 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
const auto timestamp = static_cast<const ColumnVector<T> *>(columns[0])->getData()[row_num];
|
const auto timestamp = assert_cast<const ColumnVector<T> *>(columns[0])->getData()[row_num];
|
||||||
|
|
||||||
typename Data::Events events;
|
typename Data::Events events;
|
||||||
for (const auto i : ext::range(1, arg_count))
|
for (const auto i : ext::range(1, arg_count))
|
||||||
{
|
{
|
||||||
const auto event = static_cast<const ColumnUInt8 *>(columns[i])->getData()[row_num];
|
const auto event = assert_cast<const ColumnUInt8 *>(columns[i])->getData()[row_num];
|
||||||
events.set(i - 1, event);
|
events.set(i - 1, event);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -574,7 +575,7 @@ public:
|
|||||||
auto events_it = events_begin;
|
auto events_it = events_begin;
|
||||||
|
|
||||||
bool match = this->pattern_has_time ? this->backtrackingMatch(events_it, events_end) : this->dfaMatch(events_it, events_end);
|
bool match = this->pattern_has_time ? this->backtrackingMatch(events_it, events_end) : this->dfaMatch(events_it, events_end);
|
||||||
static_cast<ColumnUInt8 &>(to).getData().push_back(match);
|
assert_cast<ColumnUInt8 &>(to).getData().push_back(match);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -594,7 +595,7 @@ public:
|
|||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
const_cast<Data &>(this->data(place)).sort();
|
const_cast<Data &>(this->data(place)).sort();
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(count(place));
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(count(place));
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
@ -3,6 +3,7 @@
|
|||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <Columns/ColumnVector.h>
|
#include <Columns/ColumnVector.h>
|
||||||
#include <Columns/ColumnTuple.h>
|
#include <Columns/ColumnTuple.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <DataTypes/DataTypeNullable.h>
|
#include <DataTypes/DataTypeNullable.h>
|
||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <DataTypes/DataTypeTuple.h>
|
#include <DataTypes/DataTypeTuple.h>
|
||||||
@ -120,8 +121,8 @@ public:
|
|||||||
Arena *
|
Arena *
|
||||||
) const override
|
) const override
|
||||||
{
|
{
|
||||||
auto col_x = static_cast<const ColumnVector<X> *>(columns[0]);
|
auto col_x = assert_cast<const ColumnVector<X> *>(columns[0]);
|
||||||
auto col_y = static_cast<const ColumnVector<Y> *>(columns[1]);
|
auto col_y = assert_cast<const ColumnVector<Y> *>(columns[1]);
|
||||||
|
|
||||||
X x = col_x->getData()[row_num];
|
X x = col_x->getData()[row_num];
|
||||||
Y y = col_y->getData()[row_num];
|
Y y = col_y->getData()[row_num];
|
||||||
@ -181,9 +182,9 @@ public:
|
|||||||
Ret k = this->data(place).getK();
|
Ret k = this->data(place).getK();
|
||||||
Ret b = this->data(place).getB(k);
|
Ret b = this->data(place).getB(k);
|
||||||
|
|
||||||
auto & col_tuple = static_cast<ColumnTuple &>(to);
|
auto & col_tuple = assert_cast<ColumnTuple &>(to);
|
||||||
auto & col_k = static_cast<ColumnVector<Ret> &>(col_tuple.getColumn(0));
|
auto & col_k = assert_cast<ColumnVector<Ret> &>(col_tuple.getColumn(0));
|
||||||
auto & col_b = static_cast<ColumnVector<Ret> &>(col_tuple.getColumn(1));
|
auto & col_b = assert_cast<ColumnVector<Ret> &>(col_tuple.getColumn(1));
|
||||||
|
|
||||||
col_k.getData().push_back(k);
|
col_k.getData().push_back(k);
|
||||||
col_b.getData().push_back(b);
|
col_b.getData().push_back(b);
|
||||||
|
@ -4,6 +4,7 @@
|
|||||||
#include <DataTypes/DataTypeAggregateFunction.h>
|
#include <DataTypes/DataTypeAggregateFunction.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <Columns/ColumnAggregateFunction.h>
|
#include <Columns/ColumnAggregateFunction.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -81,7 +82,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnAggregateFunction &>(to).getData().push_back(const_cast<AggregateDataPtr>(place));
|
assert_cast<ColumnAggregateFunction &>(to).getData().push_back(const_cast<AggregateDataPtr>(place));
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Aggregate function or aggregate function state.
|
/// Aggregate function or aggregate function state.
|
||||||
|
@ -5,9 +5,11 @@
|
|||||||
#include <DataTypes/DataTypesNumber.h>
|
#include <DataTypes/DataTypesNumber.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <cmath>
|
#include <cmath>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
{
|
{
|
||||||
|
|
||||||
@ -52,7 +54,7 @@ class AggregateFunctionVarianceData
|
|||||||
public:
|
public:
|
||||||
void update(const IColumn & column, size_t row_num)
|
void update(const IColumn & column, size_t row_num)
|
||||||
{
|
{
|
||||||
T received = static_cast<const ColumnVector<T> &>(column).getData()[row_num];
|
T received = assert_cast<const ColumnVector<T> &>(column).getData()[row_num];
|
||||||
Float64 val = static_cast<Float64>(received);
|
Float64 val = static_cast<Float64>(received);
|
||||||
Float64 delta = val - mean;
|
Float64 delta = val - mean;
|
||||||
|
|
||||||
@ -95,7 +97,7 @@ public:
|
|||||||
|
|
||||||
void publish(IColumn & to) const
|
void publish(IColumn & to) const
|
||||||
{
|
{
|
||||||
static_cast<ColumnFloat64 &>(to).getData().push_back(Op::apply(m2, count));
|
assert_cast<ColumnFloat64 &>(to).getData().push_back(Op::apply(m2, count));
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
@ -265,11 +267,11 @@ private:
|
|||||||
public:
|
public:
|
||||||
void update(const IColumn & column_left, const IColumn & column_right, size_t row_num)
|
void update(const IColumn & column_left, const IColumn & column_right, size_t row_num)
|
||||||
{
|
{
|
||||||
T left_received = static_cast<const ColumnVector<T> &>(column_left).getData()[row_num];
|
T left_received = assert_cast<const ColumnVector<T> &>(column_left).getData()[row_num];
|
||||||
Float64 left_val = static_cast<Float64>(left_received);
|
Float64 left_val = static_cast<Float64>(left_received);
|
||||||
Float64 left_delta = left_val - left_mean;
|
Float64 left_delta = left_val - left_mean;
|
||||||
|
|
||||||
U right_received = static_cast<const ColumnVector<U> &>(column_right).getData()[row_num];
|
U right_received = assert_cast<const ColumnVector<U> &>(column_right).getData()[row_num];
|
||||||
Float64 right_val = static_cast<Float64>(right_received);
|
Float64 right_val = static_cast<Float64>(right_received);
|
||||||
Float64 right_delta = right_val - right_mean;
|
Float64 right_delta = right_val - right_mean;
|
||||||
|
|
||||||
@ -345,9 +347,9 @@ public:
|
|||||||
void publish(IColumn & to) const
|
void publish(IColumn & to) const
|
||||||
{
|
{
|
||||||
if constexpr (compute_marginal_moments)
|
if constexpr (compute_marginal_moments)
|
||||||
static_cast<ColumnFloat64 &>(to).getData().push_back(Op::apply(co_moment, Base::left_m2, Base::right_m2, count));
|
assert_cast<ColumnFloat64 &>(to).getData().push_back(Op::apply(co_moment, Base::left_m2, Base::right_m2, count));
|
||||||
else
|
else
|
||||||
static_cast<ColumnFloat64 &>(to).getData().push_back(Op::apply(co_moment, count));
|
assert_cast<ColumnFloat64 &>(to).getData().push_back(Op::apply(co_moment, count));
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
@ -12,6 +12,7 @@
|
|||||||
#include <Columns/ColumnDecimal.h>
|
#include <Columns/ColumnDecimal.h>
|
||||||
|
|
||||||
#include <Common/FieldVisitors.h>
|
#include <Common/FieldVisitors.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
|
||||||
@ -83,7 +84,7 @@ public:
|
|||||||
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
// Column 0 contains array of keys of known type
|
// Column 0 contains array of keys of known type
|
||||||
const ColumnArray & array_column0 = static_cast<const ColumnArray &>(*columns[0]);
|
const ColumnArray & array_column0 = assert_cast<const ColumnArray &>(*columns[0]);
|
||||||
const IColumn::Offsets & offsets0 = array_column0.getOffsets();
|
const IColumn::Offsets & offsets0 = array_column0.getOffsets();
|
||||||
const auto & keys_vec = static_cast<const ColVecType &>(array_column0.getData());
|
const auto & keys_vec = static_cast<const ColVecType &>(array_column0.getData());
|
||||||
const size_t keys_vec_offset = offsets0[row_num - 1];
|
const size_t keys_vec_offset = offsets0[row_num - 1];
|
||||||
@ -94,7 +95,7 @@ public:
|
|||||||
for (size_t col = 0, size = values_types.size(); col < size; ++col)
|
for (size_t col = 0, size = values_types.size(); col < size; ++col)
|
||||||
{
|
{
|
||||||
Field value;
|
Field value;
|
||||||
const ColumnArray & array_column = static_cast<const ColumnArray &>(*columns[col + 1]);
|
const ColumnArray & array_column = assert_cast<const ColumnArray &>(*columns[col + 1]);
|
||||||
const IColumn::Offsets & offsets = array_column.getOffsets();
|
const IColumn::Offsets & offsets = array_column.getOffsets();
|
||||||
const size_t values_vec_offset = offsets[row_num - 1];
|
const size_t values_vec_offset = offsets[row_num - 1];
|
||||||
const size_t values_vec_size = (offsets[row_num] - values_vec_offset);
|
const size_t values_vec_size = (offsets[row_num] - values_vec_offset);
|
||||||
@ -228,8 +229,8 @@ public:
|
|||||||
|
|
||||||
size_t size = merged_maps.size();
|
size_t size = merged_maps.size();
|
||||||
|
|
||||||
auto & to_tuple = static_cast<ColumnTuple &>(to);
|
auto & to_tuple = assert_cast<ColumnTuple &>(to);
|
||||||
auto & to_keys_arr = static_cast<ColumnArray &>(to_tuple.getColumn(0));
|
auto & to_keys_arr = assert_cast<ColumnArray &>(to_tuple.getColumn(0));
|
||||||
auto & to_keys_col = to_keys_arr.getData();
|
auto & to_keys_col = to_keys_arr.getData();
|
||||||
|
|
||||||
// Advance column offsets
|
// Advance column offsets
|
||||||
@ -239,7 +240,7 @@ public:
|
|||||||
|
|
||||||
for (size_t col = 0; col < values_types.size(); ++col)
|
for (size_t col = 0; col < values_types.size(); ++col)
|
||||||
{
|
{
|
||||||
auto & to_values_arr = static_cast<ColumnArray &>(to_tuple.getColumn(col + 1));
|
auto & to_values_arr = assert_cast<ColumnArray &>(to_tuple.getColumn(col + 1));
|
||||||
auto & to_values_offsets = to_values_arr.getOffsets();
|
auto & to_values_offsets = to_values_arr.getOffsets();
|
||||||
to_values_offsets.push_back(to_values_offsets.back() + size);
|
to_values_offsets.push_back(to_values_offsets.back() + size);
|
||||||
to_values_arr.getData().reserve(size);
|
to_values_arr.getData().reserve(size);
|
||||||
@ -254,7 +255,7 @@ public:
|
|||||||
// Write 0..n arrays of values
|
// Write 0..n arrays of values
|
||||||
for (size_t col = 0; col < values_types.size(); ++col)
|
for (size_t col = 0; col < values_types.size(); ++col)
|
||||||
{
|
{
|
||||||
auto & to_values_col = static_cast<ColumnArray &>(to_tuple.getColumn(col + 1)).getData();
|
auto & to_values_col = assert_cast<ColumnArray &>(to_tuple.getColumn(col + 1)).getData();
|
||||||
to_values_col.insert(elem.second[col]);
|
to_values_col.insert(elem.second[col]);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -16,6 +16,7 @@
|
|||||||
#include <IO/ReadHelpers.h>
|
#include <IO/ReadHelpers.h>
|
||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
#include <Common/ArenaAllocator.h>
|
#include <Common/ArenaAllocator.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <ext/range.h>
|
#include <ext/range.h>
|
||||||
#include "IAggregateFunction.h"
|
#include "IAggregateFunction.h"
|
||||||
|
|
||||||
@ -235,9 +236,9 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
auto uid = static_cast<const ColumnVector<UInt64> *>(columns[0])->getData()[row_num];
|
auto uid = assert_cast<const ColumnVector<UInt64> *>(columns[0])->getData()[row_num];
|
||||||
auto ts = static_cast<const ColumnVector<Int64> *>(columns[1])->getData()[row_num];
|
auto ts = assert_cast<const ColumnVector<Int64> *>(columns[1])->getData()[row_num];
|
||||||
auto val = static_cast<const ColumnVector<Float64> *>(columns[2])->getData()[row_num];
|
auto val = assert_cast<const ColumnVector<Float64> *>(columns[2])->getData()[row_num];
|
||||||
if (uid && ts && val)
|
if (uid && ts && val)
|
||||||
{
|
{
|
||||||
this->data(place).add(uid, ts, val);
|
this->data(place).add(uid, ts, val);
|
||||||
@ -255,7 +256,7 @@ public:
|
|||||||
const auto & value = this->data(place).result;
|
const auto & value = this->data(place).result;
|
||||||
size_t size = value.size();
|
size_t size = value.size();
|
||||||
|
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
size_t old_size = offsets_to.back();
|
size_t old_size = offsets_to.back();
|
||||||
|
|
||||||
@ -264,9 +265,9 @@ public:
|
|||||||
if (size)
|
if (size)
|
||||||
{
|
{
|
||||||
typename ColumnInt64::Container & ts_to
|
typename ColumnInt64::Container & ts_to
|
||||||
= static_cast<ColumnInt64 &>(static_cast<ColumnTuple &>(arr_to.getData()).getColumn(0)).getData();
|
= assert_cast<ColumnInt64 &>(assert_cast<ColumnTuple &>(arr_to.getData()).getColumn(0)).getData();
|
||||||
typename ColumnFloat64::Container & val_to
|
typename ColumnFloat64::Container & val_to
|
||||||
= static_cast<ColumnFloat64 &>(static_cast<ColumnTuple &>(arr_to.getData()).getColumn(1)).getData();
|
= assert_cast<ColumnFloat64 &>(assert_cast<ColumnTuple &>(arr_to.getData()).getColumn(1)).getData();
|
||||||
ts_to.reserve(old_size + size);
|
ts_to.reserve(old_size + size);
|
||||||
val_to.reserve(old_size + size);
|
val_to.reserve(old_size + size);
|
||||||
size_t i = 0;
|
size_t i = 0;
|
||||||
|
@ -10,8 +10,8 @@
|
|||||||
#include <Columns/ColumnArray.h>
|
#include <Columns/ColumnArray.h>
|
||||||
|
|
||||||
#include <Common/SpaceSaving.h>
|
#include <Common/SpaceSaving.h>
|
||||||
|
|
||||||
#include <Common/FieldVisitors.h>
|
#include <Common/FieldVisitors.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -62,9 +62,9 @@ public:
|
|||||||
set.resize(reserved);
|
set.resize(reserved);
|
||||||
|
|
||||||
if constexpr (is_weighted)
|
if constexpr (is_weighted)
|
||||||
set.insert(static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num], columns[1]->getUInt(row_num));
|
set.insert(assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num], columns[1]->getUInt(row_num));
|
||||||
else
|
else
|
||||||
set.insert(static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
set.insert(assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num]);
|
||||||
}
|
}
|
||||||
|
|
||||||
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
void merge(AggregateDataPtr place, ConstAggregateDataPtr rhs, Arena *) const override
|
||||||
@ -86,7 +86,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
|
|
||||||
const typename State::Set & set = this->data(place).value;
|
const typename State::Set & set = this->data(place).value;
|
||||||
@ -95,7 +95,7 @@ public:
|
|||||||
|
|
||||||
offsets_to.push_back(offsets_to.back() + size);
|
offsets_to.push_back(offsets_to.back() + size);
|
||||||
|
|
||||||
typename ColumnVector<T>::Container & data_to = static_cast<ColumnVector<T> &>(arr_to.getData()).getData();
|
typename ColumnVector<T>::Container & data_to = assert_cast<ColumnVector<T> &>(arr_to.getData()).getData();
|
||||||
size_t old_size = data_to.size();
|
size_t old_size = data_to.size();
|
||||||
data_to.resize(old_size + size);
|
data_to.resize(old_size + size);
|
||||||
|
|
||||||
@ -215,7 +215,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
ColumnArray & arr_to = static_cast<ColumnArray &>(to);
|
ColumnArray & arr_to = assert_cast<ColumnArray &>(to);
|
||||||
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
ColumnArray::Offsets & offsets_to = arr_to.getOffsets();
|
||||||
IColumn & data_to = arr_to.getData();
|
IColumn & data_to = arr_to.getData();
|
||||||
|
|
||||||
|
@ -17,6 +17,7 @@
|
|||||||
#include <Common/HyperLogLogWithSmallSetOptimization.h>
|
#include <Common/HyperLogLogWithSmallSetOptimization.h>
|
||||||
#include <Common/CombinedCardinalityEstimator.h>
|
#include <Common/CombinedCardinalityEstimator.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/UniquesHashSet.h>
|
#include <AggregateFunctions/UniquesHashSet.h>
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
@ -170,7 +171,7 @@ struct OneAdder
|
|||||||
{
|
{
|
||||||
if constexpr (!std::is_same_v<T, String>)
|
if constexpr (!std::is_same_v<T, String>)
|
||||||
{
|
{
|
||||||
const auto & value = static_cast<const ColumnVector<T> &>(column).getData()[row_num];
|
const auto & value = assert_cast<const ColumnVector<T> &>(column).getData()[row_num];
|
||||||
data.set.insert(AggregateFunctionUniqTraits<T>::hash(value));
|
data.set.insert(AggregateFunctionUniqTraits<T>::hash(value));
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
@ -183,7 +184,7 @@ struct OneAdder
|
|||||||
{
|
{
|
||||||
if constexpr (!std::is_same_v<T, String>)
|
if constexpr (!std::is_same_v<T, String>)
|
||||||
{
|
{
|
||||||
data.set.insert(static_cast<const ColumnVector<T> &>(column).getData()[row_num]);
|
data.set.insert(assert_cast<const ColumnVector<T> &>(column).getData()[row_num]);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
@ -240,7 +241,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
@ -296,7 +297,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
|
@ -4,6 +4,7 @@
|
|||||||
#include <Common/FieldVisitors.h>
|
#include <Common/FieldVisitors.h>
|
||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <DataTypes/DataTypeTuple.h>
|
#include <DataTypes/DataTypeTuple.h>
|
||||||
#include <DataTypes/DataTypeUUID.h>
|
#include <DataTypes/DataTypeUUID.h>
|
||||||
@ -131,7 +132,7 @@ public:
|
|||||||
{
|
{
|
||||||
if constexpr (!std::is_same_v<T, String>)
|
if constexpr (!std::is_same_v<T, String>)
|
||||||
{
|
{
|
||||||
const auto & value = static_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num];
|
const auto & value = assert_cast<const ColumnVector<T> &>(*columns[0]).getData()[row_num];
|
||||||
this->data(place).set.insert(detail::AggregateFunctionUniqCombinedTraits<T>::hash(value));
|
this->data(place).set.insert(detail::AggregateFunctionUniqCombinedTraits<T>::hash(value));
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
@ -158,7 +159,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override
|
const char * getHeaderFilePath() const override
|
||||||
@ -222,7 +223,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).set.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override
|
const char * getHeaderFilePath() const override
|
||||||
|
@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
#include <Common/FieldVisitors.h>
|
#include <Common/FieldVisitors.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
#include <AggregateFunctions/UniqVariadicHash.h>
|
#include <AggregateFunctions/UniqVariadicHash.h>
|
||||||
@ -101,7 +102,7 @@ struct __attribute__((__packed__)) AggregateFunctionUniqUpToData
|
|||||||
|
|
||||||
void add(const IColumn & column, size_t row_num, UInt8 threshold)
|
void add(const IColumn & column, size_t row_num, UInt8 threshold)
|
||||||
{
|
{
|
||||||
insert(static_cast<const ColumnVector<T> &>(column).getData()[row_num], threshold);
|
insert(assert_cast<const ColumnVector<T> &>(column).getData()[row_num], threshold);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -123,7 +124,7 @@ struct AggregateFunctionUniqUpToData<UInt128> : AggregateFunctionUniqUpToData<UI
|
|||||||
{
|
{
|
||||||
void add(const IColumn & column, size_t row_num, UInt8 threshold)
|
void add(const IColumn & column, size_t row_num, UInt8 threshold)
|
||||||
{
|
{
|
||||||
UInt128 value = static_cast<const ColumnVector<UInt128> &>(column).getData()[row_num];
|
UInt128 value = assert_cast<const ColumnVector<UInt128> &>(column).getData()[row_num];
|
||||||
insert(sipHash64(value), threshold);
|
insert(sipHash64(value), threshold);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@ -176,7 +177,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).size());
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).size());
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
@ -240,7 +241,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).size());
|
assert_cast<ColumnUInt64 &>(to).getData().push_back(this->data(place).size());
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override { return __FILE__; }
|
const char * getHeaderFilePath() const override { return __FILE__; }
|
||||||
|
@ -9,6 +9,7 @@
|
|||||||
#include <IO/ReadHelpers.h>
|
#include <IO/ReadHelpers.h>
|
||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
#include <Common/ArenaAllocator.h>
|
#include <Common/ArenaAllocator.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <AggregateFunctions/IAggregateFunction.h>
|
#include <AggregateFunctions/IAggregateFunction.h>
|
||||||
|
|
||||||
@ -215,11 +216,11 @@ public:
|
|||||||
|
|
||||||
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
void add(AggregateDataPtr place, const IColumn ** columns, const size_t row_num, Arena *) const override
|
||||||
{
|
{
|
||||||
const auto timestamp = static_cast<const ColumnVector<T> *>(columns[0])->getData()[row_num];
|
const auto timestamp = assert_cast<const ColumnVector<T> *>(columns[0])->getData()[row_num];
|
||||||
// reverse iteration and stable sorting are needed for events that are qualified by more than one condition.
|
// reverse iteration and stable sorting are needed for events that are qualified by more than one condition.
|
||||||
for (auto i = events_size; i > 0; --i)
|
for (auto i = events_size; i > 0; --i)
|
||||||
{
|
{
|
||||||
auto event = static_cast<const ColumnVector<UInt8> *>(columns[i])->getData()[row_num];
|
auto event = assert_cast<const ColumnVector<UInt8> *>(columns[i])->getData()[row_num];
|
||||||
if (event)
|
if (event)
|
||||||
this->data(place).add(timestamp, i);
|
this->data(place).add(timestamp, i);
|
||||||
}
|
}
|
||||||
@ -242,7 +243,7 @@ public:
|
|||||||
|
|
||||||
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
void insertResultInto(ConstAggregateDataPtr place, IColumn & to) const override
|
||||||
{
|
{
|
||||||
static_cast<ColumnUInt8 &>(to).getData().push_back(getEventLevel(this->data(place)));
|
assert_cast<ColumnUInt8 &>(to).getData().push_back(getEventLevel(this->data(place)));
|
||||||
}
|
}
|
||||||
|
|
||||||
const char * getHeaderFilePath() const override
|
const char * getHeaderFilePath() const override
|
||||||
|
@ -4,6 +4,7 @@
|
|||||||
#include <Core/Defines.h>
|
#include <Core/Defines.h>
|
||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
#include <Common/UInt128.h>
|
#include <Common/UInt128.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <Columns/ColumnTuple.h>
|
#include <Columns/ColumnTuple.h>
|
||||||
|
|
||||||
|
|
||||||
@ -67,7 +68,7 @@ struct UniqVariadicHash<false, true>
|
|||||||
{
|
{
|
||||||
UInt64 hash;
|
UInt64 hash;
|
||||||
|
|
||||||
const auto & tuple_columns = static_cast<const ColumnTuple *>(columns[0])->getColumns();
|
const auto & tuple_columns = assert_cast<const ColumnTuple *>(columns[0])->getColumns();
|
||||||
|
|
||||||
const auto * column = tuple_columns.data();
|
const auto * column = tuple_columns.data();
|
||||||
const auto * columns_end = column + num_args;
|
const auto * columns_end = column + num_args;
|
||||||
@ -116,7 +117,7 @@ struct UniqVariadicHash<true, true>
|
|||||||
{
|
{
|
||||||
static inline UInt128 apply(size_t num_args, const IColumn ** columns, size_t row_num)
|
static inline UInt128 apply(size_t num_args, const IColumn ** columns, size_t row_num)
|
||||||
{
|
{
|
||||||
const auto & tuple_columns = static_cast<const ColumnTuple *>(columns[0])->getColumns();
|
const auto & tuple_columns = assert_cast<const ColumnTuple *>(columns[0])->getColumns();
|
||||||
|
|
||||||
const auto * column = tuple_columns.data();
|
const auto * column = tuple_columns.data();
|
||||||
const auto * columns_end = column + num_args;
|
const auto * columns_end = column + num_args;
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
#include <Columns/ColumnAggregateFunction.h>
|
#include <Columns/ColumnAggregateFunction.h>
|
||||||
#include <Columns/ColumnsCommon.h>
|
#include <Columns/ColumnsCommon.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <AggregateFunctions/AggregateFunctionState.h>
|
#include <AggregateFunctions/AggregateFunctionState.h>
|
||||||
#include <DataStreams/ColumnGathererStream.h>
|
#include <DataStreams/ColumnGathererStream.h>
|
||||||
#include <IO/WriteBufferFromArena.h>
|
#include <IO/WriteBufferFromArena.h>
|
||||||
@ -158,7 +159,7 @@ void ColumnAggregateFunction::ensureOwnership()
|
|||||||
|
|
||||||
void ColumnAggregateFunction::insertRangeFrom(const IColumn & from, size_t start, size_t length)
|
void ColumnAggregateFunction::insertRangeFrom(const IColumn & from, size_t start, size_t length)
|
||||||
{
|
{
|
||||||
const ColumnAggregateFunction & from_concrete = static_cast<const ColumnAggregateFunction &>(from);
|
const ColumnAggregateFunction & from_concrete = assert_cast<const ColumnAggregateFunction &>(from);
|
||||||
|
|
||||||
if (start + length > from_concrete.data.size())
|
if (start + length > from_concrete.data.size())
|
||||||
throw Exception("Parameters start = " + toString(start) + ", length = " + toString(length)
|
throw Exception("Parameters start = " + toString(start) + ", length = " + toString(length)
|
||||||
@ -353,7 +354,7 @@ void ColumnAggregateFunction::insertMergeFrom(ConstAggregateDataPtr place)
|
|||||||
|
|
||||||
void ColumnAggregateFunction::insertMergeFrom(const IColumn & from, size_t n)
|
void ColumnAggregateFunction::insertMergeFrom(const IColumn & from, size_t n)
|
||||||
{
|
{
|
||||||
insertMergeFrom(static_cast<const ColumnAggregateFunction &>(from).data[n]);
|
insertMergeFrom(assert_cast<const ColumnAggregateFunction &>(from).data[n]);
|
||||||
}
|
}
|
||||||
|
|
||||||
Arena & ColumnAggregateFunction::createOrGetArena()
|
Arena & ColumnAggregateFunction::createOrGetArena()
|
||||||
@ -492,7 +493,7 @@ MutableColumns ColumnAggregateFunction::scatter(IColumn::ColumnIndex num_columns
|
|||||||
}
|
}
|
||||||
|
|
||||||
for (size_t i = 0; i < num_rows; ++i)
|
for (size_t i = 0; i < num_rows; ++i)
|
||||||
static_cast<ColumnAggregateFunction &>(*columns[selector[i]]).data.push_back(data[i]);
|
assert_cast<ColumnAggregateFunction &>(*columns[selector[i]]).data.push_back(data[i]);
|
||||||
|
|
||||||
return columns;
|
return columns;
|
||||||
}
|
}
|
||||||
|
@ -16,7 +16,7 @@
|
|||||||
#include <Common/Arena.h>
|
#include <Common/Arena.h>
|
||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -227,7 +227,7 @@ void ColumnArray::insert(const Field & x)
|
|||||||
|
|
||||||
void ColumnArray::insertFrom(const IColumn & src_, size_t n)
|
void ColumnArray::insertFrom(const IColumn & src_, size_t n)
|
||||||
{
|
{
|
||||||
const ColumnArray & src = static_cast<const ColumnArray &>(src_);
|
const ColumnArray & src = assert_cast<const ColumnArray &>(src_);
|
||||||
size_t size = src.sizeAt(n);
|
size_t size = src.sizeAt(n);
|
||||||
size_t offset = src.offsetAt(n);
|
size_t offset = src.offsetAt(n);
|
||||||
|
|
||||||
@ -257,7 +257,7 @@ void ColumnArray::popBack(size_t n)
|
|||||||
|
|
||||||
int ColumnArray::compareAt(size_t n, size_t m, const IColumn & rhs_, int nan_direction_hint) const
|
int ColumnArray::compareAt(size_t n, size_t m, const IColumn & rhs_, int nan_direction_hint) const
|
||||||
{
|
{
|
||||||
const ColumnArray & rhs = static_cast<const ColumnArray &>(rhs_);
|
const ColumnArray & rhs = assert_cast<const ColumnArray &>(rhs_);
|
||||||
|
|
||||||
/// Suboptimal
|
/// Suboptimal
|
||||||
size_t lhs_size = sizeAt(n);
|
size_t lhs_size = sizeAt(n);
|
||||||
@ -373,7 +373,7 @@ void ColumnArray::insertRangeFrom(const IColumn & src, size_t start, size_t leng
|
|||||||
if (length == 0)
|
if (length == 0)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
const ColumnArray & src_concrete = static_cast<const ColumnArray &>(src);
|
const ColumnArray & src_concrete = assert_cast<const ColumnArray &>(src);
|
||||||
|
|
||||||
if (start + length > src_concrete.getOffsets().size())
|
if (start + length > src_concrete.getOffsets().size())
|
||||||
throw Exception("Parameter out of bound in ColumnArray::insertRangeFrom method. [start(" + std::to_string(start) + ") + length(" + std::to_string(length) + ") > offsets.size(" + std::to_string(src_concrete.getOffsets().size()) + ")]",
|
throw Exception("Parameter out of bound in ColumnArray::insertRangeFrom method. [start(" + std::to_string(start) + ") + length(" + std::to_string(length) + ") > offsets.size(" + std::to_string(src_concrete.getOffsets().size()) + ")]",
|
||||||
@ -429,10 +429,10 @@ ColumnPtr ColumnArray::filterNumber(const Filter & filt, ssize_t result_size_hin
|
|||||||
|
|
||||||
auto res = ColumnArray::create(data->cloneEmpty());
|
auto res = ColumnArray::create(data->cloneEmpty());
|
||||||
|
|
||||||
auto & res_elems = static_cast<ColumnVector<T> &>(res->getData()).getData();
|
auto & res_elems = assert_cast<ColumnVector<T> &>(res->getData()).getData();
|
||||||
Offsets & res_offsets = res->getOffsets();
|
Offsets & res_offsets = res->getOffsets();
|
||||||
|
|
||||||
filterArraysImpl<T>(static_cast<const ColumnVector<T> &>(*data).getData(), getOffsets(), res_elems, res_offsets, filt, result_size_hint);
|
filterArraysImpl<T>(assert_cast<const ColumnVector<T> &>(*data).getData(), getOffsets(), res_elems, res_offsets, filt, result_size_hint);
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -554,11 +554,11 @@ ColumnPtr ColumnArray::filterNullable(const Filter & filt, ssize_t result_size_h
|
|||||||
if (getOffsets().size() == 0)
|
if (getOffsets().size() == 0)
|
||||||
return ColumnArray::create(data);
|
return ColumnArray::create(data);
|
||||||
|
|
||||||
const ColumnNullable & nullable_elems = static_cast<const ColumnNullable &>(*data);
|
const ColumnNullable & nullable_elems = assert_cast<const ColumnNullable &>(*data);
|
||||||
|
|
||||||
auto array_of_nested = ColumnArray::create(nullable_elems.getNestedColumnPtr(), offsets);
|
auto array_of_nested = ColumnArray::create(nullable_elems.getNestedColumnPtr(), offsets);
|
||||||
auto filtered_array_of_nested_owner = array_of_nested->filter(filt, result_size_hint);
|
auto filtered_array_of_nested_owner = array_of_nested->filter(filt, result_size_hint);
|
||||||
auto & filtered_array_of_nested = static_cast<const ColumnArray &>(*filtered_array_of_nested_owner);
|
auto & filtered_array_of_nested = assert_cast<const ColumnArray &>(*filtered_array_of_nested_owner);
|
||||||
auto & filtered_offsets = filtered_array_of_nested.getOffsetsPtr();
|
auto & filtered_offsets = filtered_array_of_nested.getOffsetsPtr();
|
||||||
|
|
||||||
auto res_null_map = ColumnUInt8::create();
|
auto res_null_map = ColumnUInt8::create();
|
||||||
@ -577,7 +577,7 @@ ColumnPtr ColumnArray::filterTuple(const Filter & filt, ssize_t result_size_hint
|
|||||||
if (getOffsets().size() == 0)
|
if (getOffsets().size() == 0)
|
||||||
return ColumnArray::create(data);
|
return ColumnArray::create(data);
|
||||||
|
|
||||||
const ColumnTuple & tuple = static_cast<const ColumnTuple &>(*data);
|
const ColumnTuple & tuple = assert_cast<const ColumnTuple &>(*data);
|
||||||
|
|
||||||
/// Make temporary arrays for each components of Tuple, then filter and collect back.
|
/// Make temporary arrays for each components of Tuple, then filter and collect back.
|
||||||
|
|
||||||
@ -593,11 +593,11 @@ ColumnPtr ColumnArray::filterTuple(const Filter & filt, ssize_t result_size_hint
|
|||||||
|
|
||||||
Columns tuple_columns(tuple_size);
|
Columns tuple_columns(tuple_size);
|
||||||
for (size_t i = 0; i < tuple_size; ++i)
|
for (size_t i = 0; i < tuple_size; ++i)
|
||||||
tuple_columns[i] = static_cast<const ColumnArray &>(*temporary_arrays[i]).getDataPtr();
|
tuple_columns[i] = assert_cast<const ColumnArray &>(*temporary_arrays[i]).getDataPtr();
|
||||||
|
|
||||||
return ColumnArray::create(
|
return ColumnArray::create(
|
||||||
ColumnTuple::create(tuple_columns),
|
ColumnTuple::create(tuple_columns),
|
||||||
static_cast<const ColumnArray &>(*temporary_arrays.front()).getOffsetsPtr());
|
assert_cast<const ColumnArray &>(*temporary_arrays.front()).getOffsetsPtr());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -789,7 +789,7 @@ ColumnPtr ColumnArray::replicateString(const Offsets & replicate_offsets) const
|
|||||||
if (0 == col_size)
|
if (0 == col_size)
|
||||||
return res;
|
return res;
|
||||||
|
|
||||||
ColumnArray & res_ = static_cast<ColumnArray &>(*res);
|
ColumnArray & res_ = assert_cast<ColumnArray &>(*res);
|
||||||
|
|
||||||
const ColumnString & src_string = typeid_cast<const ColumnString &>(*data);
|
const ColumnString & src_string = typeid_cast<const ColumnString &>(*data);
|
||||||
const ColumnString::Chars & src_chars = src_string.getChars();
|
const ColumnString::Chars & src_chars = src_string.getChars();
|
||||||
@ -901,7 +901,7 @@ ColumnPtr ColumnArray::replicateGeneric(const Offsets & replicate_offsets) const
|
|||||||
throw Exception("Size of offsets doesn't match size of column.", ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH);
|
throw Exception("Size of offsets doesn't match size of column.", ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH);
|
||||||
|
|
||||||
MutableColumnPtr res = cloneEmpty();
|
MutableColumnPtr res = cloneEmpty();
|
||||||
ColumnArray & res_concrete = static_cast<ColumnArray &>(*res);
|
ColumnArray & res_concrete = assert_cast<ColumnArray &>(*res);
|
||||||
|
|
||||||
if (0 == col_size)
|
if (0 == col_size)
|
||||||
return res;
|
return res;
|
||||||
@ -922,7 +922,7 @@ ColumnPtr ColumnArray::replicateGeneric(const Offsets & replicate_offsets) const
|
|||||||
|
|
||||||
ColumnPtr ColumnArray::replicateNullable(const Offsets & replicate_offsets) const
|
ColumnPtr ColumnArray::replicateNullable(const Offsets & replicate_offsets) const
|
||||||
{
|
{
|
||||||
const ColumnNullable & nullable = static_cast<const ColumnNullable &>(*data);
|
const ColumnNullable & nullable = assert_cast<const ColumnNullable &>(*data);
|
||||||
|
|
||||||
/// Make temporary arrays for each components of Nullable. Then replicate them independently and collect back to result.
|
/// Make temporary arrays for each components of Nullable. Then replicate them independently and collect back to result.
|
||||||
/// NOTE Offsets are calculated twice and it is redundant.
|
/// NOTE Offsets are calculated twice and it is redundant.
|
||||||
@ -934,15 +934,15 @@ ColumnPtr ColumnArray::replicateNullable(const Offsets & replicate_offsets) cons
|
|||||||
|
|
||||||
return ColumnArray::create(
|
return ColumnArray::create(
|
||||||
ColumnNullable::create(
|
ColumnNullable::create(
|
||||||
static_cast<const ColumnArray &>(*array_of_nested).getDataPtr(),
|
assert_cast<const ColumnArray &>(*array_of_nested).getDataPtr(),
|
||||||
static_cast<const ColumnArray &>(*array_of_null_map).getDataPtr()),
|
assert_cast<const ColumnArray &>(*array_of_null_map).getDataPtr()),
|
||||||
static_cast<const ColumnArray &>(*array_of_nested).getOffsetsPtr());
|
assert_cast<const ColumnArray &>(*array_of_nested).getOffsetsPtr());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
ColumnPtr ColumnArray::replicateTuple(const Offsets & replicate_offsets) const
|
ColumnPtr ColumnArray::replicateTuple(const Offsets & replicate_offsets) const
|
||||||
{
|
{
|
||||||
const ColumnTuple & tuple = static_cast<const ColumnTuple &>(*data);
|
const ColumnTuple & tuple = assert_cast<const ColumnTuple &>(*data);
|
||||||
|
|
||||||
/// Make temporary arrays for each components of Tuple. In the same way as for Nullable.
|
/// Make temporary arrays for each components of Tuple. In the same way as for Nullable.
|
||||||
|
|
||||||
@ -958,11 +958,11 @@ ColumnPtr ColumnArray::replicateTuple(const Offsets & replicate_offsets) const
|
|||||||
|
|
||||||
Columns tuple_columns(tuple_size);
|
Columns tuple_columns(tuple_size);
|
||||||
for (size_t i = 0; i < tuple_size; ++i)
|
for (size_t i = 0; i < tuple_size; ++i)
|
||||||
tuple_columns[i] = static_cast<const ColumnArray &>(*temporary_arrays[i]).getDataPtr();
|
tuple_columns[i] = assert_cast<const ColumnArray &>(*temporary_arrays[i]).getDataPtr();
|
||||||
|
|
||||||
return ColumnArray::create(
|
return ColumnArray::create(
|
||||||
ColumnTuple::create(tuple_columns),
|
ColumnTuple::create(tuple_columns),
|
||||||
static_cast<const ColumnArray &>(*temporary_arrays.front()).getOffsetsPtr());
|
assert_cast<const ColumnArray &>(*temporary_arrays.front()).getOffsetsPtr());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -4,6 +4,7 @@
|
|||||||
#include <Columns/ColumnVector.h>
|
#include <Columns/ColumnVector.h>
|
||||||
#include <Core/Defines.h>
|
#include <Core/Defines.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -89,12 +90,12 @@ public:
|
|||||||
|
|
||||||
Offsets & ALWAYS_INLINE getOffsets()
|
Offsets & ALWAYS_INLINE getOffsets()
|
||||||
{
|
{
|
||||||
return static_cast<ColumnOffsets &>(*offsets).getData();
|
return assert_cast<ColumnOffsets &>(*offsets).getData();
|
||||||
}
|
}
|
||||||
|
|
||||||
const Offsets & ALWAYS_INLINE getOffsets() const
|
const Offsets & ALWAYS_INLINE getOffsets() const
|
||||||
{
|
{
|
||||||
return static_cast<const ColumnOffsets &>(*offsets).getData();
|
return assert_cast<const ColumnOffsets &>(*offsets).getData();
|
||||||
}
|
}
|
||||||
|
|
||||||
const ColumnPtr & getDataPtr() const { return data; }
|
const ColumnPtr & getDataPtr() const { return data; }
|
||||||
|
@ -4,6 +4,7 @@
|
|||||||
#include <Common/Exception.h>
|
#include <Common/Exception.h>
|
||||||
#include <Columns/IColumn.h>
|
#include <Columns/IColumn.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -175,7 +176,7 @@ public:
|
|||||||
|
|
||||||
int compareAt(size_t, size_t, const IColumn & rhs, int nan_direction_hint) const override
|
int compareAt(size_t, size_t, const IColumn & rhs, int nan_direction_hint) const override
|
||||||
{
|
{
|
||||||
return data->compareAt(0, 0, *static_cast<const ColumnConst &>(rhs).data, nan_direction_hint);
|
return data->compareAt(0, 0, *assert_cast<const ColumnConst &>(rhs).data, nan_direction_hint);
|
||||||
}
|
}
|
||||||
|
|
||||||
MutableColumns scatter(ColumnIndex num_columns, const Selector & selector) const override;
|
MutableColumns scatter(ColumnIndex num_columns, const Selector & selector) const override;
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
#include <Common/Exception.h>
|
#include <Common/Exception.h>
|
||||||
#include <Common/Arena.h>
|
#include <Common/Arena.h>
|
||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <common/unaligned.h>
|
#include <common/unaligned.h>
|
||||||
|
|
||||||
@ -133,7 +134,7 @@ void ColumnDecimal<T>::insertData(const char * src, size_t /*length*/)
|
|||||||
template <typename T>
|
template <typename T>
|
||||||
void ColumnDecimal<T>::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
void ColumnDecimal<T>::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
||||||
{
|
{
|
||||||
const ColumnDecimal & src_vec = static_cast<const ColumnDecimal &>(src);
|
const ColumnDecimal & src_vec = assert_cast<const ColumnDecimal &>(src);
|
||||||
|
|
||||||
if (start + length > src_vec.data.size())
|
if (start + length > src_vec.data.size())
|
||||||
throw Exception("Parameters start = " + toString(start) + ", length = " + toString(length) +
|
throw Exception("Parameters start = " + toString(start) + ", length = " + toString(length) +
|
||||||
|
@ -5,6 +5,7 @@
|
|||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
#include <Common/memcpySmall.h>
|
#include <Common/memcpySmall.h>
|
||||||
#include <Common/memcmpSmall.h>
|
#include <Common/memcmpSmall.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
#include <DataStreams/ColumnGathererStream.h>
|
#include <DataStreams/ColumnGathererStream.h>
|
||||||
|
|
||||||
@ -33,7 +34,7 @@ MutableColumnPtr ColumnFixedString::cloneResized(size_t size) const
|
|||||||
|
|
||||||
if (size > 0)
|
if (size > 0)
|
||||||
{
|
{
|
||||||
auto & new_col = static_cast<ColumnFixedString &>(*new_col_holder);
|
auto & new_col = assert_cast<ColumnFixedString &>(*new_col_holder);
|
||||||
new_col.chars.resize(size * n);
|
new_col.chars.resize(size * n);
|
||||||
|
|
||||||
size_t count = std::min(this->size(), size);
|
size_t count = std::min(this->size(), size);
|
||||||
@ -60,7 +61,7 @@ void ColumnFixedString::insert(const Field & x)
|
|||||||
|
|
||||||
void ColumnFixedString::insertFrom(const IColumn & src_, size_t index)
|
void ColumnFixedString::insertFrom(const IColumn & src_, size_t index)
|
||||||
{
|
{
|
||||||
const ColumnFixedString & src = static_cast<const ColumnFixedString &>(src_);
|
const ColumnFixedString & src = assert_cast<const ColumnFixedString &>(src_);
|
||||||
|
|
||||||
if (n != src.getN())
|
if (n != src.getN())
|
||||||
throw Exception("Size of FixedString doesn't match", ErrorCodes::SIZE_OF_FIXED_STRING_DOESNT_MATCH);
|
throw Exception("Size of FixedString doesn't match", ErrorCodes::SIZE_OF_FIXED_STRING_DOESNT_MATCH);
|
||||||
@ -140,7 +141,7 @@ void ColumnFixedString::getPermutation(bool reverse, size_t limit, int /*nan_dir
|
|||||||
|
|
||||||
void ColumnFixedString::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
void ColumnFixedString::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
||||||
{
|
{
|
||||||
const ColumnFixedString & src_concrete = static_cast<const ColumnFixedString &>(src);
|
const ColumnFixedString & src_concrete = assert_cast<const ColumnFixedString &>(src);
|
||||||
|
|
||||||
if (start + length > src_concrete.size())
|
if (start + length > src_concrete.size())
|
||||||
throw Exception("Parameters start = "
|
throw Exception("Parameters start = "
|
||||||
|
@ -3,6 +3,7 @@
|
|||||||
#include <Common/PODArray.h>
|
#include <Common/PODArray.h>
|
||||||
#include <Common/memcmpSmall.h>
|
#include <Common/memcmpSmall.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <Columns/IColumn.h>
|
#include <Columns/IColumn.h>
|
||||||
#include <Columns/ColumnVectorHelper.h>
|
#include <Columns/ColumnVectorHelper.h>
|
||||||
|
|
||||||
@ -102,7 +103,7 @@ public:
|
|||||||
|
|
||||||
int compareAt(size_t p1, size_t p2, const IColumn & rhs_, int /*nan_direction_hint*/) const override
|
int compareAt(size_t p1, size_t p2, const IColumn & rhs_, int /*nan_direction_hint*/) const override
|
||||||
{
|
{
|
||||||
const ColumnFixedString & rhs = static_cast<const ColumnFixedString &>(rhs_);
|
const ColumnFixedString & rhs = assert_cast<const ColumnFixedString &>(rhs_);
|
||||||
return memcmpSmallAllowOverflow15(chars.data() + p1 * n, rhs.chars.data() + p2 * n, n);
|
return memcmpSmallAllowOverflow15(chars.data() + p1 * n, rhs.chars.data() + p2 * n, n);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3,6 +3,8 @@
|
|||||||
#include <DataStreams/ColumnGathererStream.h>
|
#include <DataStreams/ColumnGathererStream.h>
|
||||||
#include <DataTypes/NumberTraits.h>
|
#include <DataTypes/NumberTraits.h>
|
||||||
#include <Common/HashTable/HashMap.h>
|
#include <Common/HashTable/HashMap.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
{
|
{
|
||||||
@ -244,7 +246,7 @@ MutableColumnPtr ColumnLowCardinality::cloneResized(size_t size) const
|
|||||||
|
|
||||||
int ColumnLowCardinality::compareAt(size_t n, size_t m, const IColumn & rhs, int nan_direction_hint) const
|
int ColumnLowCardinality::compareAt(size_t n, size_t m, const IColumn & rhs, int nan_direction_hint) const
|
||||||
{
|
{
|
||||||
const auto & low_cardinality_column = static_cast<const ColumnLowCardinality &>(rhs);
|
const auto & low_cardinality_column = assert_cast<const ColumnLowCardinality &>(rhs);
|
||||||
size_t n_index = getIndexes().getUInt(n);
|
size_t n_index = getIndexes().getUInt(n);
|
||||||
size_t m_index = low_cardinality_column.getIndexes().getUInt(m);
|
size_t m_index = low_cardinality_column.getIndexes().getUInt(m);
|
||||||
return getDictionary().compareAt(n_index, m_index, low_cardinality_column.getDictionary(), nan_direction_hint);
|
return getDictionary().compareAt(n_index, m_index, low_cardinality_column.getDictionary(), nan_direction_hint);
|
||||||
|
@ -2,6 +2,7 @@
|
|||||||
#include <Columns/IColumn.h>
|
#include <Columns/IColumn.h>
|
||||||
#include <Columns/IColumnUnique.h>
|
#include <Columns/IColumnUnique.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <AggregateFunctions/AggregateFunctionCount.h>
|
#include <AggregateFunctions/AggregateFunctionCount.h>
|
||||||
#include "ColumnsNumber.h"
|
#include "ColumnsNumber.h"
|
||||||
|
|
||||||
@ -166,10 +167,10 @@ public:
|
|||||||
|
|
||||||
switch (idx.getSizeOfIndexType())
|
switch (idx.getSizeOfIndexType())
|
||||||
{
|
{
|
||||||
case sizeof(UInt8): return static_cast<const ColumnUInt8 *>(indexes)->getElement(row);
|
case sizeof(UInt8): return assert_cast<const ColumnUInt8 *>(indexes)->getElement(row);
|
||||||
case sizeof(UInt16): return static_cast<const ColumnUInt16 *>(indexes)->getElement(row);
|
case sizeof(UInt16): return assert_cast<const ColumnUInt16 *>(indexes)->getElement(row);
|
||||||
case sizeof(UInt32): return static_cast<const ColumnUInt32 *>(indexes)->getElement(row);
|
case sizeof(UInt32): return assert_cast<const ColumnUInt32 *>(indexes)->getElement(row);
|
||||||
case sizeof(UInt64): return static_cast<const ColumnUInt64 *>(indexes)->getElement(row);
|
case sizeof(UInt64): return assert_cast<const ColumnUInt64 *>(indexes)->getElement(row);
|
||||||
default: throw Exception("Unexpected size of index type for low cardinality column.", ErrorCodes::LOGICAL_ERROR);
|
default: throw Exception("Unexpected size of index type for low cardinality column.", ErrorCodes::LOGICAL_ERROR);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -2,6 +2,7 @@
|
|||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
#include <Common/NaNUtils.h>
|
#include <Common/NaNUtils.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <Columns/ColumnNullable.h>
|
#include <Columns/ColumnNullable.h>
|
||||||
#include <Columns/ColumnConst.h>
|
#include <Columns/ColumnConst.h>
|
||||||
#include <DataStreams/ColumnGathererStream.h>
|
#include <DataStreams/ColumnGathererStream.h>
|
||||||
@ -129,7 +130,7 @@ const char * ColumnNullable::deserializeAndInsertFromArena(const char * pos)
|
|||||||
|
|
||||||
void ColumnNullable::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
void ColumnNullable::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
||||||
{
|
{
|
||||||
const ColumnNullable & nullable_col = static_cast<const ColumnNullable &>(src);
|
const ColumnNullable & nullable_col = assert_cast<const ColumnNullable &>(src);
|
||||||
getNullMapColumn().insertRangeFrom(*nullable_col.null_map, start, length);
|
getNullMapColumn().insertRangeFrom(*nullable_col.null_map, start, length);
|
||||||
getNestedColumn().insertRangeFrom(*nullable_col.nested_column, start, length);
|
getNestedColumn().insertRangeFrom(*nullable_col.nested_column, start, length);
|
||||||
}
|
}
|
||||||
@ -150,7 +151,7 @@ void ColumnNullable::insert(const Field & x)
|
|||||||
|
|
||||||
void ColumnNullable::insertFrom(const IColumn & src, size_t n)
|
void ColumnNullable::insertFrom(const IColumn & src, size_t n)
|
||||||
{
|
{
|
||||||
const ColumnNullable & src_concrete = static_cast<const ColumnNullable &>(src);
|
const ColumnNullable & src_concrete = assert_cast<const ColumnNullable &>(src);
|
||||||
getNestedColumn().insertFrom(src_concrete.getNestedColumn(), n);
|
getNestedColumn().insertFrom(src_concrete.getNestedColumn(), n);
|
||||||
getNullMapData().push_back(src_concrete.getNullMapData()[n]);
|
getNullMapData().push_back(src_concrete.getNullMapData()[n]);
|
||||||
}
|
}
|
||||||
@ -190,7 +191,7 @@ int ColumnNullable::compareAt(size_t n, size_t m, const IColumn & rhs_, int null
|
|||||||
/// the ordering specified by either NULLS FIRST or NULLS LAST in the
|
/// the ordering specified by either NULLS FIRST or NULLS LAST in the
|
||||||
/// ORDER BY construction.
|
/// ORDER BY construction.
|
||||||
|
|
||||||
const ColumnNullable & nullable_rhs = static_cast<const ColumnNullable &>(rhs_);
|
const ColumnNullable & nullable_rhs = assert_cast<const ColumnNullable &>(rhs_);
|
||||||
|
|
||||||
bool lval_is_null = isNullAt(n);
|
bool lval_is_null = isNullAt(n);
|
||||||
bool rval_is_null = nullable_rhs.isNullAt(m);
|
bool rval_is_null = nullable_rhs.isNullAt(m);
|
||||||
@ -458,7 +459,7 @@ ColumnPtr makeNullable(const ColumnPtr & column)
|
|||||||
return column;
|
return column;
|
||||||
|
|
||||||
if (isColumnConst(*column))
|
if (isColumnConst(*column))
|
||||||
return ColumnConst::create(makeNullable(static_cast<const ColumnConst &>(*column).getDataColumnPtr()), column->size());
|
return ColumnConst::create(makeNullable(assert_cast<const ColumnConst &>(*column).getDataColumnPtr()), column->size());
|
||||||
|
|
||||||
return ColumnNullable::create(column, ColumnUInt8::create(column->size(), 0));
|
return ColumnNullable::create(column, ColumnUInt8::create(column->size(), 0));
|
||||||
}
|
}
|
||||||
|
@ -3,6 +3,7 @@
|
|||||||
#include <Columns/IColumn.h>
|
#include <Columns/IColumn.h>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -45,7 +46,7 @@ public:
|
|||||||
std::string getName() const override { return "Nullable(" + nested_column->getName() + ")"; }
|
std::string getName() const override { return "Nullable(" + nested_column->getName() + ")"; }
|
||||||
MutableColumnPtr cloneResized(size_t size) const override;
|
MutableColumnPtr cloneResized(size_t size) const override;
|
||||||
size_t size() const override { return nested_column->size(); }
|
size_t size() const override { return nested_column->size(); }
|
||||||
bool isNullAt(size_t n) const override { return static_cast<const ColumnUInt8 &>(*null_map).getData()[n] != 0;}
|
bool isNullAt(size_t n) const override { return assert_cast<const ColumnUInt8 &>(*null_map).getData()[n] != 0;}
|
||||||
Field operator[](size_t n) const override;
|
Field operator[](size_t n) const override;
|
||||||
void get(size_t n, Field & res) const override;
|
void get(size_t n, Field & res) const override;
|
||||||
bool getBool(size_t n) const override { return isNullAt(n) ? 0 : nested_column->getBool(n); }
|
bool getBool(size_t n) const override { return isNullAt(n) ? 0 : nested_column->getBool(n); }
|
||||||
@ -116,8 +117,8 @@ public:
|
|||||||
/// Return the column that represents the byte map.
|
/// Return the column that represents the byte map.
|
||||||
const ColumnPtr & getNullMapColumnPtr() const { return null_map; }
|
const ColumnPtr & getNullMapColumnPtr() const { return null_map; }
|
||||||
|
|
||||||
ColumnUInt8 & getNullMapColumn() { return static_cast<ColumnUInt8 &>(*null_map); }
|
ColumnUInt8 & getNullMapColumn() { return assert_cast<ColumnUInt8 &>(*null_map); }
|
||||||
const ColumnUInt8 & getNullMapColumn() const { return static_cast<const ColumnUInt8 &>(*null_map); }
|
const ColumnUInt8 & getNullMapColumn() const { return assert_cast<const ColumnUInt8 &>(*null_map); }
|
||||||
|
|
||||||
NullMap & getNullMapData() { return getNullMapColumn().getData(); }
|
NullMap & getNullMapData() { return getNullMapColumn().getData(); }
|
||||||
const NullMap & getNullMapData() const { return getNullMapColumn().getData(); }
|
const NullMap & getNullMapData() const { return getNullMapColumn().getData(); }
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
#include <Core/Defines.h>
|
#include <Core/Defines.h>
|
||||||
#include <Common/Arena.h>
|
#include <Common/Arena.h>
|
||||||
#include <Common/memcmpSmall.h>
|
#include <Common/memcmpSmall.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <Columns/Collator.h>
|
#include <Columns/Collator.h>
|
||||||
#include <Columns/ColumnString.h>
|
#include <Columns/ColumnString.h>
|
||||||
#include <Columns/ColumnsCommon.h>
|
#include <Columns/ColumnsCommon.h>
|
||||||
@ -68,7 +69,7 @@ void ColumnString::insertRangeFrom(const IColumn & src, size_t start, size_t len
|
|||||||
if (length == 0)
|
if (length == 0)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
const ColumnString & src_concrete = static_cast<const ColumnString &>(src);
|
const ColumnString & src_concrete = assert_cast<const ColumnString &>(src);
|
||||||
|
|
||||||
if (start + length > src_concrete.offsets.size())
|
if (start + length > src_concrete.offsets.size())
|
||||||
throw Exception("Parameter out of bound in IColumnString::insertRangeFrom method.",
|
throw Exception("Parameter out of bound in IColumnString::insertRangeFrom method.",
|
||||||
@ -360,7 +361,7 @@ void ColumnString::getExtremes(Field & min, Field & max) const
|
|||||||
|
|
||||||
int ColumnString::compareAtWithCollation(size_t n, size_t m, const IColumn & rhs_, const Collator & collator) const
|
int ColumnString::compareAtWithCollation(size_t n, size_t m, const IColumn & rhs_, const Collator & collator) const
|
||||||
{
|
{
|
||||||
const ColumnString & rhs = static_cast<const ColumnString &>(rhs_);
|
const ColumnString & rhs = assert_cast<const ColumnString &>(rhs_);
|
||||||
|
|
||||||
return collator.compare(
|
return collator.compare(
|
||||||
reinterpret_cast<const char *>(&chars[offsetAt(n)]), sizeAt(n),
|
reinterpret_cast<const char *>(&chars[offsetAt(n)]), sizeAt(n),
|
||||||
|
@ -8,6 +8,7 @@
|
|||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
#include <Common/memcpySmall.h>
|
#include <Common/memcpySmall.h>
|
||||||
#include <Common/memcmpSmall.h>
|
#include <Common/memcmpSmall.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
class Collator;
|
class Collator;
|
||||||
@ -121,7 +122,7 @@ public:
|
|||||||
|
|
||||||
void insertFrom(const IColumn & src_, size_t n) override
|
void insertFrom(const IColumn & src_, size_t n) override
|
||||||
{
|
{
|
||||||
const ColumnString & src = static_cast<const ColumnString &>(src_);
|
const ColumnString & src = assert_cast<const ColumnString &>(src_);
|
||||||
const size_t size_to_append = src.offsets[n] - src.offsets[n - 1]; /// -1th index is Ok, see PaddedPODArray.
|
const size_t size_to_append = src.offsets[n] - src.offsets[n - 1]; /// -1th index is Ok, see PaddedPODArray.
|
||||||
|
|
||||||
if (size_to_append == 1)
|
if (size_to_append == 1)
|
||||||
@ -204,7 +205,7 @@ public:
|
|||||||
|
|
||||||
int compareAt(size_t n, size_t m, const IColumn & rhs_, int /*nan_direction_hint*/) const override
|
int compareAt(size_t n, size_t m, const IColumn & rhs_, int /*nan_direction_hint*/) const override
|
||||||
{
|
{
|
||||||
const ColumnString & rhs = static_cast<const ColumnString &>(rhs_);
|
const ColumnString & rhs = assert_cast<const ColumnString &>(rhs_);
|
||||||
return memcmpSmallAllowOverflow15(chars.data() + offsetAt(n), sizeAt(n) - 1, rhs.chars.data() + rhs.offsetAt(m), rhs.sizeAt(m) - 1);
|
return memcmpSmallAllowOverflow15(chars.data() + offsetAt(n), sizeAt(n) - 1, rhs.chars.data() + rhs.offsetAt(m), rhs.sizeAt(m) - 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -5,6 +5,7 @@
|
|||||||
#include <ext/map.h>
|
#include <ext/map.h>
|
||||||
#include <ext/range.h>
|
#include <ext/range.h>
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
|
|
||||||
|
|
||||||
namespace DB
|
namespace DB
|
||||||
@ -118,7 +119,7 @@ void ColumnTuple::insert(const Field & x)
|
|||||||
|
|
||||||
void ColumnTuple::insertFrom(const IColumn & src_, size_t n)
|
void ColumnTuple::insertFrom(const IColumn & src_, size_t n)
|
||||||
{
|
{
|
||||||
const ColumnTuple & src = static_cast<const ColumnTuple &>(src_);
|
const ColumnTuple & src = assert_cast<const ColumnTuple &>(src_);
|
||||||
|
|
||||||
const size_t tuple_size = columns.size();
|
const size_t tuple_size = columns.size();
|
||||||
if (src.columns.size() != tuple_size)
|
if (src.columns.size() != tuple_size)
|
||||||
@ -172,7 +173,7 @@ void ColumnTuple::insertRangeFrom(const IColumn & src, size_t start, size_t leng
|
|||||||
const size_t tuple_size = columns.size();
|
const size_t tuple_size = columns.size();
|
||||||
for (size_t i = 0; i < tuple_size; ++i)
|
for (size_t i = 0; i < tuple_size; ++i)
|
||||||
columns[i]->insertRangeFrom(
|
columns[i]->insertRangeFrom(
|
||||||
*static_cast<const ColumnTuple &>(src).columns[i],
|
*assert_cast<const ColumnTuple &>(src).columns[i],
|
||||||
start, length);
|
start, length);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -245,7 +246,7 @@ int ColumnTuple::compareAt(size_t n, size_t m, const IColumn & rhs, int nan_dire
|
|||||||
{
|
{
|
||||||
const size_t tuple_size = columns.size();
|
const size_t tuple_size = columns.size();
|
||||||
for (size_t i = 0; i < tuple_size; ++i)
|
for (size_t i = 0; i < tuple_size; ++i)
|
||||||
if (int res = columns[i]->compareAt(n, m, *static_cast<const ColumnTuple &>(rhs).columns[i], nan_direction_hint))
|
if (int res = columns[i]->compareAt(n, m, *assert_cast<const ColumnTuple &>(rhs).columns[i], nan_direction_hint))
|
||||||
return res;
|
return res;
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
|
@ -11,6 +11,7 @@
|
|||||||
#include <DataTypes/NumberTraits.h>
|
#include <DataTypes/NumberTraits.h>
|
||||||
|
|
||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <ext/range.h>
|
#include <ext/range.h>
|
||||||
|
|
||||||
#include <common/unaligned.h>
|
#include <common/unaligned.h>
|
||||||
@ -140,8 +141,8 @@ private:
|
|||||||
static size_t numSpecialValues(bool is_nullable) { return is_nullable ? 2 : 1; }
|
static size_t numSpecialValues(bool is_nullable) { return is_nullable ? 2 : 1; }
|
||||||
size_t numSpecialValues() const { return numSpecialValues(is_nullable); }
|
size_t numSpecialValues() const { return numSpecialValues(is_nullable); }
|
||||||
|
|
||||||
ColumnType * getRawColumnPtr() { return static_cast<ColumnType *>(column_holder.get()); }
|
ColumnType * getRawColumnPtr() { return assert_cast<ColumnType *>(column_holder.get()); }
|
||||||
const ColumnType * getRawColumnPtr() const { return static_cast<const ColumnType *>(column_holder.get()); }
|
const ColumnType * getRawColumnPtr() const { return assert_cast<const ColumnType *>(column_holder.get()); }
|
||||||
|
|
||||||
template <typename IndexType>
|
template <typename IndexType>
|
||||||
MutableColumnPtr uniqueInsertRangeImpl(
|
MutableColumnPtr uniqueInsertRangeImpl(
|
||||||
@ -232,7 +233,7 @@ void ColumnUnique<ColumnType>::updateNullMask()
|
|||||||
size_t size = getRawColumnPtr()->size();
|
size_t size = getRawColumnPtr()->size();
|
||||||
|
|
||||||
if (nested_null_mask->size() != size)
|
if (nested_null_mask->size() != size)
|
||||||
static_cast<ColumnUInt8 &>(*nested_null_mask).getData().resize_fill(size);
|
assert_cast<ColumnUInt8 &>(*nested_null_mask).getData().resize_fill(size);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -8,6 +8,7 @@
|
|||||||
#include <Common/SipHash.h>
|
#include <Common/SipHash.h>
|
||||||
#include <Common/NaNUtils.h>
|
#include <Common/NaNUtils.h>
|
||||||
#include <Common/RadixSort.h>
|
#include <Common/RadixSort.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <IO/WriteBuffer.h>
|
#include <IO/WriteBuffer.h>
|
||||||
#include <IO/WriteHelpers.h>
|
#include <IO/WriteHelpers.h>
|
||||||
#include <Columns/ColumnsCommon.h>
|
#include <Columns/ColumnsCommon.h>
|
||||||
@ -224,7 +225,7 @@ Float64 ColumnVector<T>::getFloat64(size_t n) const
|
|||||||
template <typename T>
|
template <typename T>
|
||||||
void ColumnVector<T>::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
void ColumnVector<T>::insertRangeFrom(const IColumn & src, size_t start, size_t length)
|
||||||
{
|
{
|
||||||
const ColumnVector & src_vec = static_cast<const ColumnVector &>(src);
|
const ColumnVector & src_vec = assert_cast<const ColumnVector &>(src);
|
||||||
|
|
||||||
if (start + length > src_vec.data.size())
|
if (start + length > src_vec.data.size())
|
||||||
throw Exception("Parameters start = "
|
throw Exception("Parameters start = "
|
||||||
|
@ -61,43 +61,18 @@ std::vector<size_t> countColumnsSizeInSelector(IColumn::ColumnIndex num_columns,
|
|||||||
return counts;
|
return counts;
|
||||||
}
|
}
|
||||||
|
|
||||||
/** clang 4 generates better code than gcc 6.
|
bool memoryIsByte(const void * data, size_t size, uint8_t byte)
|
||||||
* And both gcc and clang could not vectorize trivial loop by bytes automatically.
|
|
||||||
*/
|
|
||||||
bool memoryIsZero(const void * data, size_t size)
|
|
||||||
{
|
{
|
||||||
const Int8 * pos = reinterpret_cast<const Int8 *>(data);
|
if (size == 0)
|
||||||
const Int8 * end = pos + size;
|
|
||||||
|
|
||||||
#ifdef __SSE2__
|
|
||||||
const __m128 zero16 = _mm_setzero_ps();
|
|
||||||
const Int8 * end64 = pos + size / 64 * 64;
|
|
||||||
|
|
||||||
for (; pos < end64; pos += 64)
|
|
||||||
if (_mm_movemask_ps(_mm_cmpneq_ps(
|
|
||||||
_mm_loadu_ps(reinterpret_cast<const float *>(pos)),
|
|
||||||
zero16))
|
|
||||||
| _mm_movemask_ps(_mm_cmpneq_ps(
|
|
||||||
_mm_loadu_ps(reinterpret_cast<const float *>(pos + 16)),
|
|
||||||
zero16))
|
|
||||||
| _mm_movemask_ps(_mm_cmpneq_ps(
|
|
||||||
_mm_loadu_ps(reinterpret_cast<const float *>(pos + 32)),
|
|
||||||
zero16))
|
|
||||||
| _mm_movemask_ps(_mm_cmpneq_ps(
|
|
||||||
_mm_loadu_ps(reinterpret_cast<const float *>(pos + 48)),
|
|
||||||
zero16)))
|
|
||||||
return false;
|
|
||||||
|
|
||||||
/// TODO Add duff device for tail?
|
|
||||||
#endif
|
|
||||||
|
|
||||||
for (; pos < end; ++pos)
|
|
||||||
if (*pos)
|
|
||||||
return false;
|
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
|
auto ptr = reinterpret_cast<const uint8_t *>(data);
|
||||||
|
return *ptr == byte && memcmp(ptr, ptr + 1, size - 1) == 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool memoryIsZero(const void * data, size_t size)
|
||||||
|
{
|
||||||
|
return memoryIsByte(data, size, 0x0);
|
||||||
|
}
|
||||||
|
|
||||||
namespace ErrorCodes
|
namespace ErrorCodes
|
||||||
{
|
{
|
||||||
|
@ -22,7 +22,7 @@ std::vector<size_t> countColumnsSizeInSelector(IColumn::ColumnIndex num_columns,
|
|||||||
|
|
||||||
/// Returns true, if the memory contains only zeros.
|
/// Returns true, if the memory contains only zeros.
|
||||||
bool memoryIsZero(const void * data, size_t size);
|
bool memoryIsZero(const void * data, size_t size);
|
||||||
|
bool memoryIsByte(const void * data, size_t size, uint8_t byte);
|
||||||
|
|
||||||
/// The general implementation of `filter` function for ColumnArray and ColumnString.
|
/// The general implementation of `filter` function for ColumnArray and ColumnString.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
@ -1,4 +1,5 @@
|
|||||||
#include <Common/typeid_cast.h>
|
#include <Common/typeid_cast.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <Columns/FilterDescription.h>
|
#include <Columns/FilterDescription.h>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
#include <Columns/ColumnNullable.h>
|
#include <Columns/ColumnNullable.h>
|
||||||
@ -25,7 +26,7 @@ ConstantFilterDescription::ConstantFilterDescription(const IColumn & column)
|
|||||||
|
|
||||||
if (isColumnConst(column))
|
if (isColumnConst(column))
|
||||||
{
|
{
|
||||||
const ColumnConst & column_const = static_cast<const ColumnConst &>(column);
|
const ColumnConst & column_const = assert_cast<const ColumnConst &>(column);
|
||||||
ColumnPtr column_nested = column_const.getDataColumnPtr()->convertToFullColumnIfLowCardinality();
|
ColumnPtr column_nested = column_const.getDataColumnPtr()->convertToFullColumnIfLowCardinality();
|
||||||
|
|
||||||
if (!typeid_cast<const ColumnUInt8 *>(column_nested.get()))
|
if (!typeid_cast<const ColumnUInt8 *>(column_nested.get()))
|
||||||
|
@ -5,6 +5,7 @@
|
|||||||
#include <Common/HashTable/HashTableAllocator.h>
|
#include <Common/HashTable/HashTableAllocator.h>
|
||||||
#include <Columns/ColumnString.h>
|
#include <Columns/ColumnString.h>
|
||||||
#include <Columns/ColumnsNumber.h>
|
#include <Columns/ColumnsNumber.h>
|
||||||
|
#include <Common/assert_cast.h>
|
||||||
#include <ext/range.h>
|
#include <ext/range.h>
|
||||||
#include <common/unaligned.h>
|
#include <common/unaligned.h>
|
||||||
|
|
||||||
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user