From 58c83eb379f72e7033b661ddf71f1e82ef00cd58 Mon Sep 17 00:00:00 2001 From: "Bill(Shuzhou) Liu" Date: Mon, 27 Mar 2023 09:27:38 -0500 Subject: [PATCH] Increase the max BDF ID length Increase the max length from 256 to 512. Change-Id: I3114f7ce6852aafa9dfec0186f27c1121c939c69 --- src/rocm_smi_main.cc | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/rocm_smi_main.cc b/src/rocm_smi_main.cc index 16cd336e2a..52c8227c53 100755 --- a/src/rocm_smi_main.cc +++ b/src/rocm_smi_main.cc @@ -259,16 +259,17 @@ static bool bdfid_from_path(const std::string in_name, uint64_t *bdfid) { // 1 = not a good bdfid found static uint32_t ConstructBDFID(std::string path, uint64_t *bdfid) { assert(bdfid != nullptr); - char tpath[256] = {'\0'}; + const unsigned int MAX_BDF_LENGTH = 512; + char tpath[MAX_BDF_LENGTH] = {'\0'}; ssize_t ret; - memset(tpath,0,256); + memset(tpath,0,MAX_BDF_LENGTH); - ret = readlink(path.c_str(), tpath, 256); + ret = readlink(path.c_str(), tpath, MAX_BDF_LENGTH); assert(ret > 0); - assert(ret < 256); + assert(ret < MAX_BDF_LENGTH); - if (ret <= 0 || ret >= 256) { + if (ret <= 0 || ret >= MAX_BDF_LENGTH) { return 1; }