]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/cats/sql_get.c
First cut of restore
[bacula/bacula] / bacula / src / cats / sql_get.c
1 /*
2  * Bacula Catalog Database Get record interface routines
3  *  Note, these routines generally get a record by id or
4  *        by name.  If more logic is involved, the routine
5  *        should be in find.c 
6  *
7  *    Kern Sibbald, March 2000
8  *
9  *    Version $Id$
10  */
11
12 /*
13    Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
14
15    This program is free software; you can redistribute it and/or
16    modify it under the terms of the GNU General Public License as
17    published by the Free Software Foundation; either version 2 of
18    the License, or (at your option) any later version.
19
20    This program is distributed in the hope that it will be useful,
21    but WITHOUT ANY WARRANTY; without even the implied warranty of
22    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
23    General Public License for more details.
24
25    You should have received a copy of the GNU General Public
26    License along with this program; if not, write to the Free
27    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
28    MA 02111-1307, USA.
29
30  */
31
32 /* *****FIXME**** fix fixed length of select_cmd[] and insert_cmd[] */
33
34 /* The following is necessary so that we do not include
35  * the dummy external definition of DB.
36  */
37 #define __SQL_C                       /* indicate that this is sql.c */
38
39 #include "bacula.h"
40 #include "cats.h"
41
42 #if    HAVE_MYSQL || HAVE_SQLITE
43
44 /* -----------------------------------------------------------------------
45  *
46  *   Generic Routines (or almost generic)
47  *
48  * -----------------------------------------------------------------------
49  */
50
51 /* Forward referenced functions */
52 static int db_get_file_record(B_DB *mdb, FILE_DBR *fdbr);
53 static int db_get_filename_record(B_DB *mdb, char *fname);
54 static int db_get_path_record(B_DB *mdb, char *path);
55
56
57 /* Imported subroutines */
58 extern void print_result(B_DB *mdb);
59 extern int QueryDB(char *file, int line, B_DB *db, char *select_cmd);
60
61
62 /*
63  * Given a full filename (with path), look up the File record
64  * (with attributes) in the database.
65  *
66  *  Returns: 0 on failure
67  *           1 on success with the File record in FILE_DBR
68  */
69 int db_get_file_attributes_record(B_DB *mdb, char *fname, FILE_DBR *fdbr)
70 {
71    int fnl, pnl;
72    char *l, *p;
73    uint64_t id;
74    char file[MAXSTRING];
75    char spath[MAXSTRING];
76    char buf[MAXSTRING];
77    Dmsg1(20, "Enter get_file_from_catalog fname=%s \n", fname);
78
79    /* Find path without the filename.  
80     * I.e. everything after the last / is a "filename".
81     * OK, maybe it is a directory name, but we treat it like
82     * a filename. If we don't find a / then the whole name
83     * must be a path name (e.g. c:).
84     */
85    for (p=l=fname; *p; p++) {
86       if (*p == '/') {
87          l = p;
88       }
89    }
90    if (*l == '/') {                   /* did we find a slash? */
91       l++;                            /* yes, point to filename */
92    } else {                           /* no, whole thing must be path name */
93       l = p;
94    }
95
96    /* If filename doesn't exist (i.e. root directory), we
97     * simply create a blank name consisting of a single 
98     * space. This makes handling zero length filenames
99     * easier.
100     */
101    fnl = p - l;
102    if (fnl > 255) {
103       Emsg1(M_WARNING, 0, _("Filename truncated to 255 chars: %s\n"), l);
104       fnl = 255;
105    }
106    if (fnl > 0) {
107       strncpy(file, l, fnl);          /* copy filename */
108       file[fnl] = 0;
109    } else {
110       file[0] = ' ';                  /* blank filename */
111       file[1] = 0;
112       fnl = 1;
113    }
114
115    pnl = l - fname;    
116    if (pnl > 255) {
117       Emsg1(M_WARNING, 0, _("Path name truncated to 255 chars: %s\n"), fname);
118       pnl = 255;
119    }
120    strncpy(spath, fname, pnl);
121    spath[pnl] = 0;
122
123    if (pnl == 0) {
124       Mmsg1(&mdb->errmsg, _("Path length is zero. File=%s\n"), fname);
125       Emsg0(M_ERROR, 0, mdb->errmsg);
126       spath[0] = ' ';
127       spath[1] = 0;
128       pnl = 1;
129    }
130
131    Dmsg1(100, "spath=%s\n", spath);
132    Dmsg1(100, "file=%s\n", file);
133
134    db_escape_string(buf, file, fnl);
135    fdbr->FilenameId = db_get_filename_record(mdb, buf);
136    Dmsg1(50, "db_get_filename_record FilenameId=%d\n", fdbr->FilenameId);
137
138    db_escape_string(buf, spath, pnl);
139    fdbr->PathId = db_get_path_record(mdb, buf);
140    Dmsg2(50, "db_get_path_record PathId=%d path=%s\n", fdbr->PathId, buf);
141
142    id = db_get_file_record(mdb, fdbr);
143
144    return id;
145 }
146
147  
148
149 /* Get a File record   
150  * Returns: 0 on failure
151  *          1 on success
152  */
153 static
154 int db_get_file_record(B_DB *mdb, FILE_DBR *fdbr)
155 {
156    SQL_ROW row;
157    int stat = 0;
158
159    db_lock(mdb);
160    Mmsg(&mdb->cmd, 
161 "SELECT FileId, LStat, MD5 from File where File.JobId=%d and File.PathId=%d and \
162 File.FilenameId=%d", fdbr->JobId, fdbr->PathId, fdbr->FilenameId);
163
164    Dmsg3(050, "Get_file_record JobId=%d FilenameId=%d PathId=%d\n",
165       fdbr->JobId, fdbr->FilenameId, fdbr->PathId);
166       
167    Dmsg1(100, "Query=%s\n", mdb->cmd);
168
169    if (QUERY_DB(mdb, mdb->cmd)) {
170
171       mdb->num_rows = sql_num_rows(mdb);
172       Dmsg1(050, "get_file_record num_rows=%d\n", (int)mdb->num_rows);
173
174       /* 
175        * Note, we can find more than one File record with the same
176        *  filename if the file is linked.   ????????
177        */
178       if (mdb->num_rows > 1) {
179          Emsg1(M_WARNING, 0, _("get_file_record want 1 got rows=%d\n"), mdb->num_rows);
180          Emsg1(M_ERROR, 0, "%s", mdb->cmd);
181       }
182       if (mdb->num_rows >= 1) {
183          if ((row = sql_fetch_row(mdb)) == NULL) {
184             Mmsg1(&mdb->errmsg, "Error fetching row: %s\n", sql_strerror(mdb));
185          } else {
186             fdbr->FileId = (FileId_t)strtod(row[0], NULL);
187             strncpy(fdbr->LStat, row[1], sizeof(fdbr->LStat));
188             fdbr->LStat[sizeof(fdbr->LStat)] = 0;
189             strncpy(fdbr->MD5, row[2], sizeof(fdbr->MD5));
190             fdbr->MD5[sizeof(fdbr->MD5)] = 0;
191             stat = 1;
192          }
193       }
194       sql_free_result(mdb);
195    }
196    db_unlock(mdb);
197    return stat;
198
199 }
200
201 /* Get Filename record   
202  * Returns: 0 on failure
203  *          FilenameId on success
204  */
205 static int db_get_filename_record(B_DB *mdb, char *fname) 
206 {
207    SQL_ROW row;
208    int FilenameId = 0;
209
210    if (*fname == 0) {
211       Mmsg0(&mdb->errmsg, _("Null name given to db_get_filename_record\n"));
212       Emsg0(M_ABORT, 0, mdb->errmsg);
213    }
214
215    db_lock(mdb);
216    Mmsg(&mdb->cmd, "SELECT FilenameId FROM Filename WHERE Name='%s'", fname);
217    if (QUERY_DB(mdb, mdb->cmd)) {
218
219       mdb->num_rows = sql_num_rows(mdb);
220
221       if (mdb->num_rows > 1) {
222          Mmsg1(&mdb->errmsg, _("More than one Filename!: %d\n"), (int)(mdb->num_rows));
223          Emsg1(M_WARNING, 0, _("get_filename_record want 1 got rows=%d\n"), mdb->num_rows);
224          Emsg1(M_ERROR, 0, "%s", mdb->errmsg);
225       }
226       if (mdb->num_rows >= 1) {
227          if ((row = sql_fetch_row(mdb)) == NULL) {
228             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
229             Emsg1(M_ERROR, 0, "%s", mdb->errmsg);
230          } else {
231             FilenameId = atoi(row[0]);
232             if (FilenameId <= 0) {
233                Mmsg2(&mdb->errmsg, _("Get DB Filename record %s found bad record: %d\n"),
234                   mdb->cmd, FilenameId); 
235                Emsg1(M_ERROR, 0, "%s", mdb->errmsg);
236                FilenameId = 0;
237             }
238          }
239       }
240       sql_free_result(mdb);
241    }
242    db_unlock(mdb);
243    return FilenameId;
244 }
245
246 /* Get path record   
247  * Returns: 0 on failure
248  *          PathId on success
249  */
250 static int db_get_path_record(B_DB *mdb, char *path)
251 {
252    SQL_ROW row;
253    uint32_t PathId = 0;
254
255    if (*path == 0) {
256       Emsg0(M_ABORT, 0, _("Null path given to db_get_path_record\n"));
257    }
258
259    db_lock(mdb);
260
261    if (mdb->cached_path_id != 0 && strcmp(mdb->cached_path, path) == 0) {
262       db_unlock(mdb);
263       return mdb->cached_path_id;
264    }          
265
266    Mmsg(&mdb->cmd, "SELECT PathId FROM Path WHERE Path='%s'", path);
267
268    if (QUERY_DB(mdb, mdb->cmd)) {
269       char ed1[30];
270       mdb->num_rows = sql_num_rows(mdb);
271
272       if (mdb->num_rows > 1) {
273          Mmsg1(&mdb->errmsg, _("More than one Path!: %s\n"), 
274             edit_uint64(mdb->num_rows, ed1));
275       } else if (mdb->num_rows == 1) {
276          if ((row = sql_fetch_row(mdb)) == NULL) {
277             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
278          } else {
279             PathId = atoi(row[0]);
280             if (PathId <= 0) {
281                Mmsg2(&mdb->errmsg, _("Get DB path record %s found bad record: %d\n"),
282                   mdb->cmd, PathId); 
283                PathId = 0;
284             } else {
285                /* Cache path */
286                if (PathId != mdb->cached_path_id) {
287                   mdb->cached_path_id = PathId;
288                   mdb->cached_path = check_pool_memory_size(mdb->cached_path,
289                      strlen(path)+1);
290                   strcpy(mdb->cached_path, path);
291                }
292             }
293          }
294       }
295       sql_free_result(mdb);
296    }
297    db_unlock(mdb);
298    return PathId;
299 }
300
301
302 /* 
303  * Get Job record for given JobId or Job name
304  * Returns: 0 on failure
305  *          1 on success
306  */
307 int db_get_job_record(B_DB *mdb, JOB_DBR *jr)
308 {
309    SQL_ROW row;
310
311    db_lock(mdb);
312    if (jr->JobId == 0) {
313       Mmsg(&mdb->cmd, "SELECT VolSessionId, VolSessionTime, \
314 PoolId, StartTime, EndTime, JobFiles, JobBytes, JobTDate, Job, JobStatus \
315 FROM Job WHERE Job='%s'", jr->Job);
316     } else {
317       Mmsg(&mdb->cmd, "SELECT VolSessionId, VolSessionTime, \
318 PoolId, StartTime, EndTime, JobFiles, JobBytes, JobTDate, Job, JobStatus \
319 FROM Job WHERE JobId=%d", jr->JobId);
320     }
321
322    if (!QUERY_DB(mdb, mdb->cmd)) {
323       db_unlock(mdb);
324       return 0;                       /* failed */
325    }
326    if ((row = sql_fetch_row(mdb)) == NULL) {
327       Mmsg1(&mdb->errmsg, _("No Job found for JobId %d\n"), jr->JobId);
328       sql_free_result(mdb);
329       db_unlock(mdb);
330       return 0;                       /* failed */
331    }
332
333    jr->VolSessionId = atol(row[0]);
334    jr->VolSessionTime = atol(row[1]);
335    jr->PoolId = atoi(row[2]);
336    strcpy(jr->cStartTime, row[3]);
337    strcpy(jr->cEndTime, row[4]);
338    jr->JobFiles = atol(row[5]);
339    jr->JobBytes = (uint64_t)strtod(row[6], NULL);
340    jr->JobTDate = (btime_t)strtod(row[7], NULL);
341    strcpy(jr->Job, row[8]);
342    jr->JobStatus = (int)*row[9];
343    sql_free_result(mdb);
344
345    db_unlock(mdb);
346    return 1;
347 }
348
349 /*
350  * Find VolumeNames for a give JobId
351  *  Returns: 0 on error or no Volumes found
352  *           number of volumes on success
353  *              Volumes are concatenated in VolumeNames
354  *              separated by a vertical bar (|).
355  */
356 int db_get_job_volume_names(B_DB *mdb, uint32_t JobId, char *VolumeNames)
357 {
358    SQL_ROW row;
359    int stat = 0;
360    int i;
361
362    db_lock(mdb);
363    Mmsg(&mdb->cmd, 
364 "SELECT VolumeName FROM JobMedia,Media WHERE JobMedia.JobId=%d \
365 AND JobMedia.MediaId=Media.MediaId", JobId);
366
367    Dmsg1(130, "VolNam=%s\n", mdb->cmd);
368    VolumeNames[0] = 0;
369    if (QUERY_DB(mdb, mdb->cmd)) {
370       mdb->num_rows = sql_num_rows(mdb);
371       Dmsg1(130, "Num rows=%d\n", mdb->num_rows);
372       if (mdb->num_rows <= 0) {
373          Mmsg1(&mdb->errmsg, _("No volumes found for JobId=%d"), JobId);
374          stat = 0;
375       } else {
376          stat = mdb->num_rows;
377          for (i=0; i < stat; i++) {
378             if ((row = sql_fetch_row(mdb)) == NULL) {
379                Mmsg2(&mdb->errmsg, _("Error fetching row %d: ERR=%s\n"), i, sql_strerror(mdb));
380                stat = 0;
381                break;
382             } else {
383                if (VolumeNames[0] != 0) {
384                   strcat(VolumeNames, "|");
385                }
386                strcat(VolumeNames, row[0]);
387             }
388          }
389       }
390       sql_free_result(mdb);
391    }
392    db_unlock(mdb);
393    return stat;
394 }
395
396
397 /* 
398  * Get the number of pool records
399  *
400  * Returns: -1 on failure
401  *          number on success
402  */
403 int db_get_num_pool_records(B_DB *mdb)
404 {
405    int stat = 0;
406
407    db_lock(mdb);
408    Mmsg(&mdb->cmd, "SELECT count(*) from Pool");
409    stat = get_sql_record_max(mdb);
410    db_unlock(mdb);
411    return stat;
412 }
413
414 /*
415  * This function returns a list of all the Pool record ids.
416  *  The caller must free ids if non-NULL.
417  *
418  *  Returns 0: on failure
419  *          1: on success
420  */
421 int db_get_pool_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
422 {
423    SQL_ROW row;
424    int stat = 0;
425    int i = 0;
426    uint32_t *id;
427
428    db_lock(mdb);
429    *ids = NULL;
430    Mmsg(&mdb->cmd, "SELECT PoolId FROM Pool");
431    if (QUERY_DB(mdb, mdb->cmd)) {
432       *num_ids = sql_num_rows(mdb);
433       if (*num_ids > 0) {
434          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
435          while ((row = sql_fetch_row(mdb)) != NULL) {
436             id[i++] = (uint32_t)atoi(row[0]);
437          }
438          *ids = id;
439       }
440       sql_free_result(mdb);
441       stat = 1;
442    } else {
443       Mmsg(&mdb->errmsg, _("Pool id select failed: ERR=%s\n"), sql_strerror(mdb));
444       stat = 0;
445    }
446    db_unlock(mdb);
447    return stat;
448 }
449
450
451 /* Get Pool Record   
452  * If the PoolId is non-zero, we get its record,
453  *  otherwise, we search on the PoolName
454  *
455  * Returns: 0 on failure
456  *          id on success 
457  */
458 int db_get_pool_record(B_DB *mdb, POOL_DBR *pdbr)
459 {
460    SQL_ROW row;
461    int stat = 0;
462
463    db_lock(mdb);
464    if (pdbr->PoolId != 0) {               /* find by id */
465       Mmsg(&mdb->cmd, 
466 "SELECT PoolId, Name, NumVols, MaxVols, UseOnce, UseCatalog, AcceptAnyVolume, \
467 AutoPrune, Recycle, VolRetention, \
468 PoolType, LabelFormat FROM Pool WHERE Pool.PoolId=%d", pdbr->PoolId);
469    } else {                           /* find by name */
470       Mmsg(&mdb->cmd, 
471 "SELECT PoolId, Name, NumVols, MaxVols, UseOnce, UseCatalog, AcceptAnyVolume, \
472 AutoPrune, Recycle, VolRetention, \
473 PoolType, LabelFormat FROM Pool WHERE Pool.Name='%s'", pdbr->Name);
474    }  
475
476    if (QUERY_DB(mdb, mdb->cmd)) {
477       mdb->num_rows = sql_num_rows(mdb);
478       if (mdb->num_rows > 1) {
479          char ed1[30];
480          Mmsg1(&mdb->errmsg, _("More than one Pool!: %s\n"), 
481             edit_uint64(mdb->num_rows, ed1));
482       } else if (mdb->num_rows == 1) {
483          if ((row = sql_fetch_row(mdb)) == NULL) {
484             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
485          } else {
486             pdbr->PoolId = atoi(row[0]);
487             strcpy(pdbr->Name, row[1]);
488             pdbr->NumVols = atoi(row[2]);
489             pdbr->MaxVols = atoi(row[3]);
490             pdbr->UseOnce = atoi(row[4]);
491             pdbr->UseCatalog = atoi(row[5]);
492             pdbr->AcceptAnyVolume = atoi(row[6]);
493             pdbr->AutoPrune = atoi(row[7]);
494             pdbr->Recycle = atoi(row[8]);
495             pdbr->VolRetention = (btime_t)strtod(row[9], NULL);
496             strcpy(pdbr->PoolType, row[10]);
497             if (row[11]) {
498                strcpy(pdbr->LabelFormat, row[11]);
499             } else {
500                pdbr->LabelFormat[0] = 0;
501             }
502             stat = pdbr->PoolId;
503          }
504       }
505       sql_free_result(mdb);
506    }
507    db_unlock(mdb);
508    return stat;
509 }
510
511
512 /* 
513  * Get the number of Media records
514  *
515  * Returns: -1 on failure
516  *          number on success
517  */
518 int db_get_num_media_records(B_DB *mdb)
519 {
520    int stat = 0;
521
522    db_lock(mdb);
523    Mmsg(&mdb->cmd, "SELECT count(*) from Media");
524    stat = get_sql_record_max(mdb);
525    db_unlock(mdb);
526    return stat;
527 }
528
529
530 /*
531  * This function returns a list of all the Media record ids.
532  *  The caller must free ids if non-NULL.
533  *
534  *  Returns 0: on failure
535  *          1: on success
536  */
537 int db_get_media_ids(B_DB *mdb, int *num_ids, uint32_t *ids[])
538 {
539    SQL_ROW row;
540    int stat = 0;
541    int i = 0;
542    uint32_t *id;
543
544    db_lock(mdb);
545    *ids = NULL;
546    Mmsg(&mdb->cmd, "SELECT MediaId FROM Media");
547    if (QUERY_DB(mdb, mdb->cmd)) {
548       *num_ids = sql_num_rows(mdb);
549       if (*num_ids > 0) {
550          id = (uint32_t *)malloc(*num_ids * sizeof(uint32_t));
551          while ((row = sql_fetch_row(mdb)) != NULL) {
552             id[i++] = (uint32_t)atoi(row[0]);
553          }
554          *ids = id;
555       }
556       sql_free_result(mdb);
557       stat = 1;
558    } else {
559       Mmsg(&mdb->errmsg, _("Media id select failed: ERR=%s\n"), sql_strerror(mdb));
560       stat = 0;
561    }
562    db_unlock(mdb);
563    return stat;
564 }
565
566
567 /* Get Media Record   
568  *
569  * Returns: 0 on failure
570  *          id on success 
571  */
572 int db_get_media_record(B_DB *mdb, MEDIA_DBR *mr)
573 {
574    SQL_ROW row;
575    int stat = 0;
576
577    db_lock(mdb);
578    if (mr->MediaId == 0 && mr->VolumeName[0] == 0) {
579       Mmsg(&mdb->cmd, "SELECT count(*) from Media");
580       mr->MediaId = get_sql_record_max(mdb);
581       db_unlock(mdb);
582       return 1;
583    }
584    if (mr->MediaId != 0) {               /* find by id */
585       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
586 VolBytes,VolMounts,VolErrors,VolWrites,VolMaxBytes,VolCapacityBytes,\
587 MediaType,VolStatus,PoolId,VolRetention,Recycle,Slot \
588 FROM Media WHERE MediaId=%d", mr->MediaId);
589    } else {                           /* find by name */
590       Mmsg(&mdb->cmd, "SELECT MediaId,VolumeName,VolJobs,VolFiles,VolBlocks,\
591 VolBytes,VolMounts,VolErrors,VolWrites,VolMaxBytes,VolCapacityBytes,\
592 MediaType,VolStatus,PoolId,VolRetention,Recycle,Slot \
593 FROM Media WHERE VolumeName='%s'", mr->VolumeName);
594    }  
595
596    if (QUERY_DB(mdb, mdb->cmd)) {
597       mdb->num_rows = sql_num_rows(mdb);
598       if (mdb->num_rows > 1) {
599          char ed1[30];
600          Mmsg1(&mdb->errmsg, _("More than one Volume!: %s\n"), 
601             edit_uint64(mdb->num_rows, ed1));
602       } else if (mdb->num_rows == 1) {
603          if ((row = sql_fetch_row(mdb)) == NULL) {
604             Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
605          } else {
606             /* return values */
607             mr->MediaId = atoi(row[0]);
608             strcpy(mr->VolumeName, row[1]);
609             mr->VolJobs = atoi(row[2]);
610             mr->VolFiles = atoi(row[3]);
611             mr->VolBlocks = atoi(row[4]);
612             mr->VolBytes = (uint64_t)strtod(row[5], NULL);
613             mr->VolMounts = atoi(row[6]);
614             mr->VolErrors = atoi(row[7]);
615             mr->VolWrites = atoi(row[8]);
616             mr->VolMaxBytes = (uint64_t)strtod(row[9], NULL);
617             mr->VolCapacityBytes = (uint64_t)strtod(row[10], NULL);
618             strcpy(mr->MediaType, row[11]);
619             strcpy(mr->VolStatus, row[12]);
620             mr->PoolId = atoi(row[13]);
621             mr->VolRetention = (btime_t)strtod(row[14], NULL);
622             mr->Recycle = atoi(row[15]);
623             mr->Slot = atoi(row[16]);
624             stat = mr->MediaId;
625          }
626       } else {
627          Mmsg0(&mdb->errmsg, _("Media record not found.\n"));
628       }
629       sql_free_result(mdb);
630    }
631    db_unlock(mdb);
632    return stat;
633 }
634
635
636 #endif /* HAVE_MYSQL || HAVE_SQLITE */