xref: /dragonfly/sys/kern/kern_linker.c (revision 2b3f93ea6d1f70880f3e87f3c2cbe0dc0bfc9332)
1 /*-
2  * Copyright (c) 1997 Doug Rabson
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD: src/sys/kern/kern_linker.c,v 1.41.2.3 2001/11/21 17:50:35 luigi Exp $
27  */
28 
29 #include "opt_ddb.h"
30 
31 #include <sys/param.h>
32 #include <sys/kernel.h>
33 #include <sys/systm.h>
34 #include <sys/malloc.h>
35 #include <sys/sysmsg.h>
36 #include <sys/sysent.h>
37 #include <sys/proc.h>
38 #include <sys/caps.h>
39 #include <sys/lock.h>
40 #include <sys/module.h>
41 #include <sys/queue.h>
42 #include <sys/linker.h>
43 #include <sys/fcntl.h>
44 #include <sys/libkern.h>
45 #include <sys/nlookup.h>
46 #include <sys/vnode.h>
47 #include <sys/sysctl.h>
48 
49 #include <vm/vm_zone.h>
50 
51 #ifdef _KERNEL_VIRTUAL
52 #include <dlfcn.h>
53 #endif
54 
55 #ifdef KLD_DEBUG
56 int kld_debug = 1;
57 #endif
58 
59 /* Metadata from the static kernel */
60 SET_DECLARE(modmetadata_set, struct mod_metadata);
61 MALLOC_DEFINE(M_LINKER, "kld", "kernel linker");
62 
63 linker_file_t linker_current_file;
64 linker_file_t linker_kernel_file;
65 
66 static struct lock llf_lock;  /* lock for the file list */
67 static struct lock kld_lock;
68 static linker_class_list_t classes;
69 static linker_file_list_t linker_files;
70 static int next_file_id = 1;
71 
72 /* XXX wrong name; we're looking at version provision tags here, not modules */
73 typedef TAILQ_HEAD(, modlist) modlisthead_t;
74 struct modlist {
75           TAILQ_ENTRY(modlist) link;    /* chain together all modules */
76           linker_file_t   container;
77           const char          *name;
78           int             version;
79 };
80 typedef struct modlist *modlist_t;
81 static modlisthead_t found_modules;
82 
83 
84 static int linker_load_module(const char *kldname, const char *modname,
85                                     struct linker_file *parent, struct mod_depend *verinfo,
86                                     struct linker_file **lfpp);
87 
88 static char *
linker_strdup(const char * str)89 linker_strdup(const char *str)
90 {
91     char  *result;
92 
93     result = kmalloc(strlen(str) + 1, M_LINKER, M_WAITOK);
94     strcpy(result, str);
95     return(result);
96 }
97 
98 static const char *
linker_basename(const char * path)99 linker_basename(const char *path)
100 {
101     const char      *filename;
102 
103     filename = strrchr(path, '/');
104     if (filename == NULL)
105           return path;
106     if (filename[1])
107           filename++;
108     return (filename);
109 }
110 
111 static void
linker_init(void * arg)112 linker_init(void* arg)
113 {
114     lockinit(&llf_lock, "klink", 0, 0);
115     lockinit(&kld_lock, "kldlk", 0, LK_CANRECURSE);
116     TAILQ_INIT(&classes);
117     TAILQ_INIT(&linker_files);
118 }
119 
120 SYSINIT(linker, SI_BOOT2_KLD, SI_ORDER_FIRST, linker_init, 0);
121 
122 int
linker_add_class(const char * desc,void * priv,struct linker_class_ops * ops)123 linker_add_class(const char* desc, void* priv,
124                      struct linker_class_ops* ops)
125 {
126     linker_class_t lc;
127 
128     lc = kmalloc(sizeof(struct linker_class), M_LINKER, M_NOWAIT | M_ZERO);
129     if (!lc)
130           return ENOMEM;
131 
132     lc->desc = desc;
133     lc->priv = priv;
134     lc->ops = ops;
135     TAILQ_INSERT_HEAD(&classes, lc, link);
136 
137     return 0;
138 }
139 
140 static void
linker_file_sysinit(linker_file_t lf)141 linker_file_sysinit(linker_file_t lf)
142 {
143     struct sysinit** start, ** stop;
144     struct sysinit** sipp;
145     struct sysinit** xipp;
146     struct sysinit* save;
147 
148     KLD_DPF(FILE, ("linker_file_sysinit: calling SYSINITs for %s\n",
149                        lf->filename));
150 
151     if (linker_file_lookup_set(lf, "sysinit_set", &start, &stop, NULL) != 0)
152           return;
153 
154     /*
155      * Perform a bubble sort of the system initialization objects by
156      * their subsystem (primary key) and order (secondary key).
157      *
158      * Since some things care about execution order, this is the
159      * operation which ensures continued function.
160      */
161     for (sipp = start; sipp < stop; sipp++) {
162           for (xipp = sipp + 1; xipp < stop; xipp++) {
163               if ((*sipp)->subsystem < (*xipp)->subsystem ||
164                      ((*sipp)->subsystem == (*xipp)->subsystem &&
165                       (*sipp)->order <= (*xipp)->order))
166                     continue; /* skip*/
167               save = *sipp;
168               *sipp = *xipp;
169               *xipp = save;
170           }
171     }
172 
173 
174     /*
175      * Traverse the (now) ordered list of system initialization tasks.
176      * Perform each task, and continue on to the next task.
177      */
178     for (sipp = start; sipp < stop; sipp++) {
179           if ((*sipp)->subsystem == SI_SPECIAL_DUMMY)
180               continue;       /* skip dummy task(s)*/
181 
182           /* Call function */
183           (*((*sipp)->func))((*sipp)->udata);
184     }
185 }
186 
187 static void
linker_file_sysuninit(linker_file_t lf)188 linker_file_sysuninit(linker_file_t lf)
189 {
190     struct sysinit** start, ** stop;
191     struct sysinit** sipp;
192     struct sysinit** xipp;
193     struct sysinit* save;
194 
195     KLD_DPF(FILE, ("linker_file_sysuninit: calling SYSUNINITs for %s\n",
196                        lf->filename));
197 
198     if (linker_file_lookup_set(lf, "sysuninit_set", &start, &stop, NULL) != 0)
199           return;
200 
201     /*
202      * Perform a reverse bubble sort of the system initialization objects
203      * by their subsystem (primary key) and order (secondary key).
204      *
205      * Since some things care about execution order, this is the
206      * operation which ensures continued function.
207      */
208     for (sipp = start; sipp < stop; sipp++) {
209           for (xipp = sipp + 1; xipp < stop; xipp++) {
210               if ((*sipp)->subsystem > (*xipp)->subsystem ||
211                      ((*sipp)->subsystem == (*xipp)->subsystem &&
212                       (*sipp)->order >= (*xipp)->order))
213                     continue; /* skip*/
214               save = *sipp;
215               *sipp = *xipp;
216               *xipp = save;
217           }
218     }
219 
220 
221     /*
222      * Traverse the (now) ordered list of system initialization tasks.
223      * Perform each task, and continue on to the next task.
224      */
225     for (sipp = start; sipp < stop; sipp++) {
226           if ((*sipp)->subsystem == SI_SPECIAL_DUMMY)
227               continue;       /* skip dummy task(s)*/
228 
229           /* Call function */
230           (*((*sipp)->func))((*sipp)->udata);
231     }
232 }
233 
234 static void
linker_file_register_sysctls(linker_file_t lf)235 linker_file_register_sysctls(linker_file_t lf)
236 {
237     struct sysctl_oid **start, **stop, **oidp;
238 
239     KLD_DPF(FILE, ("linker_file_register_sysctls: registering SYSCTLs for %s\n",
240                        lf->filename));
241 
242     if (linker_file_lookup_set(lf, "sysctl_set", &start, &stop, NULL) != 0)
243           return;
244     for (oidp = start; oidp < stop; oidp++)
245           sysctl_register_oid(*oidp);
246 }
247 
248 static void
linker_file_unregister_sysctls(linker_file_t lf)249 linker_file_unregister_sysctls(linker_file_t lf)
250 {
251     struct sysctl_oid **start, **stop, **oidp;
252 
253     KLD_DPF(FILE, ("linker_file_unregister_sysctls: registering SYSCTLs for %s\n",
254                        lf->filename));
255 
256     if (linker_file_lookup_set(lf, "sysctl_set", &start, &stop, NULL) != 0)
257           return;
258     for (oidp = start; oidp < stop; oidp++)
259           sysctl_unregister_oid(*oidp);
260 }
261 
262 static int
linker_file_register_modules(linker_file_t lf)263 linker_file_register_modules(linker_file_t lf)
264 {
265     struct mod_metadata **start, **stop, **mdp;
266     const moduledata_t *moddata;
267     int                 first_error, error;
268 
269     KLD_DPF(FILE, ("linker_file_register_modules: registering modules in %s\n",
270                        lf->filename));
271 
272     if (linker_file_lookup_set(lf, "modmetadata_set", &start, &stop, NULL) != 0) {
273           /*
274            * This fallback should be unnecessary, but if we get booted
275            * from boot2 instead of loader and we are missing our
276            * metadata then we have to try the best we can.
277            */
278           if (lf == linker_kernel_file) {
279               start = SET_BEGIN(modmetadata_set);
280               stop = SET_LIMIT(modmetadata_set);
281           } else
282               return (0);
283     }
284     first_error = 0;
285     for (mdp = start; mdp < stop; mdp++) {
286           if ((*mdp)->md_type != MDT_MODULE)
287               continue;
288           moddata = (*mdp)->md_data;
289           KLD_DPF(FILE, ("Registering module %s in %s\n", moddata->name, lf->filename));
290           error = module_register(moddata, lf);
291           if (error) {
292               kprintf("Module %s failed to register: %d\n", moddata->name, error);
293               if (first_error == 0)
294                     first_error = error;
295           }
296     }
297     return (first_error);
298 }
299 
300 static void
linker_init_kernel_modules(void)301 linker_init_kernel_modules(void)
302 {
303 
304     linker_file_register_modules(linker_kernel_file);
305 }
306 
307 SYSINIT(linker_kernel, SI_BOOT2_KLD, SI_ORDER_ANY, linker_init_kernel_modules, 0);
308 
309 int
linker_load_file(const char * filename,linker_file_t * result)310 linker_load_file(const char *filename, linker_file_t *result)
311 {
312     linker_class_t lc;
313     linker_file_t lf;
314     int foundfile, error = 0;
315 
316     /* Refuse to load modules if securelevel raised */
317     if (securelevel > 0 || kernel_mem_readonly)
318           return EPERM;
319 
320     lf = linker_find_file_by_name(filename);
321     if (lf) {
322           KLD_DPF(FILE, ("linker_load_file: file %s is already loaded, incrementing refs\n", filename));
323           *result = lf;
324           lf->refs++;
325           goto out;
326     }
327 
328     lf = NULL;
329     foundfile = 0;
330     TAILQ_FOREACH(lc, &classes, link) {
331           KLD_DPF(FILE, ("linker_load_file: trying to load %s as %s\n",
332                            filename, lc->desc));
333 
334           error = lc->ops->load_file(filename, &lf);
335           /*
336            * If we got something other than ENOENT, then it exists but we cannot
337            * load it for some other reason.
338            */
339           if (error != ENOENT)
340               foundfile = 1;
341           if (lf) {
342               error = linker_file_register_modules(lf);
343               if (error == EEXIST) {
344                         linker_file_unload(lf /* , LINKER_UNLOAD_FORCE */);
345                         return (error);
346               }
347               linker_file_register_sysctls(lf);
348               linker_file_sysinit(lf);
349               lf->flags |= LINKER_FILE_LINKED;
350               *result = lf;
351               return (0);
352           }
353     }
354     /*
355      * Less than ideal, but tells the user whether it failed to load or
356      * the module was not found.
357      */
358     if (foundfile) {
359               /*
360                * If the file type has not been recognized by the last try
361                * printout a message before to fail.
362                */
363               if (error == ENOSYS)
364                         kprintf("linker_load_file: Unsupported file type\n");
365 
366               /*
367                * Format not recognized or otherwise unloadable.
368                * When loading a module that is statically built into
369                * the kernel EEXIST percolates back up as the return
370                * value.  Preserve this so that apps can recognize this
371                * special case.
372                */
373               if (error != EEXIST)
374                         error = ENOEXEC;
375     } else {
376           error = ENOENT;               /* Nothing found */
377     }
378 
379 out:
380     return error;
381 }
382 
383 
384 linker_file_t
linker_find_file_by_name(const char * filename)385 linker_find_file_by_name(const char* filename)
386 {
387     linker_file_t lf = NULL;
388     char *koname;
389     int i;
390 
391     for (i = strlen(filename); i > 0 && filename[i-1] != '/'; --i)
392           ;
393     filename += i;
394 
395     koname = kmalloc(strlen(filename) + 4, M_LINKER, M_WAITOK);
396     ksprintf(koname, "%s.ko", filename);
397 
398     lockmgr(&llf_lock, LK_SHARED);
399     TAILQ_FOREACH(lf, &linker_files, link) {
400           if (!strcmp(lf->filename, koname))
401               break;
402           if (!strcmp(lf->filename, filename))
403               break;
404     }
405     lockmgr(&llf_lock, LK_RELEASE);
406 
407     if (koname)
408           kfree(koname, M_LINKER);
409     return lf;
410 }
411 
412 linker_file_t
linker_find_file_by_id(int fileid)413 linker_find_file_by_id(int fileid)
414 {
415     linker_file_t lf = NULL;
416 
417     lockmgr(&llf_lock, LK_SHARED);
418     TAILQ_FOREACH(lf, &linker_files, link) {
419           if (lf->id == fileid)
420               break;
421     }
422     lockmgr(&llf_lock, LK_RELEASE);
423 
424     return lf;
425 }
426 
427 int
linker_file_foreach(linker_predicate_t * predicate,void * context)428 linker_file_foreach(linker_predicate_t *predicate, void *context)
429 {
430     linker_file_t lf;
431     int retval = 0;
432 
433     lockmgr(&llf_lock, LK_SHARED);
434     TAILQ_FOREACH(lf, &linker_files, link) {
435           retval = predicate(lf, context);
436           if (retval != 0)
437               break;
438     }
439     lockmgr(&llf_lock, LK_RELEASE);
440 
441     return (retval);
442 }
443 
444 linker_file_t
linker_make_file(const char * pathname,void * priv,struct linker_file_ops * ops)445 linker_make_file(const char* pathname, void* priv, struct linker_file_ops* ops)
446 {
447     linker_file_t lf = NULL;
448     const char *filename;
449 
450     filename = linker_basename(pathname);
451     KLD_DPF(FILE, ("linker_make_file: new file, filename=%s, pathname=%s\n",
452                        filename, pathname));
453 
454     lockmgr(&llf_lock, LK_EXCLUSIVE);
455     lf = kmalloc(sizeof(struct linker_file), M_LINKER, M_WAITOK | M_ZERO);
456     lf->refs = 1;
457     lf->userrefs = 0;
458     lf->flags = 0;
459     lf->filename = linker_strdup(filename);
460     lf->pathname = linker_strdup(pathname);
461     lf->id = next_file_id++;
462     lf->ndeps = 0;
463     lf->deps = NULL;
464     STAILQ_INIT(&lf->common);
465     TAILQ_INIT(&lf->modules);
466 
467     lf->priv = priv;
468     lf->ops = ops;
469     TAILQ_INSERT_TAIL(&linker_files, lf, link);
470 
471     lockmgr(&llf_lock, LK_RELEASE);
472 
473     return lf;
474 }
475 
476 int
linker_file_unload(linker_file_t file)477 linker_file_unload(linker_file_t file)
478 {
479     module_t mod, next;
480     modlist_t ml, nextml;
481     struct common_symbol* cp;
482     int error = 0;
483     int i;
484 
485     /* Refuse to unload modules if securelevel raised */
486     if (securelevel > 0 || kernel_mem_readonly)
487           return EPERM;
488 
489     KLD_DPF(FILE, ("linker_file_unload: lf->refs=%d\n", file->refs));
490 
491     lockmgr(&llf_lock, LK_EXCLUSIVE);
492 
493     /* Easy case of just dropping a reference. */
494     if (file->refs > 1) {
495               file->refs--;
496               lockmgr(&llf_lock, LK_RELEASE);
497               return (0);
498     }
499 
500     KLD_DPF(FILE, ("linker_file_unload: file is unloading, informing modules\n"));
501 
502     /*
503      * Inform any modules associated with this file.
504      */
505     mod = TAILQ_FIRST(&file->modules);
506     for (mod = TAILQ_FIRST(&file->modules); mod; mod = next) {
507           next = module_getfnext(mod);
508 
509           /*
510            * Give the module a chance to veto the unload.  Note that the
511            * act of unloading the module may cause other modules in the
512            * same file list to be unloaded recursively.
513            */
514           if ((error = module_unload(mod)) != 0) {
515               KLD_DPF(FILE, ("linker_file_unload: module %p vetoes unload\n",
516                                  mod));
517               lockmgr(&llf_lock, LK_RELEASE);
518               file->refs--;
519               goto out;
520           }
521           module_release(mod);
522     }
523 
524     TAILQ_FOREACH_MUTABLE(ml, &found_modules, link, nextml) {
525           if (ml->container == file) {
526               TAILQ_REMOVE(&found_modules, ml, link);
527               kfree(ml, M_LINKER);
528           }
529     }
530 
531     /* Don't try to run SYSUNINITs if we are unloaded due to a link error */
532     if (file->flags & LINKER_FILE_LINKED) {
533           file->flags &= ~LINKER_FILE_LINKED;
534           lockmgr(&llf_lock, LK_RELEASE);
535           linker_file_sysuninit(file);
536           linker_file_unregister_sysctls(file);
537           lockmgr(&llf_lock, LK_EXCLUSIVE);
538     }
539 
540     TAILQ_REMOVE(&linker_files, file, link);
541 
542     if (file->deps) {
543           lockmgr(&llf_lock, LK_RELEASE);
544           for (i = 0; i < file->ndeps; i++)
545               linker_file_unload(file->deps[i]);
546           lockmgr(&llf_lock, LK_EXCLUSIVE);
547           kfree(file->deps, M_LINKER);
548           file->deps = NULL;
549     }
550 
551     while ((cp = STAILQ_FIRST(&file->common)) != NULL) {
552           STAILQ_REMOVE_HEAD(&file->common, link);
553           kfree(cp, M_LINKER);
554     }
555 
556     file->ops->unload(file);
557 
558     if (file->filename) {
559           kfree(file->filename, M_LINKER);
560           file->filename = NULL;
561     }
562     if (file->pathname) {
563           kfree(file->pathname, M_LINKER);
564           file->pathname = NULL;
565     }
566 
567     kfree(file, M_LINKER);
568 
569     lockmgr(&llf_lock, LK_RELEASE);
570 
571 out:
572     return error;
573 }
574 
575 void
linker_file_add_dependancy(linker_file_t file,linker_file_t dep)576 linker_file_add_dependancy(linker_file_t file, linker_file_t dep)
577 {
578     linker_file_t* newdeps;
579 
580     newdeps = kmalloc((file->ndeps + 1) * sizeof(linker_file_t*),
581                          M_LINKER, M_WAITOK | M_ZERO);
582 
583     if (file->deps) {
584           bcopy(file->deps, newdeps, file->ndeps * sizeof(linker_file_t*));
585           kfree(file->deps, M_LINKER);
586     }
587     file->deps = newdeps;
588     file->deps[file->ndeps] = dep;
589     file->ndeps++;
590 }
591 
592 /*
593  * Locate a linker set and its contents.
594  * This is a helper function to avoid linker_if.h exposure elsewhere.
595  * Note: firstp and lastp are really void ***
596  */
597 int
linker_file_lookup_set(linker_file_t file,const char * name,void * firstp,void * lastp,int * countp)598 linker_file_lookup_set(linker_file_t file, const char *name,
599                       void *firstp, void *lastp, int *countp)
600 {
601     return file->ops->lookup_set(file, name, firstp, lastp, countp);
602 }
603 
604 int
linker_file_lookup_symbol(linker_file_t file,const char * name,int deps,caddr_t * raddr)605 linker_file_lookup_symbol(linker_file_t file, const char* name, int deps, caddr_t *raddr)
606 {
607     c_linker_sym_t sym;
608     linker_symval_t symval;
609     linker_file_t lf;
610     size_t common_size = 0;
611     int i;
612 
613     KLD_DPF(SYM, ("linker_file_lookup_symbol: file=%p, name=%s, deps=%d\n",
614                       file, name, deps));
615 
616     if (file->ops->lookup_symbol(file, name, &sym) == 0) {
617           file->ops->symbol_values(file, sym, &symval);
618 
619           /*
620            * XXX Assume a common symbol if its value is 0 and it has a non-zero
621            * size, otherwise it could be an absolute symbol with a value of 0.
622            */
623           if (symval.value == NULL && symval.size != 0) {
624               /*
625                * For commons, first look them up in the dependancies and
626                * only allocate space if not found there.
627                */
628               common_size = symval.size;
629           } else {
630               KLD_DPF(SYM, ("linker_file_lookup_symbol: symbol.value=%p\n", symval.value));
631               *raddr = symval.value;
632               return 0;
633           }
634     }
635     if (deps) {
636           for (i = 0; i < file->ndeps; i++) {
637               if (linker_file_lookup_symbol(file->deps[i], name, 0, raddr) == 0) {
638                     KLD_DPF(SYM, ("linker_file_lookup_symbol: deps value=%p\n", *raddr));
639                     return 0;
640               }
641           }
642 
643           /* If we have not found it in the dependencies, search globally */
644           TAILQ_FOREACH(lf, &linker_files, link) {
645               /* But skip the current file if it's on the list */
646               if (lf == file)
647                     continue;
648               /* And skip the files we searched above */
649               for (i = 0; i < file->ndeps; i++)
650                     if (lf == file->deps[i])
651                         break;
652               if (i < file->ndeps)
653                     continue;
654               if (linker_file_lookup_symbol(lf, name, 0, raddr) == 0) {
655                     KLD_DPF(SYM, ("linker_file_lookup_symbol: global value=%p\n", *raddr));
656                     return 0;
657               }
658           }
659     }
660 
661     if (common_size > 0) {
662           /*
663            * This is a common symbol which was not found in the
664            * dependancies.  We maintain a simple common symbol table in
665            * the file object.
666            */
667           struct common_symbol* cp;
668 
669           STAILQ_FOREACH(cp, &file->common, link)
670               if (!strcmp(cp->name, name)) {
671                     KLD_DPF(SYM, ("linker_file_lookup_symbol: old common value=%p\n", cp->address));
672                     *raddr = cp->address;
673                     return 0;
674               }
675 
676           /*
677            * Round the symbol size up to align.
678            */
679           common_size = (common_size + sizeof(int) - 1) & -sizeof(int);
680           cp = kmalloc(sizeof(struct common_symbol)
681                         + common_size
682                         + strlen(name) + 1,
683                         M_LINKER, M_WAITOK | M_ZERO);
684 
685           cp->address = (caddr_t) (cp + 1);
686           cp->name = cp->address + common_size;
687           strcpy(cp->name, name);
688           bzero(cp->address, common_size);
689           STAILQ_INSERT_TAIL(&file->common, cp, link);
690 
691           KLD_DPF(SYM, ("linker_file_lookup_symbol: new common value=%p\n", cp->address));
692           *raddr = cp->address;
693           return 0;
694     }
695 
696 #ifdef _KERNEL_VIRTUAL
697     *raddr = dlsym(RTLD_NEXT, name);
698     if (*raddr != NULL) {
699           KLD_DPF(SYM, ("linker_file_lookup_symbol: found dlsym=%p\n", *raddr));
700           return 0;
701     }
702 #endif
703 
704     KLD_DPF(SYM, ("linker_file_lookup_symbol: fail\n"));
705     return ENOENT;
706 }
707 
708 #ifdef DDB
709 /*
710  * DDB Helpers.  DDB has to look across multiple files with their own
711  * symbol tables and string tables.
712  *
713  * Note that we do not obey list locking protocols here.  We really don't
714  * need DDB to hang because somebody's got the lock held.  We'll take the
715  * chance that the files list is inconsistant instead.
716  */
717 
718 int
linker_ddb_lookup(const char * symstr,c_linker_sym_t * sym)719 linker_ddb_lookup(const char *symstr, c_linker_sym_t *sym)
720 {
721     linker_file_t lf;
722 
723     TAILQ_FOREACH(lf, &linker_files, link) {
724           if (lf->ops->lookup_symbol(lf, symstr, sym) == 0)
725               return 0;
726     }
727     return ENOENT;
728 }
729 
730 int
linker_ddb_search_symbol(caddr_t value,c_linker_sym_t * sym,long * diffp)731 linker_ddb_search_symbol(caddr_t value, c_linker_sym_t *sym, long *diffp)
732 {
733     linker_file_t lf;
734     u_long off = (uintptr_t)value;
735     u_long diff, bestdiff;
736     c_linker_sym_t best;
737     c_linker_sym_t es;
738 
739     best = NULL;
740     bestdiff = off;
741     TAILQ_FOREACH(lf, &linker_files, link) {
742           if (lf->ops->search_symbol(lf, value, &es, &diff) != 0)
743               continue;
744           if (es != NULL && diff < bestdiff) {
745               best = es;
746               bestdiff = diff;
747           }
748           if (bestdiff == 0)
749               break;
750     }
751     if (best) {
752           *sym = best;
753           *diffp = bestdiff;
754           return 0;
755     } else {
756           *sym = NULL;
757           *diffp = off;
758           return ENOENT;
759     }
760 }
761 
762 int
linker_ddb_symbol_values(c_linker_sym_t sym,linker_symval_t * symval)763 linker_ddb_symbol_values(c_linker_sym_t sym, linker_symval_t *symval)
764 {
765     linker_file_t lf;
766 
767     TAILQ_FOREACH(lf, &linker_files, link) {
768           if (lf->ops->symbol_values(lf, sym, symval) == 0)
769               return 0;
770     }
771     return ENOENT;
772 }
773 
774 #endif
775 
776 /*
777  * Syscalls.
778  *
779  * MPALMOSTSAFE
780  */
781 int
sys_kldload(struct sysmsg * sysmsg,const struct kldload_args * uap)782 sys_kldload(struct sysmsg *sysmsg, const struct kldload_args *uap)
783 {
784     char *file;
785     char *kldname, *modname;
786     linker_file_t lf;
787     int error = 0;
788 
789     sysmsg->sysmsg_result = -1;
790 
791     if (securelevel > 0 || kernel_mem_readonly)   /* redundant, but that's OK */
792           return EPERM;
793 
794     if ((error = caps_priv_check_self(SYSCAP_NOKLD)) != 0)
795           return error;
796 
797     file = kmalloc(MAXPATHLEN, M_TEMP, M_WAITOK);
798     if ((error = copyinstr(uap->file, file, MAXPATHLEN, NULL)) != 0)
799           goto out;
800 
801     /*
802      * If file does not contain a qualified name or any dot in it
803      * (kldname.ko, or kldname.ver.ko) treat it as an interface
804      * name.
805      */
806     if (strchr(file, '/') || strchr(file, '.')) {
807           kldname = file;
808           modname = NULL;
809     } else {
810           kldname = NULL;
811           modname = file;
812     }
813 
814     lockmgr(&kld_lock, LK_EXCLUSIVE);
815     error = linker_load_module(kldname, modname, NULL, NULL, &lf);
816     lockmgr(&kld_lock, LK_RELEASE);
817     if (error)
818           goto out;
819 
820     lf->userrefs++;
821     sysmsg->sysmsg_result = lf->id;
822 
823 out:
824     if (file)
825           kfree(file, M_TEMP);
826     return error;
827 }
828 
829 /*
830  * MPALMOSTSAFE
831  */
832 int
sys_kldunload(struct sysmsg * sysmsg,const struct kldunload_args * uap)833 sys_kldunload(struct sysmsg *sysmsg, const struct kldunload_args *uap)
834 {
835     linker_file_t lf;
836     int error = 0;
837 
838     if (securelevel > 0 || kernel_mem_readonly)   /* redundant, but that's OK */
839           return EPERM;
840 
841     if ((error = caps_priv_check_self(SYSCAP_NOKLD)) != 0)
842           return error;
843 
844     lockmgr(&kld_lock, LK_EXCLUSIVE);
845     lf = linker_find_file_by_id(uap->fileid);
846     if (lf) {
847           KLD_DPF(FILE, ("kldunload: lf->userrefs=%d\n", lf->userrefs));
848           if (lf->userrefs == 0) {
849               kprintf("linkerunload: attempt to unload file that was loaded by the kernel\n");
850               error = EBUSY;
851               goto out;
852           }
853           lf->userrefs--;
854           error = linker_file_unload(lf);
855           if (error)
856               lf->userrefs++;
857     } else {
858           error = ENOENT;
859     }
860 out:
861     lockmgr(&kld_lock, LK_RELEASE);
862 
863     return error;
864 }
865 
866 /*
867  * MPALMOSTSAFE
868  */
869 int
sys_kldfind(struct sysmsg * sysmsg,const struct kldfind_args * uap)870 sys_kldfind(struct sysmsg *sysmsg, const struct kldfind_args *uap)
871 {
872     char *pathname;
873     const char *filename;
874     linker_file_t lf;
875     int error;
876 
877     sysmsg->sysmsg_result = -1;
878 
879     pathname = kmalloc(MAXPATHLEN, M_TEMP, M_WAITOK);
880     if ((error = copyinstr(uap->file, pathname, MAXPATHLEN, NULL)) != 0)
881           goto out;
882 
883     filename = linker_basename(pathname);
884 
885     lockmgr(&kld_lock, LK_EXCLUSIVE);
886     lf = linker_find_file_by_name(filename);
887     if (lf)
888           sysmsg->sysmsg_result = lf->id;
889     else
890           error = ENOENT;
891     lockmgr(&kld_lock, LK_RELEASE);
892 
893 out:
894     if (pathname)
895           kfree(pathname, M_TEMP);
896     return error;
897 }
898 
899 /*
900  * MPALMOSTSAFE
901  */
902 int
sys_kldnext(struct sysmsg * sysmsg,const struct kldnext_args * uap)903 sys_kldnext(struct sysmsg *sysmsg, const struct kldnext_args *uap)
904 {
905     linker_file_t lf;
906     int error = 0;
907 
908     lockmgr(&kld_lock, LK_EXCLUSIVE);
909     if (uap->fileid == 0) {
910           lf = TAILQ_FIRST(&linker_files);
911     } else {
912           lf = linker_find_file_by_id(uap->fileid);
913           if (lf == NULL) {
914               error = ENOENT;
915               goto out;
916           }
917           lf = TAILQ_NEXT(lf, link);
918     }
919 
920     /* Skip partially loaded files. */
921     while (lf != NULL && !(lf->flags & LINKER_FILE_LINKED)) {
922           lf = TAILQ_NEXT(lf, link);
923     }
924 
925     if (lf)
926           sysmsg->sysmsg_result = lf->id;
927     else
928           sysmsg->sysmsg_result = 0;
929 
930 out:
931     lockmgr(&kld_lock, LK_RELEASE);
932 
933     return error;
934 }
935 
936 /*
937  * MPALMOSTSAFE
938  */
939 int
sys_kldstat(struct sysmsg * sysmsg,const struct kldstat_args * uap)940 sys_kldstat(struct sysmsg *sysmsg, const struct kldstat_args *uap)
941 {
942     linker_file_t lf;
943     int error = 0;
944     int version;
945     struct kld_file_stat* stat;
946     int namelen;
947 
948     lockmgr(&kld_lock, LK_EXCLUSIVE);
949     lf = linker_find_file_by_id(uap->fileid);
950     if (!lf) {
951           error = ENOENT;
952           goto out;
953     }
954 
955     stat = uap->stat;
956 
957     /*
958      * Check the version of the user's structure.
959      */
960     if ((error = copyin(&stat->version, &version, sizeof(version))) != 0)
961           goto out;
962     if (version != sizeof(struct kld_file_stat)) {
963           error = EINVAL;
964           goto out;
965     }
966 
967     namelen = strlen(lf->filename) + 1;
968     if (namelen > MAXPATHLEN)
969           namelen = MAXPATHLEN;
970     if ((error = copyout(lf->filename, stat->name, namelen)) != 0)
971           goto out;
972 
973     namelen = strlen(lf->pathname) + 1;
974     if (namelen > MAXPATHLEN)
975           namelen = MAXPATHLEN;
976     if ((error = copyout(lf->pathname, stat->pathname, namelen)) != 0)
977           goto out;
978 
979     if ((error = copyout(&lf->refs, &stat->refs, sizeof(int))) != 0)
980           goto out;
981     if ((error = copyout(&lf->id, &stat->id, sizeof(int))) != 0)
982           goto out;
983     if ((error = copyout(&lf->address, &stat->address, sizeof(caddr_t))) != 0)
984           goto out;
985     if ((error = copyout(&lf->size, &stat->size, sizeof(size_t))) != 0)
986           goto out;
987 
988     sysmsg->sysmsg_result = 0;
989 
990 out:
991     lockmgr(&kld_lock, LK_RELEASE);
992 
993     return error;
994 }
995 
996 /*
997  * MPALMOSTSAFE
998  */
999 int
sys_kldfirstmod(struct sysmsg * sysmsg,const struct kldfirstmod_args * uap)1000 sys_kldfirstmod(struct sysmsg *sysmsg, const struct kldfirstmod_args *uap)
1001 {
1002     linker_file_t lf;
1003     int error = 0;
1004 
1005     lockmgr(&kld_lock, LK_EXCLUSIVE);
1006     lf = linker_find_file_by_id(uap->fileid);
1007     if (lf) {
1008           if (TAILQ_FIRST(&lf->modules))
1009               sysmsg->sysmsg_result = module_getid(TAILQ_FIRST(&lf->modules));
1010           else
1011               sysmsg->sysmsg_result = 0;
1012     } else {
1013           error = ENOENT;
1014     }
1015     lockmgr(&kld_lock, LK_RELEASE);
1016 
1017     return error;
1018 }
1019 
1020 /*
1021  * MPALMOSTSAFE
1022  */
1023 int
sys_kldsym(struct sysmsg * sysmsg,const struct kldsym_args * uap)1024 sys_kldsym(struct sysmsg *sysmsg, const struct kldsym_args *uap)
1025 {
1026     char *symstr = NULL;
1027     c_linker_sym_t sym;
1028     linker_symval_t symval;
1029     linker_file_t lf;
1030     struct kld_sym_lookup lookup;
1031     int error = 0;
1032 
1033     lockmgr(&kld_lock, LK_EXCLUSIVE);
1034     if ((error = copyin(uap->data, &lookup, sizeof(lookup))) != 0)
1035           goto out;
1036     if (lookup.version != sizeof(lookup) || uap->cmd != KLDSYM_LOOKUP) {
1037           error = EINVAL;
1038           goto out;
1039     }
1040 
1041     symstr = kmalloc(MAXPATHLEN, M_TEMP, M_WAITOK);
1042     if ((error = copyinstr(lookup.symname, symstr, MAXPATHLEN, NULL)) != 0)
1043           goto out;
1044 
1045     if (uap->fileid != 0) {
1046           lf = linker_find_file_by_id(uap->fileid);
1047           if (lf == NULL) {
1048               error = ENOENT;
1049               goto out;
1050           }
1051           if (lf->ops->lookup_symbol(lf, symstr, &sym) == 0 &&
1052               lf->ops->symbol_values(lf, sym, &symval) == 0) {
1053               lookup.symvalue = (uintptr_t)symval.value;
1054               lookup.symsize = symval.size;
1055               error = copyout(&lookup, uap->data, sizeof(lookup));
1056           } else
1057               error = ENOENT;
1058     } else {
1059           TAILQ_FOREACH(lf, &linker_files, link) {
1060               if (lf->ops->lookup_symbol(lf, symstr, &sym) == 0 &&
1061                     lf->ops->symbol_values(lf, sym, &symval) == 0) {
1062                     lookup.symvalue = (uintptr_t)symval.value;
1063                     lookup.symsize = symval.size;
1064                     error = copyout(&lookup, uap->data, sizeof(lookup));
1065                     break;
1066               }
1067           }
1068           if (!lf)
1069               error = ENOENT;
1070     }
1071 out:
1072     lockmgr(&kld_lock, LK_RELEASE);
1073     if (symstr)
1074           kfree(symstr, M_TEMP);
1075 
1076     return error;
1077 }
1078 
1079 /*
1080  * Preloaded module support
1081  */
1082 
1083 static modlist_t
modlist_lookup(const char * name,int ver)1084 modlist_lookup(const char *name, int ver)
1085 {
1086     modlist_t           mod;
1087 
1088     TAILQ_FOREACH(mod, &found_modules, link) {
1089           if (strcmp(mod->name, name) == 0 && (ver == 0 || mod->version == ver))
1090               return (mod);
1091     }
1092     return (NULL);
1093 }
1094 
1095 static modlist_t
modlist_lookup2(const char * name,struct mod_depend * verinfo)1096 modlist_lookup2(const char *name, struct mod_depend *verinfo)
1097 {
1098     modlist_t           mod, bestmod;
1099     int                 ver;
1100 
1101     if (verinfo == NULL)
1102           return (modlist_lookup(name, 0));
1103     bestmod = NULL;
1104     TAILQ_FOREACH(mod, &found_modules, link) {
1105           if (strcmp(mod->name, name) != 0)
1106               continue;
1107           ver = mod->version;
1108           if (ver == verinfo->md_ver_preferred)
1109               return (mod);
1110           if (ver >= verinfo->md_ver_minimum &&
1111                     ver <= verinfo->md_ver_maximum &&
1112                     (bestmod == NULL || ver > bestmod->version))
1113               bestmod = mod;
1114     }
1115     return (bestmod);
1116 }
1117 
1118 int
linker_reference_module(const char * modname,struct mod_depend * verinfo,linker_file_t * result)1119 linker_reference_module(const char *modname, struct mod_depend *verinfo,
1120     linker_file_t *result)
1121 {
1122     modlist_t mod;
1123     int error;
1124 
1125     lockmgr(&llf_lock, LK_SHARED);
1126     if ((mod = modlist_lookup2(modname, verinfo)) != NULL) {
1127         *result = mod->container;
1128         (*result)->refs++;
1129         lockmgr(&llf_lock, LK_RELEASE);
1130         return (0);
1131     }
1132 
1133     lockmgr(&llf_lock, LK_RELEASE);
1134     /*lockmgr(&kld_lock, LK_EXCLUSIVE);*/
1135     error = linker_load_module(NULL, modname, NULL, verinfo, result);
1136     /*lockmgr(&kld_lock, LK_RELEASE);*/
1137 
1138     return (error);
1139 }
1140 
1141 int
linker_release_module(const char * modname,struct mod_depend * verinfo,linker_file_t lf)1142 linker_release_module(const char *modname, struct mod_depend *verinfo,
1143     linker_file_t lf)
1144 {
1145     modlist_t mod;
1146     int error;
1147 
1148     lockmgr(&llf_lock, LK_SHARED);
1149     if (lf == NULL) {
1150         KASSERT(modname != NULL,
1151             ("linker_release_module: no file or name"));
1152         mod = modlist_lookup2(modname, verinfo);
1153         if (mod == NULL) {
1154             lockmgr(&llf_lock, LK_RELEASE);
1155             return (ESRCH);
1156         }
1157         lf = mod->container;
1158     } else
1159         KASSERT(modname == NULL && verinfo == NULL,
1160             ("linker_release_module: both file and name"));
1161     lockmgr(&llf_lock, LK_RELEASE);
1162     /*lockmgr(&kld_lock, LK_EXCLUSIVE);*/
1163     error = linker_file_unload(lf);
1164     /*lockmgr(&kld_lock, LK_RELEASE);*/
1165 
1166     return (error);
1167 }
1168 
1169 static modlist_t
modlist_newmodule(const char * modname,int version,linker_file_t container)1170 modlist_newmodule(const char *modname, int version, linker_file_t container)
1171 {
1172     modlist_t           mod;
1173 
1174     mod = kmalloc(sizeof(struct modlist), M_LINKER, M_NOWAIT | M_ZERO);
1175     if (mod == NULL)
1176           panic("no memory for module list");
1177     mod->container = container;
1178     mod->name = modname;
1179     mod->version = version;
1180     TAILQ_INSERT_TAIL(&found_modules, mod, link);
1181     return (mod);
1182 }
1183 
1184 static void
linker_addmodules(linker_file_t lf,struct mod_metadata ** start,struct mod_metadata ** stop,int preload)1185 linker_addmodules(linker_file_t lf, struct mod_metadata **start,
1186                       struct mod_metadata **stop, int preload)
1187 {
1188     struct mod_metadata *mp, **mdp;
1189     const char     *modname;
1190     int                 ver;
1191 
1192     for (mdp = start; mdp < stop; mdp++) {
1193           mp = *mdp;
1194           if (mp->md_type != MDT_VERSION)
1195               continue;
1196           modname = mp->md_cval;
1197           ver = ((struct mod_version *)mp->md_data)->mv_version;
1198           if (modlist_lookup(modname, ver) != NULL) {
1199               kprintf("module %s already present!\n", modname);
1200               /* XXX what can we do? this is a build error. :-( */
1201               continue;
1202           }
1203           modlist_newmodule(modname, ver, lf);
1204     }
1205 }
1206 
1207 static void
linker_preload(void * arg)1208 linker_preload(void* arg)
1209 {
1210     caddr_t                   modptr;
1211     const char                *modname, *nmodname;
1212     char            *modtype;
1213     linker_file_t   lf, nlf;
1214     linker_class_t  lc;
1215     int                       error;
1216     linker_file_list_t loaded_files;
1217     linker_file_list_t depended_files;
1218     struct mod_metadata *mp, *nmp;
1219     struct mod_metadata **start, **stop, **mdp, **nmdp;
1220     struct mod_depend *verinfo;
1221     int nver;
1222     int resolves;
1223     modlist_t mod;
1224     struct sysinit  **si_start, **si_stop;
1225 
1226     TAILQ_INIT(&loaded_files);
1227     TAILQ_INIT(&depended_files);
1228     TAILQ_INIT(&found_modules);
1229 
1230     modptr = NULL;
1231     while ((modptr = preload_search_next_name(modptr)) != NULL) {
1232           modname = (char *)preload_search_info(modptr, MODINFO_NAME);
1233           modtype = (char *)preload_search_info(modptr, MODINFO_TYPE);
1234           if (modname == NULL) {
1235               kprintf("Preloaded module at %p does not have a name!\n", modptr);
1236               continue;
1237           }
1238           if (modtype == NULL) {
1239               kprintf("Preloaded module at %p does not have a type!\n", modptr);
1240               continue;
1241           }
1242 
1243           if (bootverbose)
1244               kprintf("Preloaded %s \"%s\" at %p.\n", modtype, modname, modptr);
1245           lf = NULL;
1246           TAILQ_FOREACH(lc, &classes, link) {
1247               error = lc->ops->preload_file(modname, &lf);
1248               if (!error)
1249                     break;
1250               lf = NULL;
1251           }
1252           if (lf)
1253                     TAILQ_INSERT_TAIL(&loaded_files, lf, loaded);
1254     }
1255 
1256     /*
1257      * First get a list of stuff in the kernel.
1258      */
1259     if (linker_file_lookup_set(linker_kernel_file, MDT_SETNAME, &start,
1260                                      &stop, NULL) == 0)
1261           linker_addmodules(linker_kernel_file, start, stop, 1);
1262 
1263     /*
1264      * This is a once-off kinky bubble sort to resolve relocation
1265      * dependency requirements.
1266      */
1267 restart:
1268     TAILQ_FOREACH(lf, &loaded_files, loaded) {
1269           error = linker_file_lookup_set(lf, MDT_SETNAME, &start, &stop, NULL);
1270           /*
1271            * First, look to see if we would successfully link with this
1272            * stuff.
1273            */
1274           resolves = 1;                 /* unless we know otherwise */
1275           if (!error) {
1276               for (mdp = start; mdp < stop; mdp++) {
1277                     mp = *mdp;
1278                     if (mp->md_type != MDT_DEPEND)
1279                         continue;
1280                     modname = mp->md_cval;
1281                     verinfo = mp->md_data;
1282                     for (nmdp = start; nmdp < stop; nmdp++) {
1283                         nmp = *nmdp;
1284                         if (nmp->md_type != MDT_VERSION)
1285                               continue;
1286                         nmodname = nmp->md_cval;
1287                         if (strcmp(modname, nmodname) == 0)
1288                               break;
1289                     }
1290                     if (nmdp < stop)/* it's a self reference */
1291                         continue;
1292 
1293                     /*
1294                      * ok, the module isn't here yet, we
1295                      * are not finished
1296                      */
1297                     if (modlist_lookup2(modname, verinfo) == NULL)
1298                         resolves = 0;
1299               }
1300           }
1301           /*
1302            * OK, if we found our modules, we can link.  So, "provide"
1303            * the modules inside and add it to the end of the link order
1304            * list.
1305            */
1306           if (resolves) {
1307               if (!error) {
1308                     for (mdp = start; mdp < stop; mdp++) {
1309                         mp = *mdp;
1310                         if (mp->md_type != MDT_VERSION)
1311                               continue;
1312                         modname = mp->md_cval;
1313                         nver = ((struct mod_version *)mp->md_data)->mv_version;
1314                         if (modlist_lookup(modname, nver) != NULL) {
1315                               kprintf("module %s already present!\n", modname);
1316                               TAILQ_REMOVE(&loaded_files, lf, loaded);
1317                               linker_file_unload(lf /* , LINKER_UNLOAD_FORCE */ );
1318                               /* we changed tailq next ptr */
1319                               goto restart;
1320                         }
1321                         modlist_newmodule(modname, nver, lf);
1322                     }
1323               }
1324               TAILQ_REMOVE(&loaded_files, lf, loaded);
1325               TAILQ_INSERT_TAIL(&depended_files, lf, loaded);
1326               /*
1327                * Since we provided modules, we need to restart the
1328                * sort so that the previous files that depend on us
1329                * have a chance. Also, we've busted the tailq next
1330                * pointer with the REMOVE.
1331                */
1332               goto restart;
1333           }
1334     }
1335 
1336     /*
1337      * At this point, we check to see what could not be resolved..
1338      */
1339     while ((lf = TAILQ_FIRST(&loaded_files)) != NULL) {
1340           TAILQ_REMOVE(&loaded_files, lf, loaded);
1341           kprintf("KLD file %s is missing dependencies\n", lf->filename);
1342           linker_file_unload(lf /* , LINKER_UNLOAD_FORCE */ );
1343     }
1344 
1345     /*
1346      * We made it. Finish off the linking in the order we determined.
1347      */
1348     TAILQ_FOREACH_MUTABLE(lf, &depended_files, loaded, nlf) {
1349           if (linker_kernel_file) {
1350               linker_kernel_file->refs++;
1351               linker_file_add_dependancy(lf, linker_kernel_file);
1352           }
1353           lf->userrefs++;
1354 
1355           error = linker_file_lookup_set(lf, MDT_SETNAME, &start, &stop, NULL);
1356           if (!error) {
1357               for (mdp = start; mdp < stop; mdp++) {
1358                     mp = *mdp;
1359                     if (mp->md_type != MDT_DEPEND)
1360                         continue;
1361                     modname = mp->md_cval;
1362                     verinfo = mp->md_data;
1363                     mod = modlist_lookup2(modname, verinfo);
1364                     /* Don't count self-dependencies */
1365                     if (lf == mod->container)
1366                         continue;
1367                     mod->container->refs++;
1368                     linker_file_add_dependancy(lf, mod->container);
1369               }
1370           }
1371           /*
1372            * Now do relocation etc using the symbol search paths
1373            * established by the dependencies
1374            */
1375           error = lf->ops->preload_finish(lf);
1376           if (error) {
1377               TAILQ_REMOVE(&depended_files, lf, loaded);
1378               kprintf("KLD file %s - could not finalize loading\n",
1379                         lf->filename);
1380               linker_file_unload(lf /* , LINKER_UNLOAD_FORCE */);
1381               continue;
1382           }
1383           linker_file_register_modules(lf);
1384           if (linker_file_lookup_set(lf, "sysinit_set", &si_start, &si_stop, NULL) == 0)
1385               sysinit_add(si_start, si_stop);
1386           linker_file_register_sysctls(lf);
1387           lf->flags |= LINKER_FILE_LINKED;
1388     }
1389     /* woohoo! we made it! */
1390 }
1391 
1392 SYSINIT(preload, SI_BOOT2_KLD, SI_ORDER_MIDDLE, linker_preload, 0);
1393 
1394 /*
1395  * Search for a not-loaded module by name.
1396  *
1397  * Modules may be found in the following locations:
1398  *
1399  * - preloaded (result is just the module name)
1400  * - on disk (result is full path to module)
1401  *
1402  * The search path can be manipulated via sysctl.  Note that we use the ';'
1403  * character as a separator to be consistent with the bootloader.
1404  */
1405 
1406 static char linker_path[MAXPATHLEN] = "/boot/kernel;/boot/modules.local";
1407 
1408 SYSCTL_STRING(_kern, OID_AUTO, module_path, CTLFLAG_RW, linker_path,
1409                 sizeof(linker_path), "module load search path");
1410 TUNABLE_STR("module_path", linker_path, sizeof(linker_path));
1411 
1412 char *
linker_search_path(const char * name)1413 linker_search_path(const char *name)
1414 {
1415     struct nlookupdata        nd;
1416     char            *cp, *ep, *result;
1417     char            *buf, *retbuf, *freebuf;
1418     size_t                    name_len, prefix_len;
1419     size_t                    result_len;
1420     int                       sep;
1421     int                       error;
1422     enum vtype                type;
1423     const char *exts[] = { "", ".ko", NULL };
1424     const char **ext;
1425 
1426     /* already full path */
1427     if (name[0] == '/')
1428           return linker_strdup(name);
1429 
1430     /* resolve to full path */
1431     if (strchr(name, '/') != NULL) {
1432           result = NULL;
1433 
1434           /* adapted from sys___realpath() */
1435           error = nlookup_init(&nd, name, UIO_SYSSPACE, NLC_FOLLOW);
1436           if (error == 0) {
1437               nd.nl_flags |= NLC_SHAREDLOCK;
1438               error = nlookup(&nd);
1439           }
1440           if (error == 0) {
1441               if (nd.nl_flags & NLC_NCPISLOCKED) {
1442                     nd.nl_flags &= ~NLC_NCPISLOCKED;
1443                     cache_unlock(&nd.nl_nch);
1444               }
1445               error = cache_fullpath(curproc, &nd.nl_nch, NULL,
1446                                            &retbuf, &freebuf, 0);
1447               if (error == 0) {
1448                     result = linker_strdup(retbuf);
1449                     kfree(freebuf, M_TEMP);
1450               }
1451           }
1452           nlookup_done(&nd);
1453 
1454           return (result);
1455     }
1456 
1457     /* traverse the linker path */
1458     buf = kmalloc(MAXPATHLEN, M_LINKER, M_WAITOK);
1459     cp = linker_path;
1460     name_len = strlen(name);
1461     for (;;) {
1462           /* find the end of this component */
1463           for (ep = cp; (*ep != 0) && (*ep != ';'); ep++)
1464               ;
1465           prefix_len = ep - cp;
1466           /* if this component doesn't end with a slash, add one */
1467           if (ep == cp || *(ep - 1) != '/')
1468               sep = 1;
1469           else
1470               sep = 0;
1471 
1472           result = buf;
1473           strncpy(result, cp, prefix_len);
1474           if (sep)
1475               result[prefix_len++] = '/';
1476           strcpy(result + prefix_len, name);
1477 
1478           result_len = strlen(result);
1479           for (ext = exts; *ext != NULL; ext++) {
1480               strcpy(result + result_len, *ext);
1481 
1482               /*
1483                * Attempt to open the file, and return the path if we succeed and it's
1484                * a regular file.
1485                */
1486               error = nlookup_init(&nd, result, UIO_SYSSPACE, NLC_FOLLOW|NLC_LOCKVP);
1487               if (error == 0)
1488                     error = vn_open(&nd, NULL, FREAD, 0);
1489               if (error == 0) {
1490                     type = nd.nl_open_vp->v_type;
1491                     if (type == VREG) {
1492                         nlookup_done(&nd);
1493                         return (result);
1494                     }
1495               }
1496               nlookup_done(&nd);
1497           }
1498 
1499           if (*ep == 0)
1500               break;
1501           cp = ep + 1;
1502     }
1503 
1504     kfree(buf, M_LINKER);
1505     return(NULL);
1506 }
1507 
1508 /*
1509  * Find a file which contains given module and load it, if "parent" is not
1510  * NULL, register a reference to it.
1511  */
1512 static int
linker_load_module(const char * kldname,const char * modname,struct linker_file * parent,struct mod_depend * verinfo,struct linker_file ** lfpp)1513 linker_load_module(const char *kldname, const char *modname,
1514                        struct linker_file *parent, struct mod_depend *verinfo,
1515                        struct linker_file **lfpp)
1516 {
1517     linker_file_t   lfdep;
1518     const char     *filename;
1519     char           *pathname;
1520     int                 error;
1521 
1522     if (modname == NULL) {
1523           /*
1524            * We have to load KLD
1525            */
1526           KASSERT(verinfo == NULL, ("linker_load_module: verinfo is not NULL"));
1527           pathname = linker_search_path(kldname);
1528     } else {
1529           if (modlist_lookup2(modname, verinfo) != NULL)
1530               return (EEXIST);
1531           if (kldname != NULL) {
1532               pathname = linker_strdup(kldname);
1533           } else if (rootvnode == NULL) {
1534               pathname = NULL;
1535           } else {
1536               pathname = linker_search_path(modname);
1537           }
1538 #if 0
1539           /*
1540            * Need to find a KLD with required module
1541            */
1542           pathname = linker_search_module(modname,
1543                                                   strlen(modname), verinfo);
1544 #endif
1545     }
1546     if (pathname == NULL)
1547           return (ENOENT);
1548 
1549     /*
1550      * Can't load more than one file with the same basename XXX:
1551      * Actually it should be possible to have multiple KLDs with
1552      * the same basename but different path because they can
1553      * provide different versions of the same modules.
1554      */
1555     filename = linker_basename(pathname);
1556     if (linker_find_file_by_name(filename))
1557           error = EEXIST;
1558     else
1559           do {
1560               error = linker_load_file(pathname, &lfdep);
1561               if (error)
1562                     break;
1563               if (modname && verinfo && modlist_lookup2(modname, verinfo) == NULL) {
1564                     linker_file_unload(lfdep /* , LINKER_UNLOAD_FORCE */ );
1565                     error = ENOENT;
1566                     break;
1567               }
1568               if (parent) {
1569                     linker_file_add_dependancy(parent, lfdep);
1570               }
1571               if (lfpp)
1572                     *lfpp = lfdep;
1573           } while (0);
1574     kfree(pathname, M_LINKER);
1575     return (error);
1576 }
1577 
1578 /*
1579  * This routine is responsible for finding dependencies of userland initiated
1580  * kldload(2)'s of files.
1581  */
1582 int
linker_load_dependencies(linker_file_t lf)1583 linker_load_dependencies(linker_file_t lf)
1584 {
1585     linker_file_t   lfdep;
1586     struct mod_metadata **start, **stop, **mdp, **nmdp;
1587     struct mod_metadata *mp, *nmp;
1588     struct mod_depend *verinfo;
1589     modlist_t           mod;
1590     const char     *modname, *nmodname;
1591     int                 ver, error = 0, count;
1592 
1593     /*
1594      * All files are dependant on /kernel.
1595      */
1596     if (linker_kernel_file) {
1597           linker_kernel_file->refs++;
1598           linker_file_add_dependancy(lf, linker_kernel_file);
1599     }
1600     if (linker_file_lookup_set(lf, MDT_SETNAME, &start, &stop, &count) != 0)
1601           return (0);
1602     for (mdp = start; mdp < stop; mdp++) {
1603           mp = *mdp;
1604           if (mp->md_type != MDT_VERSION)
1605               continue;
1606           modname = mp->md_cval;
1607           ver = ((struct mod_version *)mp->md_data)->mv_version;
1608           mod = modlist_lookup(modname, ver);
1609           if (mod != NULL) {
1610               kprintf("interface %s.%d already present in the KLD '%s'!\n",
1611                         modname, ver, mod->container->filename);
1612               return (EEXIST);
1613           }
1614     }
1615 
1616     for (mdp = start; mdp < stop; mdp++) {
1617           mp = *mdp;
1618           if (mp->md_type != MDT_DEPEND)
1619               continue;
1620           modname = mp->md_cval;
1621           verinfo = mp->md_data;
1622           nmodname = NULL;
1623           for (nmdp = start; nmdp < stop; nmdp++) {
1624               nmp = *nmdp;
1625               if (nmp->md_type != MDT_VERSION)
1626                     continue;
1627               nmodname = nmp->md_cval;
1628               if (strcmp(modname, nmodname) == 0)
1629                     break;
1630           }
1631           if (nmdp < stop)    /* early exit, it's a self reference */
1632               continue;
1633           mod = modlist_lookup2(modname, verinfo);
1634           if (mod) {                    /* woohoo, it's loaded already */
1635               lfdep = mod->container;
1636               lfdep->refs++;
1637               linker_file_add_dependancy(lf, lfdep);
1638               continue;
1639           }
1640           error = linker_load_module(NULL, modname, lf, verinfo, NULL);
1641           if (error) {
1642               kprintf("KLD %s: depends on %s - not available or version mismatch\n",
1643                         lf->filename, modname);
1644               break;
1645           }
1646     }
1647 
1648     if (error)
1649           return (error);
1650     linker_addmodules(lf, start, stop, 0);
1651     return (0);
1652 }
1653