fix 'download xilinx xclbin file on aws crash' issue (Xilinx#1749)

shuntarot · Jul 31, 2019 · 8654da1 · 8654da1
1 parent fbfb324
commit 8654da1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/runtime_src/driver/xclng/xrt/user_aws/shim.cpp b/src/runtime_src/driver/xclng/xrt/user_aws/shim.cpp
@@ -165,6 +165,10 @@ namespace awsbwhal {
           axlf *axlfbuffer = reinterpret_cast<axlf*>(const_cast<xclBin*> (buffer));
           fpga_mgmt_image_info orig_info;
           char* afi_id = get_afi_from_axlf(axlfbuffer);
+
+          if (!afi_id)
+              return -EINVAL;
+
           std::memset(&orig_info, 0, sizeof(struct fpga_mgmt_image_info));
           fpga_mgmt_describe_local_image(mBoardNumber, &orig_info, 0);