fix bug

axiaobiao · Jun 20, 2018 · e2a9d02 · e2a9d02
1 parent cdd5bbe
commit e2a9d02
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/WebServer/HttpData.cpp b/WebServer/HttpData.cpp
@@ -674,8 +674,15 @@ AnalysisState HttpData::analysisRequest()
  handleError(fd_, 404, "Not Found!");
  return ANALYSIS_ERROR;
  }
- char *src_addr = static_cast<char*>(mmap(NULL, sbuf.st_size, PROT_READ, MAP_PRIVATE, src_fd, 0));
+ void *mmapRet = mmap(NULL, sbuf.st_size, PROT_READ, MAP_PRIVATE, src_fd, 0);
  close(src_fd);
+ if (mmapRet == (void *)-1)
+ {
+ outBuffer_.clear();
+ handleError(fd_, 404, "Not Found!");
+ return ANALYSIS_ERROR;
+ }
+ char *src_addr = static_cast<char*>(mmapRet);
  outBuffer_ += string(src_addr, src_addr + sbuf.st_size);;
  munmap(src_addr, sbuf.st_size);
  return ANALYSIS_SUCCESS;