| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // mini_installer.exe is the first exe that is run when chrome is being | 5 // mini_installer.exe is the first exe that is run when chrome is being |
| 6 // installed or upgraded. It is designed to be extremely small (~5KB with no | 6 // installed or upgraded. It is designed to be extremely small (~5KB with no |
| 7 // extra resources linked) and it has two main jobs: | 7 // extra resources linked) and it has two main jobs: |
| 8 // 1) unpack the resources (possibly decompressing some) | 8 // 1) unpack the resources (possibly decompressing some) |
| 9 // 2) run the real installer (setup.exe) with appropriate flags. | 9 // 2) run the real installer (setup.exe) with appropriate flags. |
| 10 // | 10 // |
| (...skipping 287 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 298 reinterpret_cast<DWORD*>(exit_code))) { | 298 reinterpret_cast<DWORD*>(exit_code))) { |
| 299 ret = false; | 299 ret = false; |
| 300 } | 300 } |
| 301 } | 301 } |
| 302 | 302 |
| 303 ::CloseHandle(pi.hProcess); | 303 ::CloseHandle(pi.hProcess); |
| 304 | 304 |
| 305 return ret; | 305 return ret; |
| 306 } | 306 } |
| 307 | 307 |
| 308 // Append any command line params passed to mini_installer to the given buffer |
| 309 // so that they can be passed on to setup.exe. We do not return any error from |
| 310 // this method and simply skip making any changes in case of error. |
| 311 void AppendCommandLineFlags(CommandString* buffer) { |
| 312 PathString full_exe_path; |
| 313 size_t len = ::GetModuleFileName(NULL, full_exe_path.get(), |
| 314 full_exe_path.capacity()); |
| 315 if (!len || len >= full_exe_path.capacity()) |
| 316 return; |
| 317 |
| 318 const wchar_t* exe_name = GetNameFromPathExt(full_exe_path.get(), len); |
| 319 if (exe_name == NULL) |
| 320 return; |
| 321 |
| 322 int args_num; |
| 323 wchar_t* cmd_line = ::GetCommandLine(); |
| 324 wchar_t** args = ::CommandLineToArgvW(cmd_line, &args_num); |
| 325 if (args_num <= 0) |
| 326 return; |
| 327 |
| 328 const wchar_t* cmd_to_append = L""; |
| 329 if (!StrEndsWith(args[0], exe_name)) { |
| 330 // Current executable name not in the command line so just append |
| 331 // the whole command line. |
| 332 cmd_to_append = cmd_line; |
| 333 } else if (args_num > 1) { |
| 334 const wchar_t* tmp = SearchStringI(cmd_line, exe_name); |
| 335 tmp = SearchStringI(tmp, L" "); |
| 336 cmd_to_append = tmp; |
| 337 } |
| 338 |
| 339 buffer->append(cmd_to_append); |
| 340 |
| 341 LocalFree(args); |
| 342 } |
| 343 |
| 308 | 344 |
| 309 // Windows defined callback used in the EnumResourceNames call. For each | 345 // Windows defined callback used in the EnumResourceNames call. For each |
| 310 // matching resource found, the callback is invoked and at this point we write | 346 // matching resource found, the callback is invoked and at this point we write |
| 311 // it to disk. We expect resource names to start with 'chrome' or 'setup'. Any | 347 // it to disk. We expect resource names to start with 'chrome' or 'setup'. Any |
| 312 // other name is treated as an error. | 348 // other name is treated as an error. |
| 313 BOOL CALLBACK OnResourceFound(HMODULE module, const wchar_t* type, | 349 BOOL CALLBACK OnResourceFound(HMODULE module, const wchar_t* type, |
| 314 wchar_t* name, LONG_PTR context) { | 350 wchar_t* name, LONG_PTR context) { |
| 315 if (NULL == context) | 351 if (NULL == context) |
| 316 return FALSE; | 352 return FALSE; |
| 317 | 353 |
| (...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 388 !cmd_line.append(L"=\"") || | 424 !cmd_line.append(L"=\"") || |
| 389 !cmd_line.append(setup_path->get()) || | 425 !cmd_line.append(setup_path->get()) || |
| 390 !cmd_line.append(L"\"") || | 426 !cmd_line.append(L"\"") || |
| 391 !cmd_line.append(kCmdNewSetupExe) || | 427 !cmd_line.append(kCmdNewSetupExe) || |
| 392 !cmd_line.append(L"=\"") || | 428 !cmd_line.append(L"=\"") || |
| 393 !cmd_line.append(setup_dest_path.get()) || | 429 !cmd_line.append(setup_dest_path.get()) || |
| 394 !cmd_line.append(L"\"")) { | 430 !cmd_line.append(L"\"")) { |
| 395 success = false; | 431 success = false; |
| 396 } | 432 } |
| 397 | 433 |
| 434 // Get any command line option specified for mini_installer and pass them |
| 435 // on to setup.exe. This is important since switches such as |
| 436 // --multi-install and --chrome-frame affect where setup.exe will write |
| 437 // installer results for consumption by Google Update. |
| 438 AppendCommandLineFlags(&cmd_line); |
| 439 |
| 398 int exit_code = 0; | 440 int exit_code = 0; |
| 399 if (success && | 441 if (success && |
| 400 (!RunProcessAndWait(NULL, cmd_line.get(), &exit_code) || | 442 (!RunProcessAndWait(NULL, cmd_line.get(), &exit_code) || |
| 401 exit_code != ERROR_SUCCESS)) { | 443 exit_code != ERROR_SUCCESS)) { |
| 402 success = false; | 444 success = false; |
| 403 } | 445 } |
| 404 | 446 |
| 405 if (!success) | 447 if (!success) |
| 406 DeleteFile(setup_path->get()); | 448 DeleteFile(setup_path->get()); |
| 407 | 449 |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 448 setup_path->clear(); | 490 setup_path->clear(); |
| 449 } else if (!setup_path->assign(setup_dest_path.get())) { | 491 } else if (!setup_path->assign(setup_dest_path.get())) { |
| 450 ::DeleteFile(setup_dest_path.get()); | 492 ::DeleteFile(setup_dest_path.get()); |
| 451 } | 493 } |
| 452 } | 494 } |
| 453 } | 495 } |
| 454 | 496 |
| 455 return setup_path->length() > 0; | 497 return setup_path->length() > 0; |
| 456 } | 498 } |
| 457 | 499 |
| 458 // Append any command line params passed to mini_installer to the given buffer | |
| 459 // so that they can be passed on to setup.exe. We do not return any error from | |
| 460 // this method and simply skip making any changes in case of error. | |
| 461 void AppendCommandLineFlags(CommandString* buffer) { | |
| 462 PathString full_exe_path; | |
| 463 size_t len = ::GetModuleFileName(NULL, full_exe_path.get(), | |
| 464 full_exe_path.capacity()); | |
| 465 if (!len || len >= full_exe_path.capacity()) | |
| 466 return; | |
| 467 | |
| 468 const wchar_t* exe_name = GetNameFromPathExt(full_exe_path.get(), len); | |
| 469 if (exe_name == NULL) | |
| 470 return; | |
| 471 | |
| 472 int args_num; | |
| 473 wchar_t* cmd_line = ::GetCommandLine(); | |
| 474 wchar_t** args = ::CommandLineToArgvW(cmd_line, &args_num); | |
| 475 if (args_num <= 0) | |
| 476 return; | |
| 477 | |
| 478 const wchar_t* cmd_to_append = L""; | |
| 479 if (!StrEndsWith(args[0], exe_name)) { | |
| 480 // Current executable name not in the command line so just append | |
| 481 // the whole command line. | |
| 482 cmd_to_append = cmd_line; | |
| 483 } else if (args_num > 1) { | |
| 484 const wchar_t* tmp = SearchStringI(cmd_line, exe_name); | |
| 485 tmp = SearchStringI(tmp, L" "); | |
| 486 cmd_to_append = tmp; | |
| 487 } | |
| 488 | |
| 489 buffer->append(cmd_to_append); | |
| 490 | |
| 491 LocalFree(args); | |
| 492 } | |
| 493 | |
| 494 // Executes setup.exe, waits for it to finish and returns the exit code. | 500 // Executes setup.exe, waits for it to finish and returns the exit code. |
| 495 bool RunSetup(const wchar_t* archive_path, const wchar_t* setup_path, | 501 bool RunSetup(const wchar_t* archive_path, const wchar_t* setup_path, |
| 496 int* exit_code) { | 502 int* exit_code) { |
| 497 // There could be three full paths in the command line for setup.exe (path | 503 // There could be three full paths in the command line for setup.exe (path |
| 498 // to exe itself, path to archive and path to log file), so we declare | 504 // to exe itself, path to archive and path to log file), so we declare |
| 499 // total size as three + one additional to hold command line options. | 505 // total size as three + one additional to hold command line options. |
| 500 CommandString cmd_line; | 506 CommandString cmd_line; |
| 501 | 507 |
| 502 // Get the path to setup.exe first. | 508 // Get the path to setup.exe first. |
| 503 if (::lstrlen(setup_path) > 0) { | 509 if (::lstrlen(setup_path) > 0) { |
| (...skipping 339 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 843 case 1: | 849 case 1: |
| 844 dest8[count - 1] = c; | 850 dest8[count - 1] = c; |
| 845 } | 851 } |
| 846 | 852 |
| 847 while (adjcount-- > 0) // Copy the rest, 4 bytes/32 bits at a time | 853 while (adjcount-- > 0) // Copy the rest, 4 bytes/32 bits at a time |
| 848 *(dest32++) = fill; | 854 *(dest32++) = fill; |
| 849 | 855 |
| 850 return dest; | 856 return dest; |
| 851 } | 857 } |
| 852 } // extern "C" | 858 } // extern "C" |
| OLD | NEW |