Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(4)

Side by Side Diff: tools/win/supalink/supalink.cpp

Issue 10832161: Remove supalink (Closed) Base URL: http://git.chromium.org/git/chromium.git@trunk
Patch Set: Created 8 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « tools/win/supalink/install_supalink.py ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include <windows.h>
6
7 #include <stdio.h>
8 #include <stdlib.h>
9
10 #include <algorithm>
11 #include <iterator>
12 #include <string>
13
14 using namespace std;
15
16 // Don't use stderr for errors because VS has large buffers on them, leading
17 // to confusing error output.
18 static void Fatal(const wchar_t* msg) {
19 wprintf(L"supalink fatal error: %s\n", msg);
20 exit(1);
21 }
22
23 static wstring ErrorMessageToString(DWORD err) {
24 wchar_t* msg_buf = NULL;
25 DWORD rc = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
26 FORMAT_MESSAGE_FROM_SYSTEM,
27 NULL,
28 err,
29 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
30 reinterpret_cast<LPTSTR>(&msg_buf),
31 0,
32 NULL);
33 if (!rc)
34 return L"unknown error";
35 wstring ret(msg_buf);
36 LocalFree(msg_buf);
37 return ret;
38 }
39
40 static const wchar_t* const g_search_for[] = {
41 L"link.exe\" ",
42 L"link\" ",
43 L"link.exe ",
44 L"link ",
45 };
46
47 static void Fallback(const wchar_t* msg = NULL) {
48 if (msg) {
49 wprintf(L"supalink failed (%s), trying to fallback to standard link.\n",
50 msg);
51 wprintf(L"Original command line: %s\n", GetCommandLine());
52 fflush(stdout);
53 }
54
55 STARTUPINFO startup_info = { sizeof(STARTUPINFO) };
56 PROCESS_INFORMATION process_info;
57 DWORD exit_code;
58
59 GetStartupInfo(&startup_info);
60
61 wstring orig_cmd(GetCommandLine());
62 wstring cmd;
63 wstring replace_with = L"link.exe.supalink_orig.exe";
64 wstring orig_lowercase;
65
66 // Avoid searching for case-variations by making a lower-case copy.
67 transform(orig_cmd.begin(),
68 orig_cmd.end(),
69 back_inserter(orig_lowercase),
70 tolower);
71
72 for (size_t i = 0; i < ARRAYSIZE(g_search_for); ++i) {
73 wstring linkexe = g_search_for[i];
74 wstring::size_type at = orig_lowercase.find(linkexe, 0);
75 if (at == wstring::npos)
76 continue;
77 if (linkexe[linkexe.size() - 2] == L'"')
78 replace_with += L"\" ";
79 else
80 replace_with += L" ";
81 cmd = orig_cmd.replace(at, linkexe.size(), replace_with);
82 break;
83 }
84 if (cmd == L"") {
85 wprintf(L"Original run '%s'\n", orig_cmd.c_str());
86 Fatal(L"Couldn't find link.exe (or similar) in command line");
87 }
88
89 if (getenv("SUPALINK_DEBUG")) {
90 wprintf(L" running '%s'\n", cmd.c_str());
91 fflush(stdout);
92 }
93 if (!CreateProcess(NULL,
94 reinterpret_cast<LPWSTR>(const_cast<wchar_t *>(
95 cmd.c_str())),
96 NULL,
97 NULL,
98 TRUE,
99 0,
100 NULL,
101 NULL,
102 &startup_info, &process_info)) {
103 wstring error = ErrorMessageToString(GetLastError());
104 Fatal(error.c_str());
105 }
106 CloseHandle(process_info.hThread);
107 WaitForSingleObject(process_info.hProcess, INFINITE);
108 GetExitCodeProcess(process_info.hProcess, &exit_code);
109 CloseHandle(process_info.hProcess);
110 exit(exit_code);
111 }
112
113 wstring SlurpFile(const wchar_t* path) {
114 FILE* f = _wfopen(path, L"rb, ccs=UNICODE");
115 if (!f) Fallback(L"couldn't read file");
116 fseek(f, 0, SEEK_END);
117 long len = ftell(f);
118 rewind(f);
119 wchar_t* data = reinterpret_cast<wchar_t*>(malloc(len));
120 fread(data, 1, len, f);
121 fclose(f);
122 wstring ret(data, len/sizeof(wchar_t));
123 free(data);
124 return ret;
125 }
126
127 void DumpFile(const wchar_t* path, wstring& contents) {
128 FILE* f = _wfopen(path, L"wb, ccs=UTF-16LE");
129 if (!f) Fallback(L"couldn't write file");
130
131 fwrite(contents.c_str(), sizeof(wchar_t), contents.size(), f);
132 if (ferror(f))
133 Fatal(L"failed during response rewrite");
134 fclose(f);
135 }
136
137 // Input command line is assumed to be of the form:
138 //
139 // link.exe @C:\src\...\RSP00003045884740.rsp /NOLOGO /ERRORREPORT:PROMPT
140 //
141 // Specifically, we parse & hack the contents of argv[1] and pass the rest
142 // onwards.
143 int wmain(int argc, wchar_t** argv) {
144 ULONGLONG start_time = 0, end_time;
145
146 int rsp_file_index = -1;
147
148 if (argc < 2)
149 Fallback(L"too few commmand line args");
150
151 for (int i = 1; i < argc; ++i) {
152 if (argv[i][0] == L'@') {
153 rsp_file_index = i;
154 break;
155 }
156 }
157
158 if (rsp_file_index == -1)
159 Fallback(L"couldn't find a response file in argv");
160
161 if (_wgetenv(L"SUPALINK_DEBUG"))
162 start_time = GetTickCount64();
163
164 wstring rsp = SlurpFile(&argv[rsp_file_index][1]);
165
166 // The first line of this file is all we try to fix. It's a bunch of
167 // quoted space separated items. Simplest thing seems to be replacing " "
168 // with "\n". So, just slurp the file, replace, spit it out to the same
169 // file and continue on our way.
170
171 // Took about .5s when using the naive .replace loop to replace " " with
172 // "\r\n" so write the silly loop instead.
173 wstring fixed;
174 fixed.reserve(rsp.size() * 2);
175
176 for (const wchar_t* s = rsp.c_str(); *s;) {
177 if (*s == '"' && *(s + 1) == ' ' && *(s + 2) == '"') {
178 fixed += L"\"\r\n\"";
179 s += 3;
180 } else {
181 fixed += *s++;
182 }
183 }
184
185 DumpFile(&argv[rsp_file_index][1], fixed);
186
187 if (_wgetenv(L"SUPALINK_DEBUG")) {
188 wstring backup_copy(&argv[rsp_file_index][1]);
189 backup_copy += L".copy";
190 DumpFile(backup_copy.c_str(), fixed);
191
192 end_time = GetTickCount64();
193
194 wprintf(L" took %.2fs to modify @rsp file\n",
195 (end_time - start_time) / 1000.0);
196 }
197
198 Fallback();
199 }
OLDNEW
« no previous file with comments | « tools/win/supalink/install_supalink.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698