xref: /aosp_15_r20/external/elfutils/debuginfod/debuginfod.cxx (revision 7304104da70ce23c86437a01be71edd1a2d7f37e)
1*7304104dSAndroid Build Coastguard Worker /* Debuginfo-over-http server.
2*7304104dSAndroid Build Coastguard Worker    Copyright (C) 2019-2024 Red Hat, Inc.
3*7304104dSAndroid Build Coastguard Worker    Copyright (C) 2021, 2022 Mark J. Wielaard <[email protected]>
4*7304104dSAndroid Build Coastguard Worker    This file is part of elfutils.
5*7304104dSAndroid Build Coastguard Worker 
6*7304104dSAndroid Build Coastguard Worker    This file is free software; you can redistribute it and/or modify
7*7304104dSAndroid Build Coastguard Worker    it under the terms of the GNU General Public License as published by
8*7304104dSAndroid Build Coastguard Worker    the Free Software Foundation; either version 3 of the License, or
9*7304104dSAndroid Build Coastguard Worker    (at your option) any later version.
10*7304104dSAndroid Build Coastguard Worker 
11*7304104dSAndroid Build Coastguard Worker    elfutils is distributed in the hope that it will be useful, but
12*7304104dSAndroid Build Coastguard Worker    WITHOUT ANY WARRANTY; without even the implied warranty of
13*7304104dSAndroid Build Coastguard Worker    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14*7304104dSAndroid Build Coastguard Worker    GNU General Public License for more details.
15*7304104dSAndroid Build Coastguard Worker 
16*7304104dSAndroid Build Coastguard Worker    You should have received a copy of the GNU General Public License
17*7304104dSAndroid Build Coastguard Worker    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
18*7304104dSAndroid Build Coastguard Worker 
19*7304104dSAndroid Build Coastguard Worker 
20*7304104dSAndroid Build Coastguard Worker /* cargo-cult from libdwfl linux-kernel-modules.c */
21*7304104dSAndroid Build Coastguard Worker /* In case we have a bad fts we include this before config.h because it
22*7304104dSAndroid Build Coastguard Worker    can't handle _FILE_OFFSET_BITS.
23*7304104dSAndroid Build Coastguard Worker    Everything we need here is fine if its declarations just come first.
24*7304104dSAndroid Build Coastguard Worker    Also, include sys/types.h before fts. On some systems fts.h is not self
25*7304104dSAndroid Build Coastguard Worker    contained. */
26*7304104dSAndroid Build Coastguard Worker #ifdef BAD_FTS
27*7304104dSAndroid Build Coastguard Worker   #include <sys/types.h>
28*7304104dSAndroid Build Coastguard Worker   #include <fts.h>
29*7304104dSAndroid Build Coastguard Worker #endif
30*7304104dSAndroid Build Coastguard Worker 
31*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_CONFIG_H
32*7304104dSAndroid Build Coastguard Worker   #include "config.h"
33*7304104dSAndroid Build Coastguard Worker #endif
34*7304104dSAndroid Build Coastguard Worker 
35*7304104dSAndroid Build Coastguard Worker // #define _GNU_SOURCE
36*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_SCHED_H
37*7304104dSAndroid Build Coastguard Worker extern "C" {
38*7304104dSAndroid Build Coastguard Worker #include <sched.h>
39*7304104dSAndroid Build Coastguard Worker }
40*7304104dSAndroid Build Coastguard Worker #endif
41*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_SYS_RESOURCE_H
42*7304104dSAndroid Build Coastguard Worker extern "C" {
43*7304104dSAndroid Build Coastguard Worker #include <sys/resource.h>
44*7304104dSAndroid Build Coastguard Worker }
45*7304104dSAndroid Build Coastguard Worker #endif
46*7304104dSAndroid Build Coastguard Worker 
47*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_EXECINFO_H
48*7304104dSAndroid Build Coastguard Worker extern "C" {
49*7304104dSAndroid Build Coastguard Worker #include <execinfo.h>
50*7304104dSAndroid Build Coastguard Worker }
51*7304104dSAndroid Build Coastguard Worker #endif
52*7304104dSAndroid Build Coastguard Worker 
53*7304104dSAndroid Build Coastguard Worker #include "debuginfod.h"
54*7304104dSAndroid Build Coastguard Worker #include <dwarf.h>
55*7304104dSAndroid Build Coastguard Worker 
56*7304104dSAndroid Build Coastguard Worker #include <argp.h>
57*7304104dSAndroid Build Coastguard Worker #ifdef __GNUC__
58*7304104dSAndroid Build Coastguard Worker #undef __attribute__ /* glibc bug - rhbz 1763325 */
59*7304104dSAndroid Build Coastguard Worker #endif
60*7304104dSAndroid Build Coastguard Worker 
61*7304104dSAndroid Build Coastguard Worker #include <unistd.h>
62*7304104dSAndroid Build Coastguard Worker #include <stdlib.h>
63*7304104dSAndroid Build Coastguard Worker #include <locale.h>
64*7304104dSAndroid Build Coastguard Worker #include <pthread.h>
65*7304104dSAndroid Build Coastguard Worker #include <signal.h>
66*7304104dSAndroid Build Coastguard Worker #include <sys/stat.h>
67*7304104dSAndroid Build Coastguard Worker #include <sys/time.h>
68*7304104dSAndroid Build Coastguard Worker #include <sys/vfs.h>
69*7304104dSAndroid Build Coastguard Worker #include <unistd.h>
70*7304104dSAndroid Build Coastguard Worker #include <fcntl.h>
71*7304104dSAndroid Build Coastguard Worker #include <netdb.h>
72*7304104dSAndroid Build Coastguard Worker #include <math.h>
73*7304104dSAndroid Build Coastguard Worker #include <float.h>
74*7304104dSAndroid Build Coastguard Worker 
75*7304104dSAndroid Build Coastguard Worker 
76*7304104dSAndroid Build Coastguard Worker /* If fts.h is included before config.h, its indirect inclusions may not
77*7304104dSAndroid Build Coastguard Worker    give us the right LFS aliases of these functions, so map them manually.  */
78*7304104dSAndroid Build Coastguard Worker #ifdef BAD_FTS
79*7304104dSAndroid Build Coastguard Worker   #ifdef _FILE_OFFSET_BITS
80*7304104dSAndroid Build Coastguard Worker     #define open open64
81*7304104dSAndroid Build Coastguard Worker     #define fopen fopen64
82*7304104dSAndroid Build Coastguard Worker   #endif
83*7304104dSAndroid Build Coastguard Worker #else
84*7304104dSAndroid Build Coastguard Worker   #include <sys/types.h>
85*7304104dSAndroid Build Coastguard Worker   #include <fts.h>
86*7304104dSAndroid Build Coastguard Worker #endif
87*7304104dSAndroid Build Coastguard Worker 
88*7304104dSAndroid Build Coastguard Worker #include <cstring>
89*7304104dSAndroid Build Coastguard Worker #include <vector>
90*7304104dSAndroid Build Coastguard Worker #include <set>
91*7304104dSAndroid Build Coastguard Worker #include <unordered_set>
92*7304104dSAndroid Build Coastguard Worker #include <map>
93*7304104dSAndroid Build Coastguard Worker #include <string>
94*7304104dSAndroid Build Coastguard Worker #include <iostream>
95*7304104dSAndroid Build Coastguard Worker #include <iomanip>
96*7304104dSAndroid Build Coastguard Worker #include <ostream>
97*7304104dSAndroid Build Coastguard Worker #include <sstream>
98*7304104dSAndroid Build Coastguard Worker #include <mutex>
99*7304104dSAndroid Build Coastguard Worker #include <deque>
100*7304104dSAndroid Build Coastguard Worker #include <condition_variable>
101*7304104dSAndroid Build Coastguard Worker #include <exception>
102*7304104dSAndroid Build Coastguard Worker #include <thread>
103*7304104dSAndroid Build Coastguard Worker // #include <regex> // on rhel7 gcc 4.8, not competent
104*7304104dSAndroid Build Coastguard Worker #include <regex.h>
105*7304104dSAndroid Build Coastguard Worker // #include <algorithm>
106*7304104dSAndroid Build Coastguard Worker using namespace std;
107*7304104dSAndroid Build Coastguard Worker 
108*7304104dSAndroid Build Coastguard Worker #include <gelf.h>
109*7304104dSAndroid Build Coastguard Worker #include <libdwelf.h>
110*7304104dSAndroid Build Coastguard Worker 
111*7304104dSAndroid Build Coastguard Worker #include <microhttpd.h>
112*7304104dSAndroid Build Coastguard Worker 
113*7304104dSAndroid Build Coastguard Worker #if MHD_VERSION >= 0x00097002
114*7304104dSAndroid Build Coastguard Worker // libmicrohttpd 0.9.71 broke API
115*7304104dSAndroid Build Coastguard Worker #define MHD_RESULT enum MHD_Result
116*7304104dSAndroid Build Coastguard Worker #else
117*7304104dSAndroid Build Coastguard Worker #define MHD_RESULT int
118*7304104dSAndroid Build Coastguard Worker #endif
119*7304104dSAndroid Build Coastguard Worker 
120*7304104dSAndroid Build Coastguard Worker #include <curl/curl.h>
121*7304104dSAndroid Build Coastguard Worker #include <archive.h>
122*7304104dSAndroid Build Coastguard Worker #include <archive_entry.h>
123*7304104dSAndroid Build Coastguard Worker #include <sqlite3.h>
124*7304104dSAndroid Build Coastguard Worker 
125*7304104dSAndroid Build Coastguard Worker #ifdef __linux__
126*7304104dSAndroid Build Coastguard Worker #include <sys/syscall.h>
127*7304104dSAndroid Build Coastguard Worker #endif
128*7304104dSAndroid Build Coastguard Worker 
129*7304104dSAndroid Build Coastguard Worker #ifdef __linux__
130*7304104dSAndroid Build Coastguard Worker #define tid() syscall(SYS_gettid)
131*7304104dSAndroid Build Coastguard Worker #else
132*7304104dSAndroid Build Coastguard Worker #define tid() pthread_self()
133*7304104dSAndroid Build Coastguard Worker #endif
134*7304104dSAndroid Build Coastguard Worker 
135*7304104dSAndroid Build Coastguard Worker extern "C" {
136*7304104dSAndroid Build Coastguard Worker #include "printversion.h"
137*7304104dSAndroid Build Coastguard Worker #include "system.h"
138*7304104dSAndroid Build Coastguard Worker }
139*7304104dSAndroid Build Coastguard Worker 
140*7304104dSAndroid Build Coastguard Worker 
141*7304104dSAndroid Build Coastguard Worker inline bool
string_endswith(const string & haystack,const string & needle)142*7304104dSAndroid Build Coastguard Worker string_endswith(const string& haystack, const string& needle)
143*7304104dSAndroid Build Coastguard Worker {
144*7304104dSAndroid Build Coastguard Worker   return (haystack.size() >= needle.size() &&
145*7304104dSAndroid Build Coastguard Worker 	  equal(haystack.end()-needle.size(), haystack.end(),
146*7304104dSAndroid Build Coastguard Worker                 needle.begin()));
147*7304104dSAndroid Build Coastguard Worker }
148*7304104dSAndroid Build Coastguard Worker 
149*7304104dSAndroid Build Coastguard Worker 
150*7304104dSAndroid Build Coastguard Worker // Roll this identifier for every sqlite schema incompatibility.
151*7304104dSAndroid Build Coastguard Worker #define BUILDIDS "buildids10"
152*7304104dSAndroid Build Coastguard Worker 
153*7304104dSAndroid Build Coastguard Worker #if SQLITE_VERSION_NUMBER >= 3008000
154*7304104dSAndroid Build Coastguard Worker #define WITHOUT_ROWID "without rowid"
155*7304104dSAndroid Build Coastguard Worker #else
156*7304104dSAndroid Build Coastguard Worker #define WITHOUT_ROWID ""
157*7304104dSAndroid Build Coastguard Worker #endif
158*7304104dSAndroid Build Coastguard Worker 
159*7304104dSAndroid Build Coastguard Worker static const char DEBUGINFOD_SQLITE_DDL[] =
160*7304104dSAndroid Build Coastguard Worker   "pragma foreign_keys = on;\n"
161*7304104dSAndroid Build Coastguard Worker   "pragma synchronous = 0;\n" // disable fsync()s - this cache is disposable across a machine crash
162*7304104dSAndroid Build Coastguard Worker   "pragma journal_mode = wal;\n" // https://sqlite.org/wal.html
163*7304104dSAndroid Build Coastguard Worker   "pragma wal_checkpoint = truncate;\n" // clean out any preexisting wal file
164*7304104dSAndroid Build Coastguard Worker   "pragma journal_size_limit = 0;\n" // limit steady state file (between grooming, which also =truncate's)
165*7304104dSAndroid Build Coastguard Worker   "pragma auto_vacuum = incremental;\n" // https://sqlite.org/pragma.html
166*7304104dSAndroid Build Coastguard Worker   "pragma busy_timeout = 1000;\n" // https://sqlite.org/pragma.html
167*7304104dSAndroid Build Coastguard Worker   // NB: all these are overridable with -D option
168*7304104dSAndroid Build Coastguard Worker 
169*7304104dSAndroid Build Coastguard Worker   // Normalization table for interning file names
170*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_fileparts (\n"
171*7304104dSAndroid Build Coastguard Worker   "        id integer primary key not null,\n"
172*7304104dSAndroid Build Coastguard Worker   "        name text unique not null\n"
173*7304104dSAndroid Build Coastguard Worker   "        );\n"
174*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_files (\n"
175*7304104dSAndroid Build Coastguard Worker   "        id integer primary key not null,\n"
176*7304104dSAndroid Build Coastguard Worker   "        dirname integer not null,\n"
177*7304104dSAndroid Build Coastguard Worker   "        basename integer not null,\n"
178*7304104dSAndroid Build Coastguard Worker   "        unique (dirname, basename),\n"
179*7304104dSAndroid Build Coastguard Worker   "        foreign key (dirname) references " BUILDIDS "_fileparts(id) on delete cascade,\n"
180*7304104dSAndroid Build Coastguard Worker   "        foreign key (basename) references " BUILDIDS "_fileparts(id) on delete cascade\n"
181*7304104dSAndroid Build Coastguard Worker   "        );\n"
182*7304104dSAndroid Build Coastguard Worker   "create view if not exists " BUILDIDS "_files_v as\n" // a
183*7304104dSAndroid Build Coastguard Worker   "        select f.id, n1.name || '/' || n2.name as name\n"
184*7304104dSAndroid Build Coastguard Worker   "        from " BUILDIDS "_files f, " BUILDIDS "_fileparts n1, " BUILDIDS "_fileparts n2\n"
185*7304104dSAndroid Build Coastguard Worker   "        where f.dirname = n1.id and f.basename = n2.id;\n"
186*7304104dSAndroid Build Coastguard Worker 
187*7304104dSAndroid Build Coastguard Worker   // Normalization table for interning buildids
188*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_buildids (\n"
189*7304104dSAndroid Build Coastguard Worker   "        id integer primary key not null,\n"
190*7304104dSAndroid Build Coastguard Worker   "        hex text unique not null);\n"
191*7304104dSAndroid Build Coastguard Worker   // Track the completion of scanning of a given file & sourcetype at given time
192*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_file_mtime_scanned (\n"
193*7304104dSAndroid Build Coastguard Worker   "        mtime integer not null,\n"
194*7304104dSAndroid Build Coastguard Worker   "        file integer not null,\n"
195*7304104dSAndroid Build Coastguard Worker   "        size integer not null,\n" // in bytes
196*7304104dSAndroid Build Coastguard Worker   "        sourcetype text(1) not null\n"
197*7304104dSAndroid Build Coastguard Worker   "            check (sourcetype IN ('F', 'R')),\n"
198*7304104dSAndroid Build Coastguard Worker   "        foreign key (file) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
199*7304104dSAndroid Build Coastguard Worker   "        primary key (file, mtime, sourcetype)\n"
200*7304104dSAndroid Build Coastguard Worker   "        ) " WITHOUT_ROWID ";\n"
201*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_f_de (\n"
202*7304104dSAndroid Build Coastguard Worker   "        buildid integer not null,\n"
203*7304104dSAndroid Build Coastguard Worker   "        debuginfo_p integer not null,\n"
204*7304104dSAndroid Build Coastguard Worker   "        executable_p integer not null,\n"
205*7304104dSAndroid Build Coastguard Worker   "        file integer not null,\n"
206*7304104dSAndroid Build Coastguard Worker   "        mtime integer not null,\n"
207*7304104dSAndroid Build Coastguard Worker   "        foreign key (file) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
208*7304104dSAndroid Build Coastguard Worker   "        foreign key (buildid) references " BUILDIDS "_buildids(id) on update cascade on delete cascade,\n"
209*7304104dSAndroid Build Coastguard Worker   "        primary key (buildid, file, mtime)\n"
210*7304104dSAndroid Build Coastguard Worker   "        ) " WITHOUT_ROWID ";\n"
211*7304104dSAndroid Build Coastguard Worker   // Index for faster delete by file identifier
212*7304104dSAndroid Build Coastguard Worker   "create index if not exists " BUILDIDS "_f_de_idx on " BUILDIDS "_f_de (file, mtime);\n"
213*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_f_s (\n"
214*7304104dSAndroid Build Coastguard Worker   "        buildid integer not null,\n"
215*7304104dSAndroid Build Coastguard Worker   "        artifactsrc integer not null,\n"
216*7304104dSAndroid Build Coastguard Worker   "        file integer not null,\n" // NB: not necessarily entered into _mtime_scanned
217*7304104dSAndroid Build Coastguard Worker   "        mtime integer not null,\n"
218*7304104dSAndroid Build Coastguard Worker   "        foreign key (file) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
219*7304104dSAndroid Build Coastguard Worker   "        foreign key (artifactsrc) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
220*7304104dSAndroid Build Coastguard Worker   "        foreign key (buildid) references " BUILDIDS "_buildids(id) on update cascade on delete cascade,\n"
221*7304104dSAndroid Build Coastguard Worker   "        primary key (buildid, artifactsrc, file, mtime)\n"
222*7304104dSAndroid Build Coastguard Worker   "        ) " WITHOUT_ROWID ";\n"
223*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_r_de (\n"
224*7304104dSAndroid Build Coastguard Worker   "        buildid integer not null,\n"
225*7304104dSAndroid Build Coastguard Worker   "        debuginfo_p integer not null,\n"
226*7304104dSAndroid Build Coastguard Worker   "        executable_p integer not null,\n"
227*7304104dSAndroid Build Coastguard Worker   "        file integer not null,\n"
228*7304104dSAndroid Build Coastguard Worker   "        mtime integer not null,\n"
229*7304104dSAndroid Build Coastguard Worker   "        content integer not null,\n"
230*7304104dSAndroid Build Coastguard Worker   "        foreign key (file) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
231*7304104dSAndroid Build Coastguard Worker   "        foreign key (content) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
232*7304104dSAndroid Build Coastguard Worker   "        foreign key (buildid) references " BUILDIDS "_buildids(id) on update cascade on delete cascade,\n"
233*7304104dSAndroid Build Coastguard Worker   "        primary key (buildid, debuginfo_p, executable_p, file, content, mtime)\n"
234*7304104dSAndroid Build Coastguard Worker   "        ) " WITHOUT_ROWID ";\n"
235*7304104dSAndroid Build Coastguard Worker   // Index for faster delete by archive file identifier
236*7304104dSAndroid Build Coastguard Worker   "create index if not exists " BUILDIDS "_r_de_idx on " BUILDIDS "_r_de (file, mtime);\n"
237*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_r_sref (\n" // outgoing dwarf sourcefile references from rpm
238*7304104dSAndroid Build Coastguard Worker   "        buildid integer not null,\n"
239*7304104dSAndroid Build Coastguard Worker   "        artifactsrc integer not null,\n"
240*7304104dSAndroid Build Coastguard Worker   "        foreign key (artifactsrc) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
241*7304104dSAndroid Build Coastguard Worker   "        foreign key (buildid) references " BUILDIDS "_buildids(id) on update cascade on delete cascade,\n"
242*7304104dSAndroid Build Coastguard Worker   "        primary key (buildid, artifactsrc)\n"
243*7304104dSAndroid Build Coastguard Worker   "        ) " WITHOUT_ROWID ";\n"
244*7304104dSAndroid Build Coastguard Worker   "create table if not exists " BUILDIDS "_r_sdef (\n" // rpm contents that may satisfy sref
245*7304104dSAndroid Build Coastguard Worker   "        file integer not null,\n"
246*7304104dSAndroid Build Coastguard Worker   "        mtime integer not null,\n"
247*7304104dSAndroid Build Coastguard Worker   "        content integer not null,\n"
248*7304104dSAndroid Build Coastguard Worker   "        foreign key (file) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
249*7304104dSAndroid Build Coastguard Worker   "        foreign key (content) references " BUILDIDS "_files(id) on update cascade on delete cascade,\n"
250*7304104dSAndroid Build Coastguard Worker   "        primary key (content, file, mtime)\n"
251*7304104dSAndroid Build Coastguard Worker   "        ) " WITHOUT_ROWID ";\n"
252*7304104dSAndroid Build Coastguard Worker   // create views to glue together some of the above tables, for webapi D queries
253*7304104dSAndroid Build Coastguard Worker   "create view if not exists " BUILDIDS "_query_d as \n"
254*7304104dSAndroid Build Coastguard Worker   "select\n"
255*7304104dSAndroid Build Coastguard Worker   "        b.hex as buildid, n.mtime, 'F' as sourcetype, f0.name as source0, n.mtime as mtime, null as source1\n"
256*7304104dSAndroid Build Coastguard Worker   "        from " BUILDIDS "_buildids b, " BUILDIDS "_files_v f0, " BUILDIDS "_f_de n\n"
257*7304104dSAndroid Build Coastguard Worker   "        where b.id = n.buildid and f0.id = n.file and n.debuginfo_p = 1\n"
258*7304104dSAndroid Build Coastguard Worker   "union all select\n"
259*7304104dSAndroid Build Coastguard Worker   "        b.hex as buildid, n.mtime, 'R' as sourcetype, f0.name as source0, n.mtime as mtime, f1.name as source1\n"
260*7304104dSAndroid Build Coastguard Worker   "        from " BUILDIDS "_buildids b, " BUILDIDS "_files_v f0, " BUILDIDS "_files_v f1, " BUILDIDS "_r_de n\n"
261*7304104dSAndroid Build Coastguard Worker   "        where b.id = n.buildid and f0.id = n.file and f1.id = n.content and n.debuginfo_p = 1\n"
262*7304104dSAndroid Build Coastguard Worker   ";"
263*7304104dSAndroid Build Coastguard Worker   // ... and for E queries
264*7304104dSAndroid Build Coastguard Worker   "create view if not exists " BUILDIDS "_query_e as \n"
265*7304104dSAndroid Build Coastguard Worker   "select\n"
266*7304104dSAndroid Build Coastguard Worker   "        b.hex as buildid, n.mtime, 'F' as sourcetype, f0.name as source0, n.mtime as mtime, null as source1\n"
267*7304104dSAndroid Build Coastguard Worker   "        from " BUILDIDS "_buildids b, " BUILDIDS "_files_v f0, " BUILDIDS "_f_de n\n"
268*7304104dSAndroid Build Coastguard Worker   "        where b.id = n.buildid and f0.id = n.file and n.executable_p = 1\n"
269*7304104dSAndroid Build Coastguard Worker   "union all select\n"
270*7304104dSAndroid Build Coastguard Worker   "        b.hex as buildid, n.mtime, 'R' as sourcetype, f0.name as source0, n.mtime as mtime, f1.name as source1\n"
271*7304104dSAndroid Build Coastguard Worker   "        from " BUILDIDS "_buildids b, " BUILDIDS "_files_v f0, " BUILDIDS "_files_v f1, " BUILDIDS "_r_de n\n"
272*7304104dSAndroid Build Coastguard Worker   "        where b.id = n.buildid and f0.id = n.file and f1.id = n.content and n.executable_p = 1\n"
273*7304104dSAndroid Build Coastguard Worker   ";"
274*7304104dSAndroid Build Coastguard Worker   // ... and for S queries
275*7304104dSAndroid Build Coastguard Worker   "create view if not exists " BUILDIDS "_query_s as \n"
276*7304104dSAndroid Build Coastguard Worker   "select\n"
277*7304104dSAndroid Build Coastguard Worker   "        b.hex as buildid, fs.name as artifactsrc, 'F' as sourcetype, f0.name as source0, n.mtime as mtime, null as source1, null as source0ref\n"
278*7304104dSAndroid Build Coastguard Worker   "        from " BUILDIDS "_buildids b, " BUILDIDS "_files_v f0, " BUILDIDS "_files_v fs, " BUILDIDS "_f_s n\n"
279*7304104dSAndroid Build Coastguard Worker   "        where b.id = n.buildid and f0.id = n.file and fs.id = n.artifactsrc\n"
280*7304104dSAndroid Build Coastguard Worker   "union all select\n"
281*7304104dSAndroid Build Coastguard Worker   "        b.hex as buildid, f1.name as artifactsrc, 'R' as sourcetype, f0.name as source0, sd.mtime as mtime, f1.name as source1, fsref.name as source0ref\n"
282*7304104dSAndroid Build Coastguard Worker   "        from " BUILDIDS "_buildids b, " BUILDIDS "_files_v f0, " BUILDIDS "_files_v f1, " BUILDIDS "_files_v fsref, "
283*7304104dSAndroid Build Coastguard Worker   "        " BUILDIDS "_r_sdef sd, " BUILDIDS "_r_sref sr, " BUILDIDS "_r_de sde\n"
284*7304104dSAndroid Build Coastguard Worker   "        where b.id = sr.buildid and f0.id = sd.file and fsref.id = sde.file and f1.id = sd.content\n"
285*7304104dSAndroid Build Coastguard Worker   "        and sr.artifactsrc = sd.content and sde.buildid = sr.buildid\n"
286*7304104dSAndroid Build Coastguard Worker   ";"
287*7304104dSAndroid Build Coastguard Worker   // and for startup overview counts
288*7304104dSAndroid Build Coastguard Worker   "drop view if exists " BUILDIDS "_stats;\n"
289*7304104dSAndroid Build Coastguard Worker   "create view if not exists " BUILDIDS "_stats as\n"
290*7304104dSAndroid Build Coastguard Worker   "          select 'file d/e' as label,count(*) as quantity from " BUILDIDS "_f_de\n"
291*7304104dSAndroid Build Coastguard Worker   "union all select 'file s',count(*) from " BUILDIDS "_f_s\n"
292*7304104dSAndroid Build Coastguard Worker   "union all select 'archive d/e',count(*) from " BUILDIDS "_r_de\n"
293*7304104dSAndroid Build Coastguard Worker   "union all select 'archive sref',count(*) from " BUILDIDS "_r_sref\n"
294*7304104dSAndroid Build Coastguard Worker   "union all select 'archive sdef',count(*) from " BUILDIDS "_r_sdef\n"
295*7304104dSAndroid Build Coastguard Worker   "union all select 'buildids',count(*) from " BUILDIDS "_buildids\n"
296*7304104dSAndroid Build Coastguard Worker   "union all select 'filenames',count(*) from " BUILDIDS "_files\n"
297*7304104dSAndroid Build Coastguard Worker   "union all select 'fileparts',count(*) from " BUILDIDS "_fileparts\n"
298*7304104dSAndroid Build Coastguard Worker   "union all select 'files scanned (#)',count(*) from " BUILDIDS "_file_mtime_scanned\n"
299*7304104dSAndroid Build Coastguard Worker   "union all select 'files scanned (mb)',coalesce(sum(size)/1024/1024,0) from " BUILDIDS "_file_mtime_scanned\n"
300*7304104dSAndroid Build Coastguard Worker #if SQLITE_VERSION_NUMBER >= 3016000
301*7304104dSAndroid Build Coastguard Worker   "union all select 'index db size (mb)',page_count*page_size/1024/1024 as size FROM pragma_page_count(), pragma_page_size()\n"
302*7304104dSAndroid Build Coastguard Worker #endif
303*7304104dSAndroid Build Coastguard Worker   ";\n"
304*7304104dSAndroid Build Coastguard Worker 
305*7304104dSAndroid Build Coastguard Worker // schema change history & garbage collection
306*7304104dSAndroid Build Coastguard Worker //
307*7304104dSAndroid Build Coastguard Worker // XXX: we could have migration queries here to bring prior-schema
308*7304104dSAndroid Build Coastguard Worker // data over instead of just dropping it.  But that could incur
309*7304104dSAndroid Build Coastguard Worker // doubled storage costs.
310*7304104dSAndroid Build Coastguard Worker //
311*7304104dSAndroid Build Coastguard Worker // buildids10: split the _files table into _parts
312*7304104dSAndroid Build Coastguard Worker   "" // <<< we are here
313*7304104dSAndroid Build Coastguard Worker // buildids9: widen the mtime_scanned table
314*7304104dSAndroid Build Coastguard Worker   "DROP VIEW IF EXISTS buildids9_stats;\n"
315*7304104dSAndroid Build Coastguard Worker   "DROP INDEX IF EXISTS buildids9_r_de_idx;\n"
316*7304104dSAndroid Build Coastguard Worker   "DROP INDEX IF EXISTS buildids9_f_de_idx;\n"
317*7304104dSAndroid Build Coastguard Worker   "DROP VIEW IF EXISTS buildids9_query_s;\n"
318*7304104dSAndroid Build Coastguard Worker   "DROP VIEW IF EXISTS buildids9_query_e;\n"
319*7304104dSAndroid Build Coastguard Worker   "DROP VIEW IF EXISTS buildids9_query_d;\n"
320*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_r_sdef;\n"
321*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_r_sref;\n"
322*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_r_de;\n"
323*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_f_s;\n"
324*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_f_de;\n"
325*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_file_mtime_scanned;\n"
326*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_buildids;\n"
327*7304104dSAndroid Build Coastguard Worker   "DROP TABLE IF EXISTS buildids9_files;\n"
328*7304104dSAndroid Build Coastguard Worker // buildids8: slim the sref table
329*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_f_de;\n"
330*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_f_s;\n"
331*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_r_de;\n"
332*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_r_sref;\n"
333*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_r_sdef;\n"
334*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_file_mtime_scanned;\n"
335*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_files;\n"
336*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids8_buildids;\n"
337*7304104dSAndroid Build Coastguard Worker // buildids7: separate _norm table into dense subtype tables
338*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_f_de;\n"
339*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_f_s;\n"
340*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_r_de;\n"
341*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_r_sref;\n"
342*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_r_sdef;\n"
343*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_file_mtime_scanned;\n"
344*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_files;\n"
345*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids7_buildids;\n"
346*7304104dSAndroid Build Coastguard Worker // buildids6: drop bolo/rfolo again, represent sources / rpmcontents in main table
347*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids6_norm;\n"
348*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids6_files;\n"
349*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids6_buildids;\n"
350*7304104dSAndroid Build Coastguard Worker   "drop view if exists buildids6;\n"
351*7304104dSAndroid Build Coastguard Worker // buildids5: redefine srcfile1 column to be '.'-less (for rpms)
352*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids5_norm;\n"
353*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids5_files;\n"
354*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids5_buildids;\n"
355*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids5_bolo;\n"
356*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids5_rfolo;\n"
357*7304104dSAndroid Build Coastguard Worker   "drop view if exists buildids5;\n"
358*7304104dSAndroid Build Coastguard Worker // buildids4: introduce rpmfile RFOLO
359*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids4_norm;\n"
360*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids4_files;\n"
361*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids4_buildids;\n"
362*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids4_bolo;\n"
363*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids4_rfolo;\n"
364*7304104dSAndroid Build Coastguard Worker   "drop view if exists buildids4;\n"
365*7304104dSAndroid Build Coastguard Worker // buildids3*: split out srcfile BOLO
366*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids3_norm;\n"
367*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids3_files;\n"
368*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids3_buildids;\n"
369*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids3_bolo;\n"
370*7304104dSAndroid Build Coastguard Worker   "drop view if exists buildids3;\n"
371*7304104dSAndroid Build Coastguard Worker // buildids2: normalized buildid and filenames into interning tables;
372*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids2_norm;\n"
373*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids2_files;\n"
374*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids2_buildids;\n"
375*7304104dSAndroid Build Coastguard Worker   "drop view if exists buildids2;\n"
376*7304104dSAndroid Build Coastguard Worker   // buildids1: made buildid and artifacttype NULLable, to represent cached-negative
377*7304104dSAndroid Build Coastguard Worker //           lookups from sources, e.g. files or rpms that contain no buildid-indexable content
378*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids1;\n"
379*7304104dSAndroid Build Coastguard Worker // buildids: original
380*7304104dSAndroid Build Coastguard Worker   "drop table if exists buildids;\n"
381*7304104dSAndroid Build Coastguard Worker   ;
382*7304104dSAndroid Build Coastguard Worker 
383*7304104dSAndroid Build Coastguard Worker static const char DEBUGINFOD_SQLITE_CLEANUP_DDL[] =
384*7304104dSAndroid Build Coastguard Worker   "pragma wal_checkpoint = truncate;\n" // clean out any preexisting wal file
385*7304104dSAndroid Build Coastguard Worker   ;
386*7304104dSAndroid Build Coastguard Worker 
387*7304104dSAndroid Build Coastguard Worker 
388*7304104dSAndroid Build Coastguard Worker 
389*7304104dSAndroid Build Coastguard Worker 
390*7304104dSAndroid Build Coastguard Worker /* Name and version of program.  */
391*7304104dSAndroid Build Coastguard Worker ARGP_PROGRAM_VERSION_HOOK_DEF = print_version;
392*7304104dSAndroid Build Coastguard Worker 
393*7304104dSAndroid Build Coastguard Worker /* Bug report address.  */
394*7304104dSAndroid Build Coastguard Worker ARGP_PROGRAM_BUG_ADDRESS_DEF = PACKAGE_BUGREPORT;
395*7304104dSAndroid Build Coastguard Worker 
396*7304104dSAndroid Build Coastguard Worker /* Definitions of arguments for argp functions.  */
397*7304104dSAndroid Build Coastguard Worker static const struct argp_option options[] =
398*7304104dSAndroid Build Coastguard Worker   {
399*7304104dSAndroid Build Coastguard Worker    { NULL, 0, NULL, 0, "Scanners:", 1 },
400*7304104dSAndroid Build Coastguard Worker    { "scan-file-dir", 'F', NULL, 0, "Enable ELF/DWARF file scanning.", 0 },
401*7304104dSAndroid Build Coastguard Worker    { "scan-rpm-dir", 'R', NULL, 0, "Enable RPM scanning.", 0 },
402*7304104dSAndroid Build Coastguard Worker    { "scan-deb-dir", 'U', NULL, 0, "Enable DEB scanning.", 0 },
403*7304104dSAndroid Build Coastguard Worker    { "scan-archive", 'Z', "EXT=CMD", 0, "Enable arbitrary archive scanning.", 0 },
404*7304104dSAndroid Build Coastguard Worker    // "source-oci-imageregistry"  ...
405*7304104dSAndroid Build Coastguard Worker 
406*7304104dSAndroid Build Coastguard Worker    { NULL, 0, NULL, 0, "Options:", 2 },
407*7304104dSAndroid Build Coastguard Worker    { "logical", 'L', NULL, 0, "Follow symlinks, default=ignore.", 0 },
408*7304104dSAndroid Build Coastguard Worker    { "rescan-time", 't', "SECONDS", 0, "Number of seconds to wait between rescans, 0=disable.", 0 },
409*7304104dSAndroid Build Coastguard Worker    { "groom-time", 'g', "SECONDS", 0, "Number of seconds to wait between database grooming, 0=disable.", 0 },
410*7304104dSAndroid Build Coastguard Worker    { "maxigroom", 'G', NULL, 0, "Run a complete database groom/shrink pass at startup.", 0 },
411*7304104dSAndroid Build Coastguard Worker    { "concurrency", 'c', "NUM", 0, "Limit scanning thread concurrency to NUM, default=#CPUs.", 0 },
412*7304104dSAndroid Build Coastguard Worker    { "connection-pool", 'C', "NUM", OPTION_ARG_OPTIONAL,
413*7304104dSAndroid Build Coastguard Worker      "Use webapi connection pool with NUM threads, default=unlim.", 0 },
414*7304104dSAndroid Build Coastguard Worker    { "include", 'I', "REGEX", 0, "Include files matching REGEX, default=all.", 0 },
415*7304104dSAndroid Build Coastguard Worker    { "exclude", 'X', "REGEX", 0, "Exclude files matching REGEX, default=none.", 0 },
416*7304104dSAndroid Build Coastguard Worker    { "port", 'p', "NUM", 0, "HTTP port to listen on, default 8002.", 0 },
417*7304104dSAndroid Build Coastguard Worker    { "database", 'd', "FILE", 0, "Path to sqlite database.", 0 },
418*7304104dSAndroid Build Coastguard Worker    { "ddl", 'D', "SQL", 0, "Apply extra sqlite ddl/pragma to connection.", 0 },
419*7304104dSAndroid Build Coastguard Worker    { "verbose", 'v', NULL, 0, "Increase verbosity.", 0 },
420*7304104dSAndroid Build Coastguard Worker    { "regex-groom", 'r', NULL, 0,"Uses regexes from -I and -X arguments to groom the database.",0},
421*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_FDCACHE_FDS 0x1001
422*7304104dSAndroid Build Coastguard Worker    { "fdcache-fds", ARGP_KEY_FDCACHE_FDS, "NUM", OPTION_HIDDEN, NULL, 0 },
423*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_FDCACHE_MBS 0x1002
424*7304104dSAndroid Build Coastguard Worker    { "fdcache-mbs", ARGP_KEY_FDCACHE_MBS, "MB", 0, "Maximum total size of archive file fdcache.", 0 },
425*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_FDCACHE_PREFETCH 0x1003
426*7304104dSAndroid Build Coastguard Worker    { "fdcache-prefetch", ARGP_KEY_FDCACHE_PREFETCH, "NUM", 0, "Number of archive files to prefetch into fdcache.", 0 },
427*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_FDCACHE_MINTMP 0x1004
428*7304104dSAndroid Build Coastguard Worker    { "fdcache-mintmp", ARGP_KEY_FDCACHE_MINTMP, "NUM", 0, "Minimum free space% on tmpdir.", 0 },
429*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_FDCACHE_PREFETCH_MBS 0x1005
430*7304104dSAndroid Build Coastguard Worker    { "fdcache-prefetch-mbs", ARGP_KEY_FDCACHE_PREFETCH_MBS, "MB", OPTION_HIDDEN, NULL, 0},
431*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_FDCACHE_PREFETCH_FDS 0x1006
432*7304104dSAndroid Build Coastguard Worker    { "fdcache-prefetch-fds", ARGP_KEY_FDCACHE_PREFETCH_FDS, "NUM", OPTION_HIDDEN, NULL, 0},
433*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_FORWARDED_TTL_LIMIT 0x1007
434*7304104dSAndroid Build Coastguard Worker    {"forwarded-ttl-limit", ARGP_KEY_FORWARDED_TTL_LIMIT, "NUM", 0, "Limit of X-Forwarded-For hops, default 8.", 0},
435*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_PASSIVE 0x1008
436*7304104dSAndroid Build Coastguard Worker    { "passive", ARGP_KEY_PASSIVE, NULL, 0, "Do not scan or groom, read-only database.", 0 },
437*7304104dSAndroid Build Coastguard Worker #define ARGP_KEY_DISABLE_SOURCE_SCAN 0x1009
438*7304104dSAndroid Build Coastguard Worker    { "disable-source-scan", ARGP_KEY_DISABLE_SOURCE_SCAN, NULL, 0, "Do not scan dwarf source info.", 0 },
439*7304104dSAndroid Build Coastguard Worker #define ARGP_SCAN_CHECKPOINT 0x100A
440*7304104dSAndroid Build Coastguard Worker    { "scan-checkpoint", ARGP_SCAN_CHECKPOINT, "NUM", 0, "Number of files scanned before a WAL checkpoint.", 0 },
441*7304104dSAndroid Build Coastguard Worker    { NULL, 0, NULL, 0, NULL, 0 },
442*7304104dSAndroid Build Coastguard Worker   };
443*7304104dSAndroid Build Coastguard Worker 
444*7304104dSAndroid Build Coastguard Worker /* Short description of program.  */
445*7304104dSAndroid Build Coastguard Worker static const char doc[] = "Serve debuginfo-related content across HTTP from files under PATHs.";
446*7304104dSAndroid Build Coastguard Worker 
447*7304104dSAndroid Build Coastguard Worker /* Strings for arguments in help texts.  */
448*7304104dSAndroid Build Coastguard Worker static const char args_doc[] = "[PATH ...]";
449*7304104dSAndroid Build Coastguard Worker 
450*7304104dSAndroid Build Coastguard Worker /* Prototype for option handler.  */
451*7304104dSAndroid Build Coastguard Worker static error_t parse_opt (int key, char *arg, struct argp_state *state);
452*7304104dSAndroid Build Coastguard Worker 
453*7304104dSAndroid Build Coastguard Worker static unsigned default_concurrency();
454*7304104dSAndroid Build Coastguard Worker 
455*7304104dSAndroid Build Coastguard Worker /* Data structure to communicate with argp functions.  */
456*7304104dSAndroid Build Coastguard Worker static struct argp argp =
457*7304104dSAndroid Build Coastguard Worker   {
458*7304104dSAndroid Build Coastguard Worker    options, parse_opt, args_doc, doc, NULL, NULL, NULL
459*7304104dSAndroid Build Coastguard Worker   };
460*7304104dSAndroid Build Coastguard Worker 
461*7304104dSAndroid Build Coastguard Worker 
462*7304104dSAndroid Build Coastguard Worker static string db_path;
463*7304104dSAndroid Build Coastguard Worker static sqlite3 *db;  // single connection, serialized across all our threads!
464*7304104dSAndroid Build Coastguard Worker static sqlite3 *dbq; // webapi query-servicing readonly connection, serialized ditto!
465*7304104dSAndroid Build Coastguard Worker static unsigned verbose;
466*7304104dSAndroid Build Coastguard Worker static volatile sig_atomic_t interrupted = 0;
467*7304104dSAndroid Build Coastguard Worker static volatile sig_atomic_t forced_rescan_count = 0;
468*7304104dSAndroid Build Coastguard Worker static volatile sig_atomic_t sigusr1 = 0;
469*7304104dSAndroid Build Coastguard Worker static volatile sig_atomic_t forced_groom_count = 0;
470*7304104dSAndroid Build Coastguard Worker static volatile sig_atomic_t sigusr2 = 0;
471*7304104dSAndroid Build Coastguard Worker static unsigned http_port = 8002;
472*7304104dSAndroid Build Coastguard Worker static unsigned rescan_s = 300;
473*7304104dSAndroid Build Coastguard Worker static unsigned groom_s = 86400;
474*7304104dSAndroid Build Coastguard Worker static bool maxigroom = false;
475*7304104dSAndroid Build Coastguard Worker static unsigned concurrency = default_concurrency();
476*7304104dSAndroid Build Coastguard Worker static int connection_pool = 0;
477*7304104dSAndroid Build Coastguard Worker static set<string> source_paths;
478*7304104dSAndroid Build Coastguard Worker static bool scan_files = false;
479*7304104dSAndroid Build Coastguard Worker static map<string,string> scan_archives;
480*7304104dSAndroid Build Coastguard Worker static vector<string> extra_ddl;
481*7304104dSAndroid Build Coastguard Worker static regex_t file_include_regex;
482*7304104dSAndroid Build Coastguard Worker static regex_t file_exclude_regex;
483*7304104dSAndroid Build Coastguard Worker static bool regex_groom = false;
484*7304104dSAndroid Build Coastguard Worker static bool traverse_logical;
485*7304104dSAndroid Build Coastguard Worker static long fdcache_mbs;
486*7304104dSAndroid Build Coastguard Worker static long fdcache_prefetch;
487*7304104dSAndroid Build Coastguard Worker static long fdcache_mintmp;
488*7304104dSAndroid Build Coastguard Worker static unsigned forwarded_ttl_limit = 8;
489*7304104dSAndroid Build Coastguard Worker static bool scan_source_info = true;
490*7304104dSAndroid Build Coastguard Worker static string tmpdir;
491*7304104dSAndroid Build Coastguard Worker static bool passive_p = false;
492*7304104dSAndroid Build Coastguard Worker static long scan_checkpoint = 256;
493*7304104dSAndroid Build Coastguard Worker 
494*7304104dSAndroid Build Coastguard Worker static void set_metric(const string& key, double value);
495*7304104dSAndroid Build Coastguard Worker static void inc_metric(const string& key);
496*7304104dSAndroid Build Coastguard Worker static void add_metric(const string& metric,
497*7304104dSAndroid Build Coastguard Worker                        double value);
498*7304104dSAndroid Build Coastguard Worker static void set_metric(const string& metric,
499*7304104dSAndroid Build Coastguard Worker                        const string& lname, const string& lvalue,
500*7304104dSAndroid Build Coastguard Worker                        double value);
501*7304104dSAndroid Build Coastguard Worker static void inc_metric(const string& metric,
502*7304104dSAndroid Build Coastguard Worker                        const string& lname, const string& lvalue);
503*7304104dSAndroid Build Coastguard Worker static void add_metric(const string& metric,
504*7304104dSAndroid Build Coastguard Worker                        const string& lname, const string& lvalue,
505*7304104dSAndroid Build Coastguard Worker                        double value);
506*7304104dSAndroid Build Coastguard Worker static void inc_metric(const string& metric,
507*7304104dSAndroid Build Coastguard Worker                        const string& lname, const string& lvalue,
508*7304104dSAndroid Build Coastguard Worker                        const string& rname, const string& rvalue);
509*7304104dSAndroid Build Coastguard Worker static void add_metric(const string& metric,
510*7304104dSAndroid Build Coastguard Worker                        const string& lname, const string& lvalue,
511*7304104dSAndroid Build Coastguard Worker                        const string& rname, const string& rvalue,
512*7304104dSAndroid Build Coastguard Worker                        double value);
513*7304104dSAndroid Build Coastguard Worker 
514*7304104dSAndroid Build Coastguard Worker 
515*7304104dSAndroid Build Coastguard Worker class tmp_inc_metric { // a RAII style wrapper for exception-safe scoped increment & decrement
516*7304104dSAndroid Build Coastguard Worker   string m, n, v;
517*7304104dSAndroid Build Coastguard Worker public:
tmp_inc_metric(const string & mname,const string & lname,const string & lvalue)518*7304104dSAndroid Build Coastguard Worker   tmp_inc_metric(const string& mname, const string& lname, const string& lvalue):
519*7304104dSAndroid Build Coastguard Worker     m(mname), n(lname), v(lvalue)
520*7304104dSAndroid Build Coastguard Worker   {
521*7304104dSAndroid Build Coastguard Worker     add_metric (m, n, v, 1);
522*7304104dSAndroid Build Coastguard Worker   }
~tmp_inc_metric()523*7304104dSAndroid Build Coastguard Worker   ~tmp_inc_metric()
524*7304104dSAndroid Build Coastguard Worker   {
525*7304104dSAndroid Build Coastguard Worker     add_metric (m, n, v, -1);
526*7304104dSAndroid Build Coastguard Worker   }
527*7304104dSAndroid Build Coastguard Worker };
528*7304104dSAndroid Build Coastguard Worker 
529*7304104dSAndroid Build Coastguard Worker class tmp_ms_metric { // a RAII style wrapper for exception-safe scoped timing
530*7304104dSAndroid Build Coastguard Worker   string m, n, v;
531*7304104dSAndroid Build Coastguard Worker   struct timespec ts_start;
532*7304104dSAndroid Build Coastguard Worker public:
tmp_ms_metric(const string & mname,const string & lname,const string & lvalue)533*7304104dSAndroid Build Coastguard Worker   tmp_ms_metric(const string& mname, const string& lname, const string& lvalue):
534*7304104dSAndroid Build Coastguard Worker     m(mname), n(lname), v(lvalue)
535*7304104dSAndroid Build Coastguard Worker   {
536*7304104dSAndroid Build Coastguard Worker     clock_gettime (CLOCK_MONOTONIC, & ts_start);
537*7304104dSAndroid Build Coastguard Worker   }
~tmp_ms_metric()538*7304104dSAndroid Build Coastguard Worker   ~tmp_ms_metric()
539*7304104dSAndroid Build Coastguard Worker   {
540*7304104dSAndroid Build Coastguard Worker     struct timespec ts_end;
541*7304104dSAndroid Build Coastguard Worker     clock_gettime (CLOCK_MONOTONIC, & ts_end);
542*7304104dSAndroid Build Coastguard Worker     double deltas = (ts_end.tv_sec - ts_start.tv_sec)
543*7304104dSAndroid Build Coastguard Worker       + (ts_end.tv_nsec - ts_start.tv_nsec)/1.e9;
544*7304104dSAndroid Build Coastguard Worker 
545*7304104dSAndroid Build Coastguard Worker     add_metric (m + "_milliseconds_sum", n, v, (deltas*1000.0));
546*7304104dSAndroid Build Coastguard Worker     inc_metric (m + "_milliseconds_count", n, v);
547*7304104dSAndroid Build Coastguard Worker   }
548*7304104dSAndroid Build Coastguard Worker };
549*7304104dSAndroid Build Coastguard Worker 
550*7304104dSAndroid Build Coastguard Worker 
551*7304104dSAndroid Build Coastguard Worker /* Handle program arguments.  */
552*7304104dSAndroid Build Coastguard Worker static error_t
parse_opt(int key,char * arg,struct argp_state * state)553*7304104dSAndroid Build Coastguard Worker parse_opt (int key, char *arg,
554*7304104dSAndroid Build Coastguard Worker 	   struct argp_state *state __attribute__ ((unused)))
555*7304104dSAndroid Build Coastguard Worker {
556*7304104dSAndroid Build Coastguard Worker   int rc;
557*7304104dSAndroid Build Coastguard Worker   switch (key)
558*7304104dSAndroid Build Coastguard Worker     {
559*7304104dSAndroid Build Coastguard Worker     case 'v': verbose ++; break;
560*7304104dSAndroid Build Coastguard Worker     case 'd':
561*7304104dSAndroid Build Coastguard Worker       /* When using the in-memory database make sure it is shareable,
562*7304104dSAndroid Build Coastguard Worker 	 so we can open it twice as read/write and read-only.  */
563*7304104dSAndroid Build Coastguard Worker       if (strcmp (arg, ":memory:") == 0)
564*7304104dSAndroid Build Coastguard Worker 	db_path = "file::memory:?cache=shared";
565*7304104dSAndroid Build Coastguard Worker       else
566*7304104dSAndroid Build Coastguard Worker 	db_path = string(arg);
567*7304104dSAndroid Build Coastguard Worker       break;
568*7304104dSAndroid Build Coastguard Worker     case 'p': http_port = (unsigned) atoi(arg);
569*7304104dSAndroid Build Coastguard Worker       if (http_port == 0 || http_port > 65535)
570*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "port number");
571*7304104dSAndroid Build Coastguard Worker       break;
572*7304104dSAndroid Build Coastguard Worker     case 'F': scan_files = true; break;
573*7304104dSAndroid Build Coastguard Worker     case 'R':
574*7304104dSAndroid Build Coastguard Worker       scan_archives[".rpm"]="cat"; // libarchive groks rpm natively
575*7304104dSAndroid Build Coastguard Worker       break;
576*7304104dSAndroid Build Coastguard Worker     case 'U':
577*7304104dSAndroid Build Coastguard Worker       scan_archives[".deb"]="(bsdtar -O -x -f - data.tar\\*)<";
578*7304104dSAndroid Build Coastguard Worker       scan_archives[".ddeb"]="(bsdtar -O -x -f - data.tar\\*)<";
579*7304104dSAndroid Build Coastguard Worker       scan_archives[".ipk"]="(bsdtar -O -x -f - data.tar\\*)<";
580*7304104dSAndroid Build Coastguard Worker       // .udeb too?
581*7304104dSAndroid Build Coastguard Worker       break;
582*7304104dSAndroid Build Coastguard Worker     case 'Z':
583*7304104dSAndroid Build Coastguard Worker       {
584*7304104dSAndroid Build Coastguard Worker         char* extension = strchr(arg, '=');
585*7304104dSAndroid Build Coastguard Worker         if (arg[0] == '\0')
586*7304104dSAndroid Build Coastguard Worker           argp_failure(state, 1, EINVAL, "missing EXT");
587*7304104dSAndroid Build Coastguard Worker         else if (extension)
588*7304104dSAndroid Build Coastguard Worker           scan_archives[string(arg, (extension-arg))]=string(extension+1);
589*7304104dSAndroid Build Coastguard Worker         else
590*7304104dSAndroid Build Coastguard Worker           scan_archives[string(arg)]=string("cat");
591*7304104dSAndroid Build Coastguard Worker       }
592*7304104dSAndroid Build Coastguard Worker       break;
593*7304104dSAndroid Build Coastguard Worker     case 'L':
594*7304104dSAndroid Build Coastguard Worker       if (passive_p)
595*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-L option inconsistent with passive mode");
596*7304104dSAndroid Build Coastguard Worker       traverse_logical = true;
597*7304104dSAndroid Build Coastguard Worker       break;
598*7304104dSAndroid Build Coastguard Worker     case 'D':
599*7304104dSAndroid Build Coastguard Worker       if (passive_p)
600*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-D option inconsistent with passive mode");
601*7304104dSAndroid Build Coastguard Worker       extra_ddl.push_back(string(arg));
602*7304104dSAndroid Build Coastguard Worker       break;
603*7304104dSAndroid Build Coastguard Worker     case 't':
604*7304104dSAndroid Build Coastguard Worker       if (passive_p)
605*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-t option inconsistent with passive mode");
606*7304104dSAndroid Build Coastguard Worker       rescan_s = (unsigned) atoi(arg);
607*7304104dSAndroid Build Coastguard Worker       break;
608*7304104dSAndroid Build Coastguard Worker     case 'g':
609*7304104dSAndroid Build Coastguard Worker       if (passive_p)
610*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-g option inconsistent with passive mode");
611*7304104dSAndroid Build Coastguard Worker       groom_s = (unsigned) atoi(arg);
612*7304104dSAndroid Build Coastguard Worker       break;
613*7304104dSAndroid Build Coastguard Worker     case 'G':
614*7304104dSAndroid Build Coastguard Worker       if (passive_p)
615*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-G option inconsistent with passive mode");
616*7304104dSAndroid Build Coastguard Worker       maxigroom = true;
617*7304104dSAndroid Build Coastguard Worker       break;
618*7304104dSAndroid Build Coastguard Worker     case 'c':
619*7304104dSAndroid Build Coastguard Worker       if (passive_p)
620*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-c option inconsistent with passive mode");
621*7304104dSAndroid Build Coastguard Worker       concurrency = (unsigned) atoi(arg);
622*7304104dSAndroid Build Coastguard Worker       if (concurrency < 1) concurrency = 1;
623*7304104dSAndroid Build Coastguard Worker       break;
624*7304104dSAndroid Build Coastguard Worker     case 'C':
625*7304104dSAndroid Build Coastguard Worker       if (arg)
626*7304104dSAndroid Build Coastguard Worker         {
627*7304104dSAndroid Build Coastguard Worker           connection_pool = atoi(arg);
628*7304104dSAndroid Build Coastguard Worker           if (connection_pool < 2)
629*7304104dSAndroid Build Coastguard Worker             argp_failure(state, 1, EINVAL, "-C NUM minimum 2");
630*7304104dSAndroid Build Coastguard Worker         }
631*7304104dSAndroid Build Coastguard Worker       break;
632*7304104dSAndroid Build Coastguard Worker     case 'I':
633*7304104dSAndroid Build Coastguard Worker       // NB: no problem with unconditional free here - an earlier failed regcomp would exit program
634*7304104dSAndroid Build Coastguard Worker       if (passive_p)
635*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-I option inconsistent with passive mode");
636*7304104dSAndroid Build Coastguard Worker       regfree (&file_include_regex);
637*7304104dSAndroid Build Coastguard Worker       rc = regcomp (&file_include_regex, arg, REG_EXTENDED|REG_NOSUB);
638*7304104dSAndroid Build Coastguard Worker       if (rc != 0)
639*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "regular expression");
640*7304104dSAndroid Build Coastguard Worker       break;
641*7304104dSAndroid Build Coastguard Worker     case 'X':
642*7304104dSAndroid Build Coastguard Worker       if (passive_p)
643*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-X option inconsistent with passive mode");
644*7304104dSAndroid Build Coastguard Worker       regfree (&file_exclude_regex);
645*7304104dSAndroid Build Coastguard Worker       rc = regcomp (&file_exclude_regex, arg, REG_EXTENDED|REG_NOSUB);
646*7304104dSAndroid Build Coastguard Worker       if (rc != 0)
647*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "regular expression");
648*7304104dSAndroid Build Coastguard Worker       break;
649*7304104dSAndroid Build Coastguard Worker     case 'r':
650*7304104dSAndroid Build Coastguard Worker       if (passive_p)
651*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "-r option inconsistent with passive mode");
652*7304104dSAndroid Build Coastguard Worker       regex_groom = true;
653*7304104dSAndroid Build Coastguard Worker       break;
654*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_FDCACHE_FDS:
655*7304104dSAndroid Build Coastguard Worker       // deprecated
656*7304104dSAndroid Build Coastguard Worker       break;
657*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_FDCACHE_MBS:
658*7304104dSAndroid Build Coastguard Worker       fdcache_mbs = atol (arg);
659*7304104dSAndroid Build Coastguard Worker       break;
660*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_FDCACHE_PREFETCH:
661*7304104dSAndroid Build Coastguard Worker       fdcache_prefetch = atol (arg);
662*7304104dSAndroid Build Coastguard Worker       break;
663*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_FDCACHE_MINTMP:
664*7304104dSAndroid Build Coastguard Worker       fdcache_mintmp = atol (arg);
665*7304104dSAndroid Build Coastguard Worker       if( fdcache_mintmp > 100 || fdcache_mintmp < 0 )
666*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "fdcache mintmp percent");
667*7304104dSAndroid Build Coastguard Worker       break;
668*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_FORWARDED_TTL_LIMIT:
669*7304104dSAndroid Build Coastguard Worker       forwarded_ttl_limit = (unsigned) atoi(arg);
670*7304104dSAndroid Build Coastguard Worker       break;
671*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_ARG:
672*7304104dSAndroid Build Coastguard Worker       source_paths.insert(string(arg));
673*7304104dSAndroid Build Coastguard Worker       break;
674*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_FDCACHE_PREFETCH_FDS:
675*7304104dSAndroid Build Coastguard Worker       // deprecated
676*7304104dSAndroid Build Coastguard Worker       break;
677*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_FDCACHE_PREFETCH_MBS:
678*7304104dSAndroid Build Coastguard Worker       // deprecated
679*7304104dSAndroid Build Coastguard Worker       break;
680*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_PASSIVE:
681*7304104dSAndroid Build Coastguard Worker       passive_p = true;
682*7304104dSAndroid Build Coastguard Worker       if (source_paths.size() > 0
683*7304104dSAndroid Build Coastguard Worker           || maxigroom
684*7304104dSAndroid Build Coastguard Worker           || extra_ddl.size() > 0
685*7304104dSAndroid Build Coastguard Worker           || traverse_logical)
686*7304104dSAndroid Build Coastguard Worker         // other conflicting options tricky to check
687*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "inconsistent options with passive mode");
688*7304104dSAndroid Build Coastguard Worker       break;
689*7304104dSAndroid Build Coastguard Worker     case ARGP_KEY_DISABLE_SOURCE_SCAN:
690*7304104dSAndroid Build Coastguard Worker       scan_source_info = false;
691*7304104dSAndroid Build Coastguard Worker       break;
692*7304104dSAndroid Build Coastguard Worker     case ARGP_SCAN_CHECKPOINT:
693*7304104dSAndroid Build Coastguard Worker       scan_checkpoint = atol (arg);
694*7304104dSAndroid Build Coastguard Worker       if (scan_checkpoint < 0)
695*7304104dSAndroid Build Coastguard Worker         argp_failure(state, 1, EINVAL, "scan checkpoint");
696*7304104dSAndroid Build Coastguard Worker       break;
697*7304104dSAndroid Build Coastguard Worker       // case 'h': argp_state_help (state, stderr, ARGP_HELP_LONG|ARGP_HELP_EXIT_OK);
698*7304104dSAndroid Build Coastguard Worker     default: return ARGP_ERR_UNKNOWN;
699*7304104dSAndroid Build Coastguard Worker     }
700*7304104dSAndroid Build Coastguard Worker 
701*7304104dSAndroid Build Coastguard Worker   return 0;
702*7304104dSAndroid Build Coastguard Worker }
703*7304104dSAndroid Build Coastguard Worker 
704*7304104dSAndroid Build Coastguard Worker 
705*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
706*7304104dSAndroid Build Coastguard Worker 
707*7304104dSAndroid Build Coastguard Worker 
708*7304104dSAndroid Build Coastguard Worker static void add_mhd_response_header (struct MHD_Response *r,
709*7304104dSAndroid Build Coastguard Worker 				     const char *h, const char *v);
710*7304104dSAndroid Build Coastguard Worker 
711*7304104dSAndroid Build Coastguard Worker // represent errors that may get reported to an ostream and/or a libmicrohttpd connection
712*7304104dSAndroid Build Coastguard Worker 
713*7304104dSAndroid Build Coastguard Worker struct reportable_exception
714*7304104dSAndroid Build Coastguard Worker {
715*7304104dSAndroid Build Coastguard Worker   int code;
716*7304104dSAndroid Build Coastguard Worker   string message;
717*7304104dSAndroid Build Coastguard Worker 
reportable_exceptionreportable_exception718*7304104dSAndroid Build Coastguard Worker   reportable_exception(int c, const string& m): code(c), message(m) {}
reportable_exceptionreportable_exception719*7304104dSAndroid Build Coastguard Worker   reportable_exception(const string& m): code(503), message(m) {}
reportable_exceptionreportable_exception720*7304104dSAndroid Build Coastguard Worker   reportable_exception(): code(503), message() {}
721*7304104dSAndroid Build Coastguard Worker 
722*7304104dSAndroid Build Coastguard Worker   void report(ostream& o) const; // defined under obatched() class below
723*7304104dSAndroid Build Coastguard Worker 
mhd_send_responsereportable_exception724*7304104dSAndroid Build Coastguard Worker   MHD_RESULT mhd_send_response(MHD_Connection* c) const {
725*7304104dSAndroid Build Coastguard Worker     MHD_Response* r = MHD_create_response_from_buffer (message.size(),
726*7304104dSAndroid Build Coastguard Worker                                                        (void*) message.c_str(),
727*7304104dSAndroid Build Coastguard Worker                                                        MHD_RESPMEM_MUST_COPY);
728*7304104dSAndroid Build Coastguard Worker     add_mhd_response_header (r, "Content-Type", "text/plain");
729*7304104dSAndroid Build Coastguard Worker     MHD_RESULT rc = MHD_queue_response (c, code, r);
730*7304104dSAndroid Build Coastguard Worker     MHD_destroy_response (r);
731*7304104dSAndroid Build Coastguard Worker     return rc;
732*7304104dSAndroid Build Coastguard Worker   }
733*7304104dSAndroid Build Coastguard Worker };
734*7304104dSAndroid Build Coastguard Worker 
735*7304104dSAndroid Build Coastguard Worker 
736*7304104dSAndroid Build Coastguard Worker struct sqlite_exception: public reportable_exception
737*7304104dSAndroid Build Coastguard Worker {
sqlite_exceptionsqlite_exception738*7304104dSAndroid Build Coastguard Worker   sqlite_exception(int rc, const string& msg):
739*7304104dSAndroid Build Coastguard Worker     reportable_exception(string("sqlite3 error: ") + msg + ": " + string(sqlite3_errstr(rc) ?: "?")) {
740*7304104dSAndroid Build Coastguard Worker     inc_metric("error_count","sqlite3",sqlite3_errstr(rc));
741*7304104dSAndroid Build Coastguard Worker   }
742*7304104dSAndroid Build Coastguard Worker };
743*7304104dSAndroid Build Coastguard Worker 
744*7304104dSAndroid Build Coastguard Worker struct libc_exception: public reportable_exception
745*7304104dSAndroid Build Coastguard Worker {
libc_exceptionlibc_exception746*7304104dSAndroid Build Coastguard Worker   libc_exception(int rc, const string& msg):
747*7304104dSAndroid Build Coastguard Worker     reportable_exception(string("libc error: ") + msg + ": " + string(strerror(rc) ?: "?")) {
748*7304104dSAndroid Build Coastguard Worker     inc_metric("error_count","libc",strerror(rc));
749*7304104dSAndroid Build Coastguard Worker   }
750*7304104dSAndroid Build Coastguard Worker };
751*7304104dSAndroid Build Coastguard Worker 
752*7304104dSAndroid Build Coastguard Worker 
753*7304104dSAndroid Build Coastguard Worker struct archive_exception: public reportable_exception
754*7304104dSAndroid Build Coastguard Worker {
archive_exceptionarchive_exception755*7304104dSAndroid Build Coastguard Worker   archive_exception(const string& msg):
756*7304104dSAndroid Build Coastguard Worker     reportable_exception(string("libarchive error: ") + msg) {
757*7304104dSAndroid Build Coastguard Worker       inc_metric("error_count","libarchive",msg);
758*7304104dSAndroid Build Coastguard Worker   }
archive_exceptionarchive_exception759*7304104dSAndroid Build Coastguard Worker   archive_exception(struct archive* a, const string& msg):
760*7304104dSAndroid Build Coastguard Worker     reportable_exception(string("libarchive error: ") + msg + ": " + string(archive_error_string(a) ?: "?")) {
761*7304104dSAndroid Build Coastguard Worker     inc_metric("error_count","libarchive",msg + ": " + string(archive_error_string(a) ?: "?"));
762*7304104dSAndroid Build Coastguard Worker   }
763*7304104dSAndroid Build Coastguard Worker };
764*7304104dSAndroid Build Coastguard Worker 
765*7304104dSAndroid Build Coastguard Worker 
766*7304104dSAndroid Build Coastguard Worker struct elfutils_exception: public reportable_exception
767*7304104dSAndroid Build Coastguard Worker {
elfutils_exceptionelfutils_exception768*7304104dSAndroid Build Coastguard Worker   elfutils_exception(int rc, const string& msg):
769*7304104dSAndroid Build Coastguard Worker     reportable_exception(string("elfutils error: ") + msg + ": " + string(elf_errmsg(rc) ?: "?")) {
770*7304104dSAndroid Build Coastguard Worker     inc_metric("error_count","elfutils",elf_errmsg(rc));
771*7304104dSAndroid Build Coastguard Worker   }
772*7304104dSAndroid Build Coastguard Worker };
773*7304104dSAndroid Build Coastguard Worker 
774*7304104dSAndroid Build Coastguard Worker 
775*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
776*7304104dSAndroid Build Coastguard Worker 
777*7304104dSAndroid Build Coastguard Worker template <typename Payload>
778*7304104dSAndroid Build Coastguard Worker class workq
779*7304104dSAndroid Build Coastguard Worker {
780*7304104dSAndroid Build Coastguard Worker   unordered_set<Payload> q; // eliminate duplicates
781*7304104dSAndroid Build Coastguard Worker   mutex mtx;
782*7304104dSAndroid Build Coastguard Worker   condition_variable cv;
783*7304104dSAndroid Build Coastguard Worker   bool dead;
784*7304104dSAndroid Build Coastguard Worker   unsigned idlers;   // number of threads busy with wait_idle / done_idle
785*7304104dSAndroid Build Coastguard Worker   unsigned fronters; // number of threads busy with wait_front / done_front
786*7304104dSAndroid Build Coastguard Worker 
787*7304104dSAndroid Build Coastguard Worker public:
workq()788*7304104dSAndroid Build Coastguard Worker   workq() { dead = false; idlers = 0; fronters = 0; }
~workq()789*7304104dSAndroid Build Coastguard Worker   ~workq() {}
790*7304104dSAndroid Build Coastguard Worker 
push_back(const Payload & p)791*7304104dSAndroid Build Coastguard Worker   void push_back(const Payload& p)
792*7304104dSAndroid Build Coastguard Worker   {
793*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
794*7304104dSAndroid Build Coastguard Worker     q.insert (p);
795*7304104dSAndroid Build Coastguard Worker     set_metric("thread_work_pending","role","scan", q.size());
796*7304104dSAndroid Build Coastguard Worker     cv.notify_all();
797*7304104dSAndroid Build Coastguard Worker   }
798*7304104dSAndroid Build Coastguard Worker 
799*7304104dSAndroid Build Coastguard Worker   // kill this workqueue, wake up all idlers / scanners
nuke()800*7304104dSAndroid Build Coastguard Worker   void nuke() {
801*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
802*7304104dSAndroid Build Coastguard Worker     // optional: q.clear();
803*7304104dSAndroid Build Coastguard Worker     dead = true;
804*7304104dSAndroid Build Coastguard Worker     cv.notify_all();
805*7304104dSAndroid Build Coastguard Worker   }
806*7304104dSAndroid Build Coastguard Worker 
807*7304104dSAndroid Build Coastguard Worker   // clear the workqueue, when scanning is interrupted with USR2
clear()808*7304104dSAndroid Build Coastguard Worker   void clear() {
809*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
810*7304104dSAndroid Build Coastguard Worker     q.clear();
811*7304104dSAndroid Build Coastguard Worker     set_metric("thread_work_pending","role","scan", q.size());
812*7304104dSAndroid Build Coastguard Worker     // NB: there may still be some live fronters
813*7304104dSAndroid Build Coastguard Worker     cv.notify_all(); // maybe wake up waiting idlers
814*7304104dSAndroid Build Coastguard Worker   }
815*7304104dSAndroid Build Coastguard Worker 
816*7304104dSAndroid Build Coastguard Worker   // block this scanner thread until there is work to do and no active idler
wait_front(Payload & p)817*7304104dSAndroid Build Coastguard Worker   bool wait_front (Payload& p)
818*7304104dSAndroid Build Coastguard Worker   {
819*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
820*7304104dSAndroid Build Coastguard Worker     while (!dead && (q.size() == 0 || idlers > 0))
821*7304104dSAndroid Build Coastguard Worker       cv.wait(lock);
822*7304104dSAndroid Build Coastguard Worker     if (dead)
823*7304104dSAndroid Build Coastguard Worker       return false;
824*7304104dSAndroid Build Coastguard Worker     else
825*7304104dSAndroid Build Coastguard Worker       {
826*7304104dSAndroid Build Coastguard Worker         p = * q.begin();
827*7304104dSAndroid Build Coastguard Worker         q.erase (q.begin());
828*7304104dSAndroid Build Coastguard Worker         fronters ++; // prevent idlers from starting awhile, even if empty q
829*7304104dSAndroid Build Coastguard Worker         set_metric("thread_work_pending","role","scan", q.size());
830*7304104dSAndroid Build Coastguard Worker         // NB: don't wake up idlers yet!  The consumer is busy
831*7304104dSAndroid Build Coastguard Worker         // processing this element until it calls done_front().
832*7304104dSAndroid Build Coastguard Worker         return true;
833*7304104dSAndroid Build Coastguard Worker       }
834*7304104dSAndroid Build Coastguard Worker   }
835*7304104dSAndroid Build Coastguard Worker 
836*7304104dSAndroid Build Coastguard Worker   // notify waitq that scanner thread is done with that last item
done_front()837*7304104dSAndroid Build Coastguard Worker   void done_front ()
838*7304104dSAndroid Build Coastguard Worker   {
839*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
840*7304104dSAndroid Build Coastguard Worker     fronters --;
841*7304104dSAndroid Build Coastguard Worker     if (q.size() == 0 && fronters == 0)
842*7304104dSAndroid Build Coastguard Worker       cv.notify_all(); // maybe wake up waiting idlers
843*7304104dSAndroid Build Coastguard Worker   }
844*7304104dSAndroid Build Coastguard Worker 
845*7304104dSAndroid Build Coastguard Worker   // block this idler thread until there is no work to do
wait_idle()846*7304104dSAndroid Build Coastguard Worker   void wait_idle ()
847*7304104dSAndroid Build Coastguard Worker   {
848*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
849*7304104dSAndroid Build Coastguard Worker     cv.notify_all(); // maybe wake up waiting scanners
850*7304104dSAndroid Build Coastguard Worker     while (!dead && ((q.size() != 0) || fronters > 0))
851*7304104dSAndroid Build Coastguard Worker       cv.wait(lock);
852*7304104dSAndroid Build Coastguard Worker     idlers ++;
853*7304104dSAndroid Build Coastguard Worker   }
854*7304104dSAndroid Build Coastguard Worker 
done_idle()855*7304104dSAndroid Build Coastguard Worker   void done_idle ()
856*7304104dSAndroid Build Coastguard Worker   {
857*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
858*7304104dSAndroid Build Coastguard Worker     idlers --;
859*7304104dSAndroid Build Coastguard Worker     cv.notify_all(); // maybe wake up waiting scanners, but probably not (shutting down)
860*7304104dSAndroid Build Coastguard Worker   }
861*7304104dSAndroid Build Coastguard Worker };
862*7304104dSAndroid Build Coastguard Worker 
863*7304104dSAndroid Build Coastguard Worker typedef struct stat stat_t;
864*7304104dSAndroid Build Coastguard Worker typedef pair<string,stat_t> scan_payload;
operator <(const scan_payload & a,const scan_payload & b)865*7304104dSAndroid Build Coastguard Worker inline bool operator< (const scan_payload& a, const scan_payload& b)
866*7304104dSAndroid Build Coastguard Worker {
867*7304104dSAndroid Build Coastguard Worker   return a.first < b.first; // don't bother compare the stat fields
868*7304104dSAndroid Build Coastguard Worker }
869*7304104dSAndroid Build Coastguard Worker 
870*7304104dSAndroid Build Coastguard Worker namespace std { // https://gcc.gnu.org/bugzilla/show_bug.cgi?id=56480
871*7304104dSAndroid Build Coastguard Worker   template<> struct hash<::scan_payload>
872*7304104dSAndroid Build Coastguard Worker   {
operator ()std::hash873*7304104dSAndroid Build Coastguard Worker     std::size_t operator() (const ::scan_payload& p) const noexcept
874*7304104dSAndroid Build Coastguard Worker     {
875*7304104dSAndroid Build Coastguard Worker       return hash<string>()(p.first);
876*7304104dSAndroid Build Coastguard Worker     }
877*7304104dSAndroid Build Coastguard Worker   };
878*7304104dSAndroid Build Coastguard Worker   template<> struct equal_to<::scan_payload>
879*7304104dSAndroid Build Coastguard Worker   {
operator ()std::equal_to880*7304104dSAndroid Build Coastguard Worker     std::size_t operator() (const ::scan_payload& a, const ::scan_payload& b) const noexcept
881*7304104dSAndroid Build Coastguard Worker     {
882*7304104dSAndroid Build Coastguard Worker       return a.first == b.first;
883*7304104dSAndroid Build Coastguard Worker     }
884*7304104dSAndroid Build Coastguard Worker   };
885*7304104dSAndroid Build Coastguard Worker }
886*7304104dSAndroid Build Coastguard Worker 
887*7304104dSAndroid Build Coastguard Worker static workq<scan_payload> scanq; // just a single one
888*7304104dSAndroid Build Coastguard Worker // producer & idler: thread_main_fts_source_paths()
889*7304104dSAndroid Build Coastguard Worker // consumer: thread_main_scanner()
890*7304104dSAndroid Build Coastguard Worker // idler: thread_main_groom()
891*7304104dSAndroid Build Coastguard Worker 
892*7304104dSAndroid Build Coastguard Worker 
893*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
894*7304104dSAndroid Build Coastguard Worker 
895*7304104dSAndroid Build Coastguard Worker // Unique set is a thread-safe structure that lends 'ownership' of a value
896*7304104dSAndroid Build Coastguard Worker // to a thread.  Other threads requesting the same thing are made to wait.
897*7304104dSAndroid Build Coastguard Worker // It's like a semaphore-on-demand.
898*7304104dSAndroid Build Coastguard Worker template <typename T>
899*7304104dSAndroid Build Coastguard Worker class unique_set
900*7304104dSAndroid Build Coastguard Worker {
901*7304104dSAndroid Build Coastguard Worker private:
902*7304104dSAndroid Build Coastguard Worker   set<T> values;
903*7304104dSAndroid Build Coastguard Worker   mutex mtx;
904*7304104dSAndroid Build Coastguard Worker   condition_variable cv;
905*7304104dSAndroid Build Coastguard Worker public:
unique_set()906*7304104dSAndroid Build Coastguard Worker   unique_set() {}
~unique_set()907*7304104dSAndroid Build Coastguard Worker   ~unique_set() {}
908*7304104dSAndroid Build Coastguard Worker 
acquire(const T & value)909*7304104dSAndroid Build Coastguard Worker   void acquire(const T& value)
910*7304104dSAndroid Build Coastguard Worker   {
911*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
912*7304104dSAndroid Build Coastguard Worker     while (values.find(value) != values.end())
913*7304104dSAndroid Build Coastguard Worker       cv.wait(lock);
914*7304104dSAndroid Build Coastguard Worker     values.insert(value);
915*7304104dSAndroid Build Coastguard Worker   }
916*7304104dSAndroid Build Coastguard Worker 
release(const T & value)917*7304104dSAndroid Build Coastguard Worker   void release(const T& value)
918*7304104dSAndroid Build Coastguard Worker   {
919*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
920*7304104dSAndroid Build Coastguard Worker     // assert (values.find(value) != values.end());
921*7304104dSAndroid Build Coastguard Worker     values.erase(value);
922*7304104dSAndroid Build Coastguard Worker     cv.notify_all();
923*7304104dSAndroid Build Coastguard Worker   }
924*7304104dSAndroid Build Coastguard Worker };
925*7304104dSAndroid Build Coastguard Worker 
926*7304104dSAndroid Build Coastguard Worker 
927*7304104dSAndroid Build Coastguard Worker // This is the object that's instantiate to uniquely hold a value in a
928*7304104dSAndroid Build Coastguard Worker // RAII-pattern way.
929*7304104dSAndroid Build Coastguard Worker template <typename T>
930*7304104dSAndroid Build Coastguard Worker class unique_set_reserver
931*7304104dSAndroid Build Coastguard Worker {
932*7304104dSAndroid Build Coastguard Worker private:
933*7304104dSAndroid Build Coastguard Worker   unique_set<T>& please_hold;
934*7304104dSAndroid Build Coastguard Worker   T mine;
935*7304104dSAndroid Build Coastguard Worker public:
unique_set_reserver(unique_set<T> & t,const T & value)936*7304104dSAndroid Build Coastguard Worker   unique_set_reserver(unique_set<T>& t, const T& value):
937*7304104dSAndroid Build Coastguard Worker     please_hold(t), mine(value)  { please_hold.acquire(mine); }
~unique_set_reserver()938*7304104dSAndroid Build Coastguard Worker   ~unique_set_reserver() { please_hold.release(mine); }
939*7304104dSAndroid Build Coastguard Worker };
940*7304104dSAndroid Build Coastguard Worker 
941*7304104dSAndroid Build Coastguard Worker 
942*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
943*7304104dSAndroid Build Coastguard Worker 
944*7304104dSAndroid Build Coastguard Worker // periodic_barrier is a concurrency control object that lets N threads
945*7304104dSAndroid Build Coastguard Worker // periodically (based on counter value) agree to wait at a barrier,
946*7304104dSAndroid Build Coastguard Worker // let one of them carry out some work, then be set free
947*7304104dSAndroid Build Coastguard Worker 
948*7304104dSAndroid Build Coastguard Worker class periodic_barrier
949*7304104dSAndroid Build Coastguard Worker {
950*7304104dSAndroid Build Coastguard Worker private:
951*7304104dSAndroid Build Coastguard Worker   unsigned period; // number of count() reports to trigger barrier activation
952*7304104dSAndroid Build Coastguard Worker   unsigned threads; // number of threads participating
953*7304104dSAndroid Build Coastguard Worker   mutex mtx; // protects all the following fields
954*7304104dSAndroid Build Coastguard Worker   unsigned counter; // count of count() reports in the current generation
955*7304104dSAndroid Build Coastguard Worker   unsigned generation; // barrier activation generation
956*7304104dSAndroid Build Coastguard Worker   unsigned waiting; // number of threads waiting for barrier
957*7304104dSAndroid Build Coastguard Worker   bool dead; // bring out your
958*7304104dSAndroid Build Coastguard Worker   condition_variable cv;
959*7304104dSAndroid Build Coastguard Worker public:
periodic_barrier(unsigned t,unsigned p)960*7304104dSAndroid Build Coastguard Worker   periodic_barrier(unsigned t, unsigned p):
961*7304104dSAndroid Build Coastguard Worker     period(p), threads(t), counter(0), generation(0), waiting(0), dead(false) { }
~periodic_barrier()962*7304104dSAndroid Build Coastguard Worker   virtual ~periodic_barrier() {}
963*7304104dSAndroid Build Coastguard Worker 
964*7304104dSAndroid Build Coastguard Worker   virtual void periodic_barrier_work() noexcept = 0;
nuke()965*7304104dSAndroid Build Coastguard Worker   void nuke() {
966*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
967*7304104dSAndroid Build Coastguard Worker     dead = true;
968*7304104dSAndroid Build Coastguard Worker     cv.notify_all();
969*7304104dSAndroid Build Coastguard Worker   }
970*7304104dSAndroid Build Coastguard Worker 
count()971*7304104dSAndroid Build Coastguard Worker   void count()
972*7304104dSAndroid Build Coastguard Worker   {
973*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(mtx);
974*7304104dSAndroid Build Coastguard Worker     unsigned prev_generation = this->generation;
975*7304104dSAndroid Build Coastguard Worker     if (counter < period-1) // normal case: counter just freely running
976*7304104dSAndroid Build Coastguard Worker       {
977*7304104dSAndroid Build Coastguard Worker         counter ++;
978*7304104dSAndroid Build Coastguard Worker         return;
979*7304104dSAndroid Build Coastguard Worker       }
980*7304104dSAndroid Build Coastguard Worker     else if (counter == period-1) // we're the doer
981*7304104dSAndroid Build Coastguard Worker       {
982*7304104dSAndroid Build Coastguard Worker         counter = period; // entering barrier holding phase
983*7304104dSAndroid Build Coastguard Worker         cv.notify_all();
984*7304104dSAndroid Build Coastguard Worker         while (waiting < threads-1 && !dead)
985*7304104dSAndroid Build Coastguard Worker           cv.wait(lock);
986*7304104dSAndroid Build Coastguard Worker         // all other threads are now stuck in the barrier
987*7304104dSAndroid Build Coastguard Worker         this->periodic_barrier_work(); // NB: we're holding the mutex the whole time
988*7304104dSAndroid Build Coastguard Worker         // reset for next barrier, releasing other waiters
989*7304104dSAndroid Build Coastguard Worker         counter = 0;
990*7304104dSAndroid Build Coastguard Worker         generation ++;
991*7304104dSAndroid Build Coastguard Worker         cv.notify_all();
992*7304104dSAndroid Build Coastguard Worker         return;
993*7304104dSAndroid Build Coastguard Worker       }
994*7304104dSAndroid Build Coastguard Worker     else if (counter == period) // we're a waiter, in holding phase
995*7304104dSAndroid Build Coastguard Worker       {
996*7304104dSAndroid Build Coastguard Worker         waiting ++;
997*7304104dSAndroid Build Coastguard Worker         cv.notify_all();
998*7304104dSAndroid Build Coastguard Worker         while (counter == period && generation == prev_generation && !dead)
999*7304104dSAndroid Build Coastguard Worker           cv.wait(lock);
1000*7304104dSAndroid Build Coastguard Worker         waiting --;
1001*7304104dSAndroid Build Coastguard Worker         return;
1002*7304104dSAndroid Build Coastguard Worker       }
1003*7304104dSAndroid Build Coastguard Worker   }
1004*7304104dSAndroid Build Coastguard Worker };
1005*7304104dSAndroid Build Coastguard Worker 
1006*7304104dSAndroid Build Coastguard Worker 
1007*7304104dSAndroid Build Coastguard Worker 
1008*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
1009*7304104dSAndroid Build Coastguard Worker 
1010*7304104dSAndroid Build Coastguard Worker 
1011*7304104dSAndroid Build Coastguard Worker // Print a standard timestamp.
1012*7304104dSAndroid Build Coastguard Worker static ostream&
timestamp(ostream & o)1013*7304104dSAndroid Build Coastguard Worker timestamp (ostream &o)
1014*7304104dSAndroid Build Coastguard Worker {
1015*7304104dSAndroid Build Coastguard Worker   char datebuf[80];
1016*7304104dSAndroid Build Coastguard Worker   char *now2 = NULL;
1017*7304104dSAndroid Build Coastguard Worker   time_t now_t = time(NULL);
1018*7304104dSAndroid Build Coastguard Worker   struct tm now;
1019*7304104dSAndroid Build Coastguard Worker   struct tm *nowp = gmtime_r (&now_t, &now);
1020*7304104dSAndroid Build Coastguard Worker   if (nowp)
1021*7304104dSAndroid Build Coastguard Worker     {
1022*7304104dSAndroid Build Coastguard Worker       (void) strftime (datebuf, sizeof (datebuf), "%c", nowp);
1023*7304104dSAndroid Build Coastguard Worker       now2 = datebuf;
1024*7304104dSAndroid Build Coastguard Worker     }
1025*7304104dSAndroid Build Coastguard Worker 
1026*7304104dSAndroid Build Coastguard Worker   return o << "[" << (now2 ? now2 : "") << "] "
1027*7304104dSAndroid Build Coastguard Worker            << "(" << getpid () << "/" << tid() << "): ";
1028*7304104dSAndroid Build Coastguard Worker }
1029*7304104dSAndroid Build Coastguard Worker 
1030*7304104dSAndroid Build Coastguard Worker 
1031*7304104dSAndroid Build Coastguard Worker // A little class that impersonates an ostream to the extent that it can
1032*7304104dSAndroid Build Coastguard Worker // take << streaming operations.  It batches up the bits into an internal
1033*7304104dSAndroid Build Coastguard Worker // stringstream until it is destroyed; then flushes to the original ostream.
1034*7304104dSAndroid Build Coastguard Worker // It adds a timestamp
1035*7304104dSAndroid Build Coastguard Worker class obatched
1036*7304104dSAndroid Build Coastguard Worker {
1037*7304104dSAndroid Build Coastguard Worker private:
1038*7304104dSAndroid Build Coastguard Worker   ostream& o;
1039*7304104dSAndroid Build Coastguard Worker   stringstream stro;
1040*7304104dSAndroid Build Coastguard Worker   static mutex lock;
1041*7304104dSAndroid Build Coastguard Worker public:
obatched(ostream & oo,bool timestamp_p=true)1042*7304104dSAndroid Build Coastguard Worker   obatched(ostream& oo, bool timestamp_p = true): o(oo)
1043*7304104dSAndroid Build Coastguard Worker   {
1044*7304104dSAndroid Build Coastguard Worker     if (timestamp_p)
1045*7304104dSAndroid Build Coastguard Worker       timestamp(stro);
1046*7304104dSAndroid Build Coastguard Worker   }
~obatched()1047*7304104dSAndroid Build Coastguard Worker   ~obatched()
1048*7304104dSAndroid Build Coastguard Worker   {
1049*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> do_not_cross_the_streams(obatched::lock);
1050*7304104dSAndroid Build Coastguard Worker     o << stro.str();
1051*7304104dSAndroid Build Coastguard Worker     o.flush();
1052*7304104dSAndroid Build Coastguard Worker   }
operator ostream&()1053*7304104dSAndroid Build Coastguard Worker   operator ostream& () { return stro; }
operator <<(const T & t)1054*7304104dSAndroid Build Coastguard Worker   template <typename T> ostream& operator << (const T& t) { stro << t; return stro; }
1055*7304104dSAndroid Build Coastguard Worker };
1056*7304104dSAndroid Build Coastguard Worker mutex obatched::lock; // just the one, since cout/cerr iostreams are not thread-safe
1057*7304104dSAndroid Build Coastguard Worker 
1058*7304104dSAndroid Build Coastguard Worker 
report(ostream & o) const1059*7304104dSAndroid Build Coastguard Worker void reportable_exception::report(ostream& o) const {
1060*7304104dSAndroid Build Coastguard Worker   obatched(o) << message << endl;
1061*7304104dSAndroid Build Coastguard Worker }
1062*7304104dSAndroid Build Coastguard Worker 
1063*7304104dSAndroid Build Coastguard Worker 
1064*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
1065*7304104dSAndroid Build Coastguard Worker 
1066*7304104dSAndroid Build Coastguard Worker 
1067*7304104dSAndroid Build Coastguard Worker // RAII style sqlite prepared-statement holder that matches { } block lifetime
1068*7304104dSAndroid Build Coastguard Worker 
1069*7304104dSAndroid Build Coastguard Worker struct sqlite_ps
1070*7304104dSAndroid Build Coastguard Worker {
1071*7304104dSAndroid Build Coastguard Worker private:
1072*7304104dSAndroid Build Coastguard Worker   sqlite3* db;
1073*7304104dSAndroid Build Coastguard Worker   const string nickname;
1074*7304104dSAndroid Build Coastguard Worker   const string sql;
1075*7304104dSAndroid Build Coastguard Worker   sqlite3_stmt *pp;
1076*7304104dSAndroid Build Coastguard Worker 
1077*7304104dSAndroid Build Coastguard Worker   sqlite_ps(const sqlite_ps&); // make uncopyable
1078*7304104dSAndroid Build Coastguard Worker   sqlite_ps& operator=(const sqlite_ps &); // make unassignable
1079*7304104dSAndroid Build Coastguard Worker 
1080*7304104dSAndroid Build Coastguard Worker public:
sqlite_pssqlite_ps1081*7304104dSAndroid Build Coastguard Worker   sqlite_ps (sqlite3* d, const string& n, const string& s): db(d), nickname(n), sql(s) {
1082*7304104dSAndroid Build Coastguard Worker     // tmp_ms_metric tick("sqlite3","prep",nickname);
1083*7304104dSAndroid Build Coastguard Worker     if (verbose > 4)
1084*7304104dSAndroid Build Coastguard Worker       obatched(clog) << nickname << " prep " << sql << endl;
1085*7304104dSAndroid Build Coastguard Worker     int rc = sqlite3_prepare_v2 (db, sql.c_str(), -1 /* to \0 */, & this->pp, NULL);
1086*7304104dSAndroid Build Coastguard Worker     if (rc != SQLITE_OK)
1087*7304104dSAndroid Build Coastguard Worker       throw sqlite_exception(rc, "prepare " + sql);
1088*7304104dSAndroid Build Coastguard Worker   }
1089*7304104dSAndroid Build Coastguard Worker 
resetsqlite_ps1090*7304104dSAndroid Build Coastguard Worker   sqlite_ps& reset()
1091*7304104dSAndroid Build Coastguard Worker   {
1092*7304104dSAndroid Build Coastguard Worker     tmp_ms_metric tick("sqlite3","reset",nickname);
1093*7304104dSAndroid Build Coastguard Worker     sqlite3_reset(this->pp);
1094*7304104dSAndroid Build Coastguard Worker     return *this;
1095*7304104dSAndroid Build Coastguard Worker   }
1096*7304104dSAndroid Build Coastguard Worker 
bindsqlite_ps1097*7304104dSAndroid Build Coastguard Worker   sqlite_ps& bind(int parameter, const string& str)
1098*7304104dSAndroid Build Coastguard Worker   {
1099*7304104dSAndroid Build Coastguard Worker     if (verbose > 4)
1100*7304104dSAndroid Build Coastguard Worker       obatched(clog) << nickname << " bind " << parameter << "=" << str << endl;
1101*7304104dSAndroid Build Coastguard Worker     int rc = sqlite3_bind_text (this->pp, parameter, str.c_str(), -1, SQLITE_TRANSIENT);
1102*7304104dSAndroid Build Coastguard Worker     if (rc != SQLITE_OK)
1103*7304104dSAndroid Build Coastguard Worker       throw sqlite_exception(rc, "sqlite3 bind");
1104*7304104dSAndroid Build Coastguard Worker     return *this;
1105*7304104dSAndroid Build Coastguard Worker   }
1106*7304104dSAndroid Build Coastguard Worker 
bindsqlite_ps1107*7304104dSAndroid Build Coastguard Worker   sqlite_ps& bind(int parameter, int64_t value)
1108*7304104dSAndroid Build Coastguard Worker   {
1109*7304104dSAndroid Build Coastguard Worker     if (verbose > 4)
1110*7304104dSAndroid Build Coastguard Worker       obatched(clog) << nickname << " bind " << parameter << "=" << value << endl;
1111*7304104dSAndroid Build Coastguard Worker     int rc = sqlite3_bind_int64 (this->pp, parameter, value);
1112*7304104dSAndroid Build Coastguard Worker     if (rc != SQLITE_OK)
1113*7304104dSAndroid Build Coastguard Worker       throw sqlite_exception(rc, "sqlite3 bind");
1114*7304104dSAndroid Build Coastguard Worker     return *this;
1115*7304104dSAndroid Build Coastguard Worker   }
1116*7304104dSAndroid Build Coastguard Worker 
bindsqlite_ps1117*7304104dSAndroid Build Coastguard Worker   sqlite_ps& bind(int parameter)
1118*7304104dSAndroid Build Coastguard Worker   {
1119*7304104dSAndroid Build Coastguard Worker     if (verbose > 4)
1120*7304104dSAndroid Build Coastguard Worker       obatched(clog) << nickname << " bind " << parameter << "=" << "NULL" << endl;
1121*7304104dSAndroid Build Coastguard Worker     int rc = sqlite3_bind_null (this->pp, parameter);
1122*7304104dSAndroid Build Coastguard Worker     if (rc != SQLITE_OK)
1123*7304104dSAndroid Build Coastguard Worker       throw sqlite_exception(rc, "sqlite3 bind");
1124*7304104dSAndroid Build Coastguard Worker     return *this;
1125*7304104dSAndroid Build Coastguard Worker   }
1126*7304104dSAndroid Build Coastguard Worker 
1127*7304104dSAndroid Build Coastguard Worker 
step_ok_donesqlite_ps1128*7304104dSAndroid Build Coastguard Worker   void step_ok_done() {
1129*7304104dSAndroid Build Coastguard Worker     tmp_ms_metric tick("sqlite3","step_done",nickname);
1130*7304104dSAndroid Build Coastguard Worker     int rc = sqlite3_step (this->pp);
1131*7304104dSAndroid Build Coastguard Worker     if (verbose > 4)
1132*7304104dSAndroid Build Coastguard Worker       obatched(clog) << nickname << " step-ok-done(" << sqlite3_errstr(rc) << ") " << sql << endl;
1133*7304104dSAndroid Build Coastguard Worker     if (rc != SQLITE_OK && rc != SQLITE_DONE && rc != SQLITE_ROW)
1134*7304104dSAndroid Build Coastguard Worker       throw sqlite_exception(rc, "sqlite3 step");
1135*7304104dSAndroid Build Coastguard Worker     (void) sqlite3_reset (this->pp);
1136*7304104dSAndroid Build Coastguard Worker   }
1137*7304104dSAndroid Build Coastguard Worker 
1138*7304104dSAndroid Build Coastguard Worker 
stepsqlite_ps1139*7304104dSAndroid Build Coastguard Worker   int step() {
1140*7304104dSAndroid Build Coastguard Worker     tmp_ms_metric tick("sqlite3","step",nickname);
1141*7304104dSAndroid Build Coastguard Worker     int rc = sqlite3_step (this->pp);
1142*7304104dSAndroid Build Coastguard Worker     if (verbose > 4)
1143*7304104dSAndroid Build Coastguard Worker       obatched(clog) << nickname << " step(" << sqlite3_errstr(rc) << ") " << sql << endl;
1144*7304104dSAndroid Build Coastguard Worker     return rc;
1145*7304104dSAndroid Build Coastguard Worker   }
1146*7304104dSAndroid Build Coastguard Worker 
~sqlite_pssqlite_ps1147*7304104dSAndroid Build Coastguard Worker   ~sqlite_ps () { sqlite3_finalize (this->pp); }
operator sqlite3_stmt*sqlite_ps1148*7304104dSAndroid Build Coastguard Worker   operator sqlite3_stmt* () { return this->pp; }
1149*7304104dSAndroid Build Coastguard Worker };
1150*7304104dSAndroid Build Coastguard Worker 
1151*7304104dSAndroid Build Coastguard Worker 
1152*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
1153*7304104dSAndroid Build Coastguard Worker 
1154*7304104dSAndroid Build Coastguard Worker 
1155*7304104dSAndroid Build Coastguard Worker struct sqlite_checkpoint_pb: public periodic_barrier
1156*7304104dSAndroid Build Coastguard Worker {
1157*7304104dSAndroid Build Coastguard Worker   // NB: don't use sqlite_ps since it can throw exceptions during ctor etc.
sqlite_checkpoint_pbsqlite_checkpoint_pb1158*7304104dSAndroid Build Coastguard Worker   sqlite_checkpoint_pb(unsigned t, unsigned p):
1159*7304104dSAndroid Build Coastguard Worker     periodic_barrier(t, p) { }
1160*7304104dSAndroid Build Coastguard Worker 
periodic_barrier_worksqlite_checkpoint_pb1161*7304104dSAndroid Build Coastguard Worker   void periodic_barrier_work() noexcept
1162*7304104dSAndroid Build Coastguard Worker   {
1163*7304104dSAndroid Build Coastguard Worker     (void) sqlite3_exec (db, "pragma wal_checkpoint(truncate);", NULL, NULL, NULL);
1164*7304104dSAndroid Build Coastguard Worker   }
1165*7304104dSAndroid Build Coastguard Worker };
1166*7304104dSAndroid Build Coastguard Worker 
1167*7304104dSAndroid Build Coastguard Worker static periodic_barrier* scan_barrier = 0; // initialized in main()
1168*7304104dSAndroid Build Coastguard Worker 
1169*7304104dSAndroid Build Coastguard Worker 
1170*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
1171*7304104dSAndroid Build Coastguard Worker 
1172*7304104dSAndroid Build Coastguard Worker // RAII style templated autocloser
1173*7304104dSAndroid Build Coastguard Worker 
1174*7304104dSAndroid Build Coastguard Worker template <class Payload, class Ignore>
1175*7304104dSAndroid Build Coastguard Worker struct defer_dtor
1176*7304104dSAndroid Build Coastguard Worker {
1177*7304104dSAndroid Build Coastguard Worker public:
1178*7304104dSAndroid Build Coastguard Worker   typedef Ignore (*dtor_fn) (Payload);
1179*7304104dSAndroid Build Coastguard Worker 
1180*7304104dSAndroid Build Coastguard Worker private:
1181*7304104dSAndroid Build Coastguard Worker   Payload p;
1182*7304104dSAndroid Build Coastguard Worker   dtor_fn fn;
1183*7304104dSAndroid Build Coastguard Worker 
1184*7304104dSAndroid Build Coastguard Worker public:
defer_dtordefer_dtor1185*7304104dSAndroid Build Coastguard Worker   defer_dtor(Payload _p, dtor_fn _fn): p(_p), fn(_fn) {}
~defer_dtordefer_dtor1186*7304104dSAndroid Build Coastguard Worker   ~defer_dtor() { (void) (*fn)(p); }
1187*7304104dSAndroid Build Coastguard Worker 
1188*7304104dSAndroid Build Coastguard Worker private:
1189*7304104dSAndroid Build Coastguard Worker   defer_dtor(const defer_dtor<Payload,Ignore>&); // make uncopyable
1190*7304104dSAndroid Build Coastguard Worker   defer_dtor& operator=(const defer_dtor<Payload,Ignore> &); // make unassignable
1191*7304104dSAndroid Build Coastguard Worker };
1192*7304104dSAndroid Build Coastguard Worker 
1193*7304104dSAndroid Build Coastguard Worker 
1194*7304104dSAndroid Build Coastguard Worker 
1195*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
1196*7304104dSAndroid Build Coastguard Worker 
1197*7304104dSAndroid Build Coastguard Worker 
1198*7304104dSAndroid Build Coastguard Worker static string
header_censor(const string & str)1199*7304104dSAndroid Build Coastguard Worker header_censor(const string& str)
1200*7304104dSAndroid Build Coastguard Worker {
1201*7304104dSAndroid Build Coastguard Worker   string y;
1202*7304104dSAndroid Build Coastguard Worker   for (auto&& x : str)
1203*7304104dSAndroid Build Coastguard Worker     {
1204*7304104dSAndroid Build Coastguard Worker       if (isalnum(x) || x == '/' || x == '.' || x == ',' || x == '_' || x == ':')
1205*7304104dSAndroid Build Coastguard Worker         y += x;
1206*7304104dSAndroid Build Coastguard Worker     }
1207*7304104dSAndroid Build Coastguard Worker   return y;
1208*7304104dSAndroid Build Coastguard Worker }
1209*7304104dSAndroid Build Coastguard Worker 
1210*7304104dSAndroid Build Coastguard Worker 
1211*7304104dSAndroid Build Coastguard Worker static string
conninfo(struct MHD_Connection * conn)1212*7304104dSAndroid Build Coastguard Worker conninfo (struct MHD_Connection * conn)
1213*7304104dSAndroid Build Coastguard Worker {
1214*7304104dSAndroid Build Coastguard Worker   char hostname[256]; // RFC1035
1215*7304104dSAndroid Build Coastguard Worker   char servname[256];
1216*7304104dSAndroid Build Coastguard Worker   int sts = -1;
1217*7304104dSAndroid Build Coastguard Worker 
1218*7304104dSAndroid Build Coastguard Worker   if (conn == 0)
1219*7304104dSAndroid Build Coastguard Worker     return "internal";
1220*7304104dSAndroid Build Coastguard Worker 
1221*7304104dSAndroid Build Coastguard Worker   /* Look up client address data. */
1222*7304104dSAndroid Build Coastguard Worker   const union MHD_ConnectionInfo *u = MHD_get_connection_info (conn,
1223*7304104dSAndroid Build Coastguard Worker                                                                MHD_CONNECTION_INFO_CLIENT_ADDRESS);
1224*7304104dSAndroid Build Coastguard Worker   struct sockaddr *so = u ? u->client_addr : 0;
1225*7304104dSAndroid Build Coastguard Worker 
1226*7304104dSAndroid Build Coastguard Worker   if (so && so->sa_family == AF_INET) {
1227*7304104dSAndroid Build Coastguard Worker     sts = getnameinfo (so, sizeof (struct sockaddr_in),
1228*7304104dSAndroid Build Coastguard Worker                        hostname, sizeof (hostname),
1229*7304104dSAndroid Build Coastguard Worker                        servname, sizeof (servname),
1230*7304104dSAndroid Build Coastguard Worker                        NI_NUMERICHOST | NI_NUMERICSERV);
1231*7304104dSAndroid Build Coastguard Worker   } else if (so && so->sa_family == AF_INET6) {
1232*7304104dSAndroid Build Coastguard Worker     struct sockaddr_in6* addr6 = (struct sockaddr_in6*) so;
1233*7304104dSAndroid Build Coastguard Worker     if (IN6_IS_ADDR_V4MAPPED(&addr6->sin6_addr)) {
1234*7304104dSAndroid Build Coastguard Worker       struct sockaddr_in addr4;
1235*7304104dSAndroid Build Coastguard Worker       memset (&addr4, 0, sizeof(addr4));
1236*7304104dSAndroid Build Coastguard Worker       addr4.sin_family = AF_INET;
1237*7304104dSAndroid Build Coastguard Worker       addr4.sin_port = addr6->sin6_port;
1238*7304104dSAndroid Build Coastguard Worker       memcpy (&addr4.sin_addr.s_addr, addr6->sin6_addr.s6_addr+12, sizeof(addr4.sin_addr.s_addr));
1239*7304104dSAndroid Build Coastguard Worker       sts = getnameinfo ((struct sockaddr*) &addr4, sizeof (addr4),
1240*7304104dSAndroid Build Coastguard Worker                          hostname, sizeof (hostname),
1241*7304104dSAndroid Build Coastguard Worker                          servname, sizeof (servname),
1242*7304104dSAndroid Build Coastguard Worker                          NI_NUMERICHOST | NI_NUMERICSERV);
1243*7304104dSAndroid Build Coastguard Worker     } else {
1244*7304104dSAndroid Build Coastguard Worker       sts = getnameinfo (so, sizeof (struct sockaddr_in6),
1245*7304104dSAndroid Build Coastguard Worker                          hostname, sizeof (hostname),
1246*7304104dSAndroid Build Coastguard Worker                          servname, sizeof (servname),
1247*7304104dSAndroid Build Coastguard Worker                          NI_NUMERICHOST | NI_NUMERICSERV);
1248*7304104dSAndroid Build Coastguard Worker     }
1249*7304104dSAndroid Build Coastguard Worker   }
1250*7304104dSAndroid Build Coastguard Worker 
1251*7304104dSAndroid Build Coastguard Worker   if (sts != 0) {
1252*7304104dSAndroid Build Coastguard Worker     hostname[0] = servname[0] = '\0';
1253*7304104dSAndroid Build Coastguard Worker   }
1254*7304104dSAndroid Build Coastguard Worker 
1255*7304104dSAndroid Build Coastguard Worker   // extract headers relevant to administration
1256*7304104dSAndroid Build Coastguard Worker   const char* user_agent = MHD_lookup_connection_value (conn, MHD_HEADER_KIND, "User-Agent") ?: "";
1257*7304104dSAndroid Build Coastguard Worker   const char* x_forwarded_for = MHD_lookup_connection_value (conn, MHD_HEADER_KIND, "X-Forwarded-For") ?: "";
1258*7304104dSAndroid Build Coastguard Worker   // NB: these are untrustworthy, beware if machine-processing log files
1259*7304104dSAndroid Build Coastguard Worker 
1260*7304104dSAndroid Build Coastguard Worker   return string(hostname) + string(":") + string(servname) +
1261*7304104dSAndroid Build Coastguard Worker     string(" UA:") + header_censor(string(user_agent)) +
1262*7304104dSAndroid Build Coastguard Worker     string(" XFF:") + header_censor(string(x_forwarded_for));
1263*7304104dSAndroid Build Coastguard Worker }
1264*7304104dSAndroid Build Coastguard Worker 
1265*7304104dSAndroid Build Coastguard Worker 
1266*7304104dSAndroid Build Coastguard Worker 
1267*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
1268*7304104dSAndroid Build Coastguard Worker 
1269*7304104dSAndroid Build Coastguard Worker /* Wrapper for MHD_add_response_header that logs an error if we
1270*7304104dSAndroid Build Coastguard Worker    couldn't add the specified header.  */
1271*7304104dSAndroid Build Coastguard Worker static void
add_mhd_response_header(struct MHD_Response * r,const char * h,const char * v)1272*7304104dSAndroid Build Coastguard Worker add_mhd_response_header (struct MHD_Response *r,
1273*7304104dSAndroid Build Coastguard Worker 			 const char *h, const char *v)
1274*7304104dSAndroid Build Coastguard Worker {
1275*7304104dSAndroid Build Coastguard Worker   if (MHD_add_response_header (r, h, v) == MHD_NO)
1276*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "Error: couldn't add '" << h << "' header" << endl;
1277*7304104dSAndroid Build Coastguard Worker }
1278*7304104dSAndroid Build Coastguard Worker 
1279*7304104dSAndroid Build Coastguard Worker static void
add_mhd_last_modified(struct MHD_Response * resp,time_t mtime)1280*7304104dSAndroid Build Coastguard Worker add_mhd_last_modified (struct MHD_Response *resp, time_t mtime)
1281*7304104dSAndroid Build Coastguard Worker {
1282*7304104dSAndroid Build Coastguard Worker   struct tm now;
1283*7304104dSAndroid Build Coastguard Worker   struct tm *nowp = gmtime_r (&mtime, &now);
1284*7304104dSAndroid Build Coastguard Worker   if (nowp != NULL)
1285*7304104dSAndroid Build Coastguard Worker     {
1286*7304104dSAndroid Build Coastguard Worker       char datebuf[80];
1287*7304104dSAndroid Build Coastguard Worker       size_t rc = strftime (datebuf, sizeof (datebuf), "%a, %d %b %Y %T GMT",
1288*7304104dSAndroid Build Coastguard Worker                             nowp);
1289*7304104dSAndroid Build Coastguard Worker       if (rc > 0 && rc < sizeof (datebuf))
1290*7304104dSAndroid Build Coastguard Worker         add_mhd_response_header (resp, "Last-Modified", datebuf);
1291*7304104dSAndroid Build Coastguard Worker     }
1292*7304104dSAndroid Build Coastguard Worker 
1293*7304104dSAndroid Build Coastguard Worker   add_mhd_response_header (resp, "Cache-Control", "public");
1294*7304104dSAndroid Build Coastguard Worker }
1295*7304104dSAndroid Build Coastguard Worker 
1296*7304104dSAndroid Build Coastguard Worker // quote all questionable characters of str for safe passage through a sh -c expansion.
1297*7304104dSAndroid Build Coastguard Worker static string
shell_escape(const string & str)1298*7304104dSAndroid Build Coastguard Worker shell_escape(const string& str)
1299*7304104dSAndroid Build Coastguard Worker {
1300*7304104dSAndroid Build Coastguard Worker   string y;
1301*7304104dSAndroid Build Coastguard Worker   for (auto&& x : str)
1302*7304104dSAndroid Build Coastguard Worker     {
1303*7304104dSAndroid Build Coastguard Worker       if (! isalnum(x) && x != '/')
1304*7304104dSAndroid Build Coastguard Worker         y += "\\";
1305*7304104dSAndroid Build Coastguard Worker       y += x;
1306*7304104dSAndroid Build Coastguard Worker     }
1307*7304104dSAndroid Build Coastguard Worker   return y;
1308*7304104dSAndroid Build Coastguard Worker }
1309*7304104dSAndroid Build Coastguard Worker 
1310*7304104dSAndroid Build Coastguard Worker 
1311*7304104dSAndroid Build Coastguard Worker // PR25548: Perform POSIX / RFC3986 style path canonicalization on the input string.
1312*7304104dSAndroid Build Coastguard Worker //
1313*7304104dSAndroid Build Coastguard Worker // Namely:
1314*7304104dSAndroid Build Coastguard Worker //    //         ->   /
1315*7304104dSAndroid Build Coastguard Worker //    /foo/../   ->   /
1316*7304104dSAndroid Build Coastguard Worker //    /./        ->   /
1317*7304104dSAndroid Build Coastguard Worker //
1318*7304104dSAndroid Build Coastguard Worker // This mapping is done on dwarf-side source path names, which may
1319*7304104dSAndroid Build Coastguard Worker // include these constructs, so we can deal with debuginfod clients
1320*7304104dSAndroid Build Coastguard Worker // that accidentally canonicalize the paths.
1321*7304104dSAndroid Build Coastguard Worker //
1322*7304104dSAndroid Build Coastguard Worker // realpath(3) is close but not quite right, because it also resolves
1323*7304104dSAndroid Build Coastguard Worker // symbolic links.  Symlinks at the debuginfod server have nothing to
1324*7304104dSAndroid Build Coastguard Worker // do with the build-time symlinks, thus they must not be considered.
1325*7304104dSAndroid Build Coastguard Worker //
1326*7304104dSAndroid Build Coastguard Worker // see also curl Curl_dedotdotify() aka RFC3986, which we mostly follow here
1327*7304104dSAndroid Build Coastguard Worker // see also libc __realpath()
1328*7304104dSAndroid Build Coastguard Worker // see also llvm llvm::sys::path::remove_dots()
1329*7304104dSAndroid Build Coastguard Worker static string
canon_pathname(const string & input)1330*7304104dSAndroid Build Coastguard Worker canon_pathname (const string& input)
1331*7304104dSAndroid Build Coastguard Worker {
1332*7304104dSAndroid Build Coastguard Worker   string i = input; // 5.2.4 (1)
1333*7304104dSAndroid Build Coastguard Worker   string o;
1334*7304104dSAndroid Build Coastguard Worker 
1335*7304104dSAndroid Build Coastguard Worker   while (i.size() != 0)
1336*7304104dSAndroid Build Coastguard Worker     {
1337*7304104dSAndroid Build Coastguard Worker       // 5.2.4 (2) A
1338*7304104dSAndroid Build Coastguard Worker       if (i.substr(0,3) == "../")
1339*7304104dSAndroid Build Coastguard Worker         i = i.substr(3);
1340*7304104dSAndroid Build Coastguard Worker       else if(i.substr(0,2) == "./")
1341*7304104dSAndroid Build Coastguard Worker         i = i.substr(2);
1342*7304104dSAndroid Build Coastguard Worker 
1343*7304104dSAndroid Build Coastguard Worker       // 5.2.4 (2) B
1344*7304104dSAndroid Build Coastguard Worker       else if (i.substr(0,3) == "/./")
1345*7304104dSAndroid Build Coastguard Worker         i = i.substr(2);
1346*7304104dSAndroid Build Coastguard Worker       else if (i == "/.")
1347*7304104dSAndroid Build Coastguard Worker         i = ""; // no need to handle "/." complete-path-segment case; we're dealing with file names
1348*7304104dSAndroid Build Coastguard Worker 
1349*7304104dSAndroid Build Coastguard Worker       // 5.2.4 (2) C
1350*7304104dSAndroid Build Coastguard Worker       else if (i.substr(0,4) == "/../") {
1351*7304104dSAndroid Build Coastguard Worker         i = i.substr(3);
1352*7304104dSAndroid Build Coastguard Worker         string::size_type sl = o.rfind("/");
1353*7304104dSAndroid Build Coastguard Worker         if (sl != string::npos)
1354*7304104dSAndroid Build Coastguard Worker           o = o.substr(0, sl);
1355*7304104dSAndroid Build Coastguard Worker         else
1356*7304104dSAndroid Build Coastguard Worker           o = "";
1357*7304104dSAndroid Build Coastguard Worker       } else if (i == "/..")
1358*7304104dSAndroid Build Coastguard Worker         i = ""; // no need to handle "/.." complete-path-segment case; we're dealing with file names
1359*7304104dSAndroid Build Coastguard Worker 
1360*7304104dSAndroid Build Coastguard Worker       // 5.2.4 (2) D
1361*7304104dSAndroid Build Coastguard Worker       // no need to handle these cases; we're dealing with file names
1362*7304104dSAndroid Build Coastguard Worker       else if (i == ".")
1363*7304104dSAndroid Build Coastguard Worker         i = "";
1364*7304104dSAndroid Build Coastguard Worker       else if (i == "..")
1365*7304104dSAndroid Build Coastguard Worker         i = "";
1366*7304104dSAndroid Build Coastguard Worker 
1367*7304104dSAndroid Build Coastguard Worker       // POSIX special: map // to /
1368*7304104dSAndroid Build Coastguard Worker       else if (i.substr(0,2) == "//")
1369*7304104dSAndroid Build Coastguard Worker         i = i.substr(1);
1370*7304104dSAndroid Build Coastguard Worker 
1371*7304104dSAndroid Build Coastguard Worker       // 5.2.4 (2) E
1372*7304104dSAndroid Build Coastguard Worker       else {
1373*7304104dSAndroid Build Coastguard Worker         string::size_type next_slash = i.find("/", (i[0]=='/' ? 1 : 0)); // skip first slash
1374*7304104dSAndroid Build Coastguard Worker         o += i.substr(0, next_slash);
1375*7304104dSAndroid Build Coastguard Worker         if (next_slash == string::npos)
1376*7304104dSAndroid Build Coastguard Worker           i = "";
1377*7304104dSAndroid Build Coastguard Worker         else
1378*7304104dSAndroid Build Coastguard Worker           i = i.substr(next_slash);
1379*7304104dSAndroid Build Coastguard Worker       }
1380*7304104dSAndroid Build Coastguard Worker     }
1381*7304104dSAndroid Build Coastguard Worker 
1382*7304104dSAndroid Build Coastguard Worker   return o;
1383*7304104dSAndroid Build Coastguard Worker }
1384*7304104dSAndroid Build Coastguard Worker 
1385*7304104dSAndroid Build Coastguard Worker 
1386*7304104dSAndroid Build Coastguard Worker // Estimate available free space for a given filesystem via statfs(2).
1387*7304104dSAndroid Build Coastguard Worker // Return true if the free fraction is known to be smaller than the
1388*7304104dSAndroid Build Coastguard Worker // given minimum percentage.  Also update a related metric.
statfs_free_enough_p(const string & path,const string & label,long minfree=0)1389*7304104dSAndroid Build Coastguard Worker bool statfs_free_enough_p(const string& path, const string& label, long minfree = 0)
1390*7304104dSAndroid Build Coastguard Worker {
1391*7304104dSAndroid Build Coastguard Worker   struct statfs sfs;
1392*7304104dSAndroid Build Coastguard Worker   int rc = statfs(path.c_str(), &sfs);
1393*7304104dSAndroid Build Coastguard Worker   if (rc == 0)
1394*7304104dSAndroid Build Coastguard Worker     {
1395*7304104dSAndroid Build Coastguard Worker       double s = (double) sfs.f_bavail / (double) sfs.f_blocks;
1396*7304104dSAndroid Build Coastguard Worker       set_metric("filesys_free_ratio","purpose",label, s);
1397*7304104dSAndroid Build Coastguard Worker       return ((s * 100.0) < minfree);
1398*7304104dSAndroid Build Coastguard Worker     }
1399*7304104dSAndroid Build Coastguard Worker   return false;
1400*7304104dSAndroid Build Coastguard Worker }
1401*7304104dSAndroid Build Coastguard Worker 
1402*7304104dSAndroid Build Coastguard Worker 
1403*7304104dSAndroid Build Coastguard Worker 
1404*7304104dSAndroid Build Coastguard Worker // A map-like class that owns a cache of file descriptors (indexed by
1405*7304104dSAndroid Build Coastguard Worker // file / content names).
1406*7304104dSAndroid Build Coastguard Worker //
1407*7304104dSAndroid Build Coastguard Worker // If only it could use fd's instead of file names ... but we can't
1408*7304104dSAndroid Build Coastguard Worker // dup(2) to create independent descriptors for the same unlinked
1409*7304104dSAndroid Build Coastguard Worker // files, so would have to use some goofy linux /proc/self/fd/%d
1410*7304104dSAndroid Build Coastguard Worker // hack such as the following
1411*7304104dSAndroid Build Coastguard Worker 
1412*7304104dSAndroid Build Coastguard Worker #if 0
1413*7304104dSAndroid Build Coastguard Worker int superdup(int fd)
1414*7304104dSAndroid Build Coastguard Worker {
1415*7304104dSAndroid Build Coastguard Worker #ifdef __linux__
1416*7304104dSAndroid Build Coastguard Worker   char *fdpath = NULL;
1417*7304104dSAndroid Build Coastguard Worker   int rc = asprintf(& fdpath, "/proc/self/fd/%d", fd);
1418*7304104dSAndroid Build Coastguard Worker   int newfd;
1419*7304104dSAndroid Build Coastguard Worker   if (rc >= 0)
1420*7304104dSAndroid Build Coastguard Worker     newfd = open(fdpath, O_RDONLY);
1421*7304104dSAndroid Build Coastguard Worker   else
1422*7304104dSAndroid Build Coastguard Worker     newfd = -1;
1423*7304104dSAndroid Build Coastguard Worker   free (fdpath);
1424*7304104dSAndroid Build Coastguard Worker   return newfd;
1425*7304104dSAndroid Build Coastguard Worker #else
1426*7304104dSAndroid Build Coastguard Worker   return -1;
1427*7304104dSAndroid Build Coastguard Worker #endif
1428*7304104dSAndroid Build Coastguard Worker }
1429*7304104dSAndroid Build Coastguard Worker #endif
1430*7304104dSAndroid Build Coastguard Worker 
1431*7304104dSAndroid Build Coastguard Worker class libarchive_fdcache
1432*7304104dSAndroid Build Coastguard Worker {
1433*7304104dSAndroid Build Coastguard Worker private:
1434*7304104dSAndroid Build Coastguard Worker   mutex fdcache_lock;
1435*7304104dSAndroid Build Coastguard Worker 
1436*7304104dSAndroid Build Coastguard Worker   typedef pair<string,string> key; // archive, entry
1437*7304104dSAndroid Build Coastguard Worker   struct fdcache_entry
1438*7304104dSAndroid Build Coastguard Worker   {
1439*7304104dSAndroid Build Coastguard Worker     string fd; // file name (probably in $TMPDIR), not an actual open fd (EMFILE)
1440*7304104dSAndroid Build Coastguard Worker     double fd_size_mb; // slightly rounded up megabytes
1441*7304104dSAndroid Build Coastguard Worker     time_t freshness; // when was this entry created or requested last
1442*7304104dSAndroid Build Coastguard Worker     unsigned request_count; // how many requests were made; or 0=prefetch only
1443*7304104dSAndroid Build Coastguard Worker     double latency; // how many seconds it took to extract the file
1444*7304104dSAndroid Build Coastguard Worker   };
1445*7304104dSAndroid Build Coastguard Worker 
1446*7304104dSAndroid Build Coastguard Worker   map<key,fdcache_entry> entries; // optimized for lookup
1447*7304104dSAndroid Build Coastguard Worker   time_t last_cleaning;
1448*7304104dSAndroid Build Coastguard Worker   long max_fds;
1449*7304104dSAndroid Build Coastguard Worker   long max_mbs;
1450*7304104dSAndroid Build Coastguard Worker 
1451*7304104dSAndroid Build Coastguard Worker public:
set_metrics()1452*7304104dSAndroid Build Coastguard Worker   void set_metrics()
1453*7304104dSAndroid Build Coastguard Worker   {
1454*7304104dSAndroid Build Coastguard Worker     double fdcache_mb = 0.0;
1455*7304104dSAndroid Build Coastguard Worker     double prefetch_mb = 0.0;
1456*7304104dSAndroid Build Coastguard Worker     unsigned fdcache_count = 0;
1457*7304104dSAndroid Build Coastguard Worker     unsigned prefetch_count = 0;
1458*7304104dSAndroid Build Coastguard Worker     for (auto &i : entries) {
1459*7304104dSAndroid Build Coastguard Worker       if (i.second.request_count) {
1460*7304104dSAndroid Build Coastguard Worker         fdcache_mb += i.second.fd_size_mb;
1461*7304104dSAndroid Build Coastguard Worker         fdcache_count ++;
1462*7304104dSAndroid Build Coastguard Worker       } else {
1463*7304104dSAndroid Build Coastguard Worker         prefetch_mb += i.second.fd_size_mb;
1464*7304104dSAndroid Build Coastguard Worker         prefetch_count ++;
1465*7304104dSAndroid Build Coastguard Worker       }
1466*7304104dSAndroid Build Coastguard Worker     }
1467*7304104dSAndroid Build Coastguard Worker     set_metric("fdcache_bytes", fdcache_mb*1024.0*1024.0);
1468*7304104dSAndroid Build Coastguard Worker     set_metric("fdcache_count", fdcache_count);
1469*7304104dSAndroid Build Coastguard Worker     set_metric("fdcache_prefetch_bytes", prefetch_mb*1024.0*1024.0);
1470*7304104dSAndroid Build Coastguard Worker     set_metric("fdcache_prefetch_count", prefetch_count);
1471*7304104dSAndroid Build Coastguard Worker   }
1472*7304104dSAndroid Build Coastguard Worker 
intern(const string & a,const string & b,string fd,off_t sz,bool requested_p,double lat)1473*7304104dSAndroid Build Coastguard Worker   void intern(const string& a, const string& b, string fd, off_t sz,
1474*7304104dSAndroid Build Coastguard Worker               bool requested_p, double lat)
1475*7304104dSAndroid Build Coastguard Worker   {
1476*7304104dSAndroid Build Coastguard Worker     {
1477*7304104dSAndroid Build Coastguard Worker       unique_lock<mutex> lock(fdcache_lock);
1478*7304104dSAndroid Build Coastguard Worker       time_t now = time(NULL);
1479*7304104dSAndroid Build Coastguard Worker       // there is a chance it's already in here, just wasn't found last time
1480*7304104dSAndroid Build Coastguard Worker       // if so, there's nothing to do but count our luck
1481*7304104dSAndroid Build Coastguard Worker       auto i = entries.find(make_pair(a,b));
1482*7304104dSAndroid Build Coastguard Worker       if (i != entries.end())
1483*7304104dSAndroid Build Coastguard Worker         {
1484*7304104dSAndroid Build Coastguard Worker           inc_metric("fdcache_op_count","op","redundant_intern");
1485*7304104dSAndroid Build Coastguard Worker           if (requested_p) i->second.request_count ++; // repeat prefetch doesn't count
1486*7304104dSAndroid Build Coastguard Worker           i->second.freshness = now;
1487*7304104dSAndroid Build Coastguard Worker           // We need to nuke the temp file, since interning passes
1488*7304104dSAndroid Build Coastguard Worker           // responsibility over the path to this structure.  It is
1489*7304104dSAndroid Build Coastguard Worker           // possible that the caller still has an fd open, but that's
1490*7304104dSAndroid Build Coastguard Worker           // OK.
1491*7304104dSAndroid Build Coastguard Worker           unlink (fd.c_str());
1492*7304104dSAndroid Build Coastguard Worker           return;
1493*7304104dSAndroid Build Coastguard Worker         }
1494*7304104dSAndroid Build Coastguard Worker       double mb = (sz+65535)/1048576.0; // round up to 64K block
1495*7304104dSAndroid Build Coastguard Worker       fdcache_entry n = { .fd=fd, .fd_size_mb=mb,
1496*7304104dSAndroid Build Coastguard Worker                           .freshness=now, .request_count = requested_p?1U:0U,
1497*7304104dSAndroid Build Coastguard Worker                           .latency=lat};
1498*7304104dSAndroid Build Coastguard Worker       entries.insert(make_pair(make_pair(a,b),n));
1499*7304104dSAndroid Build Coastguard Worker 
1500*7304104dSAndroid Build Coastguard Worker       if (requested_p)
1501*7304104dSAndroid Build Coastguard Worker         inc_metric("fdcache_op_count","op","enqueue");
1502*7304104dSAndroid Build Coastguard Worker       else
1503*7304104dSAndroid Build Coastguard Worker         inc_metric("fdcache_op_count","op","prefetch_enqueue");
1504*7304104dSAndroid Build Coastguard Worker 
1505*7304104dSAndroid Build Coastguard Worker       if (verbose > 3)
1506*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "fdcache interned a=" << a << " b=" << b
1507*7304104dSAndroid Build Coastguard Worker                        << " fd=" << fd << " mb=" << mb << " front=" << requested_p
1508*7304104dSAndroid Build Coastguard Worker                        << " latency=" << lat << endl;
1509*7304104dSAndroid Build Coastguard Worker 
1510*7304104dSAndroid Build Coastguard Worker       set_metrics();
1511*7304104dSAndroid Build Coastguard Worker     }
1512*7304104dSAndroid Build Coastguard Worker 
1513*7304104dSAndroid Build Coastguard Worker     // NB: we age the cache at lookup time too
1514*7304104dSAndroid Build Coastguard Worker     if (statfs_free_enough_p(tmpdir, "tmpdir", fdcache_mintmp))
1515*7304104dSAndroid Build Coastguard Worker       {
1516*7304104dSAndroid Build Coastguard Worker         inc_metric("fdcache_op_count","op","emerg-flush");
1517*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "fdcache emergency flush for filling tmpdir" << endl;
1518*7304104dSAndroid Build Coastguard Worker         this->limit(0); // emergency flush
1519*7304104dSAndroid Build Coastguard Worker       }
1520*7304104dSAndroid Build Coastguard Worker     else // age cache normally
1521*7304104dSAndroid Build Coastguard Worker       this->limit(max_mbs);
1522*7304104dSAndroid Build Coastguard Worker   }
1523*7304104dSAndroid Build Coastguard Worker 
lookup(const string & a,const string & b)1524*7304104dSAndroid Build Coastguard Worker   int lookup(const string& a, const string& b)
1525*7304104dSAndroid Build Coastguard Worker   {
1526*7304104dSAndroid Build Coastguard Worker     int fd = -1;
1527*7304104dSAndroid Build Coastguard Worker     {
1528*7304104dSAndroid Build Coastguard Worker       unique_lock<mutex> lock(fdcache_lock);
1529*7304104dSAndroid Build Coastguard Worker       auto i = entries.find(make_pair(a,b));
1530*7304104dSAndroid Build Coastguard Worker       if (i != entries.end())
1531*7304104dSAndroid Build Coastguard Worker         {
1532*7304104dSAndroid Build Coastguard Worker           if (i->second.request_count == 0) // was a prefetch!
1533*7304104dSAndroid Build Coastguard Worker             {
1534*7304104dSAndroid Build Coastguard Worker               inc_metric("fdcache_prefetch_saved_milliseconds_count");
1535*7304104dSAndroid Build Coastguard Worker               add_metric("fdcache_prefetch_saved_milliseconds_sum", i->second.latency*1000.);
1536*7304104dSAndroid Build Coastguard Worker             }
1537*7304104dSAndroid Build Coastguard Worker           i->second.request_count ++;
1538*7304104dSAndroid Build Coastguard Worker           i->second.freshness = time(NULL);
1539*7304104dSAndroid Build Coastguard Worker           // brag about our success
1540*7304104dSAndroid Build Coastguard Worker           inc_metric("fdcache_op_count","op","prefetch_access"); // backward compat
1541*7304104dSAndroid Build Coastguard Worker           inc_metric("fdcache_saved_milliseconds_count");
1542*7304104dSAndroid Build Coastguard Worker           add_metric("fdcache_saved_milliseconds_sum", i->second.latency*1000.);
1543*7304104dSAndroid Build Coastguard Worker           fd = open(i->second.fd.c_str(), O_RDONLY);
1544*7304104dSAndroid Build Coastguard Worker         }
1545*7304104dSAndroid Build Coastguard Worker     }
1546*7304104dSAndroid Build Coastguard Worker 
1547*7304104dSAndroid Build Coastguard Worker     if (fd >= 0)
1548*7304104dSAndroid Build Coastguard Worker       inc_metric("fdcache_op_count","op","lookup_hit");
1549*7304104dSAndroid Build Coastguard Worker     else
1550*7304104dSAndroid Build Coastguard Worker       inc_metric("fdcache_op_count","op","lookup_miss");
1551*7304104dSAndroid Build Coastguard Worker 
1552*7304104dSAndroid Build Coastguard Worker     // NB: no need to age the cache after just a lookup
1553*7304104dSAndroid Build Coastguard Worker 
1554*7304104dSAndroid Build Coastguard Worker     return fd;
1555*7304104dSAndroid Build Coastguard Worker   }
1556*7304104dSAndroid Build Coastguard Worker 
probe(const string & a,const string & b)1557*7304104dSAndroid Build Coastguard Worker   int probe(const string& a, const string& b) // just a cache residency check - don't modify state, don't open
1558*7304104dSAndroid Build Coastguard Worker   {
1559*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(fdcache_lock);
1560*7304104dSAndroid Build Coastguard Worker     auto i = entries.find(make_pair(a,b));
1561*7304104dSAndroid Build Coastguard Worker     if (i != entries.end()) {
1562*7304104dSAndroid Build Coastguard Worker       inc_metric("fdcache_op_count","op","probe_hit");
1563*7304104dSAndroid Build Coastguard Worker       return true;
1564*7304104dSAndroid Build Coastguard Worker     } else {
1565*7304104dSAndroid Build Coastguard Worker       inc_metric("fdcache_op_count","op","probe_miss");
1566*7304104dSAndroid Build Coastguard Worker       return false;
1567*7304104dSAndroid Build Coastguard Worker    }
1568*7304104dSAndroid Build Coastguard Worker   }
1569*7304104dSAndroid Build Coastguard Worker 
clear(const string & a,const string & b)1570*7304104dSAndroid Build Coastguard Worker   void clear(const string& a, const string& b)
1571*7304104dSAndroid Build Coastguard Worker   {
1572*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(fdcache_lock);
1573*7304104dSAndroid Build Coastguard Worker     auto i = entries.find(make_pair(a,b));
1574*7304104dSAndroid Build Coastguard Worker     if (i != entries.end()) {
1575*7304104dSAndroid Build Coastguard Worker       inc_metric("fdcache_op_count","op",
1576*7304104dSAndroid Build Coastguard Worker                  i->second.request_count > 0 ? "clear" : "prefetch_clear");
1577*7304104dSAndroid Build Coastguard Worker       unlink (i->second.fd.c_str());
1578*7304104dSAndroid Build Coastguard Worker       entries.erase(i);
1579*7304104dSAndroid Build Coastguard Worker       set_metrics();
1580*7304104dSAndroid Build Coastguard Worker       return;
1581*7304104dSAndroid Build Coastguard Worker     }
1582*7304104dSAndroid Build Coastguard Worker   }
1583*7304104dSAndroid Build Coastguard Worker 
limit(long maxmbs,bool metrics_p=true)1584*7304104dSAndroid Build Coastguard Worker   void limit(long maxmbs, bool metrics_p = true)
1585*7304104dSAndroid Build Coastguard Worker   {
1586*7304104dSAndroid Build Coastguard Worker     time_t now = time(NULL);
1587*7304104dSAndroid Build Coastguard Worker 
1588*7304104dSAndroid Build Coastguard Worker     // avoid overly frequent limit operations
1589*7304104dSAndroid Build Coastguard Worker     if (maxmbs > 0 && (now - this->last_cleaning) < 10) // probably not worth parametrizing
1590*7304104dSAndroid Build Coastguard Worker       return;
1591*7304104dSAndroid Build Coastguard Worker     this->last_cleaning = now;
1592*7304104dSAndroid Build Coastguard Worker 
1593*7304104dSAndroid Build Coastguard Worker     if (verbose > 3 && (this->max_mbs != maxmbs))
1594*7304104dSAndroid Build Coastguard Worker       obatched(clog) << "fdcache limited to maxmbs=" << maxmbs << endl;
1595*7304104dSAndroid Build Coastguard Worker 
1596*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(fdcache_lock);
1597*7304104dSAndroid Build Coastguard Worker 
1598*7304104dSAndroid Build Coastguard Worker     this->max_mbs = maxmbs;
1599*7304104dSAndroid Build Coastguard Worker     double total_mb = 0.0;
1600*7304104dSAndroid Build Coastguard Worker 
1601*7304104dSAndroid Build Coastguard Worker     map<double, pair<string,string>> sorted_entries;
1602*7304104dSAndroid Build Coastguard Worker     for (auto &i: entries)
1603*7304104dSAndroid Build Coastguard Worker       {
1604*7304104dSAndroid Build Coastguard Worker         total_mb += i.second.fd_size_mb;
1605*7304104dSAndroid Build Coastguard Worker 
1606*7304104dSAndroid Build Coastguard Worker         // need a scalar quantity that combines these inputs in a sensible way:
1607*7304104dSAndroid Build Coastguard Worker         //
1608*7304104dSAndroid Build Coastguard Worker         // 1) freshness of this entry (last time it was accessed)
1609*7304104dSAndroid Build Coastguard Worker         // 2) size of this entry
1610*7304104dSAndroid Build Coastguard Worker         // 3) number of times it has been accessed (or if just prefetched with 0 accesses)
1611*7304104dSAndroid Build Coastguard Worker         // 4) latency it required to extract
1612*7304104dSAndroid Build Coastguard Worker         //
1613*7304104dSAndroid Build Coastguard Worker         // The lower the "score", the earlier garbage collection will
1614*7304104dSAndroid Build Coastguard Worker         // nuke it, so to prioritize entries for preservation, the
1615*7304104dSAndroid Build Coastguard Worker         // score should be higher, and vice versa.
1616*7304104dSAndroid Build Coastguard Worker         time_t factor_1_freshness = (now - i.second.freshness); // seconds
1617*7304104dSAndroid Build Coastguard Worker         double factor_2_size = i.second.fd_size_mb; // megabytes
1618*7304104dSAndroid Build Coastguard Worker         unsigned factor_3_accesscount = i.second.request_count; // units
1619*7304104dSAndroid Build Coastguard Worker         double factor_4_latency = i.second.latency; // seconds
1620*7304104dSAndroid Build Coastguard Worker 
1621*7304104dSAndroid Build Coastguard Worker         #if 0
1622*7304104dSAndroid Build Coastguard Worker         double score = - factor_1_freshness; // simple LRU
1623*7304104dSAndroid Build Coastguard Worker         #endif
1624*7304104dSAndroid Build Coastguard Worker 
1625*7304104dSAndroid Build Coastguard Worker         double score = 0.
1626*7304104dSAndroid Build Coastguard Worker           - log1p(factor_1_freshness)                // penalize old file
1627*7304104dSAndroid Build Coastguard Worker           - log1p(factor_2_size)                     // penalize large file
1628*7304104dSAndroid Build Coastguard Worker           + factor_4_latency * factor_3_accesscount; // reward slow + repeatedly read files
1629*7304104dSAndroid Build Coastguard Worker 
1630*7304104dSAndroid Build Coastguard Worker         if (verbose > 4)
1631*7304104dSAndroid Build Coastguard Worker           obatched(clog) << "fdcache scored score=" << score
1632*7304104dSAndroid Build Coastguard Worker                          << " a=" << i.first.first << " b=" << i.first.second
1633*7304104dSAndroid Build Coastguard Worker                          << " f1=" << factor_1_freshness << " f2=" << factor_2_size
1634*7304104dSAndroid Build Coastguard Worker                          << " f3=" << factor_3_accesscount << " f4=" << factor_4_latency
1635*7304104dSAndroid Build Coastguard Worker                          << endl;
1636*7304104dSAndroid Build Coastguard Worker 
1637*7304104dSAndroid Build Coastguard Worker         sorted_entries.insert(make_pair(score, i.first));
1638*7304104dSAndroid Build Coastguard Worker       }
1639*7304104dSAndroid Build Coastguard Worker 
1640*7304104dSAndroid Build Coastguard Worker     unsigned cleaned = 0;
1641*7304104dSAndroid Build Coastguard Worker     unsigned entries_original = entries.size();
1642*7304104dSAndroid Build Coastguard Worker     double cleaned_score_min = DBL_MAX;
1643*7304104dSAndroid Build Coastguard Worker     double cleaned_score_max = DBL_MIN;
1644*7304104dSAndroid Build Coastguard Worker 
1645*7304104dSAndroid Build Coastguard Worker     // drop as many entries[] as needed to bring total mb down to the threshold
1646*7304104dSAndroid Build Coastguard Worker     for (auto &i: sorted_entries) // in increasing score order!
1647*7304104dSAndroid Build Coastguard Worker       {
1648*7304104dSAndroid Build Coastguard Worker         if (this->max_mbs > 0 // if this is not a "clear entire table"
1649*7304104dSAndroid Build Coastguard Worker             && total_mb < this->max_mbs) // we've cleared enough to meet threshold
1650*7304104dSAndroid Build Coastguard Worker           break; // stop clearing
1651*7304104dSAndroid Build Coastguard Worker 
1652*7304104dSAndroid Build Coastguard Worker         auto j = entries.find(i.second);
1653*7304104dSAndroid Build Coastguard Worker         if (j == entries.end())
1654*7304104dSAndroid Build Coastguard Worker           continue; // should not happen
1655*7304104dSAndroid Build Coastguard Worker 
1656*7304104dSAndroid Build Coastguard Worker         if (cleaned == 0)
1657*7304104dSAndroid Build Coastguard Worker           cleaned_score_min = i.first;
1658*7304104dSAndroid Build Coastguard Worker         cleaned++;
1659*7304104dSAndroid Build Coastguard Worker         cleaned_score_max = i.first;
1660*7304104dSAndroid Build Coastguard Worker 
1661*7304104dSAndroid Build Coastguard Worker         if (verbose > 3)
1662*7304104dSAndroid Build Coastguard Worker           obatched(clog) << "fdcache evicted score=" << i.first
1663*7304104dSAndroid Build Coastguard Worker                          << " a=" << i.second.first << " b=" << i.second.second
1664*7304104dSAndroid Build Coastguard Worker                          << " fd=" << j->second.fd << " mb=" << j->second.fd_size_mb
1665*7304104dSAndroid Build Coastguard Worker                          << " rq=" << j->second.request_count << " lat=" << j->second.latency
1666*7304104dSAndroid Build Coastguard Worker                          << " fr=" << (now - j->second.freshness)
1667*7304104dSAndroid Build Coastguard Worker                          << endl;
1668*7304104dSAndroid Build Coastguard Worker         if (metrics_p)
1669*7304104dSAndroid Build Coastguard Worker           inc_metric("fdcache_op_count","op","evict");
1670*7304104dSAndroid Build Coastguard Worker 
1671*7304104dSAndroid Build Coastguard Worker         total_mb -= j->second.fd_size_mb;
1672*7304104dSAndroid Build Coastguard Worker         unlink (j->second.fd.c_str());
1673*7304104dSAndroid Build Coastguard Worker         entries.erase(j);
1674*7304104dSAndroid Build Coastguard Worker       }
1675*7304104dSAndroid Build Coastguard Worker 
1676*7304104dSAndroid Build Coastguard Worker     if (metrics_p)
1677*7304104dSAndroid Build Coastguard Worker       inc_metric("fdcache_op_count","op","evict_cycle");
1678*7304104dSAndroid Build Coastguard Worker 
1679*7304104dSAndroid Build Coastguard Worker     if (verbose > 1 && cleaned > 0)
1680*7304104dSAndroid Build Coastguard Worker       {
1681*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "fdcache evicted num=" << cleaned << " of=" << entries_original
1682*7304104dSAndroid Build Coastguard Worker                        << " min=" << cleaned_score_min << " max=" << cleaned_score_max
1683*7304104dSAndroid Build Coastguard Worker                        << endl;
1684*7304104dSAndroid Build Coastguard Worker       }
1685*7304104dSAndroid Build Coastguard Worker 
1686*7304104dSAndroid Build Coastguard Worker     if (metrics_p) set_metrics();
1687*7304104dSAndroid Build Coastguard Worker   }
1688*7304104dSAndroid Build Coastguard Worker 
1689*7304104dSAndroid Build Coastguard Worker 
~libarchive_fdcache()1690*7304104dSAndroid Build Coastguard Worker   ~libarchive_fdcache()
1691*7304104dSAndroid Build Coastguard Worker   {
1692*7304104dSAndroid Build Coastguard Worker     // unlink any fdcache entries in $TMPDIR
1693*7304104dSAndroid Build Coastguard Worker     // don't update metrics; those globals may be already destroyed
1694*7304104dSAndroid Build Coastguard Worker     limit(0, false);
1695*7304104dSAndroid Build Coastguard Worker   }
1696*7304104dSAndroid Build Coastguard Worker };
1697*7304104dSAndroid Build Coastguard Worker static libarchive_fdcache fdcache;
1698*7304104dSAndroid Build Coastguard Worker 
1699*7304104dSAndroid Build Coastguard Worker /* Search ELF_FD for an ELF/DWARF section with name SECTION.
1700*7304104dSAndroid Build Coastguard Worker    If found copy the section to a temporary file and return
1701*7304104dSAndroid Build Coastguard Worker    its file descriptor, otherwise return -1.
1702*7304104dSAndroid Build Coastguard Worker 
1703*7304104dSAndroid Build Coastguard Worker    The temporary file's mtime will be set to PARENT_MTIME.
1704*7304104dSAndroid Build Coastguard Worker    B_SOURCE should be a description of the parent file suitable
1705*7304104dSAndroid Build Coastguard Worker    for printing to the log.  */
1706*7304104dSAndroid Build Coastguard Worker 
1707*7304104dSAndroid Build Coastguard Worker static int
extract_section(int elf_fd,int64_t parent_mtime,const string & b_source,const string & section,const timespec & extract_begin)1708*7304104dSAndroid Build Coastguard Worker extract_section (int elf_fd, int64_t parent_mtime,
1709*7304104dSAndroid Build Coastguard Worker 		 const string& b_source, const string& section,
1710*7304104dSAndroid Build Coastguard Worker                  const timespec& extract_begin)
1711*7304104dSAndroid Build Coastguard Worker {
1712*7304104dSAndroid Build Coastguard Worker   /* Search the fdcache.  */
1713*7304104dSAndroid Build Coastguard Worker   struct stat fs;
1714*7304104dSAndroid Build Coastguard Worker   int fd = fdcache.lookup (b_source, section);
1715*7304104dSAndroid Build Coastguard Worker   if (fd >= 0)
1716*7304104dSAndroid Build Coastguard Worker     {
1717*7304104dSAndroid Build Coastguard Worker       if (fstat (fd, &fs) != 0)
1718*7304104dSAndroid Build Coastguard Worker 	{
1719*7304104dSAndroid Build Coastguard Worker 	  if (verbose)
1720*7304104dSAndroid Build Coastguard Worker 	    obatched (clog) << "cannot fstate fdcache "
1721*7304104dSAndroid Build Coastguard Worker 			    << b_source << " " << section << endl;
1722*7304104dSAndroid Build Coastguard Worker 	  close (fd);
1723*7304104dSAndroid Build Coastguard Worker 	  return -1;
1724*7304104dSAndroid Build Coastguard Worker 	}
1725*7304104dSAndroid Build Coastguard Worker       if ((int64_t) fs.st_mtime != parent_mtime)
1726*7304104dSAndroid Build Coastguard Worker 	{
1727*7304104dSAndroid Build Coastguard Worker 	  if (verbose)
1728*7304104dSAndroid Build Coastguard Worker 	    obatched(clog) << "mtime mismatch for "
1729*7304104dSAndroid Build Coastguard Worker 			   << b_source << " " << section << endl;
1730*7304104dSAndroid Build Coastguard Worker 	  close (fd);
1731*7304104dSAndroid Build Coastguard Worker 	  return -1;
1732*7304104dSAndroid Build Coastguard Worker 	}
1733*7304104dSAndroid Build Coastguard Worker       /* Success.  */
1734*7304104dSAndroid Build Coastguard Worker       return fd;
1735*7304104dSAndroid Build Coastguard Worker     }
1736*7304104dSAndroid Build Coastguard Worker 
1737*7304104dSAndroid Build Coastguard Worker   Elf *elf = elf_begin (elf_fd, ELF_C_READ_MMAP_PRIVATE, NULL);
1738*7304104dSAndroid Build Coastguard Worker   if (elf == NULL)
1739*7304104dSAndroid Build Coastguard Worker     return -1;
1740*7304104dSAndroid Build Coastguard Worker 
1741*7304104dSAndroid Build Coastguard Worker   /* Try to find the section and copy the contents into a separate file.  */
1742*7304104dSAndroid Build Coastguard Worker   try
1743*7304104dSAndroid Build Coastguard Worker     {
1744*7304104dSAndroid Build Coastguard Worker       size_t shstrndx;
1745*7304104dSAndroid Build Coastguard Worker       int rc = elf_getshdrstrndx (elf, &shstrndx);
1746*7304104dSAndroid Build Coastguard Worker       if (rc < 0)
1747*7304104dSAndroid Build Coastguard Worker 	throw elfutils_exception (rc, "getshdrstrndx");
1748*7304104dSAndroid Build Coastguard Worker 
1749*7304104dSAndroid Build Coastguard Worker       Elf_Scn *scn = NULL;
1750*7304104dSAndroid Build Coastguard Worker       while (true)
1751*7304104dSAndroid Build Coastguard Worker 	{
1752*7304104dSAndroid Build Coastguard Worker 	  scn = elf_nextscn (elf, scn);
1753*7304104dSAndroid Build Coastguard Worker 	  if (scn == NULL)
1754*7304104dSAndroid Build Coastguard Worker 	    break;
1755*7304104dSAndroid Build Coastguard Worker 	  GElf_Shdr shdr_storage;
1756*7304104dSAndroid Build Coastguard Worker 	  GElf_Shdr *shdr = gelf_getshdr (scn, &shdr_storage);
1757*7304104dSAndroid Build Coastguard Worker 	  if (shdr == NULL)
1758*7304104dSAndroid Build Coastguard Worker 	    break;
1759*7304104dSAndroid Build Coastguard Worker 
1760*7304104dSAndroid Build Coastguard Worker 	  const char *scn_name = elf_strptr (elf, shstrndx, shdr->sh_name);
1761*7304104dSAndroid Build Coastguard Worker 	  if (scn_name == NULL)
1762*7304104dSAndroid Build Coastguard Worker 	    break;
1763*7304104dSAndroid Build Coastguard Worker 	  if (scn_name == section)
1764*7304104dSAndroid Build Coastguard Worker 	    {
1765*7304104dSAndroid Build Coastguard Worker 	      Elf_Data *data = NULL;
1766*7304104dSAndroid Build Coastguard Worker 
1767*7304104dSAndroid Build Coastguard Worker 	      /* We found the desired section.  */
1768*7304104dSAndroid Build Coastguard Worker 	      data = elf_rawdata (scn, NULL);
1769*7304104dSAndroid Build Coastguard Worker 	      if (data == NULL)
1770*7304104dSAndroid Build Coastguard Worker 		throw elfutils_exception (elf_errno (), "elfraw_data");
1771*7304104dSAndroid Build Coastguard Worker 	      if (data->d_buf == NULL)
1772*7304104dSAndroid Build Coastguard Worker 		{
1773*7304104dSAndroid Build Coastguard Worker 		  obatched(clog) << "section " << section
1774*7304104dSAndroid Build Coastguard Worker 				 << " is empty" << endl;
1775*7304104dSAndroid Build Coastguard Worker 		  break;
1776*7304104dSAndroid Build Coastguard Worker 		}
1777*7304104dSAndroid Build Coastguard Worker 
1778*7304104dSAndroid Build Coastguard Worker 	      /* Create temporary file containing the section.  */
1779*7304104dSAndroid Build Coastguard Worker 	      char *tmppath = NULL;
1780*7304104dSAndroid Build Coastguard Worker 	      rc = asprintf (&tmppath, "%s/debuginfod-section.XXXXXX", tmpdir.c_str());
1781*7304104dSAndroid Build Coastguard Worker 	      if (rc < 0)
1782*7304104dSAndroid Build Coastguard Worker 		throw libc_exception (ENOMEM, "cannot allocate tmppath");
1783*7304104dSAndroid Build Coastguard Worker 	      defer_dtor<void*,void> tmmpath_freer (tmppath, free);
1784*7304104dSAndroid Build Coastguard Worker 	      fd = mkstemp (tmppath);
1785*7304104dSAndroid Build Coastguard Worker 	      if (fd < 0)
1786*7304104dSAndroid Build Coastguard Worker 		throw libc_exception (errno, "cannot create temporary file");
1787*7304104dSAndroid Build Coastguard Worker 
1788*7304104dSAndroid Build Coastguard Worker 	      ssize_t res = write_retry (fd, data->d_buf, data->d_size);
1789*7304104dSAndroid Build Coastguard Worker 	      if (res < 0 || (size_t) res != data->d_size) {
1790*7304104dSAndroid Build Coastguard Worker                 close (fd);
1791*7304104dSAndroid Build Coastguard Worker                 unlink (tmppath);
1792*7304104dSAndroid Build Coastguard Worker 		throw libc_exception (errno, "cannot write to temporary file");
1793*7304104dSAndroid Build Coastguard Worker               }
1794*7304104dSAndroid Build Coastguard Worker 
1795*7304104dSAndroid Build Coastguard Worker 	      /* Set mtime to be the same as the parent file's mtime.  */
1796*7304104dSAndroid Build Coastguard Worker 	      struct timespec tvs[2];
1797*7304104dSAndroid Build Coastguard Worker 	      if (fstat (elf_fd, &fs) != 0) {
1798*7304104dSAndroid Build Coastguard Worker                 close (fd);
1799*7304104dSAndroid Build Coastguard Worker                 unlink (tmppath);
1800*7304104dSAndroid Build Coastguard Worker 		throw libc_exception (errno, "cannot fstat file");
1801*7304104dSAndroid Build Coastguard Worker               }
1802*7304104dSAndroid Build Coastguard Worker 
1803*7304104dSAndroid Build Coastguard Worker 	      tvs[0].tv_sec = 0;
1804*7304104dSAndroid Build Coastguard Worker 	      tvs[0].tv_nsec = UTIME_OMIT;
1805*7304104dSAndroid Build Coastguard Worker 	      tvs[1] = fs.st_mtim;
1806*7304104dSAndroid Build Coastguard Worker 	      (void) futimens (fd, tvs);
1807*7304104dSAndroid Build Coastguard Worker 
1808*7304104dSAndroid Build Coastguard Worker               struct timespec extract_end;
1809*7304104dSAndroid Build Coastguard Worker               clock_gettime (CLOCK_MONOTONIC, &extract_end);
1810*7304104dSAndroid Build Coastguard Worker               double extract_time = (extract_end.tv_sec - extract_begin.tv_sec)
1811*7304104dSAndroid Build Coastguard Worker                 + (extract_end.tv_nsec - extract_begin.tv_nsec)/1.e9;
1812*7304104dSAndroid Build Coastguard Worker 
1813*7304104dSAndroid Build Coastguard Worker 	      /* Add to fdcache.  */
1814*7304104dSAndroid Build Coastguard Worker 	      fdcache.intern (b_source, section, tmppath, data->d_size, true, extract_time);
1815*7304104dSAndroid Build Coastguard Worker 	      break;
1816*7304104dSAndroid Build Coastguard Worker 	    }
1817*7304104dSAndroid Build Coastguard Worker 	}
1818*7304104dSAndroid Build Coastguard Worker     }
1819*7304104dSAndroid Build Coastguard Worker   catch (const reportable_exception &e)
1820*7304104dSAndroid Build Coastguard Worker     {
1821*7304104dSAndroid Build Coastguard Worker       e.report (clog);
1822*7304104dSAndroid Build Coastguard Worker       close (fd);
1823*7304104dSAndroid Build Coastguard Worker       fd = -1;
1824*7304104dSAndroid Build Coastguard Worker     }
1825*7304104dSAndroid Build Coastguard Worker 
1826*7304104dSAndroid Build Coastguard Worker   elf_end (elf);
1827*7304104dSAndroid Build Coastguard Worker   return fd;
1828*7304104dSAndroid Build Coastguard Worker }
1829*7304104dSAndroid Build Coastguard Worker 
1830*7304104dSAndroid Build Coastguard Worker static struct MHD_Response*
handle_buildid_f_match(bool internal_req_t,int64_t b_mtime,const string & b_source0,const string & section,int * result_fd)1831*7304104dSAndroid Build Coastguard Worker handle_buildid_f_match (bool internal_req_t,
1832*7304104dSAndroid Build Coastguard Worker                         int64_t b_mtime,
1833*7304104dSAndroid Build Coastguard Worker                         const string& b_source0,
1834*7304104dSAndroid Build Coastguard Worker                         const string& section,
1835*7304104dSAndroid Build Coastguard Worker                         int *result_fd)
1836*7304104dSAndroid Build Coastguard Worker {
1837*7304104dSAndroid Build Coastguard Worker   (void) internal_req_t; // ignored
1838*7304104dSAndroid Build Coastguard Worker 
1839*7304104dSAndroid Build Coastguard Worker   struct timespec extract_begin;
1840*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &extract_begin);
1841*7304104dSAndroid Build Coastguard Worker 
1842*7304104dSAndroid Build Coastguard Worker   int fd = open(b_source0.c_str(), O_RDONLY);
1843*7304104dSAndroid Build Coastguard Worker   if (fd < 0)
1844*7304104dSAndroid Build Coastguard Worker     throw libc_exception (errno, string("open ") + b_source0);
1845*7304104dSAndroid Build Coastguard Worker 
1846*7304104dSAndroid Build Coastguard Worker   // NB: use manual close(2) in error case instead of defer_dtor, because
1847*7304104dSAndroid Build Coastguard Worker   // in the normal case, we want to hand the fd over to libmicrohttpd for
1848*7304104dSAndroid Build Coastguard Worker   // file transfer.
1849*7304104dSAndroid Build Coastguard Worker 
1850*7304104dSAndroid Build Coastguard Worker   struct stat s;
1851*7304104dSAndroid Build Coastguard Worker   int rc = fstat(fd, &s);
1852*7304104dSAndroid Build Coastguard Worker   if (rc < 0)
1853*7304104dSAndroid Build Coastguard Worker     {
1854*7304104dSAndroid Build Coastguard Worker       close(fd);
1855*7304104dSAndroid Build Coastguard Worker       throw libc_exception (errno, string("fstat ") + b_source0);
1856*7304104dSAndroid Build Coastguard Worker     }
1857*7304104dSAndroid Build Coastguard Worker 
1858*7304104dSAndroid Build Coastguard Worker   if ((int64_t) s.st_mtime != b_mtime)
1859*7304104dSAndroid Build Coastguard Worker     {
1860*7304104dSAndroid Build Coastguard Worker       if (verbose)
1861*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "mtime mismatch for " << b_source0 << endl;
1862*7304104dSAndroid Build Coastguard Worker       close(fd);
1863*7304104dSAndroid Build Coastguard Worker       return 0;
1864*7304104dSAndroid Build Coastguard Worker     }
1865*7304104dSAndroid Build Coastguard Worker 
1866*7304104dSAndroid Build Coastguard Worker   if (!section.empty ())
1867*7304104dSAndroid Build Coastguard Worker     {
1868*7304104dSAndroid Build Coastguard Worker       int scn_fd = extract_section (fd, s.st_mtime, b_source0, section, extract_begin);
1869*7304104dSAndroid Build Coastguard Worker       close (fd);
1870*7304104dSAndroid Build Coastguard Worker 
1871*7304104dSAndroid Build Coastguard Worker       if (scn_fd >= 0)
1872*7304104dSAndroid Build Coastguard Worker 	fd = scn_fd;
1873*7304104dSAndroid Build Coastguard Worker       else
1874*7304104dSAndroid Build Coastguard Worker 	{
1875*7304104dSAndroid Build Coastguard Worker 	  if (verbose)
1876*7304104dSAndroid Build Coastguard Worker 	    obatched (clog) << "cannot find section " << section
1877*7304104dSAndroid Build Coastguard Worker 			    << " for " << b_source0 << endl;
1878*7304104dSAndroid Build Coastguard Worker 	  return 0;
1879*7304104dSAndroid Build Coastguard Worker 	}
1880*7304104dSAndroid Build Coastguard Worker 
1881*7304104dSAndroid Build Coastguard Worker       rc = fstat(fd, &s);
1882*7304104dSAndroid Build Coastguard Worker       if (rc < 0)
1883*7304104dSAndroid Build Coastguard Worker 	{
1884*7304104dSAndroid Build Coastguard Worker 	  close (fd);
1885*7304104dSAndroid Build Coastguard Worker 	  throw libc_exception (errno, string ("fstat ") + b_source0
1886*7304104dSAndroid Build Coastguard Worker 				       + string (" ") + section);
1887*7304104dSAndroid Build Coastguard Worker 	}
1888*7304104dSAndroid Build Coastguard Worker     }
1889*7304104dSAndroid Build Coastguard Worker 
1890*7304104dSAndroid Build Coastguard Worker   struct MHD_Response* r = MHD_create_response_from_fd ((uint64_t) s.st_size, fd);
1891*7304104dSAndroid Build Coastguard Worker   inc_metric ("http_responses_total","result","file");
1892*7304104dSAndroid Build Coastguard Worker   if (r == 0)
1893*7304104dSAndroid Build Coastguard Worker     {
1894*7304104dSAndroid Build Coastguard Worker       if (verbose)
1895*7304104dSAndroid Build Coastguard Worker 	obatched(clog) << "cannot create fd-response for " << b_source0
1896*7304104dSAndroid Build Coastguard Worker 		       << " section=" << section << endl;
1897*7304104dSAndroid Build Coastguard Worker       close(fd);
1898*7304104dSAndroid Build Coastguard Worker     }
1899*7304104dSAndroid Build Coastguard Worker   else
1900*7304104dSAndroid Build Coastguard Worker     {
1901*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "Content-Type", "application/octet-stream");
1902*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "X-DEBUGINFOD-SIZE",
1903*7304104dSAndroid Build Coastguard Worker 			       to_string(s.st_size).c_str());
1904*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "X-DEBUGINFOD-FILE", b_source0.c_str());
1905*7304104dSAndroid Build Coastguard Worker       add_mhd_last_modified (r, s.st_mtime);
1906*7304104dSAndroid Build Coastguard Worker       if (verbose > 1)
1907*7304104dSAndroid Build Coastguard Worker 	obatched(clog) << "serving file " << b_source0 << " section=" << section << endl;
1908*7304104dSAndroid Build Coastguard Worker       /* libmicrohttpd will close it. */
1909*7304104dSAndroid Build Coastguard Worker       if (result_fd)
1910*7304104dSAndroid Build Coastguard Worker         *result_fd = fd;
1911*7304104dSAndroid Build Coastguard Worker     }
1912*7304104dSAndroid Build Coastguard Worker 
1913*7304104dSAndroid Build Coastguard Worker   return r;
1914*7304104dSAndroid Build Coastguard Worker }
1915*7304104dSAndroid Build Coastguard Worker 
1916*7304104dSAndroid Build Coastguard Worker // For security/portability reasons, many distro-package archives have
1917*7304104dSAndroid Build Coastguard Worker // a "./" in front of path names; others have nothing, others have
1918*7304104dSAndroid Build Coastguard Worker // "/".  Canonicalize them all to a single leading "/", with the
1919*7304104dSAndroid Build Coastguard Worker // assumption that this matches the dwarf-derived file names too.
canonicalized_archive_entry_pathname(struct archive_entry * e)1920*7304104dSAndroid Build Coastguard Worker string canonicalized_archive_entry_pathname(struct archive_entry *e)
1921*7304104dSAndroid Build Coastguard Worker {
1922*7304104dSAndroid Build Coastguard Worker   string fn = archive_entry_pathname(e);
1923*7304104dSAndroid Build Coastguard Worker   if (fn.size() == 0)
1924*7304104dSAndroid Build Coastguard Worker     return fn;
1925*7304104dSAndroid Build Coastguard Worker   if (fn[0] == '/')
1926*7304104dSAndroid Build Coastguard Worker     return fn;
1927*7304104dSAndroid Build Coastguard Worker   if (fn[0] == '.')
1928*7304104dSAndroid Build Coastguard Worker     return fn.substr(1);
1929*7304104dSAndroid Build Coastguard Worker   else
1930*7304104dSAndroid Build Coastguard Worker     return string("/")+fn;
1931*7304104dSAndroid Build Coastguard Worker }
1932*7304104dSAndroid Build Coastguard Worker 
1933*7304104dSAndroid Build Coastguard Worker 
1934*7304104dSAndroid Build Coastguard Worker 
1935*7304104dSAndroid Build Coastguard Worker static struct MHD_Response*
handle_buildid_r_match(bool internal_req_p,int64_t b_mtime,const string & b_source0,const string & b_source1,const string & section,int * result_fd)1936*7304104dSAndroid Build Coastguard Worker handle_buildid_r_match (bool internal_req_p,
1937*7304104dSAndroid Build Coastguard Worker                         int64_t b_mtime,
1938*7304104dSAndroid Build Coastguard Worker                         const string& b_source0,
1939*7304104dSAndroid Build Coastguard Worker                         const string& b_source1,
1940*7304104dSAndroid Build Coastguard Worker                         const string& section,
1941*7304104dSAndroid Build Coastguard Worker                         int *result_fd)
1942*7304104dSAndroid Build Coastguard Worker {
1943*7304104dSAndroid Build Coastguard Worker   struct timespec extract_begin;
1944*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &extract_begin);
1945*7304104dSAndroid Build Coastguard Worker 
1946*7304104dSAndroid Build Coastguard Worker   struct stat fs;
1947*7304104dSAndroid Build Coastguard Worker   int rc = stat (b_source0.c_str(), &fs);
1948*7304104dSAndroid Build Coastguard Worker   if (rc != 0)
1949*7304104dSAndroid Build Coastguard Worker     throw libc_exception (errno, string("stat ") + b_source0);
1950*7304104dSAndroid Build Coastguard Worker 
1951*7304104dSAndroid Build Coastguard Worker   if ((int64_t) fs.st_mtime != b_mtime)
1952*7304104dSAndroid Build Coastguard Worker     {
1953*7304104dSAndroid Build Coastguard Worker       if (verbose)
1954*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "mtime mismatch for " << b_source0 << endl;
1955*7304104dSAndroid Build Coastguard Worker       return 0;
1956*7304104dSAndroid Build Coastguard Worker     }
1957*7304104dSAndroid Build Coastguard Worker 
1958*7304104dSAndroid Build Coastguard Worker   // check for a match in the fdcache first
1959*7304104dSAndroid Build Coastguard Worker   int fd = fdcache.lookup(b_source0, b_source1);
1960*7304104dSAndroid Build Coastguard Worker   while (fd >= 0) // got one!; NB: this is really an if() with a possible branch out to the end
1961*7304104dSAndroid Build Coastguard Worker     {
1962*7304104dSAndroid Build Coastguard Worker       rc = fstat(fd, &fs);
1963*7304104dSAndroid Build Coastguard Worker       if (rc < 0) // disappeared?
1964*7304104dSAndroid Build Coastguard Worker         {
1965*7304104dSAndroid Build Coastguard Worker           if (verbose)
1966*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "cannot fstat fdcache " << b_source0 << endl;
1967*7304104dSAndroid Build Coastguard Worker           close(fd);
1968*7304104dSAndroid Build Coastguard Worker           fdcache.clear(b_source0, b_source1);
1969*7304104dSAndroid Build Coastguard Worker           break; // branch out of if "loop", to try new libarchive fetch attempt
1970*7304104dSAndroid Build Coastguard Worker         }
1971*7304104dSAndroid Build Coastguard Worker 
1972*7304104dSAndroid Build Coastguard Worker       if (!section.empty ())
1973*7304104dSAndroid Build Coastguard Worker 	{
1974*7304104dSAndroid Build Coastguard Worker 	  int scn_fd = extract_section (fd, fs.st_mtime,
1975*7304104dSAndroid Build Coastguard Worker 					b_source0 + ":" + b_source1,
1976*7304104dSAndroid Build Coastguard Worker 					section, extract_begin);
1977*7304104dSAndroid Build Coastguard Worker 	  close (fd);
1978*7304104dSAndroid Build Coastguard Worker 	  if (scn_fd >= 0)
1979*7304104dSAndroid Build Coastguard Worker 	    fd = scn_fd;
1980*7304104dSAndroid Build Coastguard Worker 	  else
1981*7304104dSAndroid Build Coastguard Worker 	    {
1982*7304104dSAndroid Build Coastguard Worker 	      if (verbose)
1983*7304104dSAndroid Build Coastguard Worker 	        obatched (clog) << "cannot find section " << section
1984*7304104dSAndroid Build Coastguard Worker 				<< " for archive " << b_source0
1985*7304104dSAndroid Build Coastguard Worker 				<< " file " << b_source1 << endl;
1986*7304104dSAndroid Build Coastguard Worker 	      return 0;
1987*7304104dSAndroid Build Coastguard Worker 	    }
1988*7304104dSAndroid Build Coastguard Worker 
1989*7304104dSAndroid Build Coastguard Worker 	  rc = fstat(fd, &fs);
1990*7304104dSAndroid Build Coastguard Worker 	  if (rc < 0)
1991*7304104dSAndroid Build Coastguard Worker 	    {
1992*7304104dSAndroid Build Coastguard Worker 	      close (fd);
1993*7304104dSAndroid Build Coastguard Worker 	      throw libc_exception (errno,
1994*7304104dSAndroid Build Coastguard Worker 		string ("fstat archive ") + b_source0 + string (" file ") + b_source1
1995*7304104dSAndroid Build Coastguard Worker 		+ string (" section ") + section);
1996*7304104dSAndroid Build Coastguard Worker 	    }
1997*7304104dSAndroid Build Coastguard Worker 	}
1998*7304104dSAndroid Build Coastguard Worker 
1999*7304104dSAndroid Build Coastguard Worker       struct MHD_Response* r = MHD_create_response_from_fd (fs.st_size, fd);
2000*7304104dSAndroid Build Coastguard Worker       if (r == 0)
2001*7304104dSAndroid Build Coastguard Worker         {
2002*7304104dSAndroid Build Coastguard Worker           if (verbose)
2003*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "cannot create fd-response for " << b_source0 << endl;
2004*7304104dSAndroid Build Coastguard Worker           close(fd);
2005*7304104dSAndroid Build Coastguard Worker           break; // branch out of if "loop", to try new libarchive fetch attempt
2006*7304104dSAndroid Build Coastguard Worker         }
2007*7304104dSAndroid Build Coastguard Worker 
2008*7304104dSAndroid Build Coastguard Worker       inc_metric ("http_responses_total","result","archive fdcache");
2009*7304104dSAndroid Build Coastguard Worker 
2010*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "Content-Type", "application/octet-stream");
2011*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "X-DEBUGINFOD-SIZE",
2012*7304104dSAndroid Build Coastguard Worker 			       to_string(fs.st_size).c_str());
2013*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "X-DEBUGINFOD-ARCHIVE", b_source0.c_str());
2014*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "X-DEBUGINFOD-FILE", b_source1.c_str());
2015*7304104dSAndroid Build Coastguard Worker       add_mhd_last_modified (r, fs.st_mtime);
2016*7304104dSAndroid Build Coastguard Worker       if (verbose > 1)
2017*7304104dSAndroid Build Coastguard Worker 	obatched(clog) << "serving fdcache archive " << b_source0
2018*7304104dSAndroid Build Coastguard Worker 		       << " file " << b_source1
2019*7304104dSAndroid Build Coastguard Worker 		       << " section=" << section << endl;
2020*7304104dSAndroid Build Coastguard Worker       /* libmicrohttpd will close it. */
2021*7304104dSAndroid Build Coastguard Worker       if (result_fd)
2022*7304104dSAndroid Build Coastguard Worker         *result_fd = fd;
2023*7304104dSAndroid Build Coastguard Worker       return r;
2024*7304104dSAndroid Build Coastguard Worker       // NB: see, we never go around the 'loop' more than once
2025*7304104dSAndroid Build Coastguard Worker     }
2026*7304104dSAndroid Build Coastguard Worker 
2027*7304104dSAndroid Build Coastguard Worker   // no match ... grumble, must process the archive
2028*7304104dSAndroid Build Coastguard Worker   string archive_decoder = "/dev/null";
2029*7304104dSAndroid Build Coastguard Worker   string archive_extension = "";
2030*7304104dSAndroid Build Coastguard Worker   for (auto&& arch : scan_archives)
2031*7304104dSAndroid Build Coastguard Worker     if (string_endswith(b_source0, arch.first))
2032*7304104dSAndroid Build Coastguard Worker       {
2033*7304104dSAndroid Build Coastguard Worker         archive_extension = arch.first;
2034*7304104dSAndroid Build Coastguard Worker         archive_decoder = arch.second;
2035*7304104dSAndroid Build Coastguard Worker       }
2036*7304104dSAndroid Build Coastguard Worker   FILE* fp;
2037*7304104dSAndroid Build Coastguard Worker 
2038*7304104dSAndroid Build Coastguard Worker   defer_dtor<FILE*,int>::dtor_fn dfn;
2039*7304104dSAndroid Build Coastguard Worker   if (archive_decoder != "cat")
2040*7304104dSAndroid Build Coastguard Worker     {
2041*7304104dSAndroid Build Coastguard Worker       string popen_cmd = archive_decoder + " " + shell_escape(b_source0);
2042*7304104dSAndroid Build Coastguard Worker       fp = popen (popen_cmd.c_str(), "r"); // "e" O_CLOEXEC?
2043*7304104dSAndroid Build Coastguard Worker       dfn = pclose;
2044*7304104dSAndroid Build Coastguard Worker       if (fp == NULL)
2045*7304104dSAndroid Build Coastguard Worker         throw libc_exception (errno, string("popen ") + popen_cmd);
2046*7304104dSAndroid Build Coastguard Worker     }
2047*7304104dSAndroid Build Coastguard Worker   else
2048*7304104dSAndroid Build Coastguard Worker     {
2049*7304104dSAndroid Build Coastguard Worker       fp = fopen (b_source0.c_str(), "r");
2050*7304104dSAndroid Build Coastguard Worker       dfn = fclose;
2051*7304104dSAndroid Build Coastguard Worker       if (fp == NULL)
2052*7304104dSAndroid Build Coastguard Worker         throw libc_exception (errno, string("fopen ") + b_source0);
2053*7304104dSAndroid Build Coastguard Worker     }
2054*7304104dSAndroid Build Coastguard Worker   defer_dtor<FILE*,int> fp_closer (fp, dfn);
2055*7304104dSAndroid Build Coastguard Worker 
2056*7304104dSAndroid Build Coastguard Worker   struct archive *a;
2057*7304104dSAndroid Build Coastguard Worker   a = archive_read_new();
2058*7304104dSAndroid Build Coastguard Worker   if (a == NULL)
2059*7304104dSAndroid Build Coastguard Worker     throw archive_exception("cannot create archive reader");
2060*7304104dSAndroid Build Coastguard Worker   defer_dtor<struct archive*,int> archive_closer (a, archive_read_free);
2061*7304104dSAndroid Build Coastguard Worker 
2062*7304104dSAndroid Build Coastguard Worker   rc = archive_read_support_format_all(a);
2063*7304104dSAndroid Build Coastguard Worker   if (rc != ARCHIVE_OK)
2064*7304104dSAndroid Build Coastguard Worker     throw archive_exception(a, "cannot select all format");
2065*7304104dSAndroid Build Coastguard Worker   rc = archive_read_support_filter_all(a);
2066*7304104dSAndroid Build Coastguard Worker   if (rc != ARCHIVE_OK)
2067*7304104dSAndroid Build Coastguard Worker     throw archive_exception(a, "cannot select all filters");
2068*7304104dSAndroid Build Coastguard Worker 
2069*7304104dSAndroid Build Coastguard Worker   rc = archive_read_open_FILE (a, fp);
2070*7304104dSAndroid Build Coastguard Worker   if (rc != ARCHIVE_OK)
2071*7304104dSAndroid Build Coastguard Worker     {
2072*7304104dSAndroid Build Coastguard Worker       obatched(clog) << "cannot open archive from pipe " << b_source0 << endl;
2073*7304104dSAndroid Build Coastguard Worker       throw archive_exception(a, "cannot open archive from pipe");
2074*7304104dSAndroid Build Coastguard Worker     }
2075*7304104dSAndroid Build Coastguard Worker 
2076*7304104dSAndroid Build Coastguard Worker   // archive traversal is in three stages, no, four stages:
2077*7304104dSAndroid Build Coastguard Worker   // 1) skip entries whose names do not match the requested one
2078*7304104dSAndroid Build Coastguard Worker   // 2) extract the matching entry name (set r = result)
2079*7304104dSAndroid Build Coastguard Worker   // 3) extract some number of prefetched entries (just into fdcache)
2080*7304104dSAndroid Build Coastguard Worker   // 4) abort any further processing
2081*7304104dSAndroid Build Coastguard Worker   struct MHD_Response* r = 0;                 // will set in stage 2
2082*7304104dSAndroid Build Coastguard Worker   unsigned prefetch_count =
2083*7304104dSAndroid Build Coastguard Worker     internal_req_p ? 0 : fdcache_prefetch;    // will decrement in stage 3
2084*7304104dSAndroid Build Coastguard Worker 
2085*7304104dSAndroid Build Coastguard Worker   while(r == 0 || prefetch_count > 0) // stage 1, 2, or 3
2086*7304104dSAndroid Build Coastguard Worker     {
2087*7304104dSAndroid Build Coastguard Worker       if (interrupted)
2088*7304104dSAndroid Build Coastguard Worker         break;
2089*7304104dSAndroid Build Coastguard Worker 
2090*7304104dSAndroid Build Coastguard Worker       struct archive_entry *e;
2091*7304104dSAndroid Build Coastguard Worker       rc = archive_read_next_header (a, &e);
2092*7304104dSAndroid Build Coastguard Worker       if (rc != ARCHIVE_OK)
2093*7304104dSAndroid Build Coastguard Worker         break;
2094*7304104dSAndroid Build Coastguard Worker 
2095*7304104dSAndroid Build Coastguard Worker       if (! S_ISREG(archive_entry_mode (e))) // skip non-files completely
2096*7304104dSAndroid Build Coastguard Worker         continue;
2097*7304104dSAndroid Build Coastguard Worker 
2098*7304104dSAndroid Build Coastguard Worker       string fn = canonicalized_archive_entry_pathname (e);
2099*7304104dSAndroid Build Coastguard Worker       if ((r == 0) && (fn != b_source1)) // stage 1
2100*7304104dSAndroid Build Coastguard Worker         continue;
2101*7304104dSAndroid Build Coastguard Worker 
2102*7304104dSAndroid Build Coastguard Worker       if (fdcache.probe (b_source0, fn) && // skip if already interned
2103*7304104dSAndroid Build Coastguard Worker           fn != b_source1) // but only if we'd just be prefetching, PR29474
2104*7304104dSAndroid Build Coastguard Worker         continue;
2105*7304104dSAndroid Build Coastguard Worker 
2106*7304104dSAndroid Build Coastguard Worker       // extract this file to a temporary file
2107*7304104dSAndroid Build Coastguard Worker       char* tmppath = NULL;
2108*7304104dSAndroid Build Coastguard Worker       rc = asprintf (&tmppath, "%s/debuginfod-fdcache.XXXXXX", tmpdir.c_str());
2109*7304104dSAndroid Build Coastguard Worker       if (rc < 0)
2110*7304104dSAndroid Build Coastguard Worker         throw libc_exception (ENOMEM, "cannot allocate tmppath");
2111*7304104dSAndroid Build Coastguard Worker       defer_dtor<void*,void> tmmpath_freer (tmppath, free);
2112*7304104dSAndroid Build Coastguard Worker       fd = mkstemp (tmppath);
2113*7304104dSAndroid Build Coastguard Worker       if (fd < 0)
2114*7304104dSAndroid Build Coastguard Worker         throw libc_exception (errno, "cannot create temporary file");
2115*7304104dSAndroid Build Coastguard Worker       // NB: don't unlink (tmppath), as fdcache will take charge of it.
2116*7304104dSAndroid Build Coastguard Worker 
2117*7304104dSAndroid Build Coastguard Worker       // NB: this can take many uninterruptible seconds for a huge file
2118*7304104dSAndroid Build Coastguard Worker       rc = archive_read_data_into_fd (a, fd);
2119*7304104dSAndroid Build Coastguard Worker       if (rc != ARCHIVE_OK) // e.g. ENOSPC!
2120*7304104dSAndroid Build Coastguard Worker         {
2121*7304104dSAndroid Build Coastguard Worker           close (fd);
2122*7304104dSAndroid Build Coastguard Worker           unlink (tmppath);
2123*7304104dSAndroid Build Coastguard Worker           throw archive_exception(a, "cannot extract file");
2124*7304104dSAndroid Build Coastguard Worker         }
2125*7304104dSAndroid Build Coastguard Worker 
2126*7304104dSAndroid Build Coastguard Worker       // Set the mtime so the fdcache file mtimes, even prefetched ones,
2127*7304104dSAndroid Build Coastguard Worker       // propagate to future webapi clients.
2128*7304104dSAndroid Build Coastguard Worker       struct timespec tvs[2];
2129*7304104dSAndroid Build Coastguard Worker       tvs[0].tv_sec = 0;
2130*7304104dSAndroid Build Coastguard Worker       tvs[0].tv_nsec = UTIME_OMIT;
2131*7304104dSAndroid Build Coastguard Worker       tvs[1].tv_sec = archive_entry_mtime(e);
2132*7304104dSAndroid Build Coastguard Worker       tvs[1].tv_nsec = archive_entry_mtime_nsec(e);
2133*7304104dSAndroid Build Coastguard Worker       (void) futimens (fd, tvs);  /* best effort */
2134*7304104dSAndroid Build Coastguard Worker 
2135*7304104dSAndroid Build Coastguard Worker       struct timespec extract_end;
2136*7304104dSAndroid Build Coastguard Worker       clock_gettime (CLOCK_MONOTONIC, &extract_end);
2137*7304104dSAndroid Build Coastguard Worker       double extract_time = (extract_end.tv_sec - extract_begin.tv_sec)
2138*7304104dSAndroid Build Coastguard Worker         + (extract_end.tv_nsec - extract_begin.tv_nsec)/1.e9;
2139*7304104dSAndroid Build Coastguard Worker 
2140*7304104dSAndroid Build Coastguard Worker       if (r != 0) // stage 3
2141*7304104dSAndroid Build Coastguard Worker         {
2142*7304104dSAndroid Build Coastguard Worker           // NB: now we know we have a complete reusable file; make fdcache
2143*7304104dSAndroid Build Coastguard Worker           // responsible for unlinking it later.
2144*7304104dSAndroid Build Coastguard Worker           fdcache.intern(b_source0, fn,
2145*7304104dSAndroid Build Coastguard Worker                          tmppath, archive_entry_size(e),
2146*7304104dSAndroid Build Coastguard Worker                          false, extract_time); // prefetched ones go to the prefetch cache
2147*7304104dSAndroid Build Coastguard Worker           prefetch_count --;
2148*7304104dSAndroid Build Coastguard Worker           close (fd); // we're not saving this fd to make a mhd-response from!
2149*7304104dSAndroid Build Coastguard Worker           continue;
2150*7304104dSAndroid Build Coastguard Worker         }
2151*7304104dSAndroid Build Coastguard Worker 
2152*7304104dSAndroid Build Coastguard Worker       // NB: now we know we have a complete reusable file; make fdcache
2153*7304104dSAndroid Build Coastguard Worker       // responsible for unlinking it later.
2154*7304104dSAndroid Build Coastguard Worker       fdcache.intern(b_source0, b_source1,
2155*7304104dSAndroid Build Coastguard Worker                      tmppath, archive_entry_size(e),
2156*7304104dSAndroid Build Coastguard Worker                      true, extract_time); // requested ones go to the front of the line
2157*7304104dSAndroid Build Coastguard Worker 
2158*7304104dSAndroid Build Coastguard Worker       if (!section.empty ())
2159*7304104dSAndroid Build Coastguard Worker 	{
2160*7304104dSAndroid Build Coastguard Worker 	  int scn_fd = extract_section (fd, b_mtime,
2161*7304104dSAndroid Build Coastguard Worker 					b_source0 + ":" + b_source1,
2162*7304104dSAndroid Build Coastguard Worker 					section, extract_begin);
2163*7304104dSAndroid Build Coastguard Worker 	  close (fd);
2164*7304104dSAndroid Build Coastguard Worker 	  if (scn_fd >= 0)
2165*7304104dSAndroid Build Coastguard Worker 	    fd = scn_fd;
2166*7304104dSAndroid Build Coastguard Worker 	  else
2167*7304104dSAndroid Build Coastguard Worker 	    {
2168*7304104dSAndroid Build Coastguard Worker 	      if (verbose)
2169*7304104dSAndroid Build Coastguard Worker 	        obatched (clog) << "cannot find section " << section
2170*7304104dSAndroid Build Coastguard Worker 				<< " for archive " << b_source0
2171*7304104dSAndroid Build Coastguard Worker 				<< " file " << b_source1 << endl;
2172*7304104dSAndroid Build Coastguard Worker 	      return 0;
2173*7304104dSAndroid Build Coastguard Worker 	    }
2174*7304104dSAndroid Build Coastguard Worker 
2175*7304104dSAndroid Build Coastguard Worker 	  rc = fstat(fd, &fs);
2176*7304104dSAndroid Build Coastguard Worker 	  if (rc < 0)
2177*7304104dSAndroid Build Coastguard Worker 	    {
2178*7304104dSAndroid Build Coastguard Worker 	      close (fd);
2179*7304104dSAndroid Build Coastguard Worker 	      throw libc_exception (errno,
2180*7304104dSAndroid Build Coastguard Worker 		string ("fstat ") + b_source0 + string (" ") + section);
2181*7304104dSAndroid Build Coastguard Worker 	    }
2182*7304104dSAndroid Build Coastguard Worker 	  r = MHD_create_response_from_fd (fs.st_size, fd);
2183*7304104dSAndroid Build Coastguard Worker 	}
2184*7304104dSAndroid Build Coastguard Worker       else
2185*7304104dSAndroid Build Coastguard Worker 	r = MHD_create_response_from_fd (archive_entry_size(e), fd);
2186*7304104dSAndroid Build Coastguard Worker 
2187*7304104dSAndroid Build Coastguard Worker       inc_metric ("http_responses_total","result",archive_extension + " archive");
2188*7304104dSAndroid Build Coastguard Worker       if (r == 0)
2189*7304104dSAndroid Build Coastguard Worker         {
2190*7304104dSAndroid Build Coastguard Worker           if (verbose)
2191*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "cannot create fd-response for " << b_source0 << endl;
2192*7304104dSAndroid Build Coastguard Worker           close(fd);
2193*7304104dSAndroid Build Coastguard Worker           break; // assume no chance of better luck around another iteration; no other copies of same file
2194*7304104dSAndroid Build Coastguard Worker         }
2195*7304104dSAndroid Build Coastguard Worker       else
2196*7304104dSAndroid Build Coastguard Worker         {
2197*7304104dSAndroid Build Coastguard Worker           add_mhd_response_header (r, "Content-Type",
2198*7304104dSAndroid Build Coastguard Worker                                    "application/octet-stream");
2199*7304104dSAndroid Build Coastguard Worker           add_mhd_response_header (r, "X-DEBUGINFOD-SIZE",
2200*7304104dSAndroid Build Coastguard Worker                                    to_string(archive_entry_size(e)).c_str());
2201*7304104dSAndroid Build Coastguard Worker           add_mhd_response_header (r, "X-DEBUGINFOD-ARCHIVE", b_source0.c_str());
2202*7304104dSAndroid Build Coastguard Worker           add_mhd_response_header (r, "X-DEBUGINFOD-FILE", b_source1.c_str());
2203*7304104dSAndroid Build Coastguard Worker           add_mhd_last_modified (r, archive_entry_mtime(e));
2204*7304104dSAndroid Build Coastguard Worker           if (verbose > 1)
2205*7304104dSAndroid Build Coastguard Worker 	    obatched(clog) << "serving archive " << b_source0
2206*7304104dSAndroid Build Coastguard Worker 			   << " file " << b_source1
2207*7304104dSAndroid Build Coastguard Worker 			   << " section=" << section << endl;
2208*7304104dSAndroid Build Coastguard Worker           /* libmicrohttpd will close it. */
2209*7304104dSAndroid Build Coastguard Worker           if (result_fd)
2210*7304104dSAndroid Build Coastguard Worker             *result_fd = fd;
2211*7304104dSAndroid Build Coastguard Worker           continue;
2212*7304104dSAndroid Build Coastguard Worker         }
2213*7304104dSAndroid Build Coastguard Worker     }
2214*7304104dSAndroid Build Coastguard Worker 
2215*7304104dSAndroid Build Coastguard Worker   // XXX: rpm/file not found: delete this R entry?
2216*7304104dSAndroid Build Coastguard Worker   return r;
2217*7304104dSAndroid Build Coastguard Worker }
2218*7304104dSAndroid Build Coastguard Worker 
2219*7304104dSAndroid Build Coastguard Worker 
2220*7304104dSAndroid Build Coastguard Worker static struct MHD_Response*
handle_buildid_match(bool internal_req_p,int64_t b_mtime,const string & b_stype,const string & b_source0,const string & b_source1,const string & section,int * result_fd)2221*7304104dSAndroid Build Coastguard Worker handle_buildid_match (bool internal_req_p,
2222*7304104dSAndroid Build Coastguard Worker                       int64_t b_mtime,
2223*7304104dSAndroid Build Coastguard Worker                       const string& b_stype,
2224*7304104dSAndroid Build Coastguard Worker                       const string& b_source0,
2225*7304104dSAndroid Build Coastguard Worker                       const string& b_source1,
2226*7304104dSAndroid Build Coastguard Worker                       const string& section,
2227*7304104dSAndroid Build Coastguard Worker                       int *result_fd)
2228*7304104dSAndroid Build Coastguard Worker {
2229*7304104dSAndroid Build Coastguard Worker   try
2230*7304104dSAndroid Build Coastguard Worker     {
2231*7304104dSAndroid Build Coastguard Worker       if (b_stype == "F")
2232*7304104dSAndroid Build Coastguard Worker         return handle_buildid_f_match(internal_req_p, b_mtime, b_source0,
2233*7304104dSAndroid Build Coastguard Worker 				      section, result_fd);
2234*7304104dSAndroid Build Coastguard Worker       else if (b_stype == "R")
2235*7304104dSAndroid Build Coastguard Worker         return handle_buildid_r_match(internal_req_p, b_mtime, b_source0,
2236*7304104dSAndroid Build Coastguard Worker 				      b_source1, section, result_fd);
2237*7304104dSAndroid Build Coastguard Worker     }
2238*7304104dSAndroid Build Coastguard Worker   catch (const reportable_exception &e)
2239*7304104dSAndroid Build Coastguard Worker     {
2240*7304104dSAndroid Build Coastguard Worker       e.report(clog);
2241*7304104dSAndroid Build Coastguard Worker       // Report but swallow libc etc. errors here; let the caller
2242*7304104dSAndroid Build Coastguard Worker       // iterate to other matches of the content.
2243*7304104dSAndroid Build Coastguard Worker     }
2244*7304104dSAndroid Build Coastguard Worker 
2245*7304104dSAndroid Build Coastguard Worker   return 0;
2246*7304104dSAndroid Build Coastguard Worker }
2247*7304104dSAndroid Build Coastguard Worker 
2248*7304104dSAndroid Build Coastguard Worker 
2249*7304104dSAndroid Build Coastguard Worker static int
debuginfod_find_progress(debuginfod_client *,long a,long b)2250*7304104dSAndroid Build Coastguard Worker debuginfod_find_progress (debuginfod_client *, long a, long b)
2251*7304104dSAndroid Build Coastguard Worker {
2252*7304104dSAndroid Build Coastguard Worker   if (verbose > 4)
2253*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "federated debuginfod progress=" << a << "/" << b << endl;
2254*7304104dSAndroid Build Coastguard Worker 
2255*7304104dSAndroid Build Coastguard Worker   return interrupted;
2256*7304104dSAndroid Build Coastguard Worker }
2257*7304104dSAndroid Build Coastguard Worker 
2258*7304104dSAndroid Build Coastguard Worker 
2259*7304104dSAndroid Build Coastguard Worker // a little lru pool of debuginfod_client*s for reuse between query threads
2260*7304104dSAndroid Build Coastguard Worker 
2261*7304104dSAndroid Build Coastguard Worker mutex dc_pool_lock;
2262*7304104dSAndroid Build Coastguard Worker deque<debuginfod_client*> dc_pool;
2263*7304104dSAndroid Build Coastguard Worker 
debuginfod_pool_begin()2264*7304104dSAndroid Build Coastguard Worker debuginfod_client* debuginfod_pool_begin()
2265*7304104dSAndroid Build Coastguard Worker {
2266*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(dc_pool_lock);
2267*7304104dSAndroid Build Coastguard Worker   if (dc_pool.size() > 0)
2268*7304104dSAndroid Build Coastguard Worker     {
2269*7304104dSAndroid Build Coastguard Worker       inc_metric("dc_pool_op_count","op","begin-reuse");
2270*7304104dSAndroid Build Coastguard Worker       debuginfod_client *c = dc_pool.front();
2271*7304104dSAndroid Build Coastguard Worker       dc_pool.pop_front();
2272*7304104dSAndroid Build Coastguard Worker       return c;
2273*7304104dSAndroid Build Coastguard Worker     }
2274*7304104dSAndroid Build Coastguard Worker   inc_metric("dc_pool_op_count","op","begin-new");
2275*7304104dSAndroid Build Coastguard Worker   return debuginfod_begin();
2276*7304104dSAndroid Build Coastguard Worker }
2277*7304104dSAndroid Build Coastguard Worker 
2278*7304104dSAndroid Build Coastguard Worker 
debuginfod_pool_groom()2279*7304104dSAndroid Build Coastguard Worker void debuginfod_pool_groom()
2280*7304104dSAndroid Build Coastguard Worker {
2281*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(dc_pool_lock);
2282*7304104dSAndroid Build Coastguard Worker   while (dc_pool.size() > 0)
2283*7304104dSAndroid Build Coastguard Worker     {
2284*7304104dSAndroid Build Coastguard Worker       inc_metric("dc_pool_op_count","op","end");
2285*7304104dSAndroid Build Coastguard Worker       debuginfod_end(dc_pool.front());
2286*7304104dSAndroid Build Coastguard Worker       dc_pool.pop_front();
2287*7304104dSAndroid Build Coastguard Worker     }
2288*7304104dSAndroid Build Coastguard Worker }
2289*7304104dSAndroid Build Coastguard Worker 
2290*7304104dSAndroid Build Coastguard Worker 
debuginfod_pool_end(debuginfod_client * c)2291*7304104dSAndroid Build Coastguard Worker void debuginfod_pool_end(debuginfod_client* c)
2292*7304104dSAndroid Build Coastguard Worker {
2293*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(dc_pool_lock);
2294*7304104dSAndroid Build Coastguard Worker   inc_metric("dc_pool_op_count","op","end-save");
2295*7304104dSAndroid Build Coastguard Worker   dc_pool.push_front(c); // accelerate reuse, vs. push_back
2296*7304104dSAndroid Build Coastguard Worker }
2297*7304104dSAndroid Build Coastguard Worker 
2298*7304104dSAndroid Build Coastguard Worker 
2299*7304104dSAndroid Build Coastguard Worker static struct MHD_Response*
handle_buildid(MHD_Connection * conn,const string & buildid,string & artifacttype,const string & suffix,int * result_fd)2300*7304104dSAndroid Build Coastguard Worker handle_buildid (MHD_Connection* conn,
2301*7304104dSAndroid Build Coastguard Worker                 const string& buildid /* unsafe */,
2302*7304104dSAndroid Build Coastguard Worker                 string& artifacttype /* unsafe, cleanse on exception/return */,
2303*7304104dSAndroid Build Coastguard Worker                 const string& suffix /* unsafe */,
2304*7304104dSAndroid Build Coastguard Worker                 int *result_fd)
2305*7304104dSAndroid Build Coastguard Worker {
2306*7304104dSAndroid Build Coastguard Worker   // validate artifacttype
2307*7304104dSAndroid Build Coastguard Worker   string atype_code;
2308*7304104dSAndroid Build Coastguard Worker   if (artifacttype == "debuginfo") atype_code = "D";
2309*7304104dSAndroid Build Coastguard Worker   else if (artifacttype == "executable") atype_code = "E";
2310*7304104dSAndroid Build Coastguard Worker   else if (artifacttype == "source") atype_code = "S";
2311*7304104dSAndroid Build Coastguard Worker   else if (artifacttype == "section") atype_code = "I";
2312*7304104dSAndroid Build Coastguard Worker   else {
2313*7304104dSAndroid Build Coastguard Worker     artifacttype = "invalid"; // PR28242 ensure http_resposes metrics don't propagate unclean user data
2314*7304104dSAndroid Build Coastguard Worker     throw reportable_exception("invalid artifacttype");
2315*7304104dSAndroid Build Coastguard Worker   }
2316*7304104dSAndroid Build Coastguard Worker 
2317*7304104dSAndroid Build Coastguard Worker   if (conn != 0)
2318*7304104dSAndroid Build Coastguard Worker     inc_metric("http_requests_total", "type", artifacttype);
2319*7304104dSAndroid Build Coastguard Worker 
2320*7304104dSAndroid Build Coastguard Worker   string section;
2321*7304104dSAndroid Build Coastguard Worker   if (atype_code == "I")
2322*7304104dSAndroid Build Coastguard Worker     {
2323*7304104dSAndroid Build Coastguard Worker       if (suffix.size () < 2)
2324*7304104dSAndroid Build Coastguard Worker 	throw reportable_exception ("invalid section suffix");
2325*7304104dSAndroid Build Coastguard Worker 
2326*7304104dSAndroid Build Coastguard Worker       // Remove leading '/'
2327*7304104dSAndroid Build Coastguard Worker       section = suffix.substr(1);
2328*7304104dSAndroid Build Coastguard Worker     }
2329*7304104dSAndroid Build Coastguard Worker 
2330*7304104dSAndroid Build Coastguard Worker   if (atype_code == "S" && suffix == "")
2331*7304104dSAndroid Build Coastguard Worker      throw reportable_exception("invalid source suffix");
2332*7304104dSAndroid Build Coastguard Worker 
2333*7304104dSAndroid Build Coastguard Worker   // validate buildid
2334*7304104dSAndroid Build Coastguard Worker   if ((buildid.size() < 2) || // not empty
2335*7304104dSAndroid Build Coastguard Worker       (buildid.size() % 2) || // even number
2336*7304104dSAndroid Build Coastguard Worker       (buildid.find_first_not_of("0123456789abcdef") != string::npos)) // pure tasty lowercase hex
2337*7304104dSAndroid Build Coastguard Worker     throw reportable_exception("invalid buildid");
2338*7304104dSAndroid Build Coastguard Worker 
2339*7304104dSAndroid Build Coastguard Worker   if (verbose > 1)
2340*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "searching for buildid=" << buildid << " artifacttype=" << artifacttype
2341*7304104dSAndroid Build Coastguard Worker          << " suffix=" << suffix << endl;
2342*7304104dSAndroid Build Coastguard Worker 
2343*7304104dSAndroid Build Coastguard Worker   // If invoked from the scanner threads, use the scanners' read-write
2344*7304104dSAndroid Build Coastguard Worker   // connection.  Otherwise use the web query threads' read-only connection.
2345*7304104dSAndroid Build Coastguard Worker   sqlite3 *thisdb = (conn == 0) ? db : dbq;
2346*7304104dSAndroid Build Coastguard Worker 
2347*7304104dSAndroid Build Coastguard Worker   sqlite_ps *pp = 0;
2348*7304104dSAndroid Build Coastguard Worker 
2349*7304104dSAndroid Build Coastguard Worker   if (atype_code == "D")
2350*7304104dSAndroid Build Coastguard Worker     {
2351*7304104dSAndroid Build Coastguard Worker       pp = new sqlite_ps (thisdb, "mhd-query-d",
2352*7304104dSAndroid Build Coastguard Worker                           "select mtime, sourcetype, source0, source1 from " BUILDIDS "_query_d where buildid = ? "
2353*7304104dSAndroid Build Coastguard Worker                           "order by mtime desc");
2354*7304104dSAndroid Build Coastguard Worker       pp->reset();
2355*7304104dSAndroid Build Coastguard Worker       pp->bind(1, buildid);
2356*7304104dSAndroid Build Coastguard Worker     }
2357*7304104dSAndroid Build Coastguard Worker   else if (atype_code == "E")
2358*7304104dSAndroid Build Coastguard Worker     {
2359*7304104dSAndroid Build Coastguard Worker       pp = new sqlite_ps (thisdb, "mhd-query-e",
2360*7304104dSAndroid Build Coastguard Worker                           "select mtime, sourcetype, source0, source1 from " BUILDIDS "_query_e where buildid = ? "
2361*7304104dSAndroid Build Coastguard Worker                           "order by mtime desc");
2362*7304104dSAndroid Build Coastguard Worker       pp->reset();
2363*7304104dSAndroid Build Coastguard Worker       pp->bind(1, buildid);
2364*7304104dSAndroid Build Coastguard Worker     }
2365*7304104dSAndroid Build Coastguard Worker   else if (atype_code == "S")
2366*7304104dSAndroid Build Coastguard Worker     {
2367*7304104dSAndroid Build Coastguard Worker       // PR25548
2368*7304104dSAndroid Build Coastguard Worker       // Incoming source queries may come in with either dwarf-level OR canonicalized paths.
2369*7304104dSAndroid Build Coastguard Worker       // We let the query pass with either one.
2370*7304104dSAndroid Build Coastguard Worker 
2371*7304104dSAndroid Build Coastguard Worker       pp = new sqlite_ps (thisdb, "mhd-query-s",
2372*7304104dSAndroid Build Coastguard Worker                           "select mtime, sourcetype, source0, source1 from " BUILDIDS "_query_s where buildid = ? and artifactsrc in (?,?) "
2373*7304104dSAndroid Build Coastguard Worker                           "order by sharedprefix(source0,source0ref) desc, mtime desc");
2374*7304104dSAndroid Build Coastguard Worker       pp->reset();
2375*7304104dSAndroid Build Coastguard Worker       pp->bind(1, buildid);
2376*7304104dSAndroid Build Coastguard Worker       // NB: we don't store the non-canonicalized path names any more, but old databases
2377*7304104dSAndroid Build Coastguard Worker       // might have them (and no canon ones), so we keep searching for both.
2378*7304104dSAndroid Build Coastguard Worker       pp->bind(2, suffix);
2379*7304104dSAndroid Build Coastguard Worker       pp->bind(3, canon_pathname(suffix));
2380*7304104dSAndroid Build Coastguard Worker     }
2381*7304104dSAndroid Build Coastguard Worker   else if (atype_code == "I")
2382*7304104dSAndroid Build Coastguard Worker     {
2383*7304104dSAndroid Build Coastguard Worker       pp = new sqlite_ps (thisdb, "mhd-query-i",
2384*7304104dSAndroid Build Coastguard Worker 	"select mtime, sourcetype, source0, source1, 1 as debug_p from " BUILDIDS "_query_d where buildid = ? "
2385*7304104dSAndroid Build Coastguard Worker 	"union all "
2386*7304104dSAndroid Build Coastguard Worker 	"select mtime, sourcetype, source0, source1, 0 as debug_p from " BUILDIDS "_query_e where buildid = ? "
2387*7304104dSAndroid Build Coastguard Worker 	"order by debug_p desc, mtime desc");
2388*7304104dSAndroid Build Coastguard Worker       pp->reset();
2389*7304104dSAndroid Build Coastguard Worker       pp->bind(1, buildid);
2390*7304104dSAndroid Build Coastguard Worker       pp->bind(2, buildid);
2391*7304104dSAndroid Build Coastguard Worker     }
2392*7304104dSAndroid Build Coastguard Worker   unique_ptr<sqlite_ps> ps_closer(pp); // release pp if exception or return
2393*7304104dSAndroid Build Coastguard Worker 
2394*7304104dSAndroid Build Coastguard Worker   bool do_upstream_section_query = true;
2395*7304104dSAndroid Build Coastguard Worker 
2396*7304104dSAndroid Build Coastguard Worker   // consume all the rows
2397*7304104dSAndroid Build Coastguard Worker   while (1)
2398*7304104dSAndroid Build Coastguard Worker     {
2399*7304104dSAndroid Build Coastguard Worker       int rc = pp->step();
2400*7304104dSAndroid Build Coastguard Worker       if (rc == SQLITE_DONE) break;
2401*7304104dSAndroid Build Coastguard Worker       if (rc != SQLITE_ROW)
2402*7304104dSAndroid Build Coastguard Worker         throw sqlite_exception(rc, "step");
2403*7304104dSAndroid Build Coastguard Worker 
2404*7304104dSAndroid Build Coastguard Worker       int64_t b_mtime = sqlite3_column_int64 (*pp, 0);
2405*7304104dSAndroid Build Coastguard Worker       string b_stype = string((const char*) sqlite3_column_text (*pp, 1) ?: ""); /* by DDL may not be NULL */
2406*7304104dSAndroid Build Coastguard Worker       string b_source0 = string((const char*) sqlite3_column_text (*pp, 2) ?: ""); /* may be NULL */
2407*7304104dSAndroid Build Coastguard Worker       string b_source1 = string((const char*) sqlite3_column_text (*pp, 3) ?: ""); /* may be NULL */
2408*7304104dSAndroid Build Coastguard Worker 
2409*7304104dSAndroid Build Coastguard Worker       if (verbose > 1)
2410*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "found mtime=" << b_mtime << " stype=" << b_stype
2411*7304104dSAndroid Build Coastguard Worker              << " source0=" << b_source0 << " source1=" << b_source1 << endl;
2412*7304104dSAndroid Build Coastguard Worker 
2413*7304104dSAndroid Build Coastguard Worker       // Try accessing the located match.
2414*7304104dSAndroid Build Coastguard Worker       // XXX: in case of multiple matches, attempt them in parallel?
2415*7304104dSAndroid Build Coastguard Worker       auto r = handle_buildid_match (conn ? false : true,
2416*7304104dSAndroid Build Coastguard Worker                                      b_mtime, b_stype, b_source0, b_source1,
2417*7304104dSAndroid Build Coastguard Worker 				     section, result_fd);
2418*7304104dSAndroid Build Coastguard Worker       if (r)
2419*7304104dSAndroid Build Coastguard Worker         return r;
2420*7304104dSAndroid Build Coastguard Worker 
2421*7304104dSAndroid Build Coastguard Worker       // If a debuginfo file matching BUILDID was found but didn't contain
2422*7304104dSAndroid Build Coastguard Worker       // the desired section, then the section should not exist.  Don't
2423*7304104dSAndroid Build Coastguard Worker       // bother querying upstream servers.
2424*7304104dSAndroid Build Coastguard Worker       if (!section.empty () && (sqlite3_column_int (*pp, 4) == 1))
2425*7304104dSAndroid Build Coastguard Worker 	{
2426*7304104dSAndroid Build Coastguard Worker 	  struct stat st;
2427*7304104dSAndroid Build Coastguard Worker 
2428*7304104dSAndroid Build Coastguard Worker 	  // For "F" sourcetype, check if the debuginfo exists. For "R"
2429*7304104dSAndroid Build Coastguard Worker 	  // sourcetype, check if the debuginfo was interned into the fdcache.
2430*7304104dSAndroid Build Coastguard Worker 	  if ((b_stype == "F" && (stat (b_source0.c_str (), &st) == 0))
2431*7304104dSAndroid Build Coastguard Worker 	      || (b_stype == "R" && fdcache.probe (b_source0, b_source1)))
2432*7304104dSAndroid Build Coastguard Worker 	    do_upstream_section_query = false;
2433*7304104dSAndroid Build Coastguard Worker 	}
2434*7304104dSAndroid Build Coastguard Worker     }
2435*7304104dSAndroid Build Coastguard Worker   pp->reset();
2436*7304104dSAndroid Build Coastguard Worker 
2437*7304104dSAndroid Build Coastguard Worker   if (!do_upstream_section_query)
2438*7304104dSAndroid Build Coastguard Worker     throw reportable_exception(MHD_HTTP_NOT_FOUND, "not found");
2439*7304104dSAndroid Build Coastguard Worker 
2440*7304104dSAndroid Build Coastguard Worker   // We couldn't find it in the database.  Last ditch effort
2441*7304104dSAndroid Build Coastguard Worker   // is to defer to other debuginfo servers.
2442*7304104dSAndroid Build Coastguard Worker 
2443*7304104dSAndroid Build Coastguard Worker   int fd = -1;
2444*7304104dSAndroid Build Coastguard Worker   debuginfod_client *client = debuginfod_pool_begin ();
2445*7304104dSAndroid Build Coastguard Worker   if (client == NULL)
2446*7304104dSAndroid Build Coastguard Worker     throw libc_exception(errno, "debuginfod client pool alloc");
2447*7304104dSAndroid Build Coastguard Worker   defer_dtor<debuginfod_client*,void> client_closer (client, debuginfod_pool_end);
2448*7304104dSAndroid Build Coastguard Worker 
2449*7304104dSAndroid Build Coastguard Worker   debuginfod_set_progressfn (client, & debuginfod_find_progress);
2450*7304104dSAndroid Build Coastguard Worker 
2451*7304104dSAndroid Build Coastguard Worker   if (conn)
2452*7304104dSAndroid Build Coastguard Worker     {
2453*7304104dSAndroid Build Coastguard Worker       // Transcribe incoming User-Agent:
2454*7304104dSAndroid Build Coastguard Worker       string ua = MHD_lookup_connection_value (conn, MHD_HEADER_KIND, "User-Agent") ?: "";
2455*7304104dSAndroid Build Coastguard Worker       string ua_complete = string("User-Agent: ") + ua;
2456*7304104dSAndroid Build Coastguard Worker       debuginfod_add_http_header (client, ua_complete.c_str());
2457*7304104dSAndroid Build Coastguard Worker 
2458*7304104dSAndroid Build Coastguard Worker       // Compute larger XFF:, for avoiding info loss during
2459*7304104dSAndroid Build Coastguard Worker       // federation, and for future cyclicity detection.
2460*7304104dSAndroid Build Coastguard Worker       string xff = MHD_lookup_connection_value (conn, MHD_HEADER_KIND, "X-Forwarded-For") ?: "";
2461*7304104dSAndroid Build Coastguard Worker       if (xff != "")
2462*7304104dSAndroid Build Coastguard Worker         xff += string(", "); // comma separated list
2463*7304104dSAndroid Build Coastguard Worker 
2464*7304104dSAndroid Build Coastguard Worker       unsigned int xff_count = 0;
2465*7304104dSAndroid Build Coastguard Worker       for (auto&& i : xff){
2466*7304104dSAndroid Build Coastguard Worker         if (i == ',') xff_count++;
2467*7304104dSAndroid Build Coastguard Worker       }
2468*7304104dSAndroid Build Coastguard Worker 
2469*7304104dSAndroid Build Coastguard Worker       // if X-Forwarded-For: exceeds N hops,
2470*7304104dSAndroid Build Coastguard Worker       // do not delegate a local lookup miss to upstream debuginfods.
2471*7304104dSAndroid Build Coastguard Worker       if (xff_count >= forwarded_ttl_limit)
2472*7304104dSAndroid Build Coastguard Worker         throw reportable_exception(MHD_HTTP_NOT_FOUND, "not found, --forwared-ttl-limit reached \
2473*7304104dSAndroid Build Coastguard Worker and will not query the upstream servers");
2474*7304104dSAndroid Build Coastguard Worker 
2475*7304104dSAndroid Build Coastguard Worker       // Compute the client's numeric IP address only - so can't merge with conninfo()
2476*7304104dSAndroid Build Coastguard Worker       const union MHD_ConnectionInfo *u = MHD_get_connection_info (conn,
2477*7304104dSAndroid Build Coastguard Worker                                                                    MHD_CONNECTION_INFO_CLIENT_ADDRESS);
2478*7304104dSAndroid Build Coastguard Worker       struct sockaddr *so = u ? u->client_addr : 0;
2479*7304104dSAndroid Build Coastguard Worker       char hostname[256] = ""; // RFC1035
2480*7304104dSAndroid Build Coastguard Worker       if (so && so->sa_family == AF_INET) {
2481*7304104dSAndroid Build Coastguard Worker         (void) getnameinfo (so, sizeof (struct sockaddr_in), hostname, sizeof (hostname), NULL, 0,
2482*7304104dSAndroid Build Coastguard Worker                             NI_NUMERICHOST);
2483*7304104dSAndroid Build Coastguard Worker       } else if (so && so->sa_family == AF_INET6) {
2484*7304104dSAndroid Build Coastguard Worker         struct sockaddr_in6* addr6 = (struct sockaddr_in6*) so;
2485*7304104dSAndroid Build Coastguard Worker         if (IN6_IS_ADDR_V4MAPPED(&addr6->sin6_addr)) {
2486*7304104dSAndroid Build Coastguard Worker           struct sockaddr_in addr4;
2487*7304104dSAndroid Build Coastguard Worker           memset (&addr4, 0, sizeof(addr4));
2488*7304104dSAndroid Build Coastguard Worker           addr4.sin_family = AF_INET;
2489*7304104dSAndroid Build Coastguard Worker           addr4.sin_port = addr6->sin6_port;
2490*7304104dSAndroid Build Coastguard Worker           memcpy (&addr4.sin_addr.s_addr, addr6->sin6_addr.s6_addr+12, sizeof(addr4.sin_addr.s_addr));
2491*7304104dSAndroid Build Coastguard Worker           (void) getnameinfo ((struct sockaddr*) &addr4, sizeof (addr4),
2492*7304104dSAndroid Build Coastguard Worker                               hostname, sizeof (hostname), NULL, 0,
2493*7304104dSAndroid Build Coastguard Worker                               NI_NUMERICHOST);
2494*7304104dSAndroid Build Coastguard Worker         } else {
2495*7304104dSAndroid Build Coastguard Worker           (void) getnameinfo (so, sizeof (struct sockaddr_in6), hostname, sizeof (hostname), NULL, 0,
2496*7304104dSAndroid Build Coastguard Worker                               NI_NUMERICHOST);
2497*7304104dSAndroid Build Coastguard Worker         }
2498*7304104dSAndroid Build Coastguard Worker       }
2499*7304104dSAndroid Build Coastguard Worker 
2500*7304104dSAndroid Build Coastguard Worker       string xff_complete = string("X-Forwarded-For: ")+xff+string(hostname);
2501*7304104dSAndroid Build Coastguard Worker       debuginfod_add_http_header (client, xff_complete.c_str());
2502*7304104dSAndroid Build Coastguard Worker     }
2503*7304104dSAndroid Build Coastguard Worker 
2504*7304104dSAndroid Build Coastguard Worker   if (artifacttype == "debuginfo")
2505*7304104dSAndroid Build Coastguard Worker     fd = debuginfod_find_debuginfo (client,
2506*7304104dSAndroid Build Coastguard Worker                                     (const unsigned char*) buildid.c_str(),
2507*7304104dSAndroid Build Coastguard Worker                                     0, NULL);
2508*7304104dSAndroid Build Coastguard Worker   else if (artifacttype == "executable")
2509*7304104dSAndroid Build Coastguard Worker     fd = debuginfod_find_executable (client,
2510*7304104dSAndroid Build Coastguard Worker                                      (const unsigned char*) buildid.c_str(),
2511*7304104dSAndroid Build Coastguard Worker                                      0, NULL);
2512*7304104dSAndroid Build Coastguard Worker   else if (artifacttype == "source")
2513*7304104dSAndroid Build Coastguard Worker     fd = debuginfod_find_source (client,
2514*7304104dSAndroid Build Coastguard Worker                                  (const unsigned char*) buildid.c_str(),
2515*7304104dSAndroid Build Coastguard Worker                                  0, suffix.c_str(), NULL);
2516*7304104dSAndroid Build Coastguard Worker   else if (artifacttype == "section")
2517*7304104dSAndroid Build Coastguard Worker     fd = debuginfod_find_section (client,
2518*7304104dSAndroid Build Coastguard Worker                                   (const unsigned char*) buildid.c_str(),
2519*7304104dSAndroid Build Coastguard Worker                                   0, section.c_str(), NULL);
2520*7304104dSAndroid Build Coastguard Worker 
2521*7304104dSAndroid Build Coastguard Worker   if (fd >= 0)
2522*7304104dSAndroid Build Coastguard Worker     {
2523*7304104dSAndroid Build Coastguard Worker       if (conn != 0)
2524*7304104dSAndroid Build Coastguard Worker 	inc_metric ("http_responses_total","result","upstream");
2525*7304104dSAndroid Build Coastguard Worker       struct stat s;
2526*7304104dSAndroid Build Coastguard Worker       int rc = fstat (fd, &s);
2527*7304104dSAndroid Build Coastguard Worker       if (rc == 0)
2528*7304104dSAndroid Build Coastguard Worker         {
2529*7304104dSAndroid Build Coastguard Worker           auto r = MHD_create_response_from_fd ((uint64_t) s.st_size, fd);
2530*7304104dSAndroid Build Coastguard Worker           if (r)
2531*7304104dSAndroid Build Coastguard Worker             {
2532*7304104dSAndroid Build Coastguard Worker               add_mhd_response_header (r, "Content-Type",
2533*7304104dSAndroid Build Coastguard Worker 				       "application/octet-stream");
2534*7304104dSAndroid Build Coastguard Worker               // Copy the incoming headers
2535*7304104dSAndroid Build Coastguard Worker               const char * hdrs = debuginfod_get_headers(client);
2536*7304104dSAndroid Build Coastguard Worker               string header_dup;
2537*7304104dSAndroid Build Coastguard Worker               if (hdrs)
2538*7304104dSAndroid Build Coastguard Worker                 header_dup = string(hdrs);
2539*7304104dSAndroid Build Coastguard Worker               // Parse the "header: value\n" lines into (h,v) tuples and pass on
2540*7304104dSAndroid Build Coastguard Worker               while(1)
2541*7304104dSAndroid Build Coastguard Worker                 {
2542*7304104dSAndroid Build Coastguard Worker                   size_t newline = header_dup.find('\n');
2543*7304104dSAndroid Build Coastguard Worker                   if (newline == string::npos) break;
2544*7304104dSAndroid Build Coastguard Worker                   size_t colon = header_dup.find(':');
2545*7304104dSAndroid Build Coastguard Worker                   if (colon == string::npos) break;
2546*7304104dSAndroid Build Coastguard Worker                   string header = header_dup.substr(0,colon);
2547*7304104dSAndroid Build Coastguard Worker                   string value = header_dup.substr(colon+1,newline-colon-1);
2548*7304104dSAndroid Build Coastguard Worker                   // strip leading spaces from value
2549*7304104dSAndroid Build Coastguard Worker                   size_t nonspace = value.find_first_not_of(" ");
2550*7304104dSAndroid Build Coastguard Worker                   if (nonspace != string::npos)
2551*7304104dSAndroid Build Coastguard Worker                     value = value.substr(nonspace);
2552*7304104dSAndroid Build Coastguard Worker                   add_mhd_response_header(r, header.c_str(), value.c_str());
2553*7304104dSAndroid Build Coastguard Worker                   header_dup = header_dup.substr(newline+1);
2554*7304104dSAndroid Build Coastguard Worker                 }
2555*7304104dSAndroid Build Coastguard Worker 
2556*7304104dSAndroid Build Coastguard Worker               add_mhd_last_modified (r, s.st_mtime);
2557*7304104dSAndroid Build Coastguard Worker               if (verbose > 1)
2558*7304104dSAndroid Build Coastguard Worker                 obatched(clog) << "serving file from upstream debuginfod/cache" << endl;
2559*7304104dSAndroid Build Coastguard Worker               if (result_fd)
2560*7304104dSAndroid Build Coastguard Worker                 *result_fd = fd;
2561*7304104dSAndroid Build Coastguard Worker               return r; // NB: don't close fd; libmicrohttpd will
2562*7304104dSAndroid Build Coastguard Worker             }
2563*7304104dSAndroid Build Coastguard Worker         }
2564*7304104dSAndroid Build Coastguard Worker       close (fd);
2565*7304104dSAndroid Build Coastguard Worker     }
2566*7304104dSAndroid Build Coastguard Worker   else
2567*7304104dSAndroid Build Coastguard Worker     switch(fd)
2568*7304104dSAndroid Build Coastguard Worker       {
2569*7304104dSAndroid Build Coastguard Worker       case -ENOSYS:
2570*7304104dSAndroid Build Coastguard Worker         break;
2571*7304104dSAndroid Build Coastguard Worker       case -ENOENT:
2572*7304104dSAndroid Build Coastguard Worker         break;
2573*7304104dSAndroid Build Coastguard Worker       default: // some more tricky error
2574*7304104dSAndroid Build Coastguard Worker         throw libc_exception(-fd, "upstream debuginfod query failed");
2575*7304104dSAndroid Build Coastguard Worker       }
2576*7304104dSAndroid Build Coastguard Worker 
2577*7304104dSAndroid Build Coastguard Worker   throw reportable_exception(MHD_HTTP_NOT_FOUND, "not found");
2578*7304104dSAndroid Build Coastguard Worker }
2579*7304104dSAndroid Build Coastguard Worker 
2580*7304104dSAndroid Build Coastguard Worker 
2581*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
2582*7304104dSAndroid Build Coastguard Worker 
2583*7304104dSAndroid Build Coastguard Worker static map<string,double> metrics; // arbitrary data for /metrics query
2584*7304104dSAndroid Build Coastguard Worker // NB: store int64_t since all our metrics are integers; prometheus accepts double
2585*7304104dSAndroid Build Coastguard Worker static mutex metrics_lock;
2586*7304104dSAndroid Build Coastguard Worker // NB: these objects get released during the process exit via global dtors
2587*7304104dSAndroid Build Coastguard Worker // do not call them from within other global dtors
2588*7304104dSAndroid Build Coastguard Worker 
2589*7304104dSAndroid Build Coastguard Worker // utility function for assembling prometheus-compatible
2590*7304104dSAndroid Build Coastguard Worker // name="escaped-value" strings
2591*7304104dSAndroid Build Coastguard Worker // https://prometheus.io/docs/instrumenting/exposition_formats/
2592*7304104dSAndroid Build Coastguard Worker static string
metric_label(const string & name,const string & value)2593*7304104dSAndroid Build Coastguard Worker metric_label(const string& name, const string& value)
2594*7304104dSAndroid Build Coastguard Worker {
2595*7304104dSAndroid Build Coastguard Worker   string x = name + "=\"";
2596*7304104dSAndroid Build Coastguard Worker   for (auto&& c : value)
2597*7304104dSAndroid Build Coastguard Worker     switch(c)
2598*7304104dSAndroid Build Coastguard Worker       {
2599*7304104dSAndroid Build Coastguard Worker       case '\\': x += "\\\\"; break;
2600*7304104dSAndroid Build Coastguard Worker       case '\"': x += "\\\""; break;
2601*7304104dSAndroid Build Coastguard Worker       case '\n': x += "\\n"; break;
2602*7304104dSAndroid Build Coastguard Worker       default: x += c; break;
2603*7304104dSAndroid Build Coastguard Worker       }
2604*7304104dSAndroid Build Coastguard Worker   x += "\"";
2605*7304104dSAndroid Build Coastguard Worker   return x;
2606*7304104dSAndroid Build Coastguard Worker }
2607*7304104dSAndroid Build Coastguard Worker 
2608*7304104dSAndroid Build Coastguard Worker 
2609*7304104dSAndroid Build Coastguard Worker // add prometheus-format metric name + label tuple (if any) + value
2610*7304104dSAndroid Build Coastguard Worker 
2611*7304104dSAndroid Build Coastguard Worker static void
set_metric(const string & metric,double value)2612*7304104dSAndroid Build Coastguard Worker set_metric(const string& metric, double value)
2613*7304104dSAndroid Build Coastguard Worker {
2614*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2615*7304104dSAndroid Build Coastguard Worker   metrics[metric] = value;
2616*7304104dSAndroid Build Coastguard Worker }
2617*7304104dSAndroid Build Coastguard Worker static void
inc_metric(const string & metric)2618*7304104dSAndroid Build Coastguard Worker inc_metric(const string& metric)
2619*7304104dSAndroid Build Coastguard Worker {
2620*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2621*7304104dSAndroid Build Coastguard Worker   metrics[metric] ++;
2622*7304104dSAndroid Build Coastguard Worker }
2623*7304104dSAndroid Build Coastguard Worker static void
set_metric(const string & metric,const string & lname,const string & lvalue,double value)2624*7304104dSAndroid Build Coastguard Worker set_metric(const string& metric,
2625*7304104dSAndroid Build Coastguard Worker            const string& lname, const string& lvalue,
2626*7304104dSAndroid Build Coastguard Worker            double value)
2627*7304104dSAndroid Build Coastguard Worker {
2628*7304104dSAndroid Build Coastguard Worker   string key = (metric + "{" + metric_label(lname, lvalue) + "}");
2629*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2630*7304104dSAndroid Build Coastguard Worker   metrics[key] = value;
2631*7304104dSAndroid Build Coastguard Worker }
2632*7304104dSAndroid Build Coastguard Worker 
2633*7304104dSAndroid Build Coastguard Worker static void
inc_metric(const string & metric,const string & lname,const string & lvalue)2634*7304104dSAndroid Build Coastguard Worker inc_metric(const string& metric,
2635*7304104dSAndroid Build Coastguard Worker            const string& lname, const string& lvalue)
2636*7304104dSAndroid Build Coastguard Worker {
2637*7304104dSAndroid Build Coastguard Worker   string key = (metric + "{" + metric_label(lname, lvalue) + "}");
2638*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2639*7304104dSAndroid Build Coastguard Worker   metrics[key] ++;
2640*7304104dSAndroid Build Coastguard Worker }
2641*7304104dSAndroid Build Coastguard Worker static void
add_metric(const string & metric,const string & lname,const string & lvalue,double value)2642*7304104dSAndroid Build Coastguard Worker add_metric(const string& metric,
2643*7304104dSAndroid Build Coastguard Worker            const string& lname, const string& lvalue,
2644*7304104dSAndroid Build Coastguard Worker            double value)
2645*7304104dSAndroid Build Coastguard Worker {
2646*7304104dSAndroid Build Coastguard Worker   string key = (metric + "{" + metric_label(lname, lvalue) + "}");
2647*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2648*7304104dSAndroid Build Coastguard Worker   metrics[key] += value;
2649*7304104dSAndroid Build Coastguard Worker }
2650*7304104dSAndroid Build Coastguard Worker static void
add_metric(const string & metric,double value)2651*7304104dSAndroid Build Coastguard Worker add_metric(const string& metric,
2652*7304104dSAndroid Build Coastguard Worker            double value)
2653*7304104dSAndroid Build Coastguard Worker {
2654*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2655*7304104dSAndroid Build Coastguard Worker   metrics[metric] += value;
2656*7304104dSAndroid Build Coastguard Worker }
2657*7304104dSAndroid Build Coastguard Worker 
2658*7304104dSAndroid Build Coastguard Worker 
2659*7304104dSAndroid Build Coastguard Worker // and more for higher arity labels if needed
2660*7304104dSAndroid Build Coastguard Worker 
2661*7304104dSAndroid Build Coastguard Worker static void
inc_metric(const string & metric,const string & lname,const string & lvalue,const string & rname,const string & rvalue)2662*7304104dSAndroid Build Coastguard Worker inc_metric(const string& metric,
2663*7304104dSAndroid Build Coastguard Worker            const string& lname, const string& lvalue,
2664*7304104dSAndroid Build Coastguard Worker            const string& rname, const string& rvalue)
2665*7304104dSAndroid Build Coastguard Worker {
2666*7304104dSAndroid Build Coastguard Worker   string key = (metric + "{"
2667*7304104dSAndroid Build Coastguard Worker                 + metric_label(lname, lvalue) + ","
2668*7304104dSAndroid Build Coastguard Worker                 + metric_label(rname, rvalue) + "}");
2669*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2670*7304104dSAndroid Build Coastguard Worker   metrics[key] ++;
2671*7304104dSAndroid Build Coastguard Worker }
2672*7304104dSAndroid Build Coastguard Worker static void
add_metric(const string & metric,const string & lname,const string & lvalue,const string & rname,const string & rvalue,double value)2673*7304104dSAndroid Build Coastguard Worker add_metric(const string& metric,
2674*7304104dSAndroid Build Coastguard Worker            const string& lname, const string& lvalue,
2675*7304104dSAndroid Build Coastguard Worker            const string& rname, const string& rvalue,
2676*7304104dSAndroid Build Coastguard Worker            double value)
2677*7304104dSAndroid Build Coastguard Worker {
2678*7304104dSAndroid Build Coastguard Worker   string key = (metric + "{"
2679*7304104dSAndroid Build Coastguard Worker                 + metric_label(lname, lvalue) + ","
2680*7304104dSAndroid Build Coastguard Worker                 + metric_label(rname, rvalue) + "}");
2681*7304104dSAndroid Build Coastguard Worker   unique_lock<mutex> lock(metrics_lock);
2682*7304104dSAndroid Build Coastguard Worker   metrics[key] += value;
2683*7304104dSAndroid Build Coastguard Worker }
2684*7304104dSAndroid Build Coastguard Worker 
2685*7304104dSAndroid Build Coastguard Worker static struct MHD_Response*
handle_metrics(off_t * size)2686*7304104dSAndroid Build Coastguard Worker handle_metrics (off_t* size)
2687*7304104dSAndroid Build Coastguard Worker {
2688*7304104dSAndroid Build Coastguard Worker   stringstream o;
2689*7304104dSAndroid Build Coastguard Worker   {
2690*7304104dSAndroid Build Coastguard Worker     unique_lock<mutex> lock(metrics_lock);
2691*7304104dSAndroid Build Coastguard Worker     for (auto&& i : metrics)
2692*7304104dSAndroid Build Coastguard Worker       o << i.first
2693*7304104dSAndroid Build Coastguard Worker         << " "
2694*7304104dSAndroid Build Coastguard Worker         << std::setprecision(std::numeric_limits<double>::digits10 + 1)
2695*7304104dSAndroid Build Coastguard Worker         << i.second
2696*7304104dSAndroid Build Coastguard Worker         << endl;
2697*7304104dSAndroid Build Coastguard Worker   }
2698*7304104dSAndroid Build Coastguard Worker   const string& os = o.str();
2699*7304104dSAndroid Build Coastguard Worker   MHD_Response* r = MHD_create_response_from_buffer (os.size(),
2700*7304104dSAndroid Build Coastguard Worker                                                      (void*) os.c_str(),
2701*7304104dSAndroid Build Coastguard Worker                                                      MHD_RESPMEM_MUST_COPY);
2702*7304104dSAndroid Build Coastguard Worker   if (r != NULL)
2703*7304104dSAndroid Build Coastguard Worker     {
2704*7304104dSAndroid Build Coastguard Worker       *size = os.size();
2705*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "Content-Type", "text/plain");
2706*7304104dSAndroid Build Coastguard Worker     }
2707*7304104dSAndroid Build Coastguard Worker   return r;
2708*7304104dSAndroid Build Coastguard Worker }
2709*7304104dSAndroid Build Coastguard Worker 
2710*7304104dSAndroid Build Coastguard Worker static struct MHD_Response*
handle_root(off_t * size)2711*7304104dSAndroid Build Coastguard Worker handle_root (off_t* size)
2712*7304104dSAndroid Build Coastguard Worker {
2713*7304104dSAndroid Build Coastguard Worker   static string version = "debuginfod (" + string (PACKAGE_NAME) + ") "
2714*7304104dSAndroid Build Coastguard Worker 			  + string (PACKAGE_VERSION);
2715*7304104dSAndroid Build Coastguard Worker   MHD_Response* r = MHD_create_response_from_buffer (version.size (),
2716*7304104dSAndroid Build Coastguard Worker 						     (void *) version.c_str (),
2717*7304104dSAndroid Build Coastguard Worker 						     MHD_RESPMEM_PERSISTENT);
2718*7304104dSAndroid Build Coastguard Worker   if (r != NULL)
2719*7304104dSAndroid Build Coastguard Worker     {
2720*7304104dSAndroid Build Coastguard Worker       *size = version.size ();
2721*7304104dSAndroid Build Coastguard Worker       add_mhd_response_header (r, "Content-Type", "text/plain");
2722*7304104dSAndroid Build Coastguard Worker     }
2723*7304104dSAndroid Build Coastguard Worker   return r;
2724*7304104dSAndroid Build Coastguard Worker }
2725*7304104dSAndroid Build Coastguard Worker 
2726*7304104dSAndroid Build Coastguard Worker 
2727*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
2728*7304104dSAndroid Build Coastguard Worker 
2729*7304104dSAndroid Build Coastguard Worker 
2730*7304104dSAndroid Build Coastguard Worker /* libmicrohttpd callback */
2731*7304104dSAndroid Build Coastguard Worker static MHD_RESULT
handler_cb(void *,struct MHD_Connection * connection,const char * url,const char * method,const char *,const char *,size_t *,void ** ptr)2732*7304104dSAndroid Build Coastguard Worker handler_cb (void * /*cls*/,
2733*7304104dSAndroid Build Coastguard Worker             struct MHD_Connection *connection,
2734*7304104dSAndroid Build Coastguard Worker             const char *url,
2735*7304104dSAndroid Build Coastguard Worker             const char *method,
2736*7304104dSAndroid Build Coastguard Worker             const char * /*version*/,
2737*7304104dSAndroid Build Coastguard Worker             const char * /*upload_data*/,
2738*7304104dSAndroid Build Coastguard Worker             size_t * /*upload_data_size*/,
2739*7304104dSAndroid Build Coastguard Worker             void ** ptr)
2740*7304104dSAndroid Build Coastguard Worker {
2741*7304104dSAndroid Build Coastguard Worker   struct MHD_Response *r = NULL;
2742*7304104dSAndroid Build Coastguard Worker   string url_copy = url;
2743*7304104dSAndroid Build Coastguard Worker 
2744*7304104dSAndroid Build Coastguard Worker   /* libmicrohttpd always makes (at least) two callbacks: once just
2745*7304104dSAndroid Build Coastguard Worker      past the headers, and one after the request body is finished
2746*7304104dSAndroid Build Coastguard Worker      being received.  If we process things early (first callback) and
2747*7304104dSAndroid Build Coastguard Worker      queue a response, libmicrohttpd would suppress http keep-alive
2748*7304104dSAndroid Build Coastguard Worker      (via connection->read_closed = true). */
2749*7304104dSAndroid Build Coastguard Worker   static int aptr; /* just some random object to use as a flag */
2750*7304104dSAndroid Build Coastguard Worker   if (&aptr != *ptr)
2751*7304104dSAndroid Build Coastguard Worker     {
2752*7304104dSAndroid Build Coastguard Worker       /* do never respond on first call */
2753*7304104dSAndroid Build Coastguard Worker       *ptr = &aptr;
2754*7304104dSAndroid Build Coastguard Worker       return MHD_YES;
2755*7304104dSAndroid Build Coastguard Worker     }
2756*7304104dSAndroid Build Coastguard Worker   *ptr = NULL;                     /* reset when done */
2757*7304104dSAndroid Build Coastguard Worker 
2758*7304104dSAndroid Build Coastguard Worker   const char *maxsize_string = MHD_lookup_connection_value(connection, MHD_HEADER_KIND, "X-DEBUGINFOD-MAXSIZE");
2759*7304104dSAndroid Build Coastguard Worker   long maxsize = 0;
2760*7304104dSAndroid Build Coastguard Worker   if (maxsize_string != NULL && maxsize_string[0] != '\0')
2761*7304104dSAndroid Build Coastguard Worker     maxsize = atol(maxsize_string);
2762*7304104dSAndroid Build Coastguard Worker   else
2763*7304104dSAndroid Build Coastguard Worker     maxsize = 0;
2764*7304104dSAndroid Build Coastguard Worker 
2765*7304104dSAndroid Build Coastguard Worker #if MHD_VERSION >= 0x00097002
2766*7304104dSAndroid Build Coastguard Worker   enum MHD_Result rc;
2767*7304104dSAndroid Build Coastguard Worker #else
2768*7304104dSAndroid Build Coastguard Worker   int rc = MHD_NO; // mhd
2769*7304104dSAndroid Build Coastguard Worker #endif
2770*7304104dSAndroid Build Coastguard Worker   int http_code = 500;
2771*7304104dSAndroid Build Coastguard Worker   off_t http_size = -1;
2772*7304104dSAndroid Build Coastguard Worker   struct timespec ts_start, ts_end;
2773*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &ts_start);
2774*7304104dSAndroid Build Coastguard Worker   double afteryou = 0.0;
2775*7304104dSAndroid Build Coastguard Worker   string artifacttype, suffix;
2776*7304104dSAndroid Build Coastguard Worker 
2777*7304104dSAndroid Build Coastguard Worker   try
2778*7304104dSAndroid Build Coastguard Worker     {
2779*7304104dSAndroid Build Coastguard Worker       if (string(method) != "GET")
2780*7304104dSAndroid Build Coastguard Worker         throw reportable_exception(400, "we support GET only");
2781*7304104dSAndroid Build Coastguard Worker 
2782*7304104dSAndroid Build Coastguard Worker       /* Start decoding the URL. */
2783*7304104dSAndroid Build Coastguard Worker       size_t slash1 = url_copy.find('/', 1);
2784*7304104dSAndroid Build Coastguard Worker       string url1 = url_copy.substr(0, slash1); // ok even if slash1 not found
2785*7304104dSAndroid Build Coastguard Worker 
2786*7304104dSAndroid Build Coastguard Worker       if (slash1 != string::npos && url1 == "/buildid")
2787*7304104dSAndroid Build Coastguard Worker         {
2788*7304104dSAndroid Build Coastguard Worker           // PR27863: block this thread awhile if another thread is already busy
2789*7304104dSAndroid Build Coastguard Worker           // fetching the exact same thing.  This is better for Everyone.
2790*7304104dSAndroid Build Coastguard Worker           // The latecomer says "... after you!" and waits.
2791*7304104dSAndroid Build Coastguard Worker           add_metric ("thread_busy", "role", "http-buildid-after-you", 1);
2792*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_PTHREAD_SETNAME_NP
2793*7304104dSAndroid Build Coastguard Worker           (void) pthread_setname_np (pthread_self(), "mhd-buildid-after-you");
2794*7304104dSAndroid Build Coastguard Worker #endif
2795*7304104dSAndroid Build Coastguard Worker           struct timespec tsay_start, tsay_end;
2796*7304104dSAndroid Build Coastguard Worker           clock_gettime (CLOCK_MONOTONIC, &tsay_start);
2797*7304104dSAndroid Build Coastguard Worker           static unique_set<string> busy_urls;
2798*7304104dSAndroid Build Coastguard Worker           unique_set_reserver<string> after_you(busy_urls, url_copy);
2799*7304104dSAndroid Build Coastguard Worker           clock_gettime (CLOCK_MONOTONIC, &tsay_end);
2800*7304104dSAndroid Build Coastguard Worker           afteryou = (tsay_end.tv_sec - tsay_start.tv_sec) + (tsay_end.tv_nsec - tsay_start.tv_nsec)/1.e9;
2801*7304104dSAndroid Build Coastguard Worker           add_metric ("thread_busy", "role", "http-buildid-after-you", -1);
2802*7304104dSAndroid Build Coastguard Worker 
2803*7304104dSAndroid Build Coastguard Worker           tmp_inc_metric m ("thread_busy", "role", "http-buildid");
2804*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_PTHREAD_SETNAME_NP
2805*7304104dSAndroid Build Coastguard Worker           (void) pthread_setname_np (pthread_self(), "mhd-buildid");
2806*7304104dSAndroid Build Coastguard Worker #endif
2807*7304104dSAndroid Build Coastguard Worker           size_t slash2 = url_copy.find('/', slash1+1);
2808*7304104dSAndroid Build Coastguard Worker           if (slash2 == string::npos)
2809*7304104dSAndroid Build Coastguard Worker             throw reportable_exception("/buildid/ webapi error, need buildid");
2810*7304104dSAndroid Build Coastguard Worker 
2811*7304104dSAndroid Build Coastguard Worker           string buildid = url_copy.substr(slash1+1, slash2-slash1-1);
2812*7304104dSAndroid Build Coastguard Worker 
2813*7304104dSAndroid Build Coastguard Worker           size_t slash3 = url_copy.find('/', slash2+1);
2814*7304104dSAndroid Build Coastguard Worker 
2815*7304104dSAndroid Build Coastguard Worker           if (slash3 == string::npos)
2816*7304104dSAndroid Build Coastguard Worker             {
2817*7304104dSAndroid Build Coastguard Worker               artifacttype = url_copy.substr(slash2+1);
2818*7304104dSAndroid Build Coastguard Worker               suffix = "";
2819*7304104dSAndroid Build Coastguard Worker             }
2820*7304104dSAndroid Build Coastguard Worker           else
2821*7304104dSAndroid Build Coastguard Worker             {
2822*7304104dSAndroid Build Coastguard Worker               artifacttype = url_copy.substr(slash2+1, slash3-slash2-1);
2823*7304104dSAndroid Build Coastguard Worker               suffix = url_copy.substr(slash3); // include the slash in the suffix
2824*7304104dSAndroid Build Coastguard Worker             }
2825*7304104dSAndroid Build Coastguard Worker 
2826*7304104dSAndroid Build Coastguard Worker           // get the resulting fd so we can report its size
2827*7304104dSAndroid Build Coastguard Worker           int fd;
2828*7304104dSAndroid Build Coastguard Worker           r = handle_buildid(connection, buildid, artifacttype, suffix, &fd);
2829*7304104dSAndroid Build Coastguard Worker           if (r)
2830*7304104dSAndroid Build Coastguard Worker             {
2831*7304104dSAndroid Build Coastguard Worker               struct stat fs;
2832*7304104dSAndroid Build Coastguard Worker               if (fstat(fd, &fs) == 0)
2833*7304104dSAndroid Build Coastguard Worker                 http_size = fs.st_size;
2834*7304104dSAndroid Build Coastguard Worker               // libmicrohttpd will close (fd);
2835*7304104dSAndroid Build Coastguard Worker             }
2836*7304104dSAndroid Build Coastguard Worker         }
2837*7304104dSAndroid Build Coastguard Worker       else if (url1 == "/metrics")
2838*7304104dSAndroid Build Coastguard Worker         {
2839*7304104dSAndroid Build Coastguard Worker           tmp_inc_metric m ("thread_busy", "role", "http-metrics");
2840*7304104dSAndroid Build Coastguard Worker           artifacttype = "metrics";
2841*7304104dSAndroid Build Coastguard Worker           inc_metric("http_requests_total", "type", artifacttype);
2842*7304104dSAndroid Build Coastguard Worker           r = handle_metrics(& http_size);
2843*7304104dSAndroid Build Coastguard Worker         }
2844*7304104dSAndroid Build Coastguard Worker       else if (url1 == "/")
2845*7304104dSAndroid Build Coastguard Worker         {
2846*7304104dSAndroid Build Coastguard Worker           artifacttype = "/";
2847*7304104dSAndroid Build Coastguard Worker           inc_metric("http_requests_total", "type", artifacttype);
2848*7304104dSAndroid Build Coastguard Worker           r = handle_root(& http_size);
2849*7304104dSAndroid Build Coastguard Worker         }
2850*7304104dSAndroid Build Coastguard Worker       else
2851*7304104dSAndroid Build Coastguard Worker         throw reportable_exception("webapi error, unrecognized '" + url1 + "'");
2852*7304104dSAndroid Build Coastguard Worker 
2853*7304104dSAndroid Build Coastguard Worker       if (r == 0)
2854*7304104dSAndroid Build Coastguard Worker         throw reportable_exception("internal error, missing response");
2855*7304104dSAndroid Build Coastguard Worker 
2856*7304104dSAndroid Build Coastguard Worker       if (maxsize > 0 && http_size > maxsize)
2857*7304104dSAndroid Build Coastguard Worker         {
2858*7304104dSAndroid Build Coastguard Worker           MHD_destroy_response(r);
2859*7304104dSAndroid Build Coastguard Worker           throw reportable_exception(406, "File too large, max size=" + std::to_string(maxsize));
2860*7304104dSAndroid Build Coastguard Worker         }
2861*7304104dSAndroid Build Coastguard Worker 
2862*7304104dSAndroid Build Coastguard Worker       rc = MHD_queue_response (connection, MHD_HTTP_OK, r);
2863*7304104dSAndroid Build Coastguard Worker       http_code = MHD_HTTP_OK;
2864*7304104dSAndroid Build Coastguard Worker       MHD_destroy_response (r);
2865*7304104dSAndroid Build Coastguard Worker     }
2866*7304104dSAndroid Build Coastguard Worker   catch (const reportable_exception& e)
2867*7304104dSAndroid Build Coastguard Worker     {
2868*7304104dSAndroid Build Coastguard Worker       inc_metric("http_responses_total","result","error");
2869*7304104dSAndroid Build Coastguard Worker       e.report(clog);
2870*7304104dSAndroid Build Coastguard Worker       http_code = e.code;
2871*7304104dSAndroid Build Coastguard Worker       http_size = e.message.size();
2872*7304104dSAndroid Build Coastguard Worker       rc = e.mhd_send_response (connection);
2873*7304104dSAndroid Build Coastguard Worker     }
2874*7304104dSAndroid Build Coastguard Worker 
2875*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &ts_end);
2876*7304104dSAndroid Build Coastguard Worker   double deltas = (ts_end.tv_sec - ts_start.tv_sec) + (ts_end.tv_nsec - ts_start.tv_nsec)/1.e9;
2877*7304104dSAndroid Build Coastguard Worker   // afteryou: delay waiting for other client's identical query to complete
2878*7304104dSAndroid Build Coastguard Worker   // deltas: total latency, including afteryou waiting
2879*7304104dSAndroid Build Coastguard Worker   obatched(clog) << conninfo(connection)
2880*7304104dSAndroid Build Coastguard Worker                  << ' ' << method << ' ' << url
2881*7304104dSAndroid Build Coastguard Worker                  << ' ' << http_code << ' ' << http_size
2882*7304104dSAndroid Build Coastguard Worker                  << ' ' << (int)(afteryou*1000) << '+' << (int)((deltas-afteryou)*1000) << "ms"
2883*7304104dSAndroid Build Coastguard Worker                  << endl;
2884*7304104dSAndroid Build Coastguard Worker 
2885*7304104dSAndroid Build Coastguard Worker   // related prometheus metrics
2886*7304104dSAndroid Build Coastguard Worker   string http_code_str = to_string(http_code);
2887*7304104dSAndroid Build Coastguard Worker   add_metric("http_responses_transfer_bytes_sum",
2888*7304104dSAndroid Build Coastguard Worker              "code", http_code_str, "type", artifacttype, http_size);
2889*7304104dSAndroid Build Coastguard Worker   inc_metric("http_responses_transfer_bytes_count",
2890*7304104dSAndroid Build Coastguard Worker              "code", http_code_str, "type", artifacttype);
2891*7304104dSAndroid Build Coastguard Worker 
2892*7304104dSAndroid Build Coastguard Worker   add_metric("http_responses_duration_milliseconds_sum",
2893*7304104dSAndroid Build Coastguard Worker              "code", http_code_str, "type", artifacttype, deltas*1000); // prometheus prefers _seconds and floating point
2894*7304104dSAndroid Build Coastguard Worker   inc_metric("http_responses_duration_milliseconds_count",
2895*7304104dSAndroid Build Coastguard Worker              "code", http_code_str, "type", artifacttype);
2896*7304104dSAndroid Build Coastguard Worker 
2897*7304104dSAndroid Build Coastguard Worker   add_metric("http_responses_after_you_milliseconds_sum",
2898*7304104dSAndroid Build Coastguard Worker              "code", http_code_str, "type", artifacttype, afteryou*1000);
2899*7304104dSAndroid Build Coastguard Worker   inc_metric("http_responses_after_you_milliseconds_count",
2900*7304104dSAndroid Build Coastguard Worker              "code", http_code_str, "type", artifacttype);
2901*7304104dSAndroid Build Coastguard Worker 
2902*7304104dSAndroid Build Coastguard Worker   return rc;
2903*7304104dSAndroid Build Coastguard Worker }
2904*7304104dSAndroid Build Coastguard Worker 
2905*7304104dSAndroid Build Coastguard Worker 
2906*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
2907*7304104dSAndroid Build Coastguard Worker // borrowed originally from src/nm.c get_local_names()
2908*7304104dSAndroid Build Coastguard Worker 
2909*7304104dSAndroid Build Coastguard Worker static void
dwarf_extract_source_paths(Elf * elf,set<string> & debug_sourcefiles)2910*7304104dSAndroid Build Coastguard Worker dwarf_extract_source_paths (Elf *elf, set<string>& debug_sourcefiles)
2911*7304104dSAndroid Build Coastguard Worker   noexcept // no exceptions - so we can simplify the altdbg resource release at end
2912*7304104dSAndroid Build Coastguard Worker {
2913*7304104dSAndroid Build Coastguard Worker   Dwarf* dbg = dwarf_begin_elf (elf, DWARF_C_READ, NULL);
2914*7304104dSAndroid Build Coastguard Worker   if (dbg == NULL)
2915*7304104dSAndroid Build Coastguard Worker     return;
2916*7304104dSAndroid Build Coastguard Worker 
2917*7304104dSAndroid Build Coastguard Worker   Dwarf* altdbg = NULL;
2918*7304104dSAndroid Build Coastguard Worker   int    altdbg_fd = -1;
2919*7304104dSAndroid Build Coastguard Worker 
2920*7304104dSAndroid Build Coastguard Worker   // DWZ handling: if we have an unsatisfied debug-alt-link, add an
2921*7304104dSAndroid Build Coastguard Worker   // empty string into the outgoing sourcefiles set, so the caller
2922*7304104dSAndroid Build Coastguard Worker   // should know that our data is incomplete.
2923*7304104dSAndroid Build Coastguard Worker   const char *alt_name_p;
2924*7304104dSAndroid Build Coastguard Worker   const void *alt_build_id; // elfutils-owned memory
2925*7304104dSAndroid Build Coastguard Worker   ssize_t sz = dwelf_dwarf_gnu_debugaltlink (dbg, &alt_name_p, &alt_build_id);
2926*7304104dSAndroid Build Coastguard Worker   if (sz > 0) // got one!
2927*7304104dSAndroid Build Coastguard Worker     {
2928*7304104dSAndroid Build Coastguard Worker       string buildid;
2929*7304104dSAndroid Build Coastguard Worker       unsigned char* build_id_bytes = (unsigned char*) alt_build_id;
2930*7304104dSAndroid Build Coastguard Worker       for (ssize_t idx=0; idx<sz; idx++)
2931*7304104dSAndroid Build Coastguard Worker         {
2932*7304104dSAndroid Build Coastguard Worker           buildid += "0123456789abcdef"[build_id_bytes[idx] >> 4];
2933*7304104dSAndroid Build Coastguard Worker           buildid += "0123456789abcdef"[build_id_bytes[idx] & 0xf];
2934*7304104dSAndroid Build Coastguard Worker         }
2935*7304104dSAndroid Build Coastguard Worker 
2936*7304104dSAndroid Build Coastguard Worker       if (verbose > 3)
2937*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "Need altdebug buildid=" << buildid << endl;
2938*7304104dSAndroid Build Coastguard Worker 
2939*7304104dSAndroid Build Coastguard Worker       // but is it unsatisfied the normal elfutils ways?
2940*7304104dSAndroid Build Coastguard Worker       Dwarf* alt = dwarf_getalt (dbg);
2941*7304104dSAndroid Build Coastguard Worker       if (alt == NULL)
2942*7304104dSAndroid Build Coastguard Worker         {
2943*7304104dSAndroid Build Coastguard Worker           // Yup, unsatisfied the normal way.  Maybe we can satisfy it
2944*7304104dSAndroid Build Coastguard Worker           // from our own debuginfod database.
2945*7304104dSAndroid Build Coastguard Worker           int alt_fd;
2946*7304104dSAndroid Build Coastguard Worker           struct MHD_Response *r = 0;
2947*7304104dSAndroid Build Coastguard Worker           try
2948*7304104dSAndroid Build Coastguard Worker             {
2949*7304104dSAndroid Build Coastguard Worker               string artifacttype = "debuginfo";
2950*7304104dSAndroid Build Coastguard Worker               r = handle_buildid (0, buildid, artifacttype, "", &alt_fd);
2951*7304104dSAndroid Build Coastguard Worker             }
2952*7304104dSAndroid Build Coastguard Worker           catch (const reportable_exception& e)
2953*7304104dSAndroid Build Coastguard Worker             {
2954*7304104dSAndroid Build Coastguard Worker               // swallow exceptions
2955*7304104dSAndroid Build Coastguard Worker             }
2956*7304104dSAndroid Build Coastguard Worker 
2957*7304104dSAndroid Build Coastguard Worker           // NB: this is not actually recursive!  This invokes the web-query
2958*7304104dSAndroid Build Coastguard Worker           // path, which cannot get back into the scan code paths.
2959*7304104dSAndroid Build Coastguard Worker           if (r)
2960*7304104dSAndroid Build Coastguard Worker             {
2961*7304104dSAndroid Build Coastguard Worker               // Found it!
2962*7304104dSAndroid Build Coastguard Worker               altdbg_fd = dup(alt_fd); // ok if this fails, downstream failures ok
2963*7304104dSAndroid Build Coastguard Worker               alt = altdbg = dwarf_begin (altdbg_fd, DWARF_C_READ);
2964*7304104dSAndroid Build Coastguard Worker               // NB: must close this dwarf and this fd at the bottom of the function!
2965*7304104dSAndroid Build Coastguard Worker               MHD_destroy_response (r); // will close alt_fd
2966*7304104dSAndroid Build Coastguard Worker               if (alt)
2967*7304104dSAndroid Build Coastguard Worker                 dwarf_setalt (dbg, alt);
2968*7304104dSAndroid Build Coastguard Worker             }
2969*7304104dSAndroid Build Coastguard Worker         }
2970*7304104dSAndroid Build Coastguard Worker       else
2971*7304104dSAndroid Build Coastguard Worker         {
2972*7304104dSAndroid Build Coastguard Worker           // NB: dwarf_setalt(alt) inappropriate - already done!
2973*7304104dSAndroid Build Coastguard Worker           // NB: altdbg will stay 0 so nothing tries to redundantly dealloc.
2974*7304104dSAndroid Build Coastguard Worker         }
2975*7304104dSAndroid Build Coastguard Worker 
2976*7304104dSAndroid Build Coastguard Worker       if (alt)
2977*7304104dSAndroid Build Coastguard Worker         {
2978*7304104dSAndroid Build Coastguard Worker           if (verbose > 3)
2979*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "Resolved altdebug buildid=" << buildid << endl;
2980*7304104dSAndroid Build Coastguard Worker         }
2981*7304104dSAndroid Build Coastguard Worker       else // (alt == NULL) - signal possible presence of poor debuginfo
2982*7304104dSAndroid Build Coastguard Worker         {
2983*7304104dSAndroid Build Coastguard Worker           debug_sourcefiles.insert("");
2984*7304104dSAndroid Build Coastguard Worker           if (verbose > 3)
2985*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "Unresolved altdebug buildid=" << buildid << endl;
2986*7304104dSAndroid Build Coastguard Worker         }
2987*7304104dSAndroid Build Coastguard Worker     }
2988*7304104dSAndroid Build Coastguard Worker 
2989*7304104dSAndroid Build Coastguard Worker   Dwarf_Off offset = 0;
2990*7304104dSAndroid Build Coastguard Worker   Dwarf_Off old_offset;
2991*7304104dSAndroid Build Coastguard Worker   size_t hsize;
2992*7304104dSAndroid Build Coastguard Worker 
2993*7304104dSAndroid Build Coastguard Worker   while (dwarf_nextcu (dbg, old_offset = offset, &offset, &hsize, NULL, NULL, NULL) == 0)
2994*7304104dSAndroid Build Coastguard Worker     {
2995*7304104dSAndroid Build Coastguard Worker       Dwarf_Die cudie_mem;
2996*7304104dSAndroid Build Coastguard Worker       Dwarf_Die *cudie = dwarf_offdie (dbg, old_offset + hsize, &cudie_mem);
2997*7304104dSAndroid Build Coastguard Worker 
2998*7304104dSAndroid Build Coastguard Worker       if (cudie == NULL)
2999*7304104dSAndroid Build Coastguard Worker         continue;
3000*7304104dSAndroid Build Coastguard Worker       if (dwarf_tag (cudie) != DW_TAG_compile_unit)
3001*7304104dSAndroid Build Coastguard Worker         continue;
3002*7304104dSAndroid Build Coastguard Worker 
3003*7304104dSAndroid Build Coastguard Worker       const char *cuname = dwarf_diename(cudie) ?: "unknown";
3004*7304104dSAndroid Build Coastguard Worker 
3005*7304104dSAndroid Build Coastguard Worker       Dwarf_Files *files;
3006*7304104dSAndroid Build Coastguard Worker       size_t nfiles;
3007*7304104dSAndroid Build Coastguard Worker       if (dwarf_getsrcfiles (cudie, &files, &nfiles) != 0)
3008*7304104dSAndroid Build Coastguard Worker         continue;
3009*7304104dSAndroid Build Coastguard Worker 
3010*7304104dSAndroid Build Coastguard Worker       // extract DW_AT_comp_dir to resolve relative file names
3011*7304104dSAndroid Build Coastguard Worker       const char *comp_dir = "";
3012*7304104dSAndroid Build Coastguard Worker       const char *const *dirs;
3013*7304104dSAndroid Build Coastguard Worker       size_t ndirs;
3014*7304104dSAndroid Build Coastguard Worker       if (dwarf_getsrcdirs (files, &dirs, &ndirs) == 0 &&
3015*7304104dSAndroid Build Coastguard Worker           dirs[0] != NULL)
3016*7304104dSAndroid Build Coastguard Worker         comp_dir = dirs[0];
3017*7304104dSAndroid Build Coastguard Worker       if (comp_dir == NULL)
3018*7304104dSAndroid Build Coastguard Worker         comp_dir = "";
3019*7304104dSAndroid Build Coastguard Worker 
3020*7304104dSAndroid Build Coastguard Worker       if (verbose > 3)
3021*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "searching for sources for cu=" << cuname << " comp_dir=" << comp_dir
3022*7304104dSAndroid Build Coastguard Worker                        << " #files=" << nfiles << " #dirs=" << ndirs << endl;
3023*7304104dSAndroid Build Coastguard Worker 
3024*7304104dSAndroid Build Coastguard Worker       if (comp_dir[0] == '\0' && cuname[0] != '/')
3025*7304104dSAndroid Build Coastguard Worker         {
3026*7304104dSAndroid Build Coastguard Worker           if (verbose > 3)
3027*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "skipping cu=" << cuname << " due to empty comp_dir" << endl;
3028*7304104dSAndroid Build Coastguard Worker           continue;
3029*7304104dSAndroid Build Coastguard Worker         }
3030*7304104dSAndroid Build Coastguard Worker 
3031*7304104dSAndroid Build Coastguard Worker       for (size_t f = 1; f < nfiles; f++)
3032*7304104dSAndroid Build Coastguard Worker         {
3033*7304104dSAndroid Build Coastguard Worker           const char *hat = dwarf_filesrc (files, f, NULL, NULL);
3034*7304104dSAndroid Build Coastguard Worker           if (hat == NULL)
3035*7304104dSAndroid Build Coastguard Worker             continue;
3036*7304104dSAndroid Build Coastguard Worker 
3037*7304104dSAndroid Build Coastguard Worker           if (string(hat) == "<built-in>") // gcc intrinsics, don't bother record
3038*7304104dSAndroid Build Coastguard Worker             continue;
3039*7304104dSAndroid Build Coastguard Worker 
3040*7304104dSAndroid Build Coastguard Worker           string waldo;
3041*7304104dSAndroid Build Coastguard Worker           if (hat[0] == '/') // absolute
3042*7304104dSAndroid Build Coastguard Worker             waldo = (string (hat));
3043*7304104dSAndroid Build Coastguard Worker           else if (comp_dir[0] != '\0') // comp_dir relative
3044*7304104dSAndroid Build Coastguard Worker             waldo = (string (comp_dir) + string("/") + string (hat));
3045*7304104dSAndroid Build Coastguard Worker           else
3046*7304104dSAndroid Build Coastguard Worker            {
3047*7304104dSAndroid Build Coastguard Worker              if (verbose > 3)
3048*7304104dSAndroid Build Coastguard Worker                obatched(clog) << "skipping hat=" << hat << " due to empty comp_dir" << endl;
3049*7304104dSAndroid Build Coastguard Worker              continue;
3050*7304104dSAndroid Build Coastguard Worker            }
3051*7304104dSAndroid Build Coastguard Worker 
3052*7304104dSAndroid Build Coastguard Worker           // NB: this is the 'waldo' that a dbginfo client will have
3053*7304104dSAndroid Build Coastguard Worker           // to supply for us to give them the file The comp_dir
3054*7304104dSAndroid Build Coastguard Worker           // prefixing is a definite complication.  Otherwise we'd
3055*7304104dSAndroid Build Coastguard Worker           // have to return a setof comp_dirs (one per CU!) with
3056*7304104dSAndroid Build Coastguard Worker           // corresponding filesrc[] names, instead of one absolute
3057*7304104dSAndroid Build Coastguard Worker           // resoved set.  Maybe we'll have to do that anyway.  XXX
3058*7304104dSAndroid Build Coastguard Worker 
3059*7304104dSAndroid Build Coastguard Worker           if (verbose > 4)
3060*7304104dSAndroid Build Coastguard Worker             obatched(clog) << waldo
3061*7304104dSAndroid Build Coastguard Worker                            << (debug_sourcefiles.find(waldo)==debug_sourcefiles.end() ? " new" : " dup") <<  endl;
3062*7304104dSAndroid Build Coastguard Worker 
3063*7304104dSAndroid Build Coastguard Worker           debug_sourcefiles.insert (waldo);
3064*7304104dSAndroid Build Coastguard Worker         }
3065*7304104dSAndroid Build Coastguard Worker     }
3066*7304104dSAndroid Build Coastguard Worker 
3067*7304104dSAndroid Build Coastguard Worker   dwarf_end(dbg);
3068*7304104dSAndroid Build Coastguard Worker   if (altdbg)
3069*7304104dSAndroid Build Coastguard Worker     dwarf_end(altdbg);
3070*7304104dSAndroid Build Coastguard Worker   if (altdbg_fd >= 0)
3071*7304104dSAndroid Build Coastguard Worker     close(altdbg_fd);
3072*7304104dSAndroid Build Coastguard Worker }
3073*7304104dSAndroid Build Coastguard Worker 
3074*7304104dSAndroid Build Coastguard Worker 
3075*7304104dSAndroid Build Coastguard Worker 
3076*7304104dSAndroid Build Coastguard Worker static void
elf_classify(int fd,bool & executable_p,bool & debuginfo_p,string & buildid,set<string> & debug_sourcefiles)3077*7304104dSAndroid Build Coastguard Worker elf_classify (int fd, bool &executable_p, bool &debuginfo_p, string &buildid, set<string>& debug_sourcefiles)
3078*7304104dSAndroid Build Coastguard Worker {
3079*7304104dSAndroid Build Coastguard Worker   Elf *elf = elf_begin (fd, ELF_C_READ_MMAP_PRIVATE, NULL);
3080*7304104dSAndroid Build Coastguard Worker   if (elf == NULL)
3081*7304104dSAndroid Build Coastguard Worker     return;
3082*7304104dSAndroid Build Coastguard Worker 
3083*7304104dSAndroid Build Coastguard Worker   try // catch our types of errors and clean up the Elf* object
3084*7304104dSAndroid Build Coastguard Worker     {
3085*7304104dSAndroid Build Coastguard Worker       if (elf_kind (elf) != ELF_K_ELF)
3086*7304104dSAndroid Build Coastguard Worker         {
3087*7304104dSAndroid Build Coastguard Worker           elf_end (elf);
3088*7304104dSAndroid Build Coastguard Worker           return;
3089*7304104dSAndroid Build Coastguard Worker         }
3090*7304104dSAndroid Build Coastguard Worker 
3091*7304104dSAndroid Build Coastguard Worker       GElf_Ehdr ehdr_storage;
3092*7304104dSAndroid Build Coastguard Worker       GElf_Ehdr *ehdr = gelf_getehdr (elf, &ehdr_storage);
3093*7304104dSAndroid Build Coastguard Worker       if (ehdr == NULL)
3094*7304104dSAndroid Build Coastguard Worker         {
3095*7304104dSAndroid Build Coastguard Worker           elf_end (elf);
3096*7304104dSAndroid Build Coastguard Worker           return;
3097*7304104dSAndroid Build Coastguard Worker         }
3098*7304104dSAndroid Build Coastguard Worker       auto elf_type = ehdr->e_type;
3099*7304104dSAndroid Build Coastguard Worker 
3100*7304104dSAndroid Build Coastguard Worker       const void *build_id; // elfutils-owned memory
3101*7304104dSAndroid Build Coastguard Worker       ssize_t sz = dwelf_elf_gnu_build_id (elf, & build_id);
3102*7304104dSAndroid Build Coastguard Worker       if (sz <= 0)
3103*7304104dSAndroid Build Coastguard Worker         {
3104*7304104dSAndroid Build Coastguard Worker           // It's not a diagnostic-worthy error for an elf file to lack build-id.
3105*7304104dSAndroid Build Coastguard Worker           // It might just be very old.
3106*7304104dSAndroid Build Coastguard Worker           elf_end (elf);
3107*7304104dSAndroid Build Coastguard Worker           return;
3108*7304104dSAndroid Build Coastguard Worker         }
3109*7304104dSAndroid Build Coastguard Worker 
3110*7304104dSAndroid Build Coastguard Worker       // build_id is a raw byte array; convert to hexadecimal *lowercase*
3111*7304104dSAndroid Build Coastguard Worker       unsigned char* build_id_bytes = (unsigned char*) build_id;
3112*7304104dSAndroid Build Coastguard Worker       for (ssize_t idx=0; idx<sz; idx++)
3113*7304104dSAndroid Build Coastguard Worker         {
3114*7304104dSAndroid Build Coastguard Worker           buildid += "0123456789abcdef"[build_id_bytes[idx] >> 4];
3115*7304104dSAndroid Build Coastguard Worker           buildid += "0123456789abcdef"[build_id_bytes[idx] & 0xf];
3116*7304104dSAndroid Build Coastguard Worker         }
3117*7304104dSAndroid Build Coastguard Worker 
3118*7304104dSAndroid Build Coastguard Worker       // now decide whether it's an executable - namely, any allocatable section has
3119*7304104dSAndroid Build Coastguard Worker       // PROGBITS;
3120*7304104dSAndroid Build Coastguard Worker       if (elf_type == ET_EXEC || elf_type == ET_DYN)
3121*7304104dSAndroid Build Coastguard Worker         {
3122*7304104dSAndroid Build Coastguard Worker           size_t shnum;
3123*7304104dSAndroid Build Coastguard Worker           int rc = elf_getshdrnum (elf, &shnum);
3124*7304104dSAndroid Build Coastguard Worker           if (rc < 0)
3125*7304104dSAndroid Build Coastguard Worker             throw elfutils_exception(rc, "getshdrnum");
3126*7304104dSAndroid Build Coastguard Worker 
3127*7304104dSAndroid Build Coastguard Worker           executable_p = false;
3128*7304104dSAndroid Build Coastguard Worker           for (size_t sc = 0; sc < shnum; sc++)
3129*7304104dSAndroid Build Coastguard Worker             {
3130*7304104dSAndroid Build Coastguard Worker               Elf_Scn *scn = elf_getscn (elf, sc);
3131*7304104dSAndroid Build Coastguard Worker               if (scn == NULL)
3132*7304104dSAndroid Build Coastguard Worker                 continue;
3133*7304104dSAndroid Build Coastguard Worker 
3134*7304104dSAndroid Build Coastguard Worker               GElf_Shdr shdr_mem;
3135*7304104dSAndroid Build Coastguard Worker               GElf_Shdr *shdr = gelf_getshdr (scn, &shdr_mem);
3136*7304104dSAndroid Build Coastguard Worker               if (shdr == NULL)
3137*7304104dSAndroid Build Coastguard Worker                 continue;
3138*7304104dSAndroid Build Coastguard Worker 
3139*7304104dSAndroid Build Coastguard Worker               // allocated (loadable / vm-addr-assigned) section with available content?
3140*7304104dSAndroid Build Coastguard Worker               if ((shdr->sh_type == SHT_PROGBITS) && (shdr->sh_flags & SHF_ALLOC))
3141*7304104dSAndroid Build Coastguard Worker                 {
3142*7304104dSAndroid Build Coastguard Worker                   if (verbose > 4)
3143*7304104dSAndroid Build Coastguard Worker                     obatched(clog) << "executable due to SHF_ALLOC SHT_PROGBITS sc=" << sc << endl;
3144*7304104dSAndroid Build Coastguard Worker                   executable_p = true;
3145*7304104dSAndroid Build Coastguard Worker                   break; // no need to keep looking for others
3146*7304104dSAndroid Build Coastguard Worker                 }
3147*7304104dSAndroid Build Coastguard Worker             } // iterate over sections
3148*7304104dSAndroid Build Coastguard Worker         } // executable_p classification
3149*7304104dSAndroid Build Coastguard Worker 
3150*7304104dSAndroid Build Coastguard Worker       // now decide whether it's a debuginfo - namely, if it has any .debug* or .zdebug* sections
3151*7304104dSAndroid Build Coastguard Worker       // logic mostly stolen from [email protected]'s elfclassify drafts
3152*7304104dSAndroid Build Coastguard Worker       size_t shstrndx;
3153*7304104dSAndroid Build Coastguard Worker       int rc = elf_getshdrstrndx (elf, &shstrndx);
3154*7304104dSAndroid Build Coastguard Worker       if (rc < 0)
3155*7304104dSAndroid Build Coastguard Worker         throw elfutils_exception(rc, "getshdrstrndx");
3156*7304104dSAndroid Build Coastguard Worker 
3157*7304104dSAndroid Build Coastguard Worker       Elf_Scn *scn = NULL;
3158*7304104dSAndroid Build Coastguard Worker       bool symtab_p = false;
3159*7304104dSAndroid Build Coastguard Worker       bool bits_alloc_p = false;
3160*7304104dSAndroid Build Coastguard Worker       while (true)
3161*7304104dSAndroid Build Coastguard Worker         {
3162*7304104dSAndroid Build Coastguard Worker           scn = elf_nextscn (elf, scn);
3163*7304104dSAndroid Build Coastguard Worker           if (scn == NULL)
3164*7304104dSAndroid Build Coastguard Worker             break;
3165*7304104dSAndroid Build Coastguard Worker           GElf_Shdr shdr_storage;
3166*7304104dSAndroid Build Coastguard Worker           GElf_Shdr *shdr = gelf_getshdr (scn, &shdr_storage);
3167*7304104dSAndroid Build Coastguard Worker           if (shdr == NULL)
3168*7304104dSAndroid Build Coastguard Worker             break;
3169*7304104dSAndroid Build Coastguard Worker           const char *section_name = elf_strptr (elf, shstrndx, shdr->sh_name);
3170*7304104dSAndroid Build Coastguard Worker           if (section_name == NULL)
3171*7304104dSAndroid Build Coastguard Worker             break;
3172*7304104dSAndroid Build Coastguard Worker           if (startswith (section_name, ".debug_line") ||
3173*7304104dSAndroid Build Coastguard Worker               startswith (section_name, ".zdebug_line"))
3174*7304104dSAndroid Build Coastguard Worker             {
3175*7304104dSAndroid Build Coastguard Worker               debuginfo_p = true;
3176*7304104dSAndroid Build Coastguard Worker               if (scan_source_info)
3177*7304104dSAndroid Build Coastguard Worker                 dwarf_extract_source_paths (elf, debug_sourcefiles);
3178*7304104dSAndroid Build Coastguard Worker               break; // expecting only one .*debug_line, so no need to look for others
3179*7304104dSAndroid Build Coastguard Worker             }
3180*7304104dSAndroid Build Coastguard Worker           else if (startswith (section_name, ".debug_") ||
3181*7304104dSAndroid Build Coastguard Worker                    startswith (section_name, ".zdebug_"))
3182*7304104dSAndroid Build Coastguard Worker             {
3183*7304104dSAndroid Build Coastguard Worker               debuginfo_p = true;
3184*7304104dSAndroid Build Coastguard Worker               // NB: don't break; need to parse .debug_line for sources
3185*7304104dSAndroid Build Coastguard Worker             }
3186*7304104dSAndroid Build Coastguard Worker           else if (shdr->sh_type == SHT_SYMTAB)
3187*7304104dSAndroid Build Coastguard Worker             {
3188*7304104dSAndroid Build Coastguard Worker               symtab_p = true;
3189*7304104dSAndroid Build Coastguard Worker             }
3190*7304104dSAndroid Build Coastguard Worker           else if (shdr->sh_type != SHT_NOBITS
3191*7304104dSAndroid Build Coastguard Worker                    && shdr->sh_type != SHT_NOTE
3192*7304104dSAndroid Build Coastguard Worker                    && (shdr->sh_flags & SHF_ALLOC) != 0)
3193*7304104dSAndroid Build Coastguard Worker             {
3194*7304104dSAndroid Build Coastguard Worker               bits_alloc_p = true;
3195*7304104dSAndroid Build Coastguard Worker             }
3196*7304104dSAndroid Build Coastguard Worker         }
3197*7304104dSAndroid Build Coastguard Worker 
3198*7304104dSAndroid Build Coastguard Worker       // For more expansive elf/split-debuginfo classification, we
3199*7304104dSAndroid Build Coastguard Worker       // want to identify as debuginfo "strip -s"-produced files
3200*7304104dSAndroid Build Coastguard Worker       // without .debug_info* (like libicudata), but we don't want to
3201*7304104dSAndroid Build Coastguard Worker       // identify "strip -g" executables (with .symtab left there).
3202*7304104dSAndroid Build Coastguard Worker       if (symtab_p && !bits_alloc_p)
3203*7304104dSAndroid Build Coastguard Worker         debuginfo_p = true;
3204*7304104dSAndroid Build Coastguard Worker     }
3205*7304104dSAndroid Build Coastguard Worker   catch (const reportable_exception& e)
3206*7304104dSAndroid Build Coastguard Worker     {
3207*7304104dSAndroid Build Coastguard Worker       e.report(clog);
3208*7304104dSAndroid Build Coastguard Worker     }
3209*7304104dSAndroid Build Coastguard Worker   elf_end (elf);
3210*7304104dSAndroid Build Coastguard Worker }
3211*7304104dSAndroid Build Coastguard Worker 
3212*7304104dSAndroid Build Coastguard Worker 
3213*7304104dSAndroid Build Coastguard Worker // Intern the given file name in two parts (dirname & basename) and
3214*7304104dSAndroid Build Coastguard Worker // return the resulting file's id.
3215*7304104dSAndroid Build Coastguard Worker static int64_t
register_file_name(sqlite_ps & ps_upsert_fileparts,sqlite_ps & ps_upsert_file,sqlite_ps & ps_lookup_file,const string & name)3216*7304104dSAndroid Build Coastguard Worker register_file_name(sqlite_ps& ps_upsert_fileparts,
3217*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_upsert_file,
3218*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_lookup_file,
3219*7304104dSAndroid Build Coastguard Worker                    const string& name)
3220*7304104dSAndroid Build Coastguard Worker {
3221*7304104dSAndroid Build Coastguard Worker   std::size_t slash = name.rfind('/');
3222*7304104dSAndroid Build Coastguard Worker   string dirname, filename;
3223*7304104dSAndroid Build Coastguard Worker   if (slash == std::string::npos)
3224*7304104dSAndroid Build Coastguard Worker     {
3225*7304104dSAndroid Build Coastguard Worker       dirname = "";
3226*7304104dSAndroid Build Coastguard Worker       filename = name;
3227*7304104dSAndroid Build Coastguard Worker     }
3228*7304104dSAndroid Build Coastguard Worker   else
3229*7304104dSAndroid Build Coastguard Worker     {
3230*7304104dSAndroid Build Coastguard Worker       dirname = name.substr(0, slash);
3231*7304104dSAndroid Build Coastguard Worker       filename = name.substr(slash+1);
3232*7304104dSAndroid Build Coastguard Worker     }
3233*7304104dSAndroid Build Coastguard Worker 
3234*7304104dSAndroid Build Coastguard Worker   // intern the two substrings
3235*7304104dSAndroid Build Coastguard Worker   ps_upsert_fileparts
3236*7304104dSAndroid Build Coastguard Worker     .reset()
3237*7304104dSAndroid Build Coastguard Worker     .bind(1, dirname)
3238*7304104dSAndroid Build Coastguard Worker     .step_ok_done();
3239*7304104dSAndroid Build Coastguard Worker   ps_upsert_fileparts
3240*7304104dSAndroid Build Coastguard Worker     .reset()
3241*7304104dSAndroid Build Coastguard Worker     .bind(1, filename)
3242*7304104dSAndroid Build Coastguard Worker     .step_ok_done();
3243*7304104dSAndroid Build Coastguard Worker 
3244*7304104dSAndroid Build Coastguard Worker   // intern the tuple
3245*7304104dSAndroid Build Coastguard Worker   ps_upsert_file
3246*7304104dSAndroid Build Coastguard Worker     .reset()
3247*7304104dSAndroid Build Coastguard Worker     .bind(1, dirname)
3248*7304104dSAndroid Build Coastguard Worker     .bind(2, filename)
3249*7304104dSAndroid Build Coastguard Worker     .step_ok_done();
3250*7304104dSAndroid Build Coastguard Worker 
3251*7304104dSAndroid Build Coastguard Worker   // look up the tuple's id
3252*7304104dSAndroid Build Coastguard Worker   ps_lookup_file
3253*7304104dSAndroid Build Coastguard Worker     .reset()
3254*7304104dSAndroid Build Coastguard Worker     .bind(1, dirname)
3255*7304104dSAndroid Build Coastguard Worker     .bind(2, filename);
3256*7304104dSAndroid Build Coastguard Worker   int rc = ps_lookup_file.step();
3257*7304104dSAndroid Build Coastguard Worker   if (rc != SQLITE_ROW) throw sqlite_exception(rc, "step");
3258*7304104dSAndroid Build Coastguard Worker 
3259*7304104dSAndroid Build Coastguard Worker   int64_t id = sqlite3_column_int64 (ps_lookup_file, 0);
3260*7304104dSAndroid Build Coastguard Worker   ps_lookup_file.reset();
3261*7304104dSAndroid Build Coastguard Worker   return id;
3262*7304104dSAndroid Build Coastguard Worker }
3263*7304104dSAndroid Build Coastguard Worker 
3264*7304104dSAndroid Build Coastguard Worker 
3265*7304104dSAndroid Build Coastguard Worker 
3266*7304104dSAndroid Build Coastguard Worker static void
scan_source_file(const string & rps,const stat_t & st,sqlite_ps & ps_upsert_buildids,sqlite_ps & ps_upsert_fileparts,sqlite_ps & ps_upsert_file,sqlite_ps & ps_lookup_file,sqlite_ps & ps_upsert_de,sqlite_ps & ps_upsert_s,sqlite_ps & ps_query,sqlite_ps & ps_scan_done,unsigned & fts_cached,unsigned & fts_executable,unsigned & fts_debuginfo,unsigned & fts_sourcefiles)3267*7304104dSAndroid Build Coastguard Worker scan_source_file (const string& rps, const stat_t& st,
3268*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_upsert_buildids,
3269*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_upsert_fileparts,
3270*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_upsert_file,
3271*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_lookup_file,
3272*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_upsert_de,
3273*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_upsert_s,
3274*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_query,
3275*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_scan_done,
3276*7304104dSAndroid Build Coastguard Worker                   unsigned& fts_cached,
3277*7304104dSAndroid Build Coastguard Worker                   unsigned& fts_executable,
3278*7304104dSAndroid Build Coastguard Worker                   unsigned& fts_debuginfo,
3279*7304104dSAndroid Build Coastguard Worker                   unsigned& fts_sourcefiles)
3280*7304104dSAndroid Build Coastguard Worker {
3281*7304104dSAndroid Build Coastguard Worker   int64_t fileid = register_file_name(ps_upsert_fileparts, ps_upsert_file, ps_lookup_file, rps);
3282*7304104dSAndroid Build Coastguard Worker 
3283*7304104dSAndroid Build Coastguard Worker   /* See if we know of it already. */
3284*7304104dSAndroid Build Coastguard Worker   int rc = ps_query
3285*7304104dSAndroid Build Coastguard Worker     .reset()
3286*7304104dSAndroid Build Coastguard Worker     .bind(1, fileid)
3287*7304104dSAndroid Build Coastguard Worker     .bind(2, st.st_mtime)
3288*7304104dSAndroid Build Coastguard Worker     .step();
3289*7304104dSAndroid Build Coastguard Worker   ps_query.reset();
3290*7304104dSAndroid Build Coastguard Worker   if (rc == SQLITE_ROW) // i.e., a result, as opposed to DONE (no results)
3291*7304104dSAndroid Build Coastguard Worker     // no need to recheck a file/version we already know
3292*7304104dSAndroid Build Coastguard Worker     // specifically, no need to elf-begin a file we already determined is non-elf
3293*7304104dSAndroid Build Coastguard Worker     // (so is stored with buildid=NULL)
3294*7304104dSAndroid Build Coastguard Worker     {
3295*7304104dSAndroid Build Coastguard Worker       fts_cached++;
3296*7304104dSAndroid Build Coastguard Worker       return;
3297*7304104dSAndroid Build Coastguard Worker     }
3298*7304104dSAndroid Build Coastguard Worker 
3299*7304104dSAndroid Build Coastguard Worker   bool executable_p = false, debuginfo_p = false; // E and/or D
3300*7304104dSAndroid Build Coastguard Worker   string buildid;
3301*7304104dSAndroid Build Coastguard Worker   set<string> sourcefiles;
3302*7304104dSAndroid Build Coastguard Worker 
3303*7304104dSAndroid Build Coastguard Worker   int fd = open (rps.c_str(), O_RDONLY);
3304*7304104dSAndroid Build Coastguard Worker   try
3305*7304104dSAndroid Build Coastguard Worker     {
3306*7304104dSAndroid Build Coastguard Worker       if (fd >= 0)
3307*7304104dSAndroid Build Coastguard Worker         elf_classify (fd, executable_p, debuginfo_p, buildid, sourcefiles);
3308*7304104dSAndroid Build Coastguard Worker       else
3309*7304104dSAndroid Build Coastguard Worker         throw libc_exception(errno, string("open ") + rps);
3310*7304104dSAndroid Build Coastguard Worker       add_metric ("scanned_bytes_total","source","file",
3311*7304104dSAndroid Build Coastguard Worker                   st.st_size);
3312*7304104dSAndroid Build Coastguard Worker       inc_metric ("scanned_files_total","source","file");
3313*7304104dSAndroid Build Coastguard Worker     }
3314*7304104dSAndroid Build Coastguard Worker   // NB: we catch exceptions here too, so that we can
3315*7304104dSAndroid Build Coastguard Worker   // cache the corrupt-elf case (!executable_p &&
3316*7304104dSAndroid Build Coastguard Worker   // !debuginfo_p) just below, just as if we had an
3317*7304104dSAndroid Build Coastguard Worker   // EPERM error from open(2).
3318*7304104dSAndroid Build Coastguard Worker   catch (const reportable_exception& e)
3319*7304104dSAndroid Build Coastguard Worker     {
3320*7304104dSAndroid Build Coastguard Worker       e.report(clog);
3321*7304104dSAndroid Build Coastguard Worker     }
3322*7304104dSAndroid Build Coastguard Worker 
3323*7304104dSAndroid Build Coastguard Worker   if (fd >= 0)
3324*7304104dSAndroid Build Coastguard Worker     close (fd);
3325*7304104dSAndroid Build Coastguard Worker 
3326*7304104dSAndroid Build Coastguard Worker   if (buildid == "")
3327*7304104dSAndroid Build Coastguard Worker     {
3328*7304104dSAndroid Build Coastguard Worker       // no point storing an elf file without buildid
3329*7304104dSAndroid Build Coastguard Worker       executable_p = false;
3330*7304104dSAndroid Build Coastguard Worker       debuginfo_p = false;
3331*7304104dSAndroid Build Coastguard Worker     }
3332*7304104dSAndroid Build Coastguard Worker   else
3333*7304104dSAndroid Build Coastguard Worker     {
3334*7304104dSAndroid Build Coastguard Worker       // register this build-id in the interning table
3335*7304104dSAndroid Build Coastguard Worker       ps_upsert_buildids
3336*7304104dSAndroid Build Coastguard Worker         .reset()
3337*7304104dSAndroid Build Coastguard Worker         .bind(1, buildid)
3338*7304104dSAndroid Build Coastguard Worker         .step_ok_done();
3339*7304104dSAndroid Build Coastguard Worker     }
3340*7304104dSAndroid Build Coastguard Worker 
3341*7304104dSAndroid Build Coastguard Worker   if (executable_p)
3342*7304104dSAndroid Build Coastguard Worker     fts_executable ++;
3343*7304104dSAndroid Build Coastguard Worker   if (debuginfo_p)
3344*7304104dSAndroid Build Coastguard Worker     fts_debuginfo ++;
3345*7304104dSAndroid Build Coastguard Worker   if (executable_p || debuginfo_p)
3346*7304104dSAndroid Build Coastguard Worker     {
3347*7304104dSAndroid Build Coastguard Worker       ps_upsert_de
3348*7304104dSAndroid Build Coastguard Worker         .reset()
3349*7304104dSAndroid Build Coastguard Worker         .bind(1, buildid)
3350*7304104dSAndroid Build Coastguard Worker         .bind(2, debuginfo_p ? 1 : 0)
3351*7304104dSAndroid Build Coastguard Worker         .bind(3, executable_p ? 1 : 0)
3352*7304104dSAndroid Build Coastguard Worker         .bind(4, fileid)
3353*7304104dSAndroid Build Coastguard Worker         .bind(5, st.st_mtime)
3354*7304104dSAndroid Build Coastguard Worker         .step_ok_done();
3355*7304104dSAndroid Build Coastguard Worker     }
3356*7304104dSAndroid Build Coastguard Worker   if (executable_p)
3357*7304104dSAndroid Build Coastguard Worker     inc_metric("found_executable_total","source","files");
3358*7304104dSAndroid Build Coastguard Worker   if (debuginfo_p)
3359*7304104dSAndroid Build Coastguard Worker     inc_metric("found_debuginfo_total","source","files");
3360*7304104dSAndroid Build Coastguard Worker 
3361*7304104dSAndroid Build Coastguard Worker   if (sourcefiles.size() && buildid != "")
3362*7304104dSAndroid Build Coastguard Worker     {
3363*7304104dSAndroid Build Coastguard Worker       fts_sourcefiles += sourcefiles.size();
3364*7304104dSAndroid Build Coastguard Worker 
3365*7304104dSAndroid Build Coastguard Worker       for (auto&& dwarfsrc : sourcefiles)
3366*7304104dSAndroid Build Coastguard Worker         {
3367*7304104dSAndroid Build Coastguard Worker           char *srp = realpath(dwarfsrc.c_str(), NULL);
3368*7304104dSAndroid Build Coastguard Worker           if (srp == NULL) // also if DWZ unresolved dwarfsrc=""
3369*7304104dSAndroid Build Coastguard Worker             continue; // unresolvable files are not a serious problem
3370*7304104dSAndroid Build Coastguard Worker           // throw libc_exception(errno, "fts/file realpath " + srcpath);
3371*7304104dSAndroid Build Coastguard Worker           string srps = string(srp);
3372*7304104dSAndroid Build Coastguard Worker           free (srp);
3373*7304104dSAndroid Build Coastguard Worker 
3374*7304104dSAndroid Build Coastguard Worker           struct stat sfs;
3375*7304104dSAndroid Build Coastguard Worker           rc = stat(srps.c_str(), &sfs);
3376*7304104dSAndroid Build Coastguard Worker           if (rc != 0)
3377*7304104dSAndroid Build Coastguard Worker             continue;
3378*7304104dSAndroid Build Coastguard Worker 
3379*7304104dSAndroid Build Coastguard Worker           if (verbose > 2)
3380*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "recorded buildid=" << buildid << " file=" << srps
3381*7304104dSAndroid Build Coastguard Worker                            << " mtime=" << sfs.st_mtime
3382*7304104dSAndroid Build Coastguard Worker                            << " as source " << dwarfsrc << endl;
3383*7304104dSAndroid Build Coastguard Worker 
3384*7304104dSAndroid Build Coastguard Worker           // PR25548: store canonicalized dwarfsrc path
3385*7304104dSAndroid Build Coastguard Worker           string dwarfsrc_canon = canon_pathname (dwarfsrc);
3386*7304104dSAndroid Build Coastguard Worker           if (dwarfsrc_canon != dwarfsrc)
3387*7304104dSAndroid Build Coastguard Worker             {
3388*7304104dSAndroid Build Coastguard Worker               if (verbose > 3)
3389*7304104dSAndroid Build Coastguard Worker                 obatched(clog) << "canonicalized src=" << dwarfsrc << " alias=" << dwarfsrc_canon << endl;
3390*7304104dSAndroid Build Coastguard Worker             }
3391*7304104dSAndroid Build Coastguard Worker 
3392*7304104dSAndroid Build Coastguard Worker           int64_t fileid1 = register_file_name (ps_upsert_fileparts, ps_upsert_file, ps_lookup_file, dwarfsrc_canon);
3393*7304104dSAndroid Build Coastguard Worker           int64_t fileid2 = register_file_name (ps_upsert_fileparts, ps_upsert_file, ps_lookup_file, srps);
3394*7304104dSAndroid Build Coastguard Worker 
3395*7304104dSAndroid Build Coastguard Worker           ps_upsert_s
3396*7304104dSAndroid Build Coastguard Worker             .reset()
3397*7304104dSAndroid Build Coastguard Worker             .bind(1, buildid)
3398*7304104dSAndroid Build Coastguard Worker             .bind(2, fileid1)
3399*7304104dSAndroid Build Coastguard Worker             .bind(3, fileid2)
3400*7304104dSAndroid Build Coastguard Worker             .bind(4, sfs.st_mtime)
3401*7304104dSAndroid Build Coastguard Worker             .step_ok_done();
3402*7304104dSAndroid Build Coastguard Worker 
3403*7304104dSAndroid Build Coastguard Worker           inc_metric("found_sourcerefs_total","source","files");
3404*7304104dSAndroid Build Coastguard Worker         }
3405*7304104dSAndroid Build Coastguard Worker     }
3406*7304104dSAndroid Build Coastguard Worker 
3407*7304104dSAndroid Build Coastguard Worker   ps_scan_done
3408*7304104dSAndroid Build Coastguard Worker     .reset()
3409*7304104dSAndroid Build Coastguard Worker     .bind(1, fileid)
3410*7304104dSAndroid Build Coastguard Worker     .bind(2, st.st_mtime)
3411*7304104dSAndroid Build Coastguard Worker     .bind(3, st.st_size)
3412*7304104dSAndroid Build Coastguard Worker     .step_ok_done();
3413*7304104dSAndroid Build Coastguard Worker 
3414*7304104dSAndroid Build Coastguard Worker   if (verbose > 2)
3415*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "recorded buildid=" << buildid << " file=" << rps
3416*7304104dSAndroid Build Coastguard Worker                    << " mtime=" << st.st_mtime << " atype="
3417*7304104dSAndroid Build Coastguard Worker                    << (executable_p ? "E" : "")
3418*7304104dSAndroid Build Coastguard Worker                    << (debuginfo_p ? "D" : "") << endl;
3419*7304104dSAndroid Build Coastguard Worker }
3420*7304104dSAndroid Build Coastguard Worker 
3421*7304104dSAndroid Build Coastguard Worker 
3422*7304104dSAndroid Build Coastguard Worker 
3423*7304104dSAndroid Build Coastguard Worker 
3424*7304104dSAndroid Build Coastguard Worker 
3425*7304104dSAndroid Build Coastguard Worker // Analyze given archive file of given age; record buildids / exec/debuginfo-ness of its
3426*7304104dSAndroid Build Coastguard Worker // constituent files with given upsert statements.
3427*7304104dSAndroid Build Coastguard Worker static void
archive_classify(const string & rps,string & archive_extension,int64_t archiveid,sqlite_ps & ps_upsert_buildids,sqlite_ps & ps_upsert_fileparts,sqlite_ps & ps_upsert_file,sqlite_ps & ps_lookup_file,sqlite_ps & ps_upsert_de,sqlite_ps & ps_upsert_sref,sqlite_ps & ps_upsert_sdef,time_t mtime,unsigned & fts_executable,unsigned & fts_debuginfo,unsigned & fts_sref,unsigned & fts_sdef,bool & fts_sref_complete_p)3428*7304104dSAndroid Build Coastguard Worker archive_classify (const string& rps, string& archive_extension, int64_t archiveid,
3429*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_upsert_buildids, sqlite_ps& ps_upsert_fileparts, sqlite_ps& ps_upsert_file,
3430*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_lookup_file,
3431*7304104dSAndroid Build Coastguard Worker                   sqlite_ps& ps_upsert_de, sqlite_ps& ps_upsert_sref, sqlite_ps& ps_upsert_sdef,
3432*7304104dSAndroid Build Coastguard Worker                   time_t mtime,
3433*7304104dSAndroid Build Coastguard Worker                   unsigned& fts_executable, unsigned& fts_debuginfo, unsigned& fts_sref, unsigned& fts_sdef,
3434*7304104dSAndroid Build Coastguard Worker                   bool& fts_sref_complete_p)
3435*7304104dSAndroid Build Coastguard Worker {
3436*7304104dSAndroid Build Coastguard Worker   string archive_decoder = "/dev/null";
3437*7304104dSAndroid Build Coastguard Worker   for (auto&& arch : scan_archives)
3438*7304104dSAndroid Build Coastguard Worker     if (string_endswith(rps, arch.first))
3439*7304104dSAndroid Build Coastguard Worker       {
3440*7304104dSAndroid Build Coastguard Worker         archive_extension = arch.first;
3441*7304104dSAndroid Build Coastguard Worker         archive_decoder = arch.second;
3442*7304104dSAndroid Build Coastguard Worker       }
3443*7304104dSAndroid Build Coastguard Worker 
3444*7304104dSAndroid Build Coastguard Worker   FILE* fp;
3445*7304104dSAndroid Build Coastguard Worker   defer_dtor<FILE*,int>::dtor_fn dfn;
3446*7304104dSAndroid Build Coastguard Worker   if (archive_decoder != "cat")
3447*7304104dSAndroid Build Coastguard Worker     {
3448*7304104dSAndroid Build Coastguard Worker       string popen_cmd = archive_decoder + " " + shell_escape(rps);
3449*7304104dSAndroid Build Coastguard Worker       fp = popen (popen_cmd.c_str(), "r"); // "e" O_CLOEXEC?
3450*7304104dSAndroid Build Coastguard Worker       dfn = pclose;
3451*7304104dSAndroid Build Coastguard Worker       if (fp == NULL)
3452*7304104dSAndroid Build Coastguard Worker         throw libc_exception (errno, string("popen ") + popen_cmd);
3453*7304104dSAndroid Build Coastguard Worker     }
3454*7304104dSAndroid Build Coastguard Worker   else
3455*7304104dSAndroid Build Coastguard Worker     {
3456*7304104dSAndroid Build Coastguard Worker       fp = fopen (rps.c_str(), "r");
3457*7304104dSAndroid Build Coastguard Worker       dfn = fclose;
3458*7304104dSAndroid Build Coastguard Worker       if (fp == NULL)
3459*7304104dSAndroid Build Coastguard Worker         throw libc_exception (errno, string("fopen ") + rps);
3460*7304104dSAndroid Build Coastguard Worker     }
3461*7304104dSAndroid Build Coastguard Worker   defer_dtor<FILE*,int> fp_closer (fp, dfn);
3462*7304104dSAndroid Build Coastguard Worker 
3463*7304104dSAndroid Build Coastguard Worker   struct archive *a;
3464*7304104dSAndroid Build Coastguard Worker   a = archive_read_new();
3465*7304104dSAndroid Build Coastguard Worker   if (a == NULL)
3466*7304104dSAndroid Build Coastguard Worker     throw archive_exception("cannot create archive reader");
3467*7304104dSAndroid Build Coastguard Worker   defer_dtor<struct archive*,int> archive_closer (a, archive_read_free);
3468*7304104dSAndroid Build Coastguard Worker 
3469*7304104dSAndroid Build Coastguard Worker   int rc = archive_read_support_format_all(a);
3470*7304104dSAndroid Build Coastguard Worker   if (rc != ARCHIVE_OK)
3471*7304104dSAndroid Build Coastguard Worker     throw archive_exception(a, "cannot select all formats");
3472*7304104dSAndroid Build Coastguard Worker   rc = archive_read_support_filter_all(a);
3473*7304104dSAndroid Build Coastguard Worker   if (rc != ARCHIVE_OK)
3474*7304104dSAndroid Build Coastguard Worker     throw archive_exception(a, "cannot select all filters");
3475*7304104dSAndroid Build Coastguard Worker 
3476*7304104dSAndroid Build Coastguard Worker   rc = archive_read_open_FILE (a, fp);
3477*7304104dSAndroid Build Coastguard Worker   if (rc != ARCHIVE_OK)
3478*7304104dSAndroid Build Coastguard Worker     {
3479*7304104dSAndroid Build Coastguard Worker       obatched(clog) << "cannot open archive from pipe " << rps << endl;
3480*7304104dSAndroid Build Coastguard Worker       throw archive_exception(a, "cannot open archive from pipe");
3481*7304104dSAndroid Build Coastguard Worker     }
3482*7304104dSAndroid Build Coastguard Worker 
3483*7304104dSAndroid Build Coastguard Worker   if (verbose > 3)
3484*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "libarchive scanning " << rps << " id " << archiveid << endl;
3485*7304104dSAndroid Build Coastguard Worker 
3486*7304104dSAndroid Build Coastguard Worker   bool any_exceptions = false;
3487*7304104dSAndroid Build Coastguard Worker   while(1) // parse archive entries
3488*7304104dSAndroid Build Coastguard Worker     {
3489*7304104dSAndroid Build Coastguard Worker     if (interrupted)
3490*7304104dSAndroid Build Coastguard Worker       break;
3491*7304104dSAndroid Build Coastguard Worker 
3492*7304104dSAndroid Build Coastguard Worker     try
3493*7304104dSAndroid Build Coastguard Worker         {
3494*7304104dSAndroid Build Coastguard Worker           struct archive_entry *e;
3495*7304104dSAndroid Build Coastguard Worker           rc = archive_read_next_header (a, &e);
3496*7304104dSAndroid Build Coastguard Worker           if (rc != ARCHIVE_OK)
3497*7304104dSAndroid Build Coastguard Worker             break;
3498*7304104dSAndroid Build Coastguard Worker 
3499*7304104dSAndroid Build Coastguard Worker           if (! S_ISREG(archive_entry_mode (e))) // skip non-files completely
3500*7304104dSAndroid Build Coastguard Worker             continue;
3501*7304104dSAndroid Build Coastguard Worker 
3502*7304104dSAndroid Build Coastguard Worker           string fn = canonicalized_archive_entry_pathname (e);
3503*7304104dSAndroid Build Coastguard Worker 
3504*7304104dSAndroid Build Coastguard Worker           if (verbose > 3)
3505*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "libarchive checking " << fn << endl;
3506*7304104dSAndroid Build Coastguard Worker 
3507*7304104dSAndroid Build Coastguard Worker           // extract this file to a temporary file
3508*7304104dSAndroid Build Coastguard Worker           char* tmppath = NULL;
3509*7304104dSAndroid Build Coastguard Worker           rc = asprintf (&tmppath, "%s/debuginfod-classify.XXXXXX", tmpdir.c_str());
3510*7304104dSAndroid Build Coastguard Worker           if (rc < 0)
3511*7304104dSAndroid Build Coastguard Worker             throw libc_exception (ENOMEM, "cannot allocate tmppath");
3512*7304104dSAndroid Build Coastguard Worker           defer_dtor<void*,void> tmmpath_freer (tmppath, free);
3513*7304104dSAndroid Build Coastguard Worker           int fd = mkstemp (tmppath);
3514*7304104dSAndroid Build Coastguard Worker           if (fd < 0)
3515*7304104dSAndroid Build Coastguard Worker             throw libc_exception (errno, "cannot create temporary file");
3516*7304104dSAndroid Build Coastguard Worker           unlink (tmppath); // unlink now so OS will release the file as soon as we close the fd
3517*7304104dSAndroid Build Coastguard Worker           defer_dtor<int,int> minifd_closer (fd, close);
3518*7304104dSAndroid Build Coastguard Worker 
3519*7304104dSAndroid Build Coastguard Worker           rc = archive_read_data_into_fd (a, fd);
3520*7304104dSAndroid Build Coastguard Worker           if (rc != ARCHIVE_OK) {
3521*7304104dSAndroid Build Coastguard Worker             close (fd);
3522*7304104dSAndroid Build Coastguard Worker             throw archive_exception(a, "cannot extract file");
3523*7304104dSAndroid Build Coastguard Worker           }
3524*7304104dSAndroid Build Coastguard Worker 
3525*7304104dSAndroid Build Coastguard Worker           // finally ... time to run elf_classify on this bad boy and update the database
3526*7304104dSAndroid Build Coastguard Worker           bool executable_p = false, debuginfo_p = false;
3527*7304104dSAndroid Build Coastguard Worker           string buildid;
3528*7304104dSAndroid Build Coastguard Worker           set<string> sourcefiles;
3529*7304104dSAndroid Build Coastguard Worker           elf_classify (fd, executable_p, debuginfo_p, buildid, sourcefiles);
3530*7304104dSAndroid Build Coastguard Worker           // NB: might throw
3531*7304104dSAndroid Build Coastguard Worker 
3532*7304104dSAndroid Build Coastguard Worker           if (buildid != "") // intern buildid
3533*7304104dSAndroid Build Coastguard Worker             {
3534*7304104dSAndroid Build Coastguard Worker               ps_upsert_buildids
3535*7304104dSAndroid Build Coastguard Worker                 .reset()
3536*7304104dSAndroid Build Coastguard Worker                 .bind(1, buildid)
3537*7304104dSAndroid Build Coastguard Worker                 .step_ok_done();
3538*7304104dSAndroid Build Coastguard Worker             }
3539*7304104dSAndroid Build Coastguard Worker 
3540*7304104dSAndroid Build Coastguard Worker           int64_t fileid = register_file_name (ps_upsert_fileparts, ps_upsert_file, ps_lookup_file, fn);
3541*7304104dSAndroid Build Coastguard Worker 
3542*7304104dSAndroid Build Coastguard Worker           if (sourcefiles.size() > 0) // sref records needed
3543*7304104dSAndroid Build Coastguard Worker             {
3544*7304104dSAndroid Build Coastguard Worker               // NB: we intern each source file once.  Once raw, as it
3545*7304104dSAndroid Build Coastguard Worker               // appears in the DWARF file list coming back from
3546*7304104dSAndroid Build Coastguard Worker               // elf_classify() - because it'll end up in the
3547*7304104dSAndroid Build Coastguard Worker               // _norm.artifactsrc column.  We don't also put another
3548*7304104dSAndroid Build Coastguard Worker               // version with a '.' at the front, even though that's
3549*7304104dSAndroid Build Coastguard Worker               // how rpm/cpio packs names, because we hide that from
3550*7304104dSAndroid Build Coastguard Worker               // the database for storage efficiency.
3551*7304104dSAndroid Build Coastguard Worker 
3552*7304104dSAndroid Build Coastguard Worker               for (auto&& s : sourcefiles)
3553*7304104dSAndroid Build Coastguard Worker                 {
3554*7304104dSAndroid Build Coastguard Worker                   if (s == "")
3555*7304104dSAndroid Build Coastguard Worker                     {
3556*7304104dSAndroid Build Coastguard Worker                       fts_sref_complete_p = false;
3557*7304104dSAndroid Build Coastguard Worker                       continue;
3558*7304104dSAndroid Build Coastguard Worker                     }
3559*7304104dSAndroid Build Coastguard Worker 
3560*7304104dSAndroid Build Coastguard Worker                   // PR25548: store canonicalized source path
3561*7304104dSAndroid Build Coastguard Worker                   const string& dwarfsrc = s;
3562*7304104dSAndroid Build Coastguard Worker                   string dwarfsrc_canon = canon_pathname (dwarfsrc);
3563*7304104dSAndroid Build Coastguard Worker                   if (dwarfsrc_canon != dwarfsrc)
3564*7304104dSAndroid Build Coastguard Worker                     {
3565*7304104dSAndroid Build Coastguard Worker                       if (verbose > 3)
3566*7304104dSAndroid Build Coastguard Worker                         obatched(clog) << "canonicalized src=" << dwarfsrc << " alias=" << dwarfsrc_canon << endl;
3567*7304104dSAndroid Build Coastguard Worker                     }
3568*7304104dSAndroid Build Coastguard Worker 
3569*7304104dSAndroid Build Coastguard Worker                   int64_t srcfileid = register_file_name(ps_upsert_fileparts, ps_upsert_file, ps_lookup_file,
3570*7304104dSAndroid Build Coastguard Worker                                                          dwarfsrc_canon);
3571*7304104dSAndroid Build Coastguard Worker 
3572*7304104dSAndroid Build Coastguard Worker                   ps_upsert_sref
3573*7304104dSAndroid Build Coastguard Worker                     .reset()
3574*7304104dSAndroid Build Coastguard Worker                     .bind(1, buildid)
3575*7304104dSAndroid Build Coastguard Worker                     .bind(2, srcfileid)
3576*7304104dSAndroid Build Coastguard Worker                     .step_ok_done();
3577*7304104dSAndroid Build Coastguard Worker 
3578*7304104dSAndroid Build Coastguard Worker                   fts_sref ++;
3579*7304104dSAndroid Build Coastguard Worker                 }
3580*7304104dSAndroid Build Coastguard Worker             }
3581*7304104dSAndroid Build Coastguard Worker 
3582*7304104dSAndroid Build Coastguard Worker           if (executable_p)
3583*7304104dSAndroid Build Coastguard Worker             fts_executable ++;
3584*7304104dSAndroid Build Coastguard Worker           if (debuginfo_p)
3585*7304104dSAndroid Build Coastguard Worker             fts_debuginfo ++;
3586*7304104dSAndroid Build Coastguard Worker 
3587*7304104dSAndroid Build Coastguard Worker           if (executable_p || debuginfo_p)
3588*7304104dSAndroid Build Coastguard Worker             {
3589*7304104dSAndroid Build Coastguard Worker               ps_upsert_de
3590*7304104dSAndroid Build Coastguard Worker                 .reset()
3591*7304104dSAndroid Build Coastguard Worker                 .bind(1, buildid)
3592*7304104dSAndroid Build Coastguard Worker                 .bind(2, debuginfo_p ? 1 : 0)
3593*7304104dSAndroid Build Coastguard Worker                 .bind(3, executable_p ? 1 : 0)
3594*7304104dSAndroid Build Coastguard Worker                 .bind(4, archiveid)
3595*7304104dSAndroid Build Coastguard Worker                 .bind(5, mtime)
3596*7304104dSAndroid Build Coastguard Worker                 .bind(6, fileid)
3597*7304104dSAndroid Build Coastguard Worker                 .step_ok_done();
3598*7304104dSAndroid Build Coastguard Worker             }
3599*7304104dSAndroid Build Coastguard Worker           else // potential source - sdef record
3600*7304104dSAndroid Build Coastguard Worker             {
3601*7304104dSAndroid Build Coastguard Worker               fts_sdef ++;
3602*7304104dSAndroid Build Coastguard Worker               ps_upsert_sdef
3603*7304104dSAndroid Build Coastguard Worker                 .reset()
3604*7304104dSAndroid Build Coastguard Worker                 .bind(1, archiveid)
3605*7304104dSAndroid Build Coastguard Worker                 .bind(2, mtime)
3606*7304104dSAndroid Build Coastguard Worker                 .bind(3, fileid)
3607*7304104dSAndroid Build Coastguard Worker                 .step_ok_done();
3608*7304104dSAndroid Build Coastguard Worker             }
3609*7304104dSAndroid Build Coastguard Worker 
3610*7304104dSAndroid Build Coastguard Worker           if ((verbose > 2) && (executable_p || debuginfo_p))
3611*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "recorded buildid=" << buildid << " rpm=" << rps << " file=" << fn
3612*7304104dSAndroid Build Coastguard Worker                            << " mtime=" << mtime << " atype="
3613*7304104dSAndroid Build Coastguard Worker                            << (executable_p ? "E" : "")
3614*7304104dSAndroid Build Coastguard Worker                            << (debuginfo_p ? "D" : "")
3615*7304104dSAndroid Build Coastguard Worker                            << " sourcefiles=" << sourcefiles.size() << endl;
3616*7304104dSAndroid Build Coastguard Worker 
3617*7304104dSAndroid Build Coastguard Worker         }
3618*7304104dSAndroid Build Coastguard Worker       catch (const reportable_exception& e)
3619*7304104dSAndroid Build Coastguard Worker         {
3620*7304104dSAndroid Build Coastguard Worker           e.report(clog);
3621*7304104dSAndroid Build Coastguard Worker           any_exceptions = true;
3622*7304104dSAndroid Build Coastguard Worker           // NB: but we allow the libarchive iteration to continue, in
3623*7304104dSAndroid Build Coastguard Worker           // case we can still gather some useful information.  That
3624*7304104dSAndroid Build Coastguard Worker           // would allow some webapi queries to work, until later when
3625*7304104dSAndroid Build Coastguard Worker           // this archive is rescanned.  (Its vitals won't go into the
3626*7304104dSAndroid Build Coastguard Worker           // _file_mtime_scanned table until after a successful scan.)
3627*7304104dSAndroid Build Coastguard Worker         }
3628*7304104dSAndroid Build Coastguard Worker     }
3629*7304104dSAndroid Build Coastguard Worker 
3630*7304104dSAndroid Build Coastguard Worker   if (any_exceptions)
3631*7304104dSAndroid Build Coastguard Worker     throw reportable_exception("exceptions encountered during archive scan");
3632*7304104dSAndroid Build Coastguard Worker }
3633*7304104dSAndroid Build Coastguard Worker 
3634*7304104dSAndroid Build Coastguard Worker 
3635*7304104dSAndroid Build Coastguard Worker 
3636*7304104dSAndroid Build Coastguard Worker // scan for archive files such as .rpm
3637*7304104dSAndroid Build Coastguard Worker static void
scan_archive_file(const string & rps,const stat_t & st,sqlite_ps & ps_upsert_buildids,sqlite_ps & ps_upsert_fileparts,sqlite_ps & ps_upsert_file,sqlite_ps & ps_lookup_file,sqlite_ps & ps_upsert_de,sqlite_ps & ps_upsert_sref,sqlite_ps & ps_upsert_sdef,sqlite_ps & ps_query,sqlite_ps & ps_scan_done,unsigned & fts_cached,unsigned & fts_executable,unsigned & fts_debuginfo,unsigned & fts_sref,unsigned & fts_sdef)3638*7304104dSAndroid Build Coastguard Worker scan_archive_file (const string& rps, const stat_t& st,
3639*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_upsert_buildids,
3640*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_upsert_fileparts,
3641*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_upsert_file,
3642*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_lookup_file,
3643*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_upsert_de,
3644*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_upsert_sref,
3645*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_upsert_sdef,
3646*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_query,
3647*7304104dSAndroid Build Coastguard Worker                    sqlite_ps& ps_scan_done,
3648*7304104dSAndroid Build Coastguard Worker                    unsigned& fts_cached,
3649*7304104dSAndroid Build Coastguard Worker                    unsigned& fts_executable,
3650*7304104dSAndroid Build Coastguard Worker                    unsigned& fts_debuginfo,
3651*7304104dSAndroid Build Coastguard Worker                    unsigned& fts_sref,
3652*7304104dSAndroid Build Coastguard Worker                    unsigned& fts_sdef)
3653*7304104dSAndroid Build Coastguard Worker {
3654*7304104dSAndroid Build Coastguard Worker   // intern the archive file name
3655*7304104dSAndroid Build Coastguard Worker   int64_t archiveid = register_file_name (ps_upsert_fileparts, ps_upsert_file, ps_lookup_file, rps);
3656*7304104dSAndroid Build Coastguard Worker 
3657*7304104dSAndroid Build Coastguard Worker   /* See if we know of it already. */
3658*7304104dSAndroid Build Coastguard Worker   int rc = ps_query
3659*7304104dSAndroid Build Coastguard Worker     .reset()
3660*7304104dSAndroid Build Coastguard Worker     .bind(1, archiveid)
3661*7304104dSAndroid Build Coastguard Worker     .bind(2, st.st_mtime)
3662*7304104dSAndroid Build Coastguard Worker     .step();
3663*7304104dSAndroid Build Coastguard Worker   ps_query.reset();
3664*7304104dSAndroid Build Coastguard Worker   if (rc == SQLITE_ROW) // i.e., a result, as opposed to DONE (no results)
3665*7304104dSAndroid Build Coastguard Worker     // no need to recheck a file/version we already know
3666*7304104dSAndroid Build Coastguard Worker     // specifically, no need to parse this archive again, since we already have
3667*7304104dSAndroid Build Coastguard Worker     // it as a D or E or S record,
3668*7304104dSAndroid Build Coastguard Worker     // (so is stored with buildid=NULL)
3669*7304104dSAndroid Build Coastguard Worker     {
3670*7304104dSAndroid Build Coastguard Worker       fts_cached ++;
3671*7304104dSAndroid Build Coastguard Worker       return;
3672*7304104dSAndroid Build Coastguard Worker     }
3673*7304104dSAndroid Build Coastguard Worker 
3674*7304104dSAndroid Build Coastguard Worker   // extract the archive contents
3675*7304104dSAndroid Build Coastguard Worker   unsigned my_fts_executable = 0, my_fts_debuginfo = 0, my_fts_sref = 0, my_fts_sdef = 0;
3676*7304104dSAndroid Build Coastguard Worker   bool my_fts_sref_complete_p = true;
3677*7304104dSAndroid Build Coastguard Worker   bool any_exceptions = false;
3678*7304104dSAndroid Build Coastguard Worker   try
3679*7304104dSAndroid Build Coastguard Worker     {
3680*7304104dSAndroid Build Coastguard Worker       string archive_extension;
3681*7304104dSAndroid Build Coastguard Worker       archive_classify (rps, archive_extension, archiveid,
3682*7304104dSAndroid Build Coastguard Worker                         ps_upsert_buildids, ps_upsert_fileparts, ps_upsert_file, ps_lookup_file,
3683*7304104dSAndroid Build Coastguard Worker                         ps_upsert_de, ps_upsert_sref, ps_upsert_sdef, // dalt
3684*7304104dSAndroid Build Coastguard Worker                         st.st_mtime,
3685*7304104dSAndroid Build Coastguard Worker                         my_fts_executable, my_fts_debuginfo, my_fts_sref, my_fts_sdef,
3686*7304104dSAndroid Build Coastguard Worker                         my_fts_sref_complete_p);
3687*7304104dSAndroid Build Coastguard Worker       add_metric ("scanned_bytes_total","source",archive_extension + " archive",
3688*7304104dSAndroid Build Coastguard Worker                   st.st_size);
3689*7304104dSAndroid Build Coastguard Worker       inc_metric ("scanned_files_total","source",archive_extension + " archive");
3690*7304104dSAndroid Build Coastguard Worker       add_metric("found_debuginfo_total","source",archive_extension + " archive",
3691*7304104dSAndroid Build Coastguard Worker                  my_fts_debuginfo);
3692*7304104dSAndroid Build Coastguard Worker       add_metric("found_executable_total","source",archive_extension + " archive",
3693*7304104dSAndroid Build Coastguard Worker                  my_fts_executable);
3694*7304104dSAndroid Build Coastguard Worker       add_metric("found_sourcerefs_total","source",archive_extension + " archive",
3695*7304104dSAndroid Build Coastguard Worker                  my_fts_sref);
3696*7304104dSAndroid Build Coastguard Worker     }
3697*7304104dSAndroid Build Coastguard Worker   catch (const reportable_exception& e)
3698*7304104dSAndroid Build Coastguard Worker     {
3699*7304104dSAndroid Build Coastguard Worker       e.report(clog);
3700*7304104dSAndroid Build Coastguard Worker       any_exceptions = true;
3701*7304104dSAndroid Build Coastguard Worker     }
3702*7304104dSAndroid Build Coastguard Worker 
3703*7304104dSAndroid Build Coastguard Worker   if (verbose > 2)
3704*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "scanned archive=" << rps
3705*7304104dSAndroid Build Coastguard Worker                    << " mtime=" << st.st_mtime
3706*7304104dSAndroid Build Coastguard Worker                    << " executables=" << my_fts_executable
3707*7304104dSAndroid Build Coastguard Worker                    << " debuginfos=" << my_fts_debuginfo
3708*7304104dSAndroid Build Coastguard Worker                    << " srefs=" << my_fts_sref
3709*7304104dSAndroid Build Coastguard Worker                    << " sdefs=" << my_fts_sdef
3710*7304104dSAndroid Build Coastguard Worker                    << " exceptions=" << any_exceptions
3711*7304104dSAndroid Build Coastguard Worker                    << endl;
3712*7304104dSAndroid Build Coastguard Worker 
3713*7304104dSAndroid Build Coastguard Worker   fts_executable += my_fts_executable;
3714*7304104dSAndroid Build Coastguard Worker   fts_debuginfo += my_fts_debuginfo;
3715*7304104dSAndroid Build Coastguard Worker   fts_sref += my_fts_sref;
3716*7304104dSAndroid Build Coastguard Worker   fts_sdef += my_fts_sdef;
3717*7304104dSAndroid Build Coastguard Worker 
3718*7304104dSAndroid Build Coastguard Worker   if (any_exceptions)
3719*7304104dSAndroid Build Coastguard Worker     throw reportable_exception("exceptions encountered during archive scan");
3720*7304104dSAndroid Build Coastguard Worker 
3721*7304104dSAndroid Build Coastguard Worker   if (my_fts_sref_complete_p) // leave incomplete?
3722*7304104dSAndroid Build Coastguard Worker     ps_scan_done
3723*7304104dSAndroid Build Coastguard Worker       .reset()
3724*7304104dSAndroid Build Coastguard Worker       .bind(1, archiveid)
3725*7304104dSAndroid Build Coastguard Worker       .bind(2, st.st_mtime)
3726*7304104dSAndroid Build Coastguard Worker       .bind(3, st.st_size)
3727*7304104dSAndroid Build Coastguard Worker       .step_ok_done();
3728*7304104dSAndroid Build Coastguard Worker }
3729*7304104dSAndroid Build Coastguard Worker 
3730*7304104dSAndroid Build Coastguard Worker 
3731*7304104dSAndroid Build Coastguard Worker 
3732*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
3733*7304104dSAndroid Build Coastguard Worker 
3734*7304104dSAndroid Build Coastguard Worker 
3735*7304104dSAndroid Build Coastguard Worker 
3736*7304104dSAndroid Build Coastguard Worker // The thread that consumes file names off of the scanq.  We hold
3737*7304104dSAndroid Build Coastguard Worker // the persistent sqlite_ps's at this level and delegate file/archive
3738*7304104dSAndroid Build Coastguard Worker // scanning to other functions.
3739*7304104dSAndroid Build Coastguard Worker static void
scan()3740*7304104dSAndroid Build Coastguard Worker scan ()
3741*7304104dSAndroid Build Coastguard Worker {
3742*7304104dSAndroid Build Coastguard Worker   // all the prepared statements fit to use, the _f_ set:
3743*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_upsert_buildids (db, "file-buildids-intern", "insert or ignore into " BUILDIDS "_buildids VALUES (NULL, ?);");
3744*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_upsert_fileparts (db, "file-fileparts-intern", "insert or ignore into " BUILDIDS "_fileparts VALUES (NULL, ?);");
3745*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_upsert_file (db, "file-file-intern", "insert or ignore into " BUILDIDS "_files VALUES (NULL, \n"
3746*7304104dSAndroid Build Coastguard Worker                               "(select id from " BUILDIDS "_fileparts where name = ?),\n"
3747*7304104dSAndroid Build Coastguard Worker                               "(select id from " BUILDIDS "_fileparts where name = ?));");
3748*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_lookup_file (db, "file-file-lookup",
3749*7304104dSAndroid Build Coastguard Worker                               "select f.id\n"
3750*7304104dSAndroid Build Coastguard Worker                               " from " BUILDIDS "_files f, " BUILDIDS "_fileparts p1, " BUILDIDS "_fileparts p2 \n"
3751*7304104dSAndroid Build Coastguard Worker                               " where f.dirname = p1.id and f.basename = p2.id and p1.name = ? and p2.name = ?;\n");
3752*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_upsert_de (db, "file-de-upsert",
3753*7304104dSAndroid Build Coastguard Worker                           "insert or ignore into " BUILDIDS "_f_de "
3754*7304104dSAndroid Build Coastguard Worker                           "(buildid, debuginfo_p, executable_p, file, mtime) "
3755*7304104dSAndroid Build Coastguard Worker                           "values ((select id from " BUILDIDS "_buildids where hex = ?),"
3756*7304104dSAndroid Build Coastguard Worker                             "        ?,?,?,?);");
3757*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_upsert_s (db, "file-s-upsert",
3758*7304104dSAndroid Build Coastguard Worker                          "insert or ignore into " BUILDIDS "_f_s "
3759*7304104dSAndroid Build Coastguard Worker                          "(buildid, artifactsrc, file, mtime) "
3760*7304104dSAndroid Build Coastguard Worker                          "values ((select id from " BUILDIDS "_buildids where hex = ?),"
3761*7304104dSAndroid Build Coastguard Worker                          "      ?,?,?);");
3762*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_query (db, "file-negativehit-find",
3763*7304104dSAndroid Build Coastguard Worker                         "select 1 from " BUILDIDS "_file_mtime_scanned where sourcetype = 'F' "
3764*7304104dSAndroid Build Coastguard Worker                         "and file = ? and mtime = ?;");
3765*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_f_scan_done (db, "file-scanned",
3766*7304104dSAndroid Build Coastguard Worker                           "insert or ignore into " BUILDIDS "_file_mtime_scanned (sourcetype, file, mtime, size)"
3767*7304104dSAndroid Build Coastguard Worker                           "values ('F', ?,?,?);");
3768*7304104dSAndroid Build Coastguard Worker 
3769*7304104dSAndroid Build Coastguard Worker   // and now for the _r_ set
3770*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_upsert_buildids (db, "rpm-buildid-intern", "insert or ignore into " BUILDIDS "_buildids VALUES (NULL, ?);");
3771*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_upsert_fileparts (db, "rpm-fileparts-intern", "insert or ignore into " BUILDIDS "_fileparts VALUES (NULL, ?);");
3772*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_upsert_file (db, "rpm-file-intern", "insert or ignore into " BUILDIDS "_files VALUES (NULL, \n"
3773*7304104dSAndroid Build Coastguard Worker                               "(select id from " BUILDIDS "_fileparts where name = ?),\n"
3774*7304104dSAndroid Build Coastguard Worker                               "(select id from " BUILDIDS "_fileparts where name = ?));");
3775*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_lookup_file (db, "rpm-file-lookup",
3776*7304104dSAndroid Build Coastguard Worker                               "select f.id\n"
3777*7304104dSAndroid Build Coastguard Worker                               " from " BUILDIDS "_files f, " BUILDIDS "_fileparts p1, " BUILDIDS "_fileparts p2 \n"
3778*7304104dSAndroid Build Coastguard Worker                               " where f.dirname = p1.id and f.basename = p2.id and p1.name = ? and p2.name = ?;\n");
3779*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_upsert_de (db, "rpm-de-insert",
3780*7304104dSAndroid Build Coastguard Worker                           "insert or ignore into " BUILDIDS "_r_de (buildid, debuginfo_p, executable_p, file, mtime, content) values ("
3781*7304104dSAndroid Build Coastguard Worker                           "(select id from " BUILDIDS "_buildids where hex = ?), ?, ?, ?, ?, ?);");
3782*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_upsert_sref (db, "rpm-sref-insert",
3783*7304104dSAndroid Build Coastguard Worker                             "insert or ignore into " BUILDIDS "_r_sref (buildid, artifactsrc) values ("
3784*7304104dSAndroid Build Coastguard Worker                             "(select id from " BUILDIDS "_buildids where hex = ?), "
3785*7304104dSAndroid Build Coastguard Worker                             "?);");
3786*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_upsert_sdef (db, "rpm-sdef-insert",
3787*7304104dSAndroid Build Coastguard Worker                             "insert or ignore into " BUILDIDS "_r_sdef (file, mtime, content) values ("
3788*7304104dSAndroid Build Coastguard Worker                             "?, ?, ?);");
3789*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_query (db, "rpm-negativehit-query",
3790*7304104dSAndroid Build Coastguard Worker                       "select 1 from " BUILDIDS "_file_mtime_scanned where "
3791*7304104dSAndroid Build Coastguard Worker                       "sourcetype = 'R' and file = ? and mtime = ?;");
3792*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_r_scan_done (db, "rpm-scanned",
3793*7304104dSAndroid Build Coastguard Worker                           "insert or ignore into " BUILDIDS "_file_mtime_scanned (sourcetype, file, mtime, size)"
3794*7304104dSAndroid Build Coastguard Worker                           "values ('R', ?, ?, ?);");
3795*7304104dSAndroid Build Coastguard Worker 
3796*7304104dSAndroid Build Coastguard Worker 
3797*7304104dSAndroid Build Coastguard Worker   unsigned fts_cached = 0, fts_executable = 0, fts_debuginfo = 0, fts_sourcefiles = 0;
3798*7304104dSAndroid Build Coastguard Worker   unsigned fts_sref = 0, fts_sdef = 0;
3799*7304104dSAndroid Build Coastguard Worker 
3800*7304104dSAndroid Build Coastguard Worker   add_metric("thread_count", "role", "scan", 1);
3801*7304104dSAndroid Build Coastguard Worker   add_metric("thread_busy", "role", "scan", 1);
3802*7304104dSAndroid Build Coastguard Worker   while (! interrupted)
3803*7304104dSAndroid Build Coastguard Worker     {
3804*7304104dSAndroid Build Coastguard Worker       scan_payload p;
3805*7304104dSAndroid Build Coastguard Worker 
3806*7304104dSAndroid Build Coastguard Worker       add_metric("thread_busy", "role", "scan", -1);
3807*7304104dSAndroid Build Coastguard Worker       // NB: threads may be blocked within either of these two waiting
3808*7304104dSAndroid Build Coastguard Worker       // states, if the work queue happens to run dry.  That's OK.
3809*7304104dSAndroid Build Coastguard Worker       if (scan_barrier) scan_barrier->count();
3810*7304104dSAndroid Build Coastguard Worker       bool gotone = scanq.wait_front(p);
3811*7304104dSAndroid Build Coastguard Worker       add_metric("thread_busy", "role", "scan", 1);
3812*7304104dSAndroid Build Coastguard Worker 
3813*7304104dSAndroid Build Coastguard Worker       if (! gotone) continue; // go back to waiting
3814*7304104dSAndroid Build Coastguard Worker 
3815*7304104dSAndroid Build Coastguard Worker       try
3816*7304104dSAndroid Build Coastguard Worker         {
3817*7304104dSAndroid Build Coastguard Worker           bool scan_archive = false;
3818*7304104dSAndroid Build Coastguard Worker           for (auto&& arch : scan_archives)
3819*7304104dSAndroid Build Coastguard Worker             if (string_endswith(p.first, arch.first))
3820*7304104dSAndroid Build Coastguard Worker               scan_archive = true;
3821*7304104dSAndroid Build Coastguard Worker 
3822*7304104dSAndroid Build Coastguard Worker           if (scan_archive)
3823*7304104dSAndroid Build Coastguard Worker             scan_archive_file (p.first, p.second,
3824*7304104dSAndroid Build Coastguard Worker                                ps_r_upsert_buildids,
3825*7304104dSAndroid Build Coastguard Worker                                ps_r_upsert_fileparts,
3826*7304104dSAndroid Build Coastguard Worker                                ps_r_upsert_file,
3827*7304104dSAndroid Build Coastguard Worker                                ps_r_lookup_file,
3828*7304104dSAndroid Build Coastguard Worker                                ps_r_upsert_de,
3829*7304104dSAndroid Build Coastguard Worker                                ps_r_upsert_sref,
3830*7304104dSAndroid Build Coastguard Worker                                ps_r_upsert_sdef,
3831*7304104dSAndroid Build Coastguard Worker                                ps_r_query,
3832*7304104dSAndroid Build Coastguard Worker                                ps_r_scan_done,
3833*7304104dSAndroid Build Coastguard Worker                                fts_cached,
3834*7304104dSAndroid Build Coastguard Worker                                fts_executable,
3835*7304104dSAndroid Build Coastguard Worker                                fts_debuginfo,
3836*7304104dSAndroid Build Coastguard Worker                                fts_sref,
3837*7304104dSAndroid Build Coastguard Worker                                fts_sdef);
3838*7304104dSAndroid Build Coastguard Worker 
3839*7304104dSAndroid Build Coastguard Worker           if (scan_files) // NB: maybe "else if" ?
3840*7304104dSAndroid Build Coastguard Worker             scan_source_file (p.first, p.second,
3841*7304104dSAndroid Build Coastguard Worker                               ps_f_upsert_buildids,
3842*7304104dSAndroid Build Coastguard Worker                               ps_f_upsert_fileparts,
3843*7304104dSAndroid Build Coastguard Worker                               ps_f_upsert_file,
3844*7304104dSAndroid Build Coastguard Worker                               ps_f_lookup_file,
3845*7304104dSAndroid Build Coastguard Worker                               ps_f_upsert_de,
3846*7304104dSAndroid Build Coastguard Worker                               ps_f_upsert_s,
3847*7304104dSAndroid Build Coastguard Worker                               ps_f_query,
3848*7304104dSAndroid Build Coastguard Worker                               ps_f_scan_done,
3849*7304104dSAndroid Build Coastguard Worker                               fts_cached, fts_executable, fts_debuginfo, fts_sourcefiles);
3850*7304104dSAndroid Build Coastguard Worker         }
3851*7304104dSAndroid Build Coastguard Worker       catch (const reportable_exception& e)
3852*7304104dSAndroid Build Coastguard Worker         {
3853*7304104dSAndroid Build Coastguard Worker           e.report(cerr);
3854*7304104dSAndroid Build Coastguard Worker         }
3855*7304104dSAndroid Build Coastguard Worker 
3856*7304104dSAndroid Build Coastguard Worker       scanq.done_front(); // let idlers run
3857*7304104dSAndroid Build Coastguard Worker 
3858*7304104dSAndroid Build Coastguard Worker       if (fts_cached || fts_executable || fts_debuginfo || fts_sourcefiles || fts_sref || fts_sdef)
3859*7304104dSAndroid Build Coastguard Worker         {} // NB: not just if a successful scan - we might have encountered -ENOSPC & failed
3860*7304104dSAndroid Build Coastguard Worker       (void) statfs_free_enough_p(db_path, "database"); // report sqlite filesystem size
3861*7304104dSAndroid Build Coastguard Worker       (void) statfs_free_enough_p(tmpdir, "tmpdir"); // this too, in case of fdcache/tmpfile usage
3862*7304104dSAndroid Build Coastguard Worker 
3863*7304104dSAndroid Build Coastguard Worker       // finished a scanning step -- not a "loop", because we just
3864*7304104dSAndroid Build Coastguard Worker       // consume the traversal loop's work, whenever
3865*7304104dSAndroid Build Coastguard Worker       inc_metric("thread_work_total","role","scan");
3866*7304104dSAndroid Build Coastguard Worker     }
3867*7304104dSAndroid Build Coastguard Worker 
3868*7304104dSAndroid Build Coastguard Worker   add_metric("thread_busy", "role", "scan", -1);
3869*7304104dSAndroid Build Coastguard Worker }
3870*7304104dSAndroid Build Coastguard Worker 
3871*7304104dSAndroid Build Coastguard Worker 
3872*7304104dSAndroid Build Coastguard Worker // Use this function as the thread entry point, so it can catch our
3873*7304104dSAndroid Build Coastguard Worker // fleet of exceptions (incl. the sqlite_ps ctors) and report.
3874*7304104dSAndroid Build Coastguard Worker static void*
thread_main_scanner(void * arg)3875*7304104dSAndroid Build Coastguard Worker thread_main_scanner (void* arg)
3876*7304104dSAndroid Build Coastguard Worker {
3877*7304104dSAndroid Build Coastguard Worker   (void) arg;
3878*7304104dSAndroid Build Coastguard Worker   while (! interrupted)
3879*7304104dSAndroid Build Coastguard Worker     try
3880*7304104dSAndroid Build Coastguard Worker       {
3881*7304104dSAndroid Build Coastguard Worker         scan();
3882*7304104dSAndroid Build Coastguard Worker       }
3883*7304104dSAndroid Build Coastguard Worker     catch (const reportable_exception& e)
3884*7304104dSAndroid Build Coastguard Worker       {
3885*7304104dSAndroid Build Coastguard Worker         e.report(cerr);
3886*7304104dSAndroid Build Coastguard Worker       }
3887*7304104dSAndroid Build Coastguard Worker   return 0;
3888*7304104dSAndroid Build Coastguard Worker }
3889*7304104dSAndroid Build Coastguard Worker 
3890*7304104dSAndroid Build Coastguard Worker 
3891*7304104dSAndroid Build Coastguard Worker 
3892*7304104dSAndroid Build Coastguard Worker // The thread that traverses all the source_paths and enqueues all the
3893*7304104dSAndroid Build Coastguard Worker // matching files into the file/archive scan queue.
3894*7304104dSAndroid Build Coastguard Worker static void
scan_source_paths()3895*7304104dSAndroid Build Coastguard Worker scan_source_paths()
3896*7304104dSAndroid Build Coastguard Worker {
3897*7304104dSAndroid Build Coastguard Worker   // NB: fedora 31 glibc/fts(3) crashes inside fts_read() on empty
3898*7304104dSAndroid Build Coastguard Worker   // path list.
3899*7304104dSAndroid Build Coastguard Worker   if (source_paths.empty())
3900*7304104dSAndroid Build Coastguard Worker     return;
3901*7304104dSAndroid Build Coastguard Worker 
3902*7304104dSAndroid Build Coastguard Worker   // Turn the source_paths into an fts(3)-compatible char**.  Since
3903*7304104dSAndroid Build Coastguard Worker   // source_paths[] does not change after argv processing, the
3904*7304104dSAndroid Build Coastguard Worker   // c_str()'s are safe to keep around awile.
3905*7304104dSAndroid Build Coastguard Worker   vector<const char *> sps;
3906*7304104dSAndroid Build Coastguard Worker   for (auto&& sp: source_paths)
3907*7304104dSAndroid Build Coastguard Worker     sps.push_back(sp.c_str());
3908*7304104dSAndroid Build Coastguard Worker   sps.push_back(NULL);
3909*7304104dSAndroid Build Coastguard Worker 
3910*7304104dSAndroid Build Coastguard Worker   FTS *fts = fts_open ((char * const *)sps.data(),
3911*7304104dSAndroid Build Coastguard Worker                       (traverse_logical ? FTS_LOGICAL : FTS_PHYSICAL|FTS_XDEV)
3912*7304104dSAndroid Build Coastguard Worker                       | FTS_NOCHDIR /* multithreaded */,
3913*7304104dSAndroid Build Coastguard Worker                       NULL);
3914*7304104dSAndroid Build Coastguard Worker   if (fts == NULL)
3915*7304104dSAndroid Build Coastguard Worker     throw libc_exception(errno, "cannot fts_open");
3916*7304104dSAndroid Build Coastguard Worker   defer_dtor<FTS*,int> fts_cleanup (fts, fts_close);
3917*7304104dSAndroid Build Coastguard Worker 
3918*7304104dSAndroid Build Coastguard Worker   struct timespec ts_start, ts_end;
3919*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &ts_start);
3920*7304104dSAndroid Build Coastguard Worker   unsigned fts_scanned = 0, fts_regex = 0;
3921*7304104dSAndroid Build Coastguard Worker 
3922*7304104dSAndroid Build Coastguard Worker   FTSENT *f;
3923*7304104dSAndroid Build Coastguard Worker   while ((f = fts_read (fts)) != NULL)
3924*7304104dSAndroid Build Coastguard Worker   {
3925*7304104dSAndroid Build Coastguard Worker     if (interrupted) break;
3926*7304104dSAndroid Build Coastguard Worker 
3927*7304104dSAndroid Build Coastguard Worker     if (sigusr2 != forced_groom_count) // stop early if groom triggered
3928*7304104dSAndroid Build Coastguard Worker       {
3929*7304104dSAndroid Build Coastguard Worker         scanq.clear(); // clear previously issued work for scanner threads
3930*7304104dSAndroid Build Coastguard Worker         break;
3931*7304104dSAndroid Build Coastguard Worker       }
3932*7304104dSAndroid Build Coastguard Worker 
3933*7304104dSAndroid Build Coastguard Worker     fts_scanned ++;
3934*7304104dSAndroid Build Coastguard Worker 
3935*7304104dSAndroid Build Coastguard Worker     if (verbose > 2)
3936*7304104dSAndroid Build Coastguard Worker       obatched(clog) << "fts traversing " << f->fts_path << endl;
3937*7304104dSAndroid Build Coastguard Worker 
3938*7304104dSAndroid Build Coastguard Worker     switch (f->fts_info)
3939*7304104dSAndroid Build Coastguard Worker       {
3940*7304104dSAndroid Build Coastguard Worker       case FTS_F:
3941*7304104dSAndroid Build Coastguard Worker         {
3942*7304104dSAndroid Build Coastguard Worker           /* Found a file.  Convert it to an absolute path, so
3943*7304104dSAndroid Build Coastguard Worker              the buildid database does not have relative path
3944*7304104dSAndroid Build Coastguard Worker              names that are unresolvable from a subsequent run
3945*7304104dSAndroid Build Coastguard Worker              in a different cwd. */
3946*7304104dSAndroid Build Coastguard Worker           char *rp = realpath(f->fts_path, NULL);
3947*7304104dSAndroid Build Coastguard Worker           if (rp == NULL)
3948*7304104dSAndroid Build Coastguard Worker             continue; // ignore dangling symlink or such
3949*7304104dSAndroid Build Coastguard Worker           string rps = string(rp);
3950*7304104dSAndroid Build Coastguard Worker           free (rp);
3951*7304104dSAndroid Build Coastguard Worker 
3952*7304104dSAndroid Build Coastguard Worker           bool ri = !regexec (&file_include_regex, rps.c_str(), 0, 0, 0);
3953*7304104dSAndroid Build Coastguard Worker           bool rx = !regexec (&file_exclude_regex, rps.c_str(), 0, 0, 0);
3954*7304104dSAndroid Build Coastguard Worker           if (!ri || rx)
3955*7304104dSAndroid Build Coastguard Worker             {
3956*7304104dSAndroid Build Coastguard Worker               if (verbose > 3)
3957*7304104dSAndroid Build Coastguard Worker                 obatched(clog) << "fts skipped by regex "
3958*7304104dSAndroid Build Coastguard Worker                                << (!ri ? "I" : "") << (rx ? "X" : "") << endl;
3959*7304104dSAndroid Build Coastguard Worker               fts_regex ++;
3960*7304104dSAndroid Build Coastguard Worker               if (!ri)
3961*7304104dSAndroid Build Coastguard Worker                 inc_metric("traversed_total","type","file-skipped-I");
3962*7304104dSAndroid Build Coastguard Worker               if (rx)
3963*7304104dSAndroid Build Coastguard Worker                 inc_metric("traversed_total","type","file-skipped-X");
3964*7304104dSAndroid Build Coastguard Worker             }
3965*7304104dSAndroid Build Coastguard Worker           else
3966*7304104dSAndroid Build Coastguard Worker             {
3967*7304104dSAndroid Build Coastguard Worker               scanq.push_back (make_pair(rps, *f->fts_statp));
3968*7304104dSAndroid Build Coastguard Worker               inc_metric("traversed_total","type","file");
3969*7304104dSAndroid Build Coastguard Worker             }
3970*7304104dSAndroid Build Coastguard Worker         }
3971*7304104dSAndroid Build Coastguard Worker         break;
3972*7304104dSAndroid Build Coastguard Worker 
3973*7304104dSAndroid Build Coastguard Worker       case FTS_ERR:
3974*7304104dSAndroid Build Coastguard Worker       case FTS_NS:
3975*7304104dSAndroid Build Coastguard Worker         // report on some types of errors because they may reflect fixable misconfiguration
3976*7304104dSAndroid Build Coastguard Worker         {
3977*7304104dSAndroid Build Coastguard Worker           auto x = libc_exception(f->fts_errno, string("fts traversal ") + string(f->fts_path));
3978*7304104dSAndroid Build Coastguard Worker           x.report(cerr);
3979*7304104dSAndroid Build Coastguard Worker         }
3980*7304104dSAndroid Build Coastguard Worker         inc_metric("traversed_total","type","error");
3981*7304104dSAndroid Build Coastguard Worker         break;
3982*7304104dSAndroid Build Coastguard Worker 
3983*7304104dSAndroid Build Coastguard Worker       case FTS_SL: // ignore, but count because debuginfod -L would traverse these
3984*7304104dSAndroid Build Coastguard Worker         inc_metric("traversed_total","type","symlink");
3985*7304104dSAndroid Build Coastguard Worker         break;
3986*7304104dSAndroid Build Coastguard Worker 
3987*7304104dSAndroid Build Coastguard Worker       case FTS_D: // ignore
3988*7304104dSAndroid Build Coastguard Worker         inc_metric("traversed_total","type","directory");
3989*7304104dSAndroid Build Coastguard Worker         break;
3990*7304104dSAndroid Build Coastguard Worker 
3991*7304104dSAndroid Build Coastguard Worker       default: // ignore
3992*7304104dSAndroid Build Coastguard Worker         inc_metric("traversed_total","type","other");
3993*7304104dSAndroid Build Coastguard Worker         break;
3994*7304104dSAndroid Build Coastguard Worker       }
3995*7304104dSAndroid Build Coastguard Worker   }
3996*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &ts_end);
3997*7304104dSAndroid Build Coastguard Worker   double deltas = (ts_end.tv_sec - ts_start.tv_sec) + (ts_end.tv_nsec - ts_start.tv_nsec)/1.e9;
3998*7304104dSAndroid Build Coastguard Worker 
3999*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "fts traversed source paths in " << deltas << "s, scanned=" << fts_scanned
4000*7304104dSAndroid Build Coastguard Worker                  << ", regex-skipped=" << fts_regex << endl;
4001*7304104dSAndroid Build Coastguard Worker }
4002*7304104dSAndroid Build Coastguard Worker 
4003*7304104dSAndroid Build Coastguard Worker 
4004*7304104dSAndroid Build Coastguard Worker static void*
thread_main_fts_source_paths(void * arg)4005*7304104dSAndroid Build Coastguard Worker thread_main_fts_source_paths (void* arg)
4006*7304104dSAndroid Build Coastguard Worker {
4007*7304104dSAndroid Build Coastguard Worker   (void) arg; // ignore; we operate on global data
4008*7304104dSAndroid Build Coastguard Worker 
4009*7304104dSAndroid Build Coastguard Worker   set_metric("thread_tid", "role","traverse", tid());
4010*7304104dSAndroid Build Coastguard Worker   add_metric("thread_count", "role", "traverse", 1);
4011*7304104dSAndroid Build Coastguard Worker 
4012*7304104dSAndroid Build Coastguard Worker   time_t last_rescan = 0;
4013*7304104dSAndroid Build Coastguard Worker 
4014*7304104dSAndroid Build Coastguard Worker   while (! interrupted)
4015*7304104dSAndroid Build Coastguard Worker     {
4016*7304104dSAndroid Build Coastguard Worker       sleep (1);
4017*7304104dSAndroid Build Coastguard Worker       scanq.wait_idle(); // don't start a new traversal while scanners haven't finished the job
4018*7304104dSAndroid Build Coastguard Worker       scanq.done_idle(); // release the hounds
4019*7304104dSAndroid Build Coastguard Worker       if (interrupted) break;
4020*7304104dSAndroid Build Coastguard Worker 
4021*7304104dSAndroid Build Coastguard Worker       time_t now = time(NULL);
4022*7304104dSAndroid Build Coastguard Worker       bool rescan_now = false;
4023*7304104dSAndroid Build Coastguard Worker       if (last_rescan == 0) // at least one initial rescan is documented even for -t0
4024*7304104dSAndroid Build Coastguard Worker         rescan_now = true;
4025*7304104dSAndroid Build Coastguard Worker       if (rescan_s > 0 && (long)now > (long)(last_rescan + rescan_s))
4026*7304104dSAndroid Build Coastguard Worker         rescan_now = true;
4027*7304104dSAndroid Build Coastguard Worker       if (sigusr1 != forced_rescan_count)
4028*7304104dSAndroid Build Coastguard Worker         {
4029*7304104dSAndroid Build Coastguard Worker           forced_rescan_count = sigusr1;
4030*7304104dSAndroid Build Coastguard Worker           rescan_now = true;
4031*7304104dSAndroid Build Coastguard Worker         }
4032*7304104dSAndroid Build Coastguard Worker       if (rescan_now)
4033*7304104dSAndroid Build Coastguard Worker         {
4034*7304104dSAndroid Build Coastguard Worker           set_metric("thread_busy", "role","traverse", 1);
4035*7304104dSAndroid Build Coastguard Worker           try
4036*7304104dSAndroid Build Coastguard Worker             {
4037*7304104dSAndroid Build Coastguard Worker               scan_source_paths();
4038*7304104dSAndroid Build Coastguard Worker             }
4039*7304104dSAndroid Build Coastguard Worker           catch (const reportable_exception& e)
4040*7304104dSAndroid Build Coastguard Worker             {
4041*7304104dSAndroid Build Coastguard Worker               e.report(cerr);
4042*7304104dSAndroid Build Coastguard Worker             }
4043*7304104dSAndroid Build Coastguard Worker           last_rescan = time(NULL); // NB: now was before scanning
4044*7304104dSAndroid Build Coastguard Worker           // finished a traversal loop
4045*7304104dSAndroid Build Coastguard Worker           inc_metric("thread_work_total", "role","traverse");
4046*7304104dSAndroid Build Coastguard Worker           set_metric("thread_busy", "role","traverse", 0);
4047*7304104dSAndroid Build Coastguard Worker         }
4048*7304104dSAndroid Build Coastguard Worker     }
4049*7304104dSAndroid Build Coastguard Worker 
4050*7304104dSAndroid Build Coastguard Worker   return 0;
4051*7304104dSAndroid Build Coastguard Worker }
4052*7304104dSAndroid Build Coastguard Worker 
4053*7304104dSAndroid Build Coastguard Worker 
4054*7304104dSAndroid Build Coastguard Worker 
4055*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
4056*7304104dSAndroid Build Coastguard Worker 
4057*7304104dSAndroid Build Coastguard Worker static void
database_stats_report()4058*7304104dSAndroid Build Coastguard Worker database_stats_report()
4059*7304104dSAndroid Build Coastguard Worker {
4060*7304104dSAndroid Build Coastguard Worker   sqlite_ps ps_query (db, "database-overview",
4061*7304104dSAndroid Build Coastguard Worker                       "select label,quantity from " BUILDIDS "_stats");
4062*7304104dSAndroid Build Coastguard Worker 
4063*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "database record counts:" << endl;
4064*7304104dSAndroid Build Coastguard Worker   while (1)
4065*7304104dSAndroid Build Coastguard Worker     {
4066*7304104dSAndroid Build Coastguard Worker       if (interrupted) break;
4067*7304104dSAndroid Build Coastguard Worker       if (sigusr1 != forced_rescan_count) // stop early if scan triggered
4068*7304104dSAndroid Build Coastguard Worker         break;
4069*7304104dSAndroid Build Coastguard Worker 
4070*7304104dSAndroid Build Coastguard Worker       int rc = ps_query.step();
4071*7304104dSAndroid Build Coastguard Worker       if (rc == SQLITE_DONE) break;
4072*7304104dSAndroid Build Coastguard Worker       if (rc != SQLITE_ROW)
4073*7304104dSAndroid Build Coastguard Worker         throw sqlite_exception(rc, "step");
4074*7304104dSAndroid Build Coastguard Worker 
4075*7304104dSAndroid Build Coastguard Worker       obatched(clog)
4076*7304104dSAndroid Build Coastguard Worker         << ((const char*) sqlite3_column_text(ps_query, 0) ?: (const char*) "NULL")
4077*7304104dSAndroid Build Coastguard Worker         << " "
4078*7304104dSAndroid Build Coastguard Worker         << (sqlite3_column_text(ps_query, 1) ?: (const unsigned char*) "NULL")
4079*7304104dSAndroid Build Coastguard Worker         << endl;
4080*7304104dSAndroid Build Coastguard Worker 
4081*7304104dSAndroid Build Coastguard Worker       set_metric("groom", "statistic",
4082*7304104dSAndroid Build Coastguard Worker                  ((const char*) sqlite3_column_text(ps_query, 0) ?: (const char*) "NULL"),
4083*7304104dSAndroid Build Coastguard Worker                  (sqlite3_column_double(ps_query, 1)));
4084*7304104dSAndroid Build Coastguard Worker     }
4085*7304104dSAndroid Build Coastguard Worker }
4086*7304104dSAndroid Build Coastguard Worker 
4087*7304104dSAndroid Build Coastguard Worker 
4088*7304104dSAndroid Build Coastguard Worker // Do a round of database grooming that might take many minutes to run.
groom()4089*7304104dSAndroid Build Coastguard Worker void groom()
4090*7304104dSAndroid Build Coastguard Worker {
4091*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "grooming database" << endl;
4092*7304104dSAndroid Build Coastguard Worker 
4093*7304104dSAndroid Build Coastguard Worker   struct timespec ts_start, ts_end;
4094*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &ts_start);
4095*7304104dSAndroid Build Coastguard Worker 
4096*7304104dSAndroid Build Coastguard Worker   // scan for files that have disappeared
4097*7304104dSAndroid Build Coastguard Worker   sqlite_ps files (db, "check old files",
4098*7304104dSAndroid Build Coastguard Worker                    "select distinct s.mtime, s.file, f.name from "
4099*7304104dSAndroid Build Coastguard Worker                    BUILDIDS "_file_mtime_scanned s, " BUILDIDS "_files_v f "
4100*7304104dSAndroid Build Coastguard Worker                    "where f.id = s.file");
4101*7304104dSAndroid Build Coastguard Worker   // NB: Because _ftime_mtime_scanned can contain both F and
4102*7304104dSAndroid Build Coastguard Worker   // R records for the same file, this query would return duplicates if the
4103*7304104dSAndroid Build Coastguard Worker   // DISTINCT qualifier were not there.
4104*7304104dSAndroid Build Coastguard Worker   files.reset();
4105*7304104dSAndroid Build Coastguard Worker 
4106*7304104dSAndroid Build Coastguard Worker   // DECISION TIME - we enumerate stale fileids/mtimes
4107*7304104dSAndroid Build Coastguard Worker   deque<pair<int64_t,int64_t> > stale_fileid_mtime;
4108*7304104dSAndroid Build Coastguard Worker 
4109*7304104dSAndroid Build Coastguard Worker   time_t time_start = time(NULL);
4110*7304104dSAndroid Build Coastguard Worker   while(1)
4111*7304104dSAndroid Build Coastguard Worker     {
4112*7304104dSAndroid Build Coastguard Worker       // PR28514: limit grooming iteration to O(rescan time), to avoid
4113*7304104dSAndroid Build Coastguard Worker       // slow filesystem tests over many files locking out rescans for
4114*7304104dSAndroid Build Coastguard Worker       // too long.
4115*7304104dSAndroid Build Coastguard Worker       if (rescan_s > 0 && (long)time(NULL) > (long)(time_start + rescan_s))
4116*7304104dSAndroid Build Coastguard Worker         {
4117*7304104dSAndroid Build Coastguard Worker           inc_metric("groomed_total", "decision", "aborted");
4118*7304104dSAndroid Build Coastguard Worker           break;
4119*7304104dSAndroid Build Coastguard Worker         }
4120*7304104dSAndroid Build Coastguard Worker 
4121*7304104dSAndroid Build Coastguard Worker       if (interrupted) break;
4122*7304104dSAndroid Build Coastguard Worker 
4123*7304104dSAndroid Build Coastguard Worker       int rc = files.step();
4124*7304104dSAndroid Build Coastguard Worker       if (rc != SQLITE_ROW)
4125*7304104dSAndroid Build Coastguard Worker         break;
4126*7304104dSAndroid Build Coastguard Worker 
4127*7304104dSAndroid Build Coastguard Worker       int64_t mtime = sqlite3_column_int64 (files, 0);
4128*7304104dSAndroid Build Coastguard Worker       int64_t fileid = sqlite3_column_int64 (files, 1);
4129*7304104dSAndroid Build Coastguard Worker       const char* filename = ((const char*) sqlite3_column_text (files, 2) ?: "");
4130*7304104dSAndroid Build Coastguard Worker       struct stat s;
4131*7304104dSAndroid Build Coastguard Worker       bool regex_file_drop = 0;
4132*7304104dSAndroid Build Coastguard Worker 
4133*7304104dSAndroid Build Coastguard Worker       if (regex_groom)
4134*7304104dSAndroid Build Coastguard Worker         {
4135*7304104dSAndroid Build Coastguard Worker           bool reg_include = !regexec (&file_include_regex, filename, 0, 0, 0);
4136*7304104dSAndroid Build Coastguard Worker           bool reg_exclude = !regexec (&file_exclude_regex, filename, 0, 0, 0);
4137*7304104dSAndroid Build Coastguard Worker           regex_file_drop = !reg_include || reg_exclude; // match logic of scan_source_paths
4138*7304104dSAndroid Build Coastguard Worker         }
4139*7304104dSAndroid Build Coastguard Worker 
4140*7304104dSAndroid Build Coastguard Worker       rc = stat(filename, &s);
4141*7304104dSAndroid Build Coastguard Worker       if ( regex_file_drop ||  rc < 0 || (mtime != (int64_t) s.st_mtime) )
4142*7304104dSAndroid Build Coastguard Worker         {
4143*7304104dSAndroid Build Coastguard Worker           if (verbose > 2)
4144*7304104dSAndroid Build Coastguard Worker             obatched(clog) << "groom: stale file=" << filename << " mtime=" << mtime << endl;
4145*7304104dSAndroid Build Coastguard Worker           stale_fileid_mtime.push_back(make_pair(fileid,mtime));
4146*7304104dSAndroid Build Coastguard Worker           inc_metric("groomed_total", "decision", "stale");
4147*7304104dSAndroid Build Coastguard Worker           set_metric("thread_work_pending","role","groom", stale_fileid_mtime.size());
4148*7304104dSAndroid Build Coastguard Worker         }
4149*7304104dSAndroid Build Coastguard Worker       else
4150*7304104dSAndroid Build Coastguard Worker         inc_metric("groomed_total", "decision", "fresh");
4151*7304104dSAndroid Build Coastguard Worker 
4152*7304104dSAndroid Build Coastguard Worker       if (sigusr1 != forced_rescan_count) // stop early if scan triggered
4153*7304104dSAndroid Build Coastguard Worker         break;
4154*7304104dSAndroid Build Coastguard Worker     }
4155*7304104dSAndroid Build Coastguard Worker   files.reset();
4156*7304104dSAndroid Build Coastguard Worker 
4157*7304104dSAndroid Build Coastguard Worker   // ACTION TIME
4158*7304104dSAndroid Build Coastguard Worker 
4159*7304104dSAndroid Build Coastguard Worker   // Now that we know which file/mtime tuples are stale, actually do
4160*7304104dSAndroid Build Coastguard Worker   // the deletion from the database.  Doing this during the SELECT
4161*7304104dSAndroid Build Coastguard Worker   // iteration above results in undefined behaviour in sqlite, as per
4162*7304104dSAndroid Build Coastguard Worker   // https://www.sqlite.org/isolation.html
4163*7304104dSAndroid Build Coastguard Worker 
4164*7304104dSAndroid Build Coastguard Worker   // We could shuffle stale_fileid_mtime[] here.  It'd let aborted
4165*7304104dSAndroid Build Coastguard Worker   // sequences of nuke operations resume at random locations, instead
4166*7304104dSAndroid Build Coastguard Worker   // of just starting over.  But it doesn't matter much either way,
4167*7304104dSAndroid Build Coastguard Worker   // as long as we make progress.
4168*7304104dSAndroid Build Coastguard Worker 
4169*7304104dSAndroid Build Coastguard Worker   sqlite_ps files_del_f_de (db, "nuke f_de", "delete from " BUILDIDS "_f_de where file = ? and mtime = ?");
4170*7304104dSAndroid Build Coastguard Worker   sqlite_ps files_del_r_de (db, "nuke r_de", "delete from " BUILDIDS "_r_de where file = ? and mtime = ?");
4171*7304104dSAndroid Build Coastguard Worker   sqlite_ps files_del_scan (db, "nuke f_m_s", "delete from " BUILDIDS "_file_mtime_scanned "
4172*7304104dSAndroid Build Coastguard Worker                             "where file = ? and mtime = ?");
4173*7304104dSAndroid Build Coastguard Worker 
4174*7304104dSAndroid Build Coastguard Worker   while (! stale_fileid_mtime.empty())
4175*7304104dSAndroid Build Coastguard Worker     {
4176*7304104dSAndroid Build Coastguard Worker       auto stale = stale_fileid_mtime.front();
4177*7304104dSAndroid Build Coastguard Worker       stale_fileid_mtime.pop_front();
4178*7304104dSAndroid Build Coastguard Worker       set_metric("thread_work_pending","role","groom", stale_fileid_mtime.size());
4179*7304104dSAndroid Build Coastguard Worker 
4180*7304104dSAndroid Build Coastguard Worker       // PR28514: limit grooming iteration to O(rescan time), to avoid
4181*7304104dSAndroid Build Coastguard Worker       // slow nuke_* queries over many files locking out rescans for too
4182*7304104dSAndroid Build Coastguard Worker       // long.  We iterate over the files in random() sequence to avoid
4183*7304104dSAndroid Build Coastguard Worker       // partial checks going over the same set.
4184*7304104dSAndroid Build Coastguard Worker       if (rescan_s > 0 && (long)time(NULL) > (long)(time_start + rescan_s))
4185*7304104dSAndroid Build Coastguard Worker         {
4186*7304104dSAndroid Build Coastguard Worker           inc_metric("groomed_total", "action", "aborted");
4187*7304104dSAndroid Build Coastguard Worker           break;
4188*7304104dSAndroid Build Coastguard Worker         }
4189*7304104dSAndroid Build Coastguard Worker 
4190*7304104dSAndroid Build Coastguard Worker       if (interrupted) break;
4191*7304104dSAndroid Build Coastguard Worker 
4192*7304104dSAndroid Build Coastguard Worker       int64_t fileid = stale.first;
4193*7304104dSAndroid Build Coastguard Worker       int64_t mtime = stale.second;
4194*7304104dSAndroid Build Coastguard Worker       files_del_f_de.reset().bind(1,fileid).bind(2,mtime).step_ok_done();
4195*7304104dSAndroid Build Coastguard Worker       files_del_r_de.reset().bind(1,fileid).bind(2,mtime).step_ok_done();
4196*7304104dSAndroid Build Coastguard Worker       files_del_scan.reset().bind(1,fileid).bind(2,mtime).step_ok_done();
4197*7304104dSAndroid Build Coastguard Worker       inc_metric("groomed_total", "action", "cleaned");
4198*7304104dSAndroid Build Coastguard Worker 
4199*7304104dSAndroid Build Coastguard Worker        if (sigusr1 != forced_rescan_count) // stop early if scan triggered
4200*7304104dSAndroid Build Coastguard Worker         break;
4201*7304104dSAndroid Build Coastguard Worker     }
4202*7304104dSAndroid Build Coastguard Worker   stale_fileid_mtime.clear(); // no need for this any longer
4203*7304104dSAndroid Build Coastguard Worker   set_metric("thread_work_pending","role","groom", stale_fileid_mtime.size());
4204*7304104dSAndroid Build Coastguard Worker 
4205*7304104dSAndroid Build Coastguard Worker   // delete buildids with no references in _r_de or _f_de tables;
4206*7304104dSAndroid Build Coastguard Worker   // cascades to _r_sref & _f_s records
4207*7304104dSAndroid Build Coastguard Worker   sqlite_ps buildids_del (db, "nuke orphan buildids",
4208*7304104dSAndroid Build Coastguard Worker                           "delete from " BUILDIDS "_buildids "
4209*7304104dSAndroid Build Coastguard Worker                           "where not exists (select 1 from " BUILDIDS "_f_de d where " BUILDIDS "_buildids.id = d.buildid) "
4210*7304104dSAndroid Build Coastguard Worker                           "and not exists (select 1 from " BUILDIDS "_r_de d where " BUILDIDS "_buildids.id = d.buildid)");
4211*7304104dSAndroid Build Coastguard Worker   buildids_del.reset().step_ok_done();
4212*7304104dSAndroid Build Coastguard Worker 
4213*7304104dSAndroid Build Coastguard Worker   if (interrupted) return;
4214*7304104dSAndroid Build Coastguard Worker 
4215*7304104dSAndroid Build Coastguard Worker   // NB: "vacuum" is too heavy for even daily runs: it rewrites the entire db, so is done as maxigroom -G
4216*7304104dSAndroid Build Coastguard Worker   sqlite_ps g1 (db, "incremental vacuum", "pragma incremental_vacuum");
4217*7304104dSAndroid Build Coastguard Worker   g1.reset().step_ok_done();
4218*7304104dSAndroid Build Coastguard Worker   sqlite_ps g2 (db, "optimize", "pragma optimize");
4219*7304104dSAndroid Build Coastguard Worker   g2.reset().step_ok_done();
4220*7304104dSAndroid Build Coastguard Worker   sqlite_ps g3 (db, "wal checkpoint", "pragma wal_checkpoint=truncate");
4221*7304104dSAndroid Build Coastguard Worker   g3.reset().step_ok_done();
4222*7304104dSAndroid Build Coastguard Worker 
4223*7304104dSAndroid Build Coastguard Worker   database_stats_report();
4224*7304104dSAndroid Build Coastguard Worker 
4225*7304104dSAndroid Build Coastguard Worker   (void) statfs_free_enough_p(db_path, "database"); // report sqlite filesystem size
4226*7304104dSAndroid Build Coastguard Worker 
4227*7304104dSAndroid Build Coastguard Worker   sqlite3_db_release_memory(db); // shrink the process if possible
4228*7304104dSAndroid Build Coastguard Worker   sqlite3_db_release_memory(dbq); // ... for both connections
4229*7304104dSAndroid Build Coastguard Worker   debuginfod_pool_groom(); // and release any debuginfod_client objects we've been holding onto
4230*7304104dSAndroid Build Coastguard Worker 
4231*7304104dSAndroid Build Coastguard Worker #if 0 /* PR31265: don't jettison cache unnecessarily */
4232*7304104dSAndroid Build Coastguard Worker 
4233*7304104dSAndroid Build Coastguard Worker   fdcache.limit(0); // release the fdcache contents
4234*7304104dSAndroid Build Coastguard Worker   fdcache.limit(fdcache_mbs); // restore status quo parameters
4235*7304104dSAndroid Build Coastguard Worker #endif
4236*7304104dSAndroid Build Coastguard Worker 
4237*7304104dSAndroid Build Coastguard Worker   clock_gettime (CLOCK_MONOTONIC, &ts_end);
4238*7304104dSAndroid Build Coastguard Worker   double deltas = (ts_end.tv_sec - ts_start.tv_sec) + (ts_end.tv_nsec - ts_start.tv_nsec)/1.e9;
4239*7304104dSAndroid Build Coastguard Worker 
4240*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "groomed database in " << deltas << "s" << endl;
4241*7304104dSAndroid Build Coastguard Worker }
4242*7304104dSAndroid Build Coastguard Worker 
4243*7304104dSAndroid Build Coastguard Worker 
4244*7304104dSAndroid Build Coastguard Worker static void*
thread_main_groom(void *)4245*7304104dSAndroid Build Coastguard Worker thread_main_groom (void* /*arg*/)
4246*7304104dSAndroid Build Coastguard Worker {
4247*7304104dSAndroid Build Coastguard Worker   set_metric("thread_tid", "role", "groom", tid());
4248*7304104dSAndroid Build Coastguard Worker   add_metric("thread_count", "role", "groom", 1);
4249*7304104dSAndroid Build Coastguard Worker 
4250*7304104dSAndroid Build Coastguard Worker   time_t last_groom = 0;
4251*7304104dSAndroid Build Coastguard Worker 
4252*7304104dSAndroid Build Coastguard Worker   while (1)
4253*7304104dSAndroid Build Coastguard Worker     {
4254*7304104dSAndroid Build Coastguard Worker       sleep (1);
4255*7304104dSAndroid Build Coastguard Worker       scanq.wait_idle(); // PR25394: block scanners during grooming!
4256*7304104dSAndroid Build Coastguard Worker       if (interrupted) break;
4257*7304104dSAndroid Build Coastguard Worker 
4258*7304104dSAndroid Build Coastguard Worker       time_t now = time(NULL);
4259*7304104dSAndroid Build Coastguard Worker       bool groom_now = false;
4260*7304104dSAndroid Build Coastguard Worker       if (last_groom == 0) // at least one initial groom is documented even for -g0
4261*7304104dSAndroid Build Coastguard Worker         groom_now = true;
4262*7304104dSAndroid Build Coastguard Worker       if (groom_s > 0 && (long)now > (long)(last_groom + groom_s))
4263*7304104dSAndroid Build Coastguard Worker         groom_now = true;
4264*7304104dSAndroid Build Coastguard Worker       if (sigusr2 != forced_groom_count)
4265*7304104dSAndroid Build Coastguard Worker         {
4266*7304104dSAndroid Build Coastguard Worker           forced_groom_count = sigusr2;
4267*7304104dSAndroid Build Coastguard Worker           groom_now = true;
4268*7304104dSAndroid Build Coastguard Worker         }
4269*7304104dSAndroid Build Coastguard Worker       if (groom_now)
4270*7304104dSAndroid Build Coastguard Worker         {
4271*7304104dSAndroid Build Coastguard Worker           set_metric("thread_busy", "role", "groom", 1);
4272*7304104dSAndroid Build Coastguard Worker           try
4273*7304104dSAndroid Build Coastguard Worker             {
4274*7304104dSAndroid Build Coastguard Worker               groom ();
4275*7304104dSAndroid Build Coastguard Worker             }
4276*7304104dSAndroid Build Coastguard Worker           catch (const sqlite_exception& e)
4277*7304104dSAndroid Build Coastguard Worker             {
4278*7304104dSAndroid Build Coastguard Worker               obatched(cerr) << e.message << endl;
4279*7304104dSAndroid Build Coastguard Worker             }
4280*7304104dSAndroid Build Coastguard Worker           last_groom = time(NULL); // NB: now was before grooming
4281*7304104dSAndroid Build Coastguard Worker           // finished a grooming loop
4282*7304104dSAndroid Build Coastguard Worker           inc_metric("thread_work_total", "role", "groom");
4283*7304104dSAndroid Build Coastguard Worker           set_metric("thread_busy", "role", "groom", 0);
4284*7304104dSAndroid Build Coastguard Worker         }
4285*7304104dSAndroid Build Coastguard Worker 
4286*7304104dSAndroid Build Coastguard Worker       scanq.done_idle();
4287*7304104dSAndroid Build Coastguard Worker     }
4288*7304104dSAndroid Build Coastguard Worker 
4289*7304104dSAndroid Build Coastguard Worker   return 0;
4290*7304104dSAndroid Build Coastguard Worker }
4291*7304104dSAndroid Build Coastguard Worker 
4292*7304104dSAndroid Build Coastguard Worker 
4293*7304104dSAndroid Build Coastguard Worker ////////////////////////////////////////////////////////////////////////
4294*7304104dSAndroid Build Coastguard Worker 
4295*7304104dSAndroid Build Coastguard Worker 
4296*7304104dSAndroid Build Coastguard Worker static void
signal_handler(int)4297*7304104dSAndroid Build Coastguard Worker signal_handler (int /* sig */)
4298*7304104dSAndroid Build Coastguard Worker {
4299*7304104dSAndroid Build Coastguard Worker   interrupted ++;
4300*7304104dSAndroid Build Coastguard Worker 
4301*7304104dSAndroid Build Coastguard Worker   if (db)
4302*7304104dSAndroid Build Coastguard Worker     sqlite3_interrupt (db);
4303*7304104dSAndroid Build Coastguard Worker   if (dbq)
4304*7304104dSAndroid Build Coastguard Worker     sqlite3_interrupt (dbq);
4305*7304104dSAndroid Build Coastguard Worker 
4306*7304104dSAndroid Build Coastguard Worker   // NB: don't do anything else in here
4307*7304104dSAndroid Build Coastguard Worker }
4308*7304104dSAndroid Build Coastguard Worker 
4309*7304104dSAndroid Build Coastguard Worker static void
sigusr1_handler(int)4310*7304104dSAndroid Build Coastguard Worker sigusr1_handler (int /* sig */)
4311*7304104dSAndroid Build Coastguard Worker {
4312*7304104dSAndroid Build Coastguard Worker    sigusr1 ++;
4313*7304104dSAndroid Build Coastguard Worker   // NB: don't do anything else in here
4314*7304104dSAndroid Build Coastguard Worker }
4315*7304104dSAndroid Build Coastguard Worker 
4316*7304104dSAndroid Build Coastguard Worker static void
sigusr2_handler(int)4317*7304104dSAndroid Build Coastguard Worker sigusr2_handler (int /* sig */)
4318*7304104dSAndroid Build Coastguard Worker {
4319*7304104dSAndroid Build Coastguard Worker    sigusr2 ++;
4320*7304104dSAndroid Build Coastguard Worker   // NB: don't do anything else in here
4321*7304104dSAndroid Build Coastguard Worker }
4322*7304104dSAndroid Build Coastguard Worker 
4323*7304104dSAndroid Build Coastguard Worker 
4324*7304104dSAndroid Build Coastguard Worker static void // error logging callback from libmicrohttpd internals
error_cb(void * arg,const char * fmt,va_list ap)4325*7304104dSAndroid Build Coastguard Worker error_cb (void *arg, const char *fmt, va_list ap)
4326*7304104dSAndroid Build Coastguard Worker {
4327*7304104dSAndroid Build Coastguard Worker   (void) arg;
4328*7304104dSAndroid Build Coastguard Worker   inc_metric("error_count","libmicrohttpd",fmt);
4329*7304104dSAndroid Build Coastguard Worker   char errmsg[512];
4330*7304104dSAndroid Build Coastguard Worker   (void) vsnprintf (errmsg, sizeof(errmsg), fmt, ap); // ok if slightly truncated
4331*7304104dSAndroid Build Coastguard Worker   obatched(cerr) << "libmicrohttpd error: " << errmsg; // MHD_DLOG calls already include \n
4332*7304104dSAndroid Build Coastguard Worker }
4333*7304104dSAndroid Build Coastguard Worker 
4334*7304104dSAndroid Build Coastguard Worker 
4335*7304104dSAndroid Build Coastguard Worker // A user-defined sqlite function, to score the sharedness of the
4336*7304104dSAndroid Build Coastguard Worker // prefix of two strings.  This is used to compare candidate debuginfo
4337*7304104dSAndroid Build Coastguard Worker // / source-rpm names, so that the closest match
4338*7304104dSAndroid Build Coastguard Worker // (directory-topology-wise closest) is found.  This is important in
4339*7304104dSAndroid Build Coastguard Worker // case the same sref (source file name) is in many -debuginfo or
4340*7304104dSAndroid Build Coastguard Worker // -debugsource RPMs, such as when multiple versions/releases of the
4341*7304104dSAndroid Build Coastguard Worker // same package are in the database.
4342*7304104dSAndroid Build Coastguard Worker 
sqlite3_sharedprefix_fn(sqlite3_context * c,int argc,sqlite3_value ** argv)4343*7304104dSAndroid Build Coastguard Worker static void sqlite3_sharedprefix_fn (sqlite3_context* c, int argc, sqlite3_value** argv)
4344*7304104dSAndroid Build Coastguard Worker {
4345*7304104dSAndroid Build Coastguard Worker   if (argc != 2)
4346*7304104dSAndroid Build Coastguard Worker     sqlite3_result_error(c, "expect 2 string arguments", -1);
4347*7304104dSAndroid Build Coastguard Worker   else if ((sqlite3_value_type(argv[0]) != SQLITE_TEXT) ||
4348*7304104dSAndroid Build Coastguard Worker            (sqlite3_value_type(argv[1]) != SQLITE_TEXT))
4349*7304104dSAndroid Build Coastguard Worker     sqlite3_result_null(c);
4350*7304104dSAndroid Build Coastguard Worker   else
4351*7304104dSAndroid Build Coastguard Worker     {
4352*7304104dSAndroid Build Coastguard Worker       const unsigned char* a = sqlite3_value_text (argv[0]);
4353*7304104dSAndroid Build Coastguard Worker       const unsigned char* b = sqlite3_value_text (argv[1]);
4354*7304104dSAndroid Build Coastguard Worker       int i = 0;
4355*7304104dSAndroid Build Coastguard Worker       while (*a != '\0' && *b != '\0' && *a++ == *b++)
4356*7304104dSAndroid Build Coastguard Worker         i++;
4357*7304104dSAndroid Build Coastguard Worker       sqlite3_result_int (c, i);
4358*7304104dSAndroid Build Coastguard Worker     }
4359*7304104dSAndroid Build Coastguard Worker }
4360*7304104dSAndroid Build Coastguard Worker 
4361*7304104dSAndroid Build Coastguard Worker 
4362*7304104dSAndroid Build Coastguard Worker static unsigned
default_concurrency()4363*7304104dSAndroid Build Coastguard Worker default_concurrency() // guaranteed >= 1
4364*7304104dSAndroid Build Coastguard Worker {
4365*7304104dSAndroid Build Coastguard Worker   // Prior to PR29975 & PR29976, we'd just use this:
4366*7304104dSAndroid Build Coastguard Worker   unsigned sth = std::thread::hardware_concurrency();
4367*7304104dSAndroid Build Coastguard Worker   // ... but on many-CPU boxes, admins or distros may throttle
4368*7304104dSAndroid Build Coastguard Worker   // resources in such a way that debuginfod would mysteriously fail.
4369*7304104dSAndroid Build Coastguard Worker   // So we reduce the defaults:
4370*7304104dSAndroid Build Coastguard Worker 
4371*7304104dSAndroid Build Coastguard Worker   unsigned aff = 0;
4372*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_SCHED_GETAFFINITY
4373*7304104dSAndroid Build Coastguard Worker   {
4374*7304104dSAndroid Build Coastguard Worker     int ret;
4375*7304104dSAndroid Build Coastguard Worker     cpu_set_t mask;
4376*7304104dSAndroid Build Coastguard Worker     CPU_ZERO(&mask);
4377*7304104dSAndroid Build Coastguard Worker     ret = sched_getaffinity(0, sizeof(mask), &mask);
4378*7304104dSAndroid Build Coastguard Worker     if (ret == 0)
4379*7304104dSAndroid Build Coastguard Worker       aff = CPU_COUNT(&mask);
4380*7304104dSAndroid Build Coastguard Worker   }
4381*7304104dSAndroid Build Coastguard Worker #endif
4382*7304104dSAndroid Build Coastguard Worker 
4383*7304104dSAndroid Build Coastguard Worker   unsigned fn = 0;
4384*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_GETRLIMIT
4385*7304104dSAndroid Build Coastguard Worker   {
4386*7304104dSAndroid Build Coastguard Worker     struct rlimit rlim;
4387*7304104dSAndroid Build Coastguard Worker     int rc = getrlimit(RLIMIT_NOFILE, &rlim);
4388*7304104dSAndroid Build Coastguard Worker     if (rc == 0)
4389*7304104dSAndroid Build Coastguard Worker       fn = max((rlim_t)1, (rlim.rlim_cur - 100) / 4);
4390*7304104dSAndroid Build Coastguard Worker     // at least 2 fds are used by each listener thread etc.
4391*7304104dSAndroid Build Coastguard Worker     // plus a bunch to account for shared libraries and such
4392*7304104dSAndroid Build Coastguard Worker   }
4393*7304104dSAndroid Build Coastguard Worker #endif
4394*7304104dSAndroid Build Coastguard Worker 
4395*7304104dSAndroid Build Coastguard Worker   unsigned d = min(max(sth, 1U),
4396*7304104dSAndroid Build Coastguard Worker                    min(max(aff, 1U),
4397*7304104dSAndroid Build Coastguard Worker                        max(fn, 1U)));
4398*7304104dSAndroid Build Coastguard Worker   return d;
4399*7304104dSAndroid Build Coastguard Worker }
4400*7304104dSAndroid Build Coastguard Worker 
4401*7304104dSAndroid Build Coastguard Worker 
4402*7304104dSAndroid Build Coastguard Worker // 30879: Something to help out in case of an uncaught exception.
my_terminate_handler()4403*7304104dSAndroid Build Coastguard Worker void my_terminate_handler()
4404*7304104dSAndroid Build Coastguard Worker {
4405*7304104dSAndroid Build Coastguard Worker #if defined(__GLIBC__)
4406*7304104dSAndroid Build Coastguard Worker   void *array[40];
4407*7304104dSAndroid Build Coastguard Worker   int size = backtrace (array, 40);
4408*7304104dSAndroid Build Coastguard Worker   backtrace_symbols_fd (array, size, STDERR_FILENO);
4409*7304104dSAndroid Build Coastguard Worker #endif
4410*7304104dSAndroid Build Coastguard Worker #if defined(__GLIBCXX__) || defined(__GLIBCPP__)
4411*7304104dSAndroid Build Coastguard Worker   __gnu_cxx::__verbose_terminate_handler();
4412*7304104dSAndroid Build Coastguard Worker #endif
4413*7304104dSAndroid Build Coastguard Worker   abort();
4414*7304104dSAndroid Build Coastguard Worker }
4415*7304104dSAndroid Build Coastguard Worker 
4416*7304104dSAndroid Build Coastguard Worker 
4417*7304104dSAndroid Build Coastguard Worker int
main(int argc,char * argv[])4418*7304104dSAndroid Build Coastguard Worker main (int argc, char *argv[])
4419*7304104dSAndroid Build Coastguard Worker {
4420*7304104dSAndroid Build Coastguard Worker   (void) setlocale (LC_ALL, "");
4421*7304104dSAndroid Build Coastguard Worker   (void) bindtextdomain (PACKAGE_TARNAME, LOCALEDIR);
4422*7304104dSAndroid Build Coastguard Worker   (void) textdomain (PACKAGE_TARNAME);
4423*7304104dSAndroid Build Coastguard Worker 
4424*7304104dSAndroid Build Coastguard Worker   std::set_terminate(& my_terminate_handler);
4425*7304104dSAndroid Build Coastguard Worker 
4426*7304104dSAndroid Build Coastguard Worker   /* Tell the library which version we are expecting.  */
4427*7304104dSAndroid Build Coastguard Worker   elf_version (EV_CURRENT);
4428*7304104dSAndroid Build Coastguard Worker 
4429*7304104dSAndroid Build Coastguard Worker   tmpdir = string(getenv("TMPDIR") ?: "/tmp");
4430*7304104dSAndroid Build Coastguard Worker 
4431*7304104dSAndroid Build Coastguard Worker   /* Set computed default values. */
4432*7304104dSAndroid Build Coastguard Worker   db_path = string(getenv("HOME") ?: "/") + string("/.debuginfod.sqlite"); /* XDG? */
4433*7304104dSAndroid Build Coastguard Worker   int rc = regcomp (& file_include_regex, ".*", REG_EXTENDED|REG_NOSUB); // match everything
4434*7304104dSAndroid Build Coastguard Worker   if (rc != 0)
4435*7304104dSAndroid Build Coastguard Worker     error (EXIT_FAILURE, 0, "regcomp failure: %d", rc);
4436*7304104dSAndroid Build Coastguard Worker   rc = regcomp (& file_exclude_regex, "^$", REG_EXTENDED|REG_NOSUB); // match nothing
4437*7304104dSAndroid Build Coastguard Worker   if (rc != 0)
4438*7304104dSAndroid Build Coastguard Worker     error (EXIT_FAILURE, 0, "regcomp failure: %d", rc);
4439*7304104dSAndroid Build Coastguard Worker 
4440*7304104dSAndroid Build Coastguard Worker   // default parameters for fdcache are computed from system stats
4441*7304104dSAndroid Build Coastguard Worker   struct statfs sfs;
4442*7304104dSAndroid Build Coastguard Worker   rc = statfs(tmpdir.c_str(), &sfs);
4443*7304104dSAndroid Build Coastguard Worker   if (rc < 0)
4444*7304104dSAndroid Build Coastguard Worker     fdcache_mbs = 1024; // 1 gigabyte
4445*7304104dSAndroid Build Coastguard Worker   else
4446*7304104dSAndroid Build Coastguard Worker     fdcache_mbs = sfs.f_bavail * sfs.f_bsize / 1024 / 1024 / 4; // 25% of free space
4447*7304104dSAndroid Build Coastguard Worker   fdcache_mintmp = 25; // emergency flush at 25% remaining (75% full)
4448*7304104dSAndroid Build Coastguard Worker   fdcache_prefetch = 64; // guesstimate storage is this much less costly than re-decompression
4449*7304104dSAndroid Build Coastguard Worker 
4450*7304104dSAndroid Build Coastguard Worker   /* Parse and process arguments.  */
4451*7304104dSAndroid Build Coastguard Worker   int remaining;
4452*7304104dSAndroid Build Coastguard Worker   (void) argp_parse (&argp, argc, argv, ARGP_IN_ORDER, &remaining, NULL);
4453*7304104dSAndroid Build Coastguard Worker   if (remaining != argc)
4454*7304104dSAndroid Build Coastguard Worker       error (EXIT_FAILURE, 0,
4455*7304104dSAndroid Build Coastguard Worker              "unexpected argument: %s", argv[remaining]);
4456*7304104dSAndroid Build Coastguard Worker 
4457*7304104dSAndroid Build Coastguard Worker   if (scan_archives.size()==0 && !scan_files && source_paths.size()>0)
4458*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "warning: without -F -R -U -Z, ignoring PATHs" << endl;
4459*7304104dSAndroid Build Coastguard Worker 
4460*7304104dSAndroid Build Coastguard Worker   fdcache.limit(fdcache_mbs);
4461*7304104dSAndroid Build Coastguard Worker 
4462*7304104dSAndroid Build Coastguard Worker   (void) signal (SIGPIPE, SIG_IGN); // microhttpd can generate it incidentally, ignore
4463*7304104dSAndroid Build Coastguard Worker   (void) signal (SIGINT, signal_handler); // ^C
4464*7304104dSAndroid Build Coastguard Worker   (void) signal (SIGHUP, signal_handler); // EOF
4465*7304104dSAndroid Build Coastguard Worker   (void) signal (SIGTERM, signal_handler); // systemd
4466*7304104dSAndroid Build Coastguard Worker   (void) signal (SIGUSR1, sigusr1_handler); // end-user
4467*7304104dSAndroid Build Coastguard Worker   (void) signal (SIGUSR2, sigusr2_handler); // end-user
4468*7304104dSAndroid Build Coastguard Worker 
4469*7304104dSAndroid Build Coastguard Worker   /* Get database ready. */
4470*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4471*7304104dSAndroid Build Coastguard Worker     {
4472*7304104dSAndroid Build Coastguard Worker       rc = sqlite3_open_v2 (db_path.c_str(), &db, (SQLITE_OPEN_READWRITE
4473*7304104dSAndroid Build Coastguard Worker                                                    |SQLITE_OPEN_URI
4474*7304104dSAndroid Build Coastguard Worker                                                    |SQLITE_OPEN_PRIVATECACHE
4475*7304104dSAndroid Build Coastguard Worker                                                    |SQLITE_OPEN_CREATE
4476*7304104dSAndroid Build Coastguard Worker                                                    |SQLITE_OPEN_FULLMUTEX), /* thread-safe */
4477*7304104dSAndroid Build Coastguard Worker                             NULL);
4478*7304104dSAndroid Build Coastguard Worker       if (rc == SQLITE_CORRUPT)
4479*7304104dSAndroid Build Coastguard Worker         {
4480*7304104dSAndroid Build Coastguard Worker           (void) unlink (db_path.c_str());
4481*7304104dSAndroid Build Coastguard Worker           error (EXIT_FAILURE, 0,
4482*7304104dSAndroid Build Coastguard Worker                  "cannot open %s, deleted database: %s", db_path.c_str(), sqlite3_errmsg(db));
4483*7304104dSAndroid Build Coastguard Worker         }
4484*7304104dSAndroid Build Coastguard Worker       else if (rc)
4485*7304104dSAndroid Build Coastguard Worker         {
4486*7304104dSAndroid Build Coastguard Worker           error (EXIT_FAILURE, 0,
4487*7304104dSAndroid Build Coastguard Worker                  "cannot open %s, consider deleting database: %s", db_path.c_str(), sqlite3_errmsg(db));
4488*7304104dSAndroid Build Coastguard Worker         }
4489*7304104dSAndroid Build Coastguard Worker     }
4490*7304104dSAndroid Build Coastguard Worker 
4491*7304104dSAndroid Build Coastguard Worker   // open the readonly query variant
4492*7304104dSAndroid Build Coastguard Worker   // NB: PRIVATECACHE allows web queries to operate in parallel with
4493*7304104dSAndroid Build Coastguard Worker   // much other grooming/scanning operation.
4494*7304104dSAndroid Build Coastguard Worker   rc = sqlite3_open_v2 (db_path.c_str(), &dbq, (SQLITE_OPEN_READONLY
4495*7304104dSAndroid Build Coastguard Worker                                                 |SQLITE_OPEN_URI
4496*7304104dSAndroid Build Coastguard Worker                                                 |SQLITE_OPEN_PRIVATECACHE
4497*7304104dSAndroid Build Coastguard Worker                                                 |SQLITE_OPEN_FULLMUTEX), /* thread-safe */
4498*7304104dSAndroid Build Coastguard Worker                         NULL);
4499*7304104dSAndroid Build Coastguard Worker   if (rc)
4500*7304104dSAndroid Build Coastguard Worker     {
4501*7304104dSAndroid Build Coastguard Worker       error (EXIT_FAILURE, 0,
4502*7304104dSAndroid Build Coastguard Worker              "cannot open %s, consider deleting database: %s", db_path.c_str(), sqlite3_errmsg(dbq));
4503*7304104dSAndroid Build Coastguard Worker     }
4504*7304104dSAndroid Build Coastguard Worker 
4505*7304104dSAndroid Build Coastguard Worker 
4506*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "opened database " << db_path
4507*7304104dSAndroid Build Coastguard Worker                  << (db?" rw":"") << (dbq?" ro":"") << endl;
4508*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "sqlite version " << sqlite3_version << endl;
4509*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "service mode " << (passive_p ? "passive":"active") << endl;
4510*7304104dSAndroid Build Coastguard Worker 
4511*7304104dSAndroid Build Coastguard Worker   // add special string-prefix-similarity function used in rpm sref/sdef resolution
4512*7304104dSAndroid Build Coastguard Worker   rc = sqlite3_create_function(dbq, "sharedprefix", 2, SQLITE_UTF8, NULL,
4513*7304104dSAndroid Build Coastguard Worker                                & sqlite3_sharedprefix_fn, NULL, NULL);
4514*7304104dSAndroid Build Coastguard Worker   if (rc != SQLITE_OK)
4515*7304104dSAndroid Build Coastguard Worker     error (EXIT_FAILURE, 0,
4516*7304104dSAndroid Build Coastguard Worker            "cannot create sharedprefix function: %s", sqlite3_errmsg(dbq));
4517*7304104dSAndroid Build Coastguard Worker 
4518*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4519*7304104dSAndroid Build Coastguard Worker     {
4520*7304104dSAndroid Build Coastguard Worker       if (verbose > 3)
4521*7304104dSAndroid Build Coastguard Worker         obatched(clog) << "ddl: " << DEBUGINFOD_SQLITE_DDL << endl;
4522*7304104dSAndroid Build Coastguard Worker       rc = sqlite3_exec (db, DEBUGINFOD_SQLITE_DDL, NULL, NULL, NULL);
4523*7304104dSAndroid Build Coastguard Worker       if (rc != SQLITE_OK)
4524*7304104dSAndroid Build Coastguard Worker         {
4525*7304104dSAndroid Build Coastguard Worker           error (EXIT_FAILURE, 0,
4526*7304104dSAndroid Build Coastguard Worker                  "cannot run database schema ddl: %s", sqlite3_errmsg(db));
4527*7304104dSAndroid Build Coastguard Worker         }
4528*7304104dSAndroid Build Coastguard Worker     }
4529*7304104dSAndroid Build Coastguard Worker 
4530*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "libmicrohttpd version " << MHD_get_version() << endl;
4531*7304104dSAndroid Build Coastguard Worker 
4532*7304104dSAndroid Build Coastguard Worker   /* If '-C' wasn't given or was given with no arg, pick a reasonable default
4533*7304104dSAndroid Build Coastguard Worker      for the number of worker threads.  */
4534*7304104dSAndroid Build Coastguard Worker   if (connection_pool == 0)
4535*7304104dSAndroid Build Coastguard Worker     connection_pool = default_concurrency();
4536*7304104dSAndroid Build Coastguard Worker 
4537*7304104dSAndroid Build Coastguard Worker   /* Note that MHD_USE_EPOLL and MHD_USE_THREAD_PER_CONNECTION don't
4538*7304104dSAndroid Build Coastguard Worker      work together.  */
4539*7304104dSAndroid Build Coastguard Worker   unsigned int use_epoll = 0;
4540*7304104dSAndroid Build Coastguard Worker #if MHD_VERSION >= 0x00095100
4541*7304104dSAndroid Build Coastguard Worker   use_epoll = MHD_USE_EPOLL;
4542*7304104dSAndroid Build Coastguard Worker #endif
4543*7304104dSAndroid Build Coastguard Worker 
4544*7304104dSAndroid Build Coastguard Worker   unsigned int mhd_flags = (
4545*7304104dSAndroid Build Coastguard Worker #if MHD_VERSION >= 0x00095300
4546*7304104dSAndroid Build Coastguard Worker 			    MHD_USE_INTERNAL_POLLING_THREAD
4547*7304104dSAndroid Build Coastguard Worker #else
4548*7304104dSAndroid Build Coastguard Worker 			    MHD_USE_SELECT_INTERNALLY
4549*7304104dSAndroid Build Coastguard Worker #endif
4550*7304104dSAndroid Build Coastguard Worker 			    | MHD_USE_DUAL_STACK
4551*7304104dSAndroid Build Coastguard Worker 			    | use_epoll
4552*7304104dSAndroid Build Coastguard Worker #if MHD_VERSION >= 0x00095200
4553*7304104dSAndroid Build Coastguard Worker 			    | MHD_USE_ITC
4554*7304104dSAndroid Build Coastguard Worker #endif
4555*7304104dSAndroid Build Coastguard Worker 			    | MHD_USE_DEBUG); /* report errors to stderr */
4556*7304104dSAndroid Build Coastguard Worker 
4557*7304104dSAndroid Build Coastguard Worker   // Start httpd server threads.  Use a single dual-homed pool.
4558*7304104dSAndroid Build Coastguard Worker   MHD_Daemon *d46 = MHD_start_daemon (mhd_flags, http_port,
4559*7304104dSAndroid Build Coastguard Worker 				      NULL, NULL, /* default accept policy */
4560*7304104dSAndroid Build Coastguard Worker 				      handler_cb, NULL, /* handler callback */
4561*7304104dSAndroid Build Coastguard Worker 				      MHD_OPTION_EXTERNAL_LOGGER,
4562*7304104dSAndroid Build Coastguard Worker 				      error_cb, NULL,
4563*7304104dSAndroid Build Coastguard Worker 				      MHD_OPTION_THREAD_POOL_SIZE,
4564*7304104dSAndroid Build Coastguard Worker 				      (int)connection_pool,
4565*7304104dSAndroid Build Coastguard Worker 				      MHD_OPTION_END);
4566*7304104dSAndroid Build Coastguard Worker 
4567*7304104dSAndroid Build Coastguard Worker   MHD_Daemon *d4 = NULL;
4568*7304104dSAndroid Build Coastguard Worker   if (d46 == NULL)
4569*7304104dSAndroid Build Coastguard Worker     {
4570*7304104dSAndroid Build Coastguard Worker       // Cannot use dual_stack, use ipv4 only
4571*7304104dSAndroid Build Coastguard Worker       mhd_flags &= ~(MHD_USE_DUAL_STACK);
4572*7304104dSAndroid Build Coastguard Worker       d4 = MHD_start_daemon (mhd_flags, http_port,
4573*7304104dSAndroid Build Coastguard Worker 			     NULL, NULL, /* default accept policy */
4574*7304104dSAndroid Build Coastguard Worker 			     handler_cb, NULL, /* handler callback */
4575*7304104dSAndroid Build Coastguard Worker 			     MHD_OPTION_EXTERNAL_LOGGER,
4576*7304104dSAndroid Build Coastguard Worker 			     error_cb, NULL,
4577*7304104dSAndroid Build Coastguard Worker 			     (connection_pool
4578*7304104dSAndroid Build Coastguard Worker 			      ? MHD_OPTION_THREAD_POOL_SIZE
4579*7304104dSAndroid Build Coastguard Worker 			      : MHD_OPTION_END),
4580*7304104dSAndroid Build Coastguard Worker 			     (connection_pool
4581*7304104dSAndroid Build Coastguard Worker 			      ? (int)connection_pool
4582*7304104dSAndroid Build Coastguard Worker 			      : MHD_OPTION_END),
4583*7304104dSAndroid Build Coastguard Worker 			     MHD_OPTION_END);
4584*7304104dSAndroid Build Coastguard Worker       if (d4 == NULL)
4585*7304104dSAndroid Build Coastguard Worker 	{
4586*7304104dSAndroid Build Coastguard Worker 	  sqlite3 *database = db;
4587*7304104dSAndroid Build Coastguard Worker 	  sqlite3 *databaseq = dbq;
4588*7304104dSAndroid Build Coastguard Worker 	  db = dbq = 0; // for signal_handler not to freak
4589*7304104dSAndroid Build Coastguard Worker 	  sqlite3_close (databaseq);
4590*7304104dSAndroid Build Coastguard Worker 	  sqlite3_close (database);
4591*7304104dSAndroid Build Coastguard Worker 	  error (EXIT_FAILURE, 0, "cannot start http server at port %d",
4592*7304104dSAndroid Build Coastguard Worker 		 http_port);
4593*7304104dSAndroid Build Coastguard Worker 	}
4594*7304104dSAndroid Build Coastguard Worker 
4595*7304104dSAndroid Build Coastguard Worker     }
4596*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "started http server on"
4597*7304104dSAndroid Build Coastguard Worker                  << (d4 != NULL ? " IPv4 " : " IPv4 IPv6 ")
4598*7304104dSAndroid Build Coastguard Worker                  << "port=" << http_port << endl;
4599*7304104dSAndroid Build Coastguard Worker 
4600*7304104dSAndroid Build Coastguard Worker   // add maxigroom sql if -G given
4601*7304104dSAndroid Build Coastguard Worker   if (maxigroom)
4602*7304104dSAndroid Build Coastguard Worker     {
4603*7304104dSAndroid Build Coastguard Worker       obatched(clog) << "maxigrooming database, please wait." << endl;
4604*7304104dSAndroid Build Coastguard Worker       extra_ddl.push_back("create index if not exists " BUILDIDS "_r_sref_arc on " BUILDIDS "_r_sref(artifactsrc);");
4605*7304104dSAndroid Build Coastguard Worker       extra_ddl.push_back("delete from " BUILDIDS "_r_sdef where not exists (select 1 from " BUILDIDS "_r_sref b where " BUILDIDS "_r_sdef.content = b.artifactsrc);");
4606*7304104dSAndroid Build Coastguard Worker       extra_ddl.push_back("drop index if exists " BUILDIDS "_r_sref_arc;");
4607*7304104dSAndroid Build Coastguard Worker 
4608*7304104dSAndroid Build Coastguard Worker       // NB: we don't maxigroom the _files interning table.  It'd require a temp index on all the
4609*7304104dSAndroid Build Coastguard Worker       // tables that have file foreign-keys, which is a lot.
4610*7304104dSAndroid Build Coastguard Worker 
4611*7304104dSAndroid Build Coastguard Worker       // NB: with =delete, may take up 3x disk space total during vacuum process
4612*7304104dSAndroid Build Coastguard Worker       //     vs.  =off (only 2x but may corrupt database if program dies mid-vacuum)
4613*7304104dSAndroid Build Coastguard Worker       //     vs.  =wal (>3x observed, but safe)
4614*7304104dSAndroid Build Coastguard Worker       extra_ddl.push_back("pragma journal_mode=delete;");
4615*7304104dSAndroid Build Coastguard Worker       extra_ddl.push_back("vacuum;");
4616*7304104dSAndroid Build Coastguard Worker       extra_ddl.push_back("pragma journal_mode=wal;");
4617*7304104dSAndroid Build Coastguard Worker     }
4618*7304104dSAndroid Build Coastguard Worker 
4619*7304104dSAndroid Build Coastguard Worker   // run extra -D sql if given
4620*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4621*7304104dSAndroid Build Coastguard Worker     for (auto&& i: extra_ddl)
4622*7304104dSAndroid Build Coastguard Worker       {
4623*7304104dSAndroid Build Coastguard Worker         if (verbose > 1)
4624*7304104dSAndroid Build Coastguard Worker           obatched(clog) << "extra ddl:\n" << i << endl;
4625*7304104dSAndroid Build Coastguard Worker         rc = sqlite3_exec (db, i.c_str(), NULL, NULL, NULL);
4626*7304104dSAndroid Build Coastguard Worker         if (rc != SQLITE_OK && rc != SQLITE_DONE && rc != SQLITE_ROW)
4627*7304104dSAndroid Build Coastguard Worker           error (0, 0,
4628*7304104dSAndroid Build Coastguard Worker                  "warning: cannot run database extra ddl %s: %s", i.c_str(), sqlite3_errmsg(db));
4629*7304104dSAndroid Build Coastguard Worker 
4630*7304104dSAndroid Build Coastguard Worker         if (maxigroom)
4631*7304104dSAndroid Build Coastguard Worker           obatched(clog) << "maxigroomed database" << endl;
4632*7304104dSAndroid Build Coastguard Worker       }
4633*7304104dSAndroid Build Coastguard Worker 
4634*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4635*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "search concurrency " << concurrency << endl;
4636*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "webapi connection pool " << connection_pool
4637*7304104dSAndroid Build Coastguard Worker                  << (connection_pool ? "" : " (unlimited)") << endl;
4638*7304104dSAndroid Build Coastguard Worker   if (! passive_p) {
4639*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "rescan time " << rescan_s << endl;
4640*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "scan checkpoint " << scan_checkpoint << endl;
4641*7304104dSAndroid Build Coastguard Worker   }
4642*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "fdcache mbs " << fdcache_mbs << endl;
4643*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "fdcache prefetch " << fdcache_prefetch << endl;
4644*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "fdcache tmpdir " << tmpdir << endl;
4645*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "fdcache tmpdir min% " << fdcache_mintmp << endl;
4646*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4647*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "groom time " << groom_s << endl;
4648*7304104dSAndroid Build Coastguard Worker   obatched(clog) << "forwarded ttl limit " << forwarded_ttl_limit << endl;
4649*7304104dSAndroid Build Coastguard Worker 
4650*7304104dSAndroid Build Coastguard Worker   if (scan_archives.size()>0)
4651*7304104dSAndroid Build Coastguard Worker     {
4652*7304104dSAndroid Build Coastguard Worker       obatched ob(clog);
4653*7304104dSAndroid Build Coastguard Worker       auto& o = ob << "accepting archive types ";
4654*7304104dSAndroid Build Coastguard Worker       for (auto&& arch : scan_archives)
4655*7304104dSAndroid Build Coastguard Worker 	o << arch.first << "(" << arch.second << ") ";
4656*7304104dSAndroid Build Coastguard Worker       o << endl;
4657*7304104dSAndroid Build Coastguard Worker     }
4658*7304104dSAndroid Build Coastguard Worker   const char* du = getenv(DEBUGINFOD_URLS_ENV_VAR);
4659*7304104dSAndroid Build Coastguard Worker   if (du && du[0] != '\0') // set to non-empty string?
4660*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "upstream debuginfod servers: " << du << endl;
4661*7304104dSAndroid Build Coastguard Worker 
4662*7304104dSAndroid Build Coastguard Worker   vector<pthread_t> all_threads;
4663*7304104dSAndroid Build Coastguard Worker 
4664*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4665*7304104dSAndroid Build Coastguard Worker     {
4666*7304104dSAndroid Build Coastguard Worker       pthread_t pt;
4667*7304104dSAndroid Build Coastguard Worker       rc = pthread_create (& pt, NULL, thread_main_groom, NULL);
4668*7304104dSAndroid Build Coastguard Worker       if (rc)
4669*7304104dSAndroid Build Coastguard Worker         error (EXIT_FAILURE, rc, "cannot spawn thread to groom database\n");
4670*7304104dSAndroid Build Coastguard Worker       else
4671*7304104dSAndroid Build Coastguard Worker         {
4672*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_PTHREAD_SETNAME_NP
4673*7304104dSAndroid Build Coastguard Worker           (void) pthread_setname_np (pt, "groom");
4674*7304104dSAndroid Build Coastguard Worker #endif
4675*7304104dSAndroid Build Coastguard Worker           all_threads.push_back(pt);
4676*7304104dSAndroid Build Coastguard Worker         }
4677*7304104dSAndroid Build Coastguard Worker 
4678*7304104dSAndroid Build Coastguard Worker       if (scan_files || scan_archives.size() > 0)
4679*7304104dSAndroid Build Coastguard Worker         {
4680*7304104dSAndroid Build Coastguard Worker           if (scan_checkpoint > 0)
4681*7304104dSAndroid Build Coastguard Worker             scan_barrier = new sqlite_checkpoint_pb(concurrency, (unsigned) scan_checkpoint);
4682*7304104dSAndroid Build Coastguard Worker 
4683*7304104dSAndroid Build Coastguard Worker           rc = pthread_create (& pt, NULL, thread_main_fts_source_paths, NULL);
4684*7304104dSAndroid Build Coastguard Worker           if (rc)
4685*7304104dSAndroid Build Coastguard Worker             error (EXIT_FAILURE, rc, "cannot spawn thread to traverse source paths\n");
4686*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_PTHREAD_SETNAME_NP
4687*7304104dSAndroid Build Coastguard Worker           (void) pthread_setname_np (pt, "traverse");
4688*7304104dSAndroid Build Coastguard Worker #endif
4689*7304104dSAndroid Build Coastguard Worker           all_threads.push_back(pt);
4690*7304104dSAndroid Build Coastguard Worker 
4691*7304104dSAndroid Build Coastguard Worker           for (unsigned i=0; i<concurrency; i++)
4692*7304104dSAndroid Build Coastguard Worker             {
4693*7304104dSAndroid Build Coastguard Worker               rc = pthread_create (& pt, NULL, thread_main_scanner, NULL);
4694*7304104dSAndroid Build Coastguard Worker               if (rc)
4695*7304104dSAndroid Build Coastguard Worker                 error (EXIT_FAILURE, rc, "cannot spawn thread to scan source files / archives\n");
4696*7304104dSAndroid Build Coastguard Worker #ifdef HAVE_PTHREAD_SETNAME_NP
4697*7304104dSAndroid Build Coastguard Worker               (void) pthread_setname_np (pt, "scan");
4698*7304104dSAndroid Build Coastguard Worker #endif
4699*7304104dSAndroid Build Coastguard Worker               all_threads.push_back(pt);
4700*7304104dSAndroid Build Coastguard Worker             }
4701*7304104dSAndroid Build Coastguard Worker         }
4702*7304104dSAndroid Build Coastguard Worker     }
4703*7304104dSAndroid Build Coastguard Worker 
4704*7304104dSAndroid Build Coastguard Worker   /* Trivial main loop! */
4705*7304104dSAndroid Build Coastguard Worker   set_metric("ready", 1);
4706*7304104dSAndroid Build Coastguard Worker   while (! interrupted)
4707*7304104dSAndroid Build Coastguard Worker     pause ();
4708*7304104dSAndroid Build Coastguard Worker   scanq.nuke(); // wake up any remaining scanq-related threads, let them die
4709*7304104dSAndroid Build Coastguard Worker   if (scan_barrier) scan_barrier->nuke(); // ... in case they're stuck in a barrier
4710*7304104dSAndroid Build Coastguard Worker   set_metric("ready", 0);
4711*7304104dSAndroid Build Coastguard Worker 
4712*7304104dSAndroid Build Coastguard Worker   if (verbose)
4713*7304104dSAndroid Build Coastguard Worker     obatched(clog) << "stopping" << endl;
4714*7304104dSAndroid Build Coastguard Worker 
4715*7304104dSAndroid Build Coastguard Worker   /* Join all our threads. */
4716*7304104dSAndroid Build Coastguard Worker   for (auto&& it : all_threads)
4717*7304104dSAndroid Build Coastguard Worker     pthread_join (it, NULL);
4718*7304104dSAndroid Build Coastguard Worker 
4719*7304104dSAndroid Build Coastguard Worker   /* Stop all the web service threads. */
4720*7304104dSAndroid Build Coastguard Worker   if (d46) MHD_stop_daemon (d46);
4721*7304104dSAndroid Build Coastguard Worker   if (d4) MHD_stop_daemon (d4);
4722*7304104dSAndroid Build Coastguard Worker 
4723*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4724*7304104dSAndroid Build Coastguard Worker     {
4725*7304104dSAndroid Build Coastguard Worker       /* With all threads known dead, we can clean up the global resources. */
4726*7304104dSAndroid Build Coastguard Worker       rc = sqlite3_exec (db, DEBUGINFOD_SQLITE_CLEANUP_DDL, NULL, NULL, NULL);
4727*7304104dSAndroid Build Coastguard Worker       if (rc != SQLITE_OK)
4728*7304104dSAndroid Build Coastguard Worker         {
4729*7304104dSAndroid Build Coastguard Worker           error (0, 0,
4730*7304104dSAndroid Build Coastguard Worker                  "warning: cannot run database cleanup ddl: %s", sqlite3_errmsg(db));
4731*7304104dSAndroid Build Coastguard Worker         }
4732*7304104dSAndroid Build Coastguard Worker     }
4733*7304104dSAndroid Build Coastguard Worker 
4734*7304104dSAndroid Build Coastguard Worker   debuginfod_pool_groom ();
4735*7304104dSAndroid Build Coastguard Worker   delete scan_barrier;
4736*7304104dSAndroid Build Coastguard Worker 
4737*7304104dSAndroid Build Coastguard Worker   // NB: no problem with unconditional free here - an earlier failed regcomp would exit program
4738*7304104dSAndroid Build Coastguard Worker   (void) regfree (& file_include_regex);
4739*7304104dSAndroid Build Coastguard Worker   (void) regfree (& file_exclude_regex);
4740*7304104dSAndroid Build Coastguard Worker 
4741*7304104dSAndroid Build Coastguard Worker   sqlite3 *database = db;
4742*7304104dSAndroid Build Coastguard Worker   sqlite3 *databaseq = dbq;
4743*7304104dSAndroid Build Coastguard Worker   db = dbq = 0; // for signal_handler not to freak
4744*7304104dSAndroid Build Coastguard Worker   (void) sqlite3_close (databaseq);
4745*7304104dSAndroid Build Coastguard Worker   if (! passive_p)
4746*7304104dSAndroid Build Coastguard Worker     (void) sqlite3_close (database);
4747*7304104dSAndroid Build Coastguard Worker 
4748*7304104dSAndroid Build Coastguard Worker   return 0;
4749*7304104dSAndroid Build Coastguard Worker }
4750