Fix some memory leaks
This commit is contained in:
		
							parent
							
								
									28f22b4653
								
							
						
					
					
						commit
						f52b6c944e
					
				
					 3 changed files with 27 additions and 35 deletions
				
			
		|  | @ -415,18 +415,6 @@ static void commonChildInit(Pipe & logPipe) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| /* Convert a string list to an array of char pointers.  Careful: the
 |  | ||||||
|    string list should outlive the array. */ |  | ||||||
| const char * * strings2CharPtrs(const Strings & ss) |  | ||||||
| { |  | ||||||
|     const char * * arr = new const char * [ss.size() + 1]; |  | ||||||
|     const char * * p = arr; |  | ||||||
|     foreach (Strings::const_iterator, i, ss) *p++ = i->c_str(); |  | ||||||
|     *p = 0; |  | ||||||
|     return arr; |  | ||||||
| } |  | ||||||
| 
 |  | ||||||
| 
 |  | ||||||
| //////////////////////////////////////////////////////////////////////
 | //////////////////////////////////////////////////////////////////////
 | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
|  | @ -2124,11 +2112,7 @@ void DerivationGoal::runChild() | ||||||
|         Strings envStrs; |         Strings envStrs; | ||||||
|         foreach (Environment::const_iterator, i, env) |         foreach (Environment::const_iterator, i, env) | ||||||
|             envStrs.push_back(rewriteHashes(i->first + "=" + i->second, rewritesToTmp)); |             envStrs.push_back(rewriteHashes(i->first + "=" + i->second, rewritesToTmp)); | ||||||
|         const char * * envArr = strings2CharPtrs(envStrs); |         auto envArr = stringsToCharPtrs(envStrs); | ||||||
| 
 |  | ||||||
|         Path program = drv.builder.c_str(); |  | ||||||
|         std::vector<const char *> args; /* careful with c_str()! */ |  | ||||||
|         string user; /* must be here for its c_str()! */ |  | ||||||
| 
 | 
 | ||||||
|         /* If we are running in `build-users' mode, then switch to the
 |         /* If we are running in `build-users' mode, then switch to the
 | ||||||
|            user we allocated above.  Make sure that we drop all root |            user we allocated above.  Make sure that we drop all root | ||||||
|  | @ -2152,16 +2136,12 @@ void DerivationGoal::runChild() | ||||||
|         } |         } | ||||||
| 
 | 
 | ||||||
|         /* Fill in the arguments. */ |         /* Fill in the arguments. */ | ||||||
|  |         Strings args; | ||||||
|         string builderBasename = baseNameOf(drv.builder); |         string builderBasename = baseNameOf(drv.builder); | ||||||
|         args.push_back(builderBasename.c_str()); |         args.push_back(builderBasename); | ||||||
|         foreach (Strings::iterator, i, drv.args) { |         foreach (Strings::iterator, i, drv.args) | ||||||
|             auto re = rewriteHashes(*i, rewritesToTmp); |             args.push_back(rewriteHashes(*i, rewritesToTmp)); | ||||||
|             auto cstr = new char[re.length()+1]; |         auto argArr = stringsToCharPtrs(args); | ||||||
|             std::strcpy(cstr, re.c_str()); |  | ||||||
| 
 |  | ||||||
|             args.push_back(cstr); |  | ||||||
|         } |  | ||||||
|         args.push_back(0); |  | ||||||
| 
 | 
 | ||||||
|         restoreSIGPIPE(); |         restoreSIGPIPE(); | ||||||
| 
 | 
 | ||||||
|  | @ -2169,7 +2149,7 @@ void DerivationGoal::runChild() | ||||||
|         writeFull(STDERR_FILENO, "\n"); |         writeFull(STDERR_FILENO, "\n"); | ||||||
| 
 | 
 | ||||||
|         /* Execute the program.  This should not return. */ |         /* Execute the program.  This should not return. */ | ||||||
|         execve(program.c_str(), (char * *) &args[0], (char * *) envArr); |         execve(drv.builder.c_str(), (char * *) &argArr[0], (char * *) &envArr[0]); | ||||||
| 
 | 
 | ||||||
|         throw SysError(format("executing ‘%1%’") % drv.builder); |         throw SysError(format("executing ‘%1%’") % drv.builder); | ||||||
| 
 | 
 | ||||||
|  | @ -2797,7 +2777,7 @@ void SubstitutionGoal::tryToRun() | ||||||
|     args.push_back("--substitute"); |     args.push_back("--substitute"); | ||||||
|     args.push_back(storePath); |     args.push_back(storePath); | ||||||
|     args.push_back(destPath); |     args.push_back(destPath); | ||||||
|     const char * * argArr = strings2CharPtrs(args); |     auto argArr = stringsToCharPtrs(args); | ||||||
| 
 | 
 | ||||||
|     /* Fork the substitute program. */ |     /* Fork the substitute program. */ | ||||||
|     pid = startProcess([&]() { |     pid = startProcess([&]() { | ||||||
|  | @ -2807,7 +2787,7 @@ void SubstitutionGoal::tryToRun() | ||||||
|         if (dup2(outPipe.writeSide, STDOUT_FILENO) == -1) |         if (dup2(outPipe.writeSide, STDOUT_FILENO) == -1) | ||||||
|             throw SysError("cannot dup output pipe into stdout"); |             throw SysError("cannot dup output pipe into stdout"); | ||||||
| 
 | 
 | ||||||
|         execv(sub.c_str(), (char * *) argArr); |         execv(sub.c_str(), (char * *) &argArr[0]); | ||||||
| 
 | 
 | ||||||
|         throw SysError(format("executing ‘%1%’") % sub); |         throw SysError(format("executing ‘%1%’") % sub); | ||||||
|     }); |     }); | ||||||
|  |  | ||||||
|  | @ -912,16 +912,19 @@ pid_t startProcess(std::function<void()> fun, const ProcessOptions & options) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
|  | std::vector<const char *> stringsToCharPtrs(const Strings & ss) | ||||||
|  | { | ||||||
|  |     std::vector<const char *> res; | ||||||
|  |     for (auto & s : ss) res.push_back(s.c_str()); | ||||||
|  |     res.push_back(0); | ||||||
|  |     return res; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
| string runProgram(Path program, bool searchPath, const Strings & args) | string runProgram(Path program, bool searchPath, const Strings & args) | ||||||
| { | { | ||||||
|     checkInterrupt(); |     checkInterrupt(); | ||||||
| 
 | 
 | ||||||
|     std::vector<const char *> cargs; /* careful with c_str()! */ |  | ||||||
|     cargs.push_back(program.c_str()); |  | ||||||
|     for (Strings::const_iterator i = args.begin(); i != args.end(); ++i) |  | ||||||
|         cargs.push_back(i->c_str()); |  | ||||||
|     cargs.push_back(0); |  | ||||||
| 
 |  | ||||||
|     /* Create a pipe. */ |     /* Create a pipe. */ | ||||||
|     Pipe pipe; |     Pipe pipe; | ||||||
|     pipe.create(); |     pipe.create(); | ||||||
|  | @ -931,6 +934,10 @@ string runProgram(Path program, bool searchPath, const Strings & args) | ||||||
|         if (dup2(pipe.writeSide, STDOUT_FILENO) == -1) |         if (dup2(pipe.writeSide, STDOUT_FILENO) == -1) | ||||||
|             throw SysError("dupping stdout"); |             throw SysError("dupping stdout"); | ||||||
| 
 | 
 | ||||||
|  |         Strings args_(args); | ||||||
|  |         args_.push_front(program); | ||||||
|  |         auto cargs = stringsToCharPtrs(args_); | ||||||
|  | 
 | ||||||
|         if (searchPath) |         if (searchPath) | ||||||
|             execvp(program.c_str(), (char * *) &cargs[0]); |             execvp(program.c_str(), (char * *) &cargs[0]); | ||||||
|         else |         else | ||||||
|  |  | ||||||
|  | @ -289,6 +289,11 @@ string runProgram(Path program, bool searchPath = false, | ||||||
| 
 | 
 | ||||||
| MakeError(ExecError, Error) | MakeError(ExecError, Error) | ||||||
| 
 | 
 | ||||||
|  | /* Convert a list of strings to a null-terminated vector of char
 | ||||||
|  |    *'s. The result must not be accessed beyond the lifetime of the | ||||||
|  |    list of strings. */ | ||||||
|  | std::vector<const char *> stringsToCharPtrs(const Strings & ss); | ||||||
|  | 
 | ||||||
| /* Close all file descriptors except stdin, stdout, stderr, and those
 | /* Close all file descriptors except stdin, stdout, stderr, and those
 | ||||||
|    listed in the given set.  Good practice in child processes. */ |    listed in the given set.  Good practice in child processes. */ | ||||||
| void closeMostFDs(const set<int> & exceptions); | void closeMostFDs(const set<int> & exceptions); | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue