[Fuzzer] SetThreadName implementation for Windows

devnexen · devnexen · commit cf76ddcbeb10 · 2023-08-28T13:14:47.000+01:00
Api available since Windows Server 2016/Windows 10 1607 Reviewers: vitalybuka Reviewed-By: vitalybuka Differential Revison: https://reviews.llvm.org/D156317
diff --git a/compiler-rt/lib/fuzzer/FuzzerUtilWindows.cpp b/compiler-rt/lib/fuzzer/FuzzerUtilWindows.cpp
@@ -234,8 +234,13 @@ size_t PageSize() {
 }
 
 void SetThreadName(std::thread &thread, const std::string &name) {
-  // TODO ?
-  // to UTF-8 then SetThreadDescription ?
+  std::wstring wname;
+  auto sz = MultiByteToWideChar(CP_UTF8, 0, name.data(), name.size(), nullptr, 0);
+  wname.resize(sz);
+  if (sz == MultiByteToWideChar(CP_UTF8, 0, name.data, name.size, &wname[0], sz) > 0) {
+      wname.resize(sz - 1);
+      (void)SetThreadDescription(thread.native_handle(), wname.c_str());
+  }
 }
 
 } // namespace fuzzer

Original file line number	Diff line number	Diff line change
`@@ -234,8 +234,13 @@ size_t PageSize() {`
`234`	`234`	`}`
`235`	`235`
`236`	`236`	`void SetThreadName(std::thread &thread, const std::string &name) {`
`237`		`- // TODO ?`
`238`		`- // to UTF-8 then SetThreadDescription ?`
	`237`	`+ std::wstring wname;`
	`238`	`+ auto sz = MultiByteToWideChar(CP_UTF8, 0, name.data(), name.size(), nullptr, 0);`
	`239`	`+ wname.resize(sz);`
	`240`	`+ if (sz == MultiByteToWideChar(CP_UTF8, 0, name.data, name.size, &wname[0], sz) > 0) {`
	`241`	`+ wname.resize(sz - 1);`
	`242`	`+ (void)SetThreadDescription(thread.native_handle(), wname.c_str());`
	`243`	`+ }`
`239`	`244`	`}`
`240`	`245`
`241`	`246`	`} // namespace fuzzer`