开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
snmpconf
资源名称：net-snmp-5.2.2.rar　[点击查看]
上传用户：wxp200602
上传日期：2007-10-30
资源大小：4028k
文件大小：26k
源码类别：

SNMP编程

开发平台：

Unix_Linux

snmpconf：源码内容
  1. #!/usr/bin/perl -w
  3. #
  4. # A simple configuration file builder based on questions listed in
  5. # its own configuration file.  It would certainly be easy to use this
  6. # for other (non-snmp) programs as well.
  7. #
  9. use Getopt::Std;
  10. use Term::ReadLine;
  11. use IO::File;
  12. use Data::Dumper;
  13. use File::Copy;
  14. if ($^O eq 'MSWin32') {
  15.   eval 'require Win32::Registry;';
  16.   if ($@) {
  17.     print "nWarning: Perl module Win32::Registry is not installed.  This module isn";
  18.     print "         required to read the SNMPSHAREPATH and SNMPCONFPATH values from n";
  19.     print "         the registry.  To use snmpconf without the module you need ton";
  20.     print "         define SNMPSHAREPATH and SNMPCONFPATH as environment variablesn";
  21.     print "         or use the -c and -I command line options.n";
  22.   }
  23. }
  25. # globals
  26. %tokenitems=qw(line 1 info 1 comment 1);
  27. %arrayitems=qw(question 1 validanswer 1);
  29. # default folder for snmpconf-data
  30. if (defined(&my_getenv("SNMPSHAREPATH"))) {
  31.   $opts{'c'} = &my_getenv("SNMPSHAREPATH") . "/snmpconf-data";
  32. }
  33. else {
  34.   $opts{'c'} = "/usr/local/share/snmp/snmpconf-data";
  35. }
  36.  
  37. # default config file path
  38. if (defined(&my_getenv("SNMPCONFPATH"))) {
  39.   $confpath = &my_getenv("SNMPCONFPATH");
  40. }
  41. else {
  42.   $confpath = "/usr/local/share/snmp";
  43. }
  45. # home environment variable
  46. if (defined(&my_getenv("HOME"))) {
  47.   $home = &my_getenv("HOME") . "/.snmp";
  48. }
  49. else {
  50.   $home = "(HOME dir - n/a)";
  51. }
  53. # read the argument string
  54. getopts("qadhfc:piI:r:R:g:G", %opts);
  56. # display help
  57. if ($opts{'h'}) {
  58.     print "$0 [options] [FILETOCREATE...]n";
  59.     print "options:n";
  60.     print "  -f           overwrite existing files without promptingn";
  61.     print "  -i           install created files into $confpath.n";
  62.     print "  -p           install created files into $home.n";
  63.     print "  -I DIR       install created files into DIR.n";
  64.     print "  -a           Don't ask any questions, just read in currentn";
  65.     print "                   current .conf files and comment themn";
  66.     print "  -r all|none  Read in all or none of the .conf files found.n";
  67.     print "  -R file,...  Read in a particular list of .conf files.n";
  68.     print "  -g GROUP     Ask a series of GROUPed questions.n";
  69.     print "  -G           List known GROUPs.n";
  70.     print "  -c conf_dir  use alternate configuration directory.n";
  71.     print "  -q           run more quietly with less advice.n";
  72.     print "  -d           turn on debugging output.n";
  73.     print "  -D           turn on debugging dumper output.n";
  74.     exit;
  75. }
  77. # setup terminal interface.
  78. $ENV{'PERL_RL'}='o=0' if (!exists($ENV{'PERL_RL'}));
  79. $term = new Term::ReadLine 'snmpconf';
  81. # read in configuration file set
  82. read_config_files($opts{'c'}, %filetypes);
  83. debug(my_Dumper(%filetypes));
  85. if ($opts{'G'}) {
  86.     Print("nKnown GROUPs of tokens:nn");
  87.     foreach my $group (keys(%groups)) {
  88. print "  $groupn";
  89.     }
  90.     Print("n");
  91.     exit;
  92. }
  94. #
  95. # Expand the search path in case it contains multiple directories
  96. # separated by : (Unix) or ; (Win32)
  97. #
  98. my $ENV_SEPARATOR = ':';
  99. if ($^O eq 'MSWin32') {
  100.   $ENV_SEPARATOR = ';';
  101. }
  102. my @searchpath = split(/$ENV_SEPARATOR/, $confpath);
  103. push @searchpath, "/usr/local/etc/snmp";
  104. push @searchpath, ".";
  105. push @searchpath, "$home";
  107. # Remove trailing /'s or 's
  108. for (my $i=0; $i <= $#searchpath; $i++) {
  109.   $searchpath[$i] =~ /(.*?)([/\])*$/;
  110.   $searchpath[$i] = $1;
  111. }
  113. # Determine persistent directory.  Order of preference:
  114. # 
  115. # file in SNMP_PERSISTENT_FILE environment variable
  116. # directory defined by persistentDir snmp.conf variable
  117. # directory in SNMP_PERSISTENT_DIR environment variable
  118. # default PERSISTENT_DIRECTORY directory
  119. my $persistentDir = "";
  120. my $persistentFile = "";
  122. # SNMP_PERSISTENT_FILE environment variable
  123. if (defined(&my_getenv("SNMP_PERSISTENT_FILE"))) {
  124.   $persistentFile = &my_getenv("SNMP_PERSISTENT_FILE");
  125.   debug ("persistent file: SNMP_PERSISTENT_FILE environment variable setn");
  126. }
  128. # snmp.conf persistentDir
  129. if (!($persistentDir) && !($persistentFile)) {
  130.   foreach my $i (@searchpath) {
  131.     debug ("Searching file $i/snmp.conf for persistentDirn");
  132.     my $temp = get_persistentDir("$i/snmp.conf");
  133.     if ($temp) {
  134.       debug("persistent directory: set to $temp in $i/snmp.confn");
  135.       $persistentDir = $temp;
  136.       last;
  137.     }
  138.   }
  139. }
  141. # SNMP_PERSISTENT_DIR environment variable
  142. if (!($persistentDir) && !($persistentFile)) {
  143.   if (&my_getenv("SNMP_PERSISTENT_DIR")) {
  144.     $persistentDir = &my_getenv("SNMP_PERSISTENT_DIR");
  145.     debug ("persistent directory: SNMP_PERSISTENT_DIR environment variable setn");
  146.   }
  147. }
  149. # PERSISTENT_DIRECTORY default variable
  150. if (!($persistentDir) && !($persistentFile)) {
  151.   $persistentDir = "/var/net-snmp";
  152.   debug ("persistent directory: Using default valuen");
  153. }
  155. # Rebuild search path without persistent folder
  156. # Note:  persistent file handled in Find existing 
  157. # files to possibly read in section
  158. if ($persistentDir) {
  159.   # Remove trailing /'s or 's
  160.   $persistentDir =~ /(.*?)([/\])*$/;
  161.   $persistentDir = $1;
  162.   debug ("persistent directory: $persistentDirn");
  164.   my @searchpath_old = @searchpath;
  165.   @searchpath = ();
  166.   foreach my $path_temp (@searchpath_old) {
  167.     if ($path_temp eq $persistentDir) {
  168.       debug("skipping persistent directory $path_tempn");
  169.       next;
  170.     }
  171.     push @searchpath, $path_temp;
  172.   }
  173. }
  175. # Reset $confpath to the first path
  176. $confpath = $searchpath[0];
  178. #
  179. # Find existing files to possibly read in.
  180. #
  181. push @searchpath, $opts{I} if ($opts{I});
  182. foreach my $i (@searchpath) {
  183.     debug("searching $in");
  184.     foreach my $ft (keys(%filetypes)) {
  185.         if ("$i/$ft" eq $persistentFile) {
  186.           debug("skipping persistent file $i/$ftn");
  187.           next;
  188.         }
  189. debug("searching for $i/$ftn");
  190. $knownfiles{"$i/$ft"} = $ft if (-f "$i/$ft");
  191. my $localft = $ft;
  192. $localft =~ s/.conf/.local.conf/;
  193. $knownfiles{"$i/$localft"} = $ft if (-f "$i/$localft");
  194.     }
  195. }
  197. #
  198. # Ask the user if they want them to be read in and read them
  199. #
  200. if (keys(%knownfiles)) {
  201.     my @files;
  202.     if (defined($opts{'r'})) {
  203. if ($opts{'r'} eq "all" || $opts{'r'} eq "a") {
  204.     @files = keys(%knownfiles);
  205. } elsif ($opts{'r'} ne "none" && $opts{'r'} ne "n") {
  206.     print "unknown argument to -r: $opts{'r'}n";
  207.     exit(1);
  208. }
  209.     } elsif(defined($opts{'R'})) {
  210. @files = split(/s*,s*/,$opts{'R'});
  211. foreach my $i (@files) {
  212.     my $x = $i;
  213.     $x =~ s/.*/([^/]+)$/$1/;
  214.     $knownfiles{$i} = $x;
  215. }
  216. Print("reading: ", join(",",@files),"n");
  217.     } else {
  218. @files = display_menu(-head => "The following installed configuration files were found:n",
  219.       -tail => "Would you like me to read them in?  Their content will be merged with thenoutput files created by this session.nnValid answer examples: "all", "none","3","1,2,5"n",
  220.       -multiple => 1,
  221.       -question => 'Read in which',
  222.       -defaultvalue => 'all',
  223.       sort keys(%knownfiles));
  224.     }
  225.     foreach my $i (@files) {
  226. debug("reading $in");
  227. read_config($i, $knownfiles{$i});
  228.     }
  229. }
  231. if ($opts{'g'}) {
  232.     my @groups = split(/,:s/,$opts{'g'});
  233.     foreach my $group (@groups) {
  234. do_group($group);
  235.     }
  236. } elsif ($#ARGV >= 0) {
  237.     #
  238.     # loop through requested files.
  239.     #
  240.     foreach my $i (@ARGV) {
  241. if (!defined($filetypes{$i})) {
  242.     warn "invalid file: $in";
  243. } else {
  244.     if ($opts{'a'}) {
  245. $didfile{$i} = 1;
  246.     } else {
  247. build_file($term, $i, $filetypes{$i});
  248.     }
  249. }
  250.     }
  251. } else {
  252.     #
  253.     # ask user to select file type to operate on.
  254.     #
  255.     while(1) {
  256. my $line = display_menu(-head => "I can create the following types of configuration files for you.nSelect the file type you wish to create:n(you can create more than one as you run this program)n",
  257. -question => 'Select File',
  258. -otheranswers => ['quit'],
  259. -mapanswers => { 'q' => 'quit' },
  260. keys(%filetypes));
  261. last if ($line eq "quit");
  262. debug("file selected: $linen");
  263. build_file($term, $line, $filetypes{$line});
  264.     }
  265. }
  267. #
  268. # Write out the results to the output files.
  269. #
  270. output_files(%filetypes, $term);
  273. #
  274. # Display the files that have been created for the user.
  275. #
  276. Print("nnThe following files were created:nn");
  277. @didfiles = keys(%didfile);
  278. foreach my $i (@didfiles) {
  279.     if ($didfile{$i} ne "1") {
  280. if ($opts{'i'} || $opts{'I'}) {
  281.           $opts{'I'} = "$confpath" if (!$opts{'I'});
  283.           if (! (-d "$opts{'I'}") && ! (mkdir ("$opts{'I'}", 0755))) {
  284.     print "nCould not create $opts{'I'} directory: $!n";
  285.     print ("File $didfile{$i} left in current directoryn");
  286.   }
  287.   else {
  288.             move ("$opts{'I'}/$i", "$opts{'I'}/$i.bak") if (-f "$opts{'I'}/$i");
  289.             if (move ("$didfile{$i}", "$opts{'I'}")) {
  290.               print("  $didfile{$i} installed in $opts{'I'}n");     
  291.             }
  292.             else {
  293.               print "nCould not move file $didfile{$i} to $opts{'I'}/$i: $!n";
  294.               print ("File $didfile{$i} left in current directoryn");
  295.             }
  296.           }
  297. } elsif ($opts{'p'}) {
  298.   if (! (-d "$home") && ! (mkdir ("$home", 0755))) {
  299.     print "nCould not create $home directory: $!n";
  300.     print ("File $didfile{$i} left in current directoryn");
  301.   }
  302.   else {     
  303.     move ("$home/$i", "$home/$i.bak") if (-f "$home/$i");
  304.     if (move ("$didfile{$i}", "$home")) {
  305.       print("  $didfile{$i} installed in $homen");
  306.     }
  307.     else {
  308.       print "nCould not move file $didfile{$i} to $home: $!n";
  309.       print ("File $didfile{$i} left in current directoryn");
  310.        }
  311.   }
  312.    } else {
  313.     Print("  $didfile{$i} ",
  314.     ($i ne $didfile{$i})?"[ from $i specifications]":" ","n");
  315.     if ($opts{'d'}) {
  316. open(I,$didfile{$i});
  317. debug("    " . join("    ",<I>) . "n");
  318. close(I);
  319.     }
  320. }
  321.     }
  322. }
  324. if (!$opts{'p'} && !$opts{'i'} && !$opts{'I'}) {
  325.     Print("nThese files should be moved to $confpath if you
  326. want them used by everyone on the system.  In the future, if you add 
  327. the -i option to the command line I'll copy them there automatically for you.
  329. Or, if you want them for your personal use only, copy them to
  330. $home .  In the future, if you add the -p option to the
  331. command line I'll copy them there automatically for you.
  333. ");
  334. }
  336. ###########################################################################
  337. # Functions
  338. ###########################################################################
  340. sub Print {
  341.     print @_ if (!$opts{'q'});
  342. }
  343. #
  344. # handle a group of questions
  345. #
  346. sub get_yn_maybe {
  347.     my $question = shift;
  348.     my $ans = "y";
  349.     if ($question ne "") {
  350. $ans = get_answer($term, $question,
  351.   valid_answers(qw(yes y no n)), 'y');
  352.     }
  353.     return ($ans =~ /^y/)?1:0;
  354. }
  356. sub do_group {
  357.     my $group = shift;
  358.     die "no such group $groupn" if (!$groups{$group});
  359.     foreach my $token (@{$groups{$group}}) {
  360. if ($token->[0] eq "message") {
  361.     Print ("$token->[1] $token->[2]n");
  362. } elsif ($token->[0] eq "subgroup") {
  363.     do_group($token->[1]) if (get_yn_maybe($token->[2]));
  364. } elsif (defined($tokenmap{$token->[1]})) {
  365.     if (get_yn_maybe($token->[2])) {
  366. do {
  367.     do_line($token->[1], $tokenmap{$token->[1]});
  368. } until ($token->[0] ne "multiple" ||
  369.  get_answer($term, "Do another $token->[1] line?",
  370.     valid_answers(qw(yes y no n)), 'y')
  371.  =~ /n/);
  372.     }
  373. } elsif (defined($filetypes{$token->[1]})) {
  374.     $didfile{$token->[1]} = 1;
  375. } else {
  376.     die "invalid member $token->[1] of group $groupn";
  377. }
  378.     }
  379. }
  381. #
  382. # build a particular type of file by operating on sections
  383. #
  384. sub build_file {
  385.     my ($term, $filename, $fileconf) = @_;
  386.     $didfile{$filename} = 1;
  387.     my (@lines);
  388.     while(1) {
  389. my $line = display_menu(-head => "The configuration information which can be put into $filename is dividedninto sections.  Select a configuration section for $filenamenthat you wish to create:n",
  390. -otheranswers => ['finished'],
  391. -mapanswers => { 'f' => 'finished' },
  392. -question => "Select section",
  393. -numeric => 1,
  394. map { $_->{'title'}[0] } @$fileconf);
  396. return @lines if ($line eq "finished");
  397. do_section($fileconf->[$line-1]);
  398.     }
  399. }
  401. #
  402. # configure a particular section by operating on token types
  403. #
  404. sub do_section {
  405.     my $confsect = shift;
  406.     my @lines;
  407.     while(1) {
  408. Print ("nSection: $confsect->{'title'}[0]n");
  409. Print ("Description:n");
  410. Print ("  ", join("n  ",@{$confsect->{'description'}}),"n");
  411. my $line =
  412.     display_menu(-head => "Select from:n",
  413.  -otheranswers => ['finished','list'],
  414.  -mapanswers => { 'f' => 'finished',
  415.   'l' => 'list' },
  416.  -question => 'Select section',
  417.  -descriptions => [map { $confsect->{$_}{info}[0] } 
  418.    @{$confsect->{'thetokens'}}],
  419.  @{$confsect->{'thetokens'}});
  420. return @lines if ($line eq "finished");
  421. if ($line eq "list") {
  422.     print "Lines defined for section "$confsect->{title}[0]" so far:n";
  423.     foreach my $i (@{$confsect->{'thetokens'}}) {
  424. if ($#{$confsect->{$i}{'results'}} >= 0) {
  425.     print "  ",join("n  ",@{$confsect->{$i}{'results'}}),"n";
  426. }
  427.     }
  428.     next;
  429. }
  430. do_line($line, $confsect->{$line});
  431.     }
  432.     return;
  433. }
  435. #
  436. # Ask all the questions related to a particular line type
  437. #
  438. sub do_line {
  439.     my $token = shift;
  440.     my $confline = shift;
  441.     my (@answers, $counter, $i);
  442. #    debug(my_Dumper($confline));
  443.     Print ("nConfiguring: $tokenn");
  444.     Print ("Description:n  ",join("n    ",@{$confline->{'info'}}),"nn");
  445.     for($i=0; $i <= $#{$confline->{'question'}}; $i++) {
  446. if (defined($confline->{'question'}[$i]) &&
  447.     $confline->{'question'}[$i] ne "") {
  448.     my $q = $confline->{'question'}[$i];
  449.     $q =~ s/$(d+)/$answers[$1]/g;
  450.     debug("after: $term, $q, ",$confline->{'validanswer'}[$i],"n");
  451.     $answers[$i] = get_answer($term, $q,
  452.       $confline->{'validanswer'}[$i]);
  453.     $answers[$i] =~ s/"/\"/g;
  454.     $answers[$i] = '"' . $answers[$i] . '"' if ($answers[$i] =~ /s/);
  455. }
  456.     }
  457.     if ($#{$confline->{'line'}} == -1) {
  458. my ($i,$line);
  459. for($i=0; $i <= $#{$confline->{'question'}}; $i++) {
  460.     next if (!defined($confline->{'question'}[$i]) ||
  461.      $confline->{'question'}[$i] eq "");
  462.     $line .= " $" . $i;
  463. }
  464. push @{$confline->{'line'}}, $line;
  465.     }
  467.     foreach my $line (@{$confline->{'line'}}) {
  468. my $finished = $line;
  469. debug("preline: $finishedn");
  470. debug("answers: ",my_Dumper(@answers));
  471. $finished =~ s/$(d+)/$answers[$1]/g;
  472. if ($line =~ s/^evals+//) {
  473.     debug("eval: $finishedn");
  474.     $finished = eval $finished;
  475.     debug("eval results: $finishedn");
  476. }
  477. $finished = $token . " " . $finished;
  478. Print ("nFinished Output: $finishedn");
  479. push @{$confline->{'results'}},$finished;
  480.     }
  481. }
  483. #
  484. # read all sets of config files in the various subdirectories.
  485. #
  486. sub read_config_files {
  487.     my $readdir = shift;
  488.     my $filetypes = shift;
  489.     opendir(DH, $readdir) || die "no such directory $readdir, did you run make install?n";
  490.     my $dir;
  491.     my $configfilename="snmpconf-config";
  493.     while(defined($dir = readdir(DH))) {
  494. next if ($dir =~ /^./);
  495. next if ($dir =~ /CVS/);
  496. debug("dir entry: $dirn");
  497. if (-d "$readdir/$dir" && -f "$readdir/$dir/$configfilename") {
  499.     my $conffile;
  501.     # read the top level configuration inforamation about the direcotry.
  502.     open(I, "$readdir/$dir/$configfilename");
  503.     while(<I>) {
  504. $conffile = $1 if (/forconffile: (.*)/);
  505.     }
  506.     close(I);
  508.     # no README informatino.
  509.     if ($conffile eq "") {
  510. print STDERR "Warning: No 'forconffile' information in $readdir/$dir/$configfilenamen";
  511. next;
  512.     }
  514.     # read all the daat in the directory
  515.     $filetypes->{$conffile} = read_config_items("$readdir/$dir", $conffile);
  516. } else {
  517.     # no README informatino.
  518.     print STDERR "Warning: No $configfilename file found in $readdir/$dirn";
  519. }
  520.     }
  521.     closedir DH;
  522. }
  524. #
  525. # read each configuration file in a directory
  526. #
  527. sub read_config_items {
  528.     my $itemdir = shift;
  529.     my $type = shift;
  530.     opendir(ITEMS, $itemdir);
  531.     my $file;
  532.     my @results;
  533.     while(defined($file = readdir(ITEMS))) {
  534. next if ($file =~ /~$/);
  535. next if ($file =~ /^snmpconf-config$/);
  536. if (-f "$itemdir/$file") {
  537.     my $res = read_config_item("$itemdir/$file", $type);
  538.     if (scalar(keys(%$res)) > 0) {
  539. push @results, $res;
  540.     }
  541. }
  542.     }
  543.     closedir(ITEMS);
  544.     return @results;
  545. }
  547. #
  548. # mark a list of tokens as a special "group"
  549. #
  550. sub read_config_group {
  551.     my ($fh, $group, $type) = @_;
  552.     my $line;
  553.     debug("handling group $groupn");
  554.     push (@{$groups{$group}},['filetype', $type]);
  555.     while($line = <$fh>) {
  556. chomp($line);
  557. next if ($line =~ /^s*$/);
  558. next if ($line =~ /^#/);
  559. return $line if ($line !~ /^(single|multiple|message|filetype|subgroup)/);
  560. my ($type, $token, $rest) = ($line =~ /^(w+)s+([^s]+)s*(.*)/);
  561. debug ("reading group $group : $type -> $token -> $restn");
  562. push (@{$groups{$group}}, [$type, $token, $rest]);
  563.     }
  564.     return;
  565. }
  568. #
  569. # Parse one file
  570. #
  571. sub read_config_item {
  572.     my $itemfile = shift;
  573.     my $itemcount;
  574.     my $type = shift;
  575.     my $fh = new IO::File($itemfile);
  576.     return if (!defined($fh));
  577.     my (%results, $curtoken);
  578.     debug("tokenitems:  ", my_Dumper(%tokenitems));
  579.   topwhile:
  580.     while($line = <$fh>) {
  581. next if ($line =~ /^s*#/);
  582. my ($token, $rest) = ($line =~ /^(w+)s+(.*)/);
  583. next if (!defined($token) || !defined($rest));
  584. while ($token eq 'group') {
  585.     # handle special group list
  586.     my $next = read_config_group($fh, $rest,$type);
  587.     if ($next) {
  588. ($token, $rest) = ($next =~ /^(w+)s+(.*)/);
  589.     } else {
  590. next topwhile;
  591.     }
  592. }
  593. debug("token: $token => $restn");
  594. if ($token eq 'steal') {
  595.     foreach my $stealfrom (keys(%{$results{$rest}})) {
  596. if (!defined($results{$curtoken}{$stealfrom})) {
  597.     @{$results{$curtoken}{$stealfrom}} = 
  598. @{$results{$rest}{$stealfrom}};
  599. }
  600.     }
  601. } elsif (defined($tokenitems{$token})) {
  602.     if (!defined($curtoken)) {
  603. die "error in configuration file $itemfile, no token setn";
  604.     }
  605.     $rest =~ s/^#//;
  606.     push @{$results{$curtoken}{$token}},$rest;
  607. } elsif (defined($arrayitems{$token})) {
  608.     if (!defined($curtoken)) {
  609. die "error in configuration file $itemfile, no token setn";
  610.     }
  611.     my ($num, $newrest) = ($rest =~ /^(d+)s+(.*)/);
  612.     if (!defined($num) || !defined($newrest)) {
  613. warn "invalid config line: $linen";
  614.     } else {
  615. $results{$curtoken}{$token}[$num] = $newrest;
  616.     }
  617. } elsif ($token =~ /^tokens*$/) {
  618.     $rest = lc($rest);
  619.     $curtoken = $rest;
  620.     if (! exists $results{$curtoken}{'defined'}) {
  621. push @{$results{'thetokens'}}, $curtoken;
  622. $results{$curtoken}{'defined'} = 1;
  623.     }
  624.     $tokenmap{$curtoken} = $results{$curtoken};
  625.     debug("current token set to $tokenn");
  626. } else {
  627.     push @{$results{$token}},$rest;
  628. }
  629.     }
  630.     return %results;
  631. }
  633. sub debug {
  634.     print @_ if ($opts{'d'});
  635. }
  637. sub output_files {
  638.     my $filetypes = shift;
  639.     my $term = shift;
  640.     foreach my $ft (keys(%$filetypes)) {
  641. next if (!$didfile{$ft});
  642. my $outputf = $ft;
  643. if (-f $outputf && !$opts{'f'}) {
  644.     print "nError: An $outputf file already exists in this directory.nn";
  645.     my $ans = get_answer($term,"'overwrite', 'skip', 'rename' or 'append'? ",valid_answers(qw(o overwrite r rename s skip a append)));
  646.     next if ($ans =~ /^(s|skip)$/i);
  647.     if ($ans =~ /^(a|append)/) {
  648. $outputf = ">$outputf";
  649.     } elsif ($ans =~ /^(r|rename)$/i) {
  650. # default to rename for error conditions
  651. $outputf = $term->readline("Save to what new file name instead (or 'skip')? ");
  652.     }
  653. }
  654. $didfile{$ft} = $outputf;
  655. open(O,">$outputf") || warn "couldn't write to $outputfn";
  656. print O "#" x 75,"n";
  657. print O "#n# $ftn";
  658. print O "#n#   - created by the snmpconf configuration programn#n";
  659. foreach my $sect (@{$filetypes->{$ft}}) {
  660.     my $secthelp = 0;
  661.     foreach my $token (@{$sect->{'thetokens'}}) {
  662. if ($#{$sect->{$token}{'results'}} >= 0) {
  663.     if ($secthelp++ == 0) {
  664. print O "#" x 75,"n# SECTION: ",
  665. join("n#          ", @{$sect->{title}}), "n#n";
  666. print O "#   ", join("n#   ",@{$sect->{description}}),
  667. "n";
  668.     }
  669.     print O "n# $token: ",
  670.     join("n#   ",@{$sect->{$token}{info}}), "nn";
  671.     foreach my $result (@{$sect->{$token}{'results'}}) {
  672. print O "$resultn";
  673.     }
  674. }
  675.     }
  676.     print O "nnn";
  677. }
  678. if ($#{$unknown{$ft}} > -1) {
  679.     print O "#n# Unknown directives read in from other files by snmpconfn#n";
  680.     foreach my $unknown (@{$unknown{$ft}}) {
  681. print O $unknown,"n";
  682.     }
  683. }
  684. close(O);
  685.     }
  686. }
  688. sub get_answer {
  689.     my ($term, $question, $regexp, $defaultval) = @_;
  690.     $question .= " (default = $defaultval)" if (defined($defaultval) && $defaultval ne "");
  691.     $question .= ": ";
  692.     my $ans = $term->readline($question);
  693.     return $defaultval if ($ans eq "" && defined($defaultval) && 
  694.    $defaultval ne "");
  695.     while (!(!defined($regexp) ||
  696.      $regexp eq "" ||
  697.      $ans =~ /$regexp/)) {
  698. print "invalid answer!  It must match this regular expression: $regexpn";
  699. $ans = $term->readline($question);
  700.     }
  701.     return $defaultval if ($ans eq "" && defined($defaultval) && 
  702.    $defaultval ne "");
  703.     return $ans;
  704. }
  705.     
  706. sub valid_answers {
  707.     my @list;
  708.     foreach $i (@_) {
  709. push @list, $i if ($i);
  710.     }
  711.     return "^(" . join("|",@list) . ")$";
  712. }
  714. sub read_config {
  715.     my $file = shift;
  716.     my $filetype = shift;
  717.     return if (!defined($filetypes{$filetype}));
  718.     if (! -f $file) {
  719. warn "$file does not existn";
  720. return;
  721.     }
  722.     open(I,$file);
  723.     while(<I>) {
  724. next if (/^s*#/);
  725. next if (/^s*$/);
  726. chomp;
  727. my ($token, $rest) = /^s*(w+)s+(.*)/;
  728. $token = lc($token);
  729. next if (defined($alllines{$_})); # drop duplicate lines
  730. if (defined($tokenmap{$token})) {
  731.     push @{$tokenmap{$token}{'results'}},$_;
  732. } else {
  733.     push @{$unknown{$filetype}},$_;
  734. }
  735. $alllines{$_}++;
  736.     }
  737.     close(I);
  738. }
  740. sub display_menu {
  741.     my %config;
  743.     while ($#_ > -1 && $_[0] =~ /^-/) {
  744. my $key = shift;
  745. $config{$key} = shift;
  746.     }
  748.     my $count=1;
  749.     print "n" if (!defined($config{'-dense'}));
  750.     if ($config{'-head'}) {
  751. print $config{'-head'};
  752. print "n" if (!defined($config{'-dense'}));
  753.     }
  754.     my @answers = @_;
  755.     my @list;
  756.     if (defined($config{'-descriptions'}) && 
  757. ref($config{'-descriptions'}) eq "ARRAY") {
  758. @list = @{$config{'-descriptions'}}
  759.     } else {
  760. @list = @_;
  761.     }
  762.     foreach my $i (@list) {
  763. printf "  %2d:  $in", $count++ if ($i);
  764.     }
  765.     print "n" if (!defined($config{'-dense'}));
  766.     if (defined($config{'-otheranswers'})) {
  767. if (ref($config{'-otheranswers'}) eq 'ARRAY') {
  768.     print "Other options: ", join(", ",
  769.   @{$config{'-otheranswers'}}), "n";
  770.     push @answers, @{$config{'-otheranswers'}};
  771.     push @answers, keys(%{$config{'-mapanswers'}});
  772. } else {
  773.     my $maxlen = 0;
  774.     push @answers,keys(%{$config{'-otheranswers'}});
  775.     foreach my $i (keys(%{$config{'-otheranswers'}})) {
  776. $maxlen = length($i) if (length($i) > $maxlen);
  777.     }
  778.     foreach my $i (keys(%{$config{'-otheranswers'}})) {
  779. printf("  %-" . $maxlen . "s: %sn", $i, 
  780.        $config{'-otheranswers'}{$i});
  781.     }
  782. }
  783. print "n" if (!defined($config{'-dense'}));
  784.     }
  785.     if ($config{'-tail'}) {
  786. print $config{'-tail'};
  787. print "n" if (!defined($config{'-dense'}));
  788.     }
  790.     if (defined($config{'-question'})) {
  791. while(1) {
  792.     my $numexpr;
  793.     if ($config{'-multiple'}) {
  794. $numexpr = '[ds,]+|all|a|none|n';
  795.     } else {
  796. $numexpr = 'd+';
  797.     }
  798.     push @answers,"" if ($config{'-defaultvalue'});
  799.     $ans = get_answer($term, $config{'-question'},
  800.       valid_answers($numexpr,@answers),
  801.       $config{'-defaultvalue'});
  802.     if ($config{'-mapanswers'}{$ans}) {
  803. $ans = $config{'-mapanswers'}{$ans};
  804.     }
  806.     if ($ans =~ /^$numexpr$/) {
  807. if ($config{'-multiple'}) {
  808.     my @list = split(/s*,s*/,$ans);
  809.     my @ret;
  810.     $count = 0;
  811.     foreach my $i (@_) {
  812. $count++;
  813. if ($ans eq "all" || $ans eq "a" 
  814.     || grep(/^$count$/,@list)) {
  815.     push @ret, $i;
  816. }
  817.     }
  818.     return @ret;
  819. } else {
  820.     if ($ans <= 0 || $ans > $#_+1) {
  821. warn "invalid selection: $ans [must be 1-" .
  822.     ($#_+1) . "]n";
  823.     } else {
  824. return $ans if ($config{'-numeric'});
  825. $count = 0;
  826. foreach my $i (@_) {
  827.     $count++;
  828.     if ($ans eq $count) {
  829. return $i;
  830.     }
  831. }
  832.     }
  833. }
  834.     } else {
  835. return $ans;
  836.     }
  837. }
  838.     }
  839. }
  841. sub my_Dumper {
  842.     if ($opts{'D'}) {
  843. return Dumper(@_);
  844.     } else {
  845. return "n";
  846.     }
  847. }
  849. sub get_persistentDir {
  850.     my $file = shift;
  851.     my $result = 0;
  852.     if (! -f $file) {
  853. return 0;
  854.     }
  855.     open(I,$file);
  856.     while(<I>) {
  857. next if (/^s*#/);
  858. next if (/^s*$/);
  859. chomp;
  860. my ($token, $rest) = /^s*(w+)s+(.*)/;
  861.         if (lc($token) eq "persistentdir") {
  862.           $result = $rest;
  863.         }
  864.         next;
  865.     }
  866.     close(I);
  867.     return $result;
  868. }
  870. # Usage:   &win32_reg_read("key", "value")
  871. # Example: &win32_reg_read("SOFTWARE\Net-SNMP","SNMPSHAREPATH");
  872. # Returns: Value if found in HKCU or HCLM.  Otherwise an empty string.
  873. sub win32_reg_read {
  874.   my $sub_key = shift;
  875.   my $value = shift;
  877.   require Win32::Registry;
  878.    
  879.   my ($hkey, %key_values, $temp, $no_warn);
  880.  
  881.   # Try HKCU first
  882.   $no_warn = $HKEY_CURRENT_USER;
  883.   if ($HKEY_CURRENT_USER->Open($sub_key, $hkey))
  884.   {
  885.     $hkey->GetValues(%key_values);   
  886.     foreach $temp (sort keys %key_values) {
  887.       if ($temp eq $value) {
  888.         return $key_values{$temp}[2];
  889.       }
  890.     }
  891.     $hkey->Close(); 
  892.   }
  894.   # Try HKLM second
  895.   $no_warn = $HKEY_LOCAL_MACHINE;
  896.   if ($HKEY_LOCAL_MACHINE->Open($sub_key, $hkey))
  897.   {
  898.     $hkey->GetValues(%key_values);
  899.     foreach $temp (sort keys %key_values) {
  900.       if ($temp eq $value) {
  901.         return $key_values{$temp}[2];
  902.       }
  903.     }
  904.     $hkey->Close(); 
  905.   }
  906.   return "";
  907. }
  909. # Usage:   &my_getenv("key")
  910. # Example: &my_getenv("SNMPSHAREPATH");
  911. # Returns: Unix:  Environment variable value (undef if not defined)
  912. #          Win32: HKCUSoftwareNet-SNMP(key) or
  913. #          Win32: HKLMSoftwareNet-SNMP(key) or
  914. #          Win32: Environment variable value (undef if not defined)
  915. sub my_getenv {
  916.   my $key = shift;
  917.   
  918.   # Unix
  919.   if ($^O ne 'MSWin32') {
  920.     return $ENV{$key};
  921.   }
  922.   # Windows
  923.   else {
  924.     my $temp = &win32_reg_read("SOFTWARE\Net-SNMP","$key");
  925.     if ($temp ne "") {
  926.       return $temp;
  927.     }
  928.     else {
  929.       return $ENV{$key};
  930.     }
  931.   }
  932. }