home *** CD-ROM | disk | FTP | other *** search
/ Cricao de Sites - 650 Layouts Prontos / WebMasters.iso / Servidores / xampp-win32-1.6.7-installer.exe / mysql / scripts / mysql_setpermission < prev    next >
Encoding:
Text File  |  2008-04-17  |  16.3 KB  |  655 lines
  1. #!@PERL@
  2. ## Emacs, this is -*- perl -*- mode? :-)
  3. ##
  4. ##        Permission setter for MySQL
  5. ##
  6. ##        mady by Luuk de Boer (luuk@wxs.nl) 1998.
  7. ##        it's made under GPL ...:-))
  8. ##
  9. ##
  10. ############################################################################
  11. ## History
  12. ##
  13. ## 1.0 first start of the program
  14. ## 1.1 some changes from monty and after that
  15. ##     initial release in mysql 3.22.10 (nov 1998)
  16. ## 1.2 begin screen now in a loop + quit is using 0 instead of 9
  17. ##     after ideas of Paul DuBois.
  18. ## 1.2a Add Grant, References, Index and Alter privilege handling (Monty)
  19. ## 1.3 Applied patch provided by Martin Mokrejs <mmokrejs@natur.cuni.cz>
  20. ##     (General code cleanup, use the GRANT statement instead of updating
  21. ##     the privilege tables directly, added option to revoke privileges)
  22. ## 1.4 Remove option 6 which attempted to erroneously grant global privileges
  23.  
  24. #### TODO
  25. #
  26. # empty ... suggestions ... mail them to me ...
  27.  
  28.  
  29. $version="1.4";
  30.  
  31. use DBI;
  32. use Getopt::Long;
  33. use strict;
  34. use vars qw($dbh $sth $hostname $opt_user $opt_password $opt_help $opt_host
  35.         $opt_socket $opt_port $host $version);
  36.  
  37. my $sqlhost = "";
  38. my $user = "";
  39.  
  40. $dbh=$host=$opt_user= $opt_password= $opt_help= $opt_host= $opt_socket= "";
  41. $opt_port=0;
  42.  
  43. read_my_cnf();        # Read options from ~/.my.cnf
  44.  
  45. GetOptions("user=s","password=s","help","host=s","socket=s","port=i");
  46.  
  47. usage() if ($opt_help); # the help function
  48.  
  49. if ($opt_host eq '')
  50. {
  51.   $sqlhost = "localhost";
  52. }
  53. else
  54. {
  55.   $sqlhost = $opt_host;
  56. }
  57.  
  58. # ask for a password if no password is set already
  59. if ($opt_password eq '')
  60. {
  61.   system "stty -echo";
  62.   print "Password for user $opt_user to connect to MySQL: ";
  63.   $opt_password = <STDIN>;
  64.   chomp($opt_password);
  65.   system "stty echo";
  66.   print "\n";
  67. }
  68.  
  69.  
  70. # make the connection to MySQL
  71. $dbh= DBI->connect("DBI:mysql:mysql:host=$sqlhost:port=$opt_port:mysql_socket=$opt_socket",$opt_user,$opt_password, {PrintError => 0}) ||
  72.   die("Can't make a connection to the mysql server.\n The error: $DBI::errstr");
  73.  
  74. # the start of the program
  75. &q1();
  76. exit(0); # the end...
  77.  
  78. #####
  79. # below all subroutines of the program
  80. #####
  81.  
  82. ###
  83. # the beginning of the program
  84. ###
  85. sub q1 { # first question ...
  86.   my ($answer,$end);
  87.   while (! $end) {
  88.     print "#"x70;
  89.     print "\n";
  90.     print "## Welcome to the permission setter $version for MySQL.\n";
  91.     print "## made by Luuk de Boer\n";
  92.     print "#"x70;
  93.     print "\n";
  94.     print "What would you like to do:\n";
  95.     print "  1. Set password for an existing user.\n";
  96.     print "  2. Create a database + user privilege for that database\n";
  97.         print "     and host combination (user can only do SELECT)\n";
  98.     print "  3. Create/append user privilege for an existing database\n";
  99.         print "     and host combination (user can only do SELECT)\n";
  100.     print "  4. Create/append broader user privileges for an existing\n";
  101.         print "     database and host combination\n";
  102.         print "     (user can do SELECT,INSERT,UPDATE,DELETE)\n";
  103.     print "  5. Create/append quite extended user privileges for an\n";
  104.         print "     existing database and host combination (user can do\n";
  105.         print "     SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,\n";
  106.         print "     LOCK TABLES,CREATE TEMPORARY TABLES)\n";
  107.     print "  6. Create/append full privileges for an existing database\n";
  108.         print "     and host combination (user has FULL privilege)\n";
  109.     print "  7. Remove all privileges for for an existing database and\n";
  110.         print "     host combination.\n";
  111.     print "     (user will have all permission fields set to N)\n";
  112.     print "  0. exit this program\n";
  113.     print "\nMake your choice [1,2,3,4,5,6,7,0]: ";
  114.     while (<STDIN>) {
  115.       $answer = $_;
  116.       chomp($answer);
  117.       if ($answer =~ /^[1234567]$/) {
  118.         if ($answer == 1) {
  119.        setpwd();
  120.         } elsif ($answer =~ /^[234567]$/) {
  121.        addall($answer);
  122.     } else {
  123.           print "Sorry, something went wrong. With such option number you should not get here.\n\n";
  124.           $end = 1;
  125.         }
  126.       } elsif ($answer == 0) {
  127.         print "We hope we can help you next time \n\n";
  128.     $end = 1;
  129.       } else {
  130.         print "Your answer was $answer\n";
  131.         print "and that's wrong .... Try again\n";
  132.       }
  133.       last;
  134.     }
  135.   }
  136. }
  137.  
  138. ###
  139. # set a password for a user
  140. ###
  141. sub setpwd
  142. {
  143.   my ($user,$pass,$host) = "";
  144.   print "\n\nSetting a (new) password for a user.\n";
  145.  
  146.   $user = user();
  147.   $pass = newpass($user);
  148.   $host = hosts($user);
  149.  
  150.   print "#"x70;
  151.   print "\n\n";
  152.   print "That was it ... here is an overview of what you gave to me:\n";
  153.   print "The username         : $user\n";
  154. #  print "The password        : $pass\n";
  155.   print "The host        : $host\n";
  156.   print "#"x70;
  157.   print "\n\n";
  158.   print "Are you pretty sure you would like to implement this [yes/no]: ";
  159.   my $no = <STDIN>;
  160.   chomp($no);
  161.   if ($no =~ /n/i)
  162.   {
  163.     print "Okay .. that was it then ... See ya\n\n";
  164.     return(0);
  165.   }
  166.   else
  167.   {
  168.     print "Okay ... let's go then ...\n\n";
  169.   }
  170.   $user = $dbh->quote($user);
  171.   $host = $dbh->quote($host);
  172.   if ($pass eq '')
  173.   {
  174.     $pass = "''";
  175.   }
  176.   else
  177.   {
  178.     $pass = "PASSWORD(". $dbh->quote($pass) . ")";
  179.   }
  180.   my $sth = $dbh->prepare("update user set Password=$pass where User = $user and Host = $host") || die $dbh->errstr;
  181.   $sth->execute || die $dbh->errstr;
  182.   $sth->finish;
  183.   print "The password is set for user $user.\n\n";
  184.  
  185. }
  186.  
  187. ###
  188. # all things which will be added are done here
  189. ###
  190. sub addall {
  191.   my ($todo) = @_;
  192.   my ($answer,$good,$db,$user,$pass,$host,$priv);
  193.  
  194.   if ($todo == 2) {
  195.     $db = newdatabase();
  196.   } else {
  197.     $db = database();
  198.   }
  199.  
  200.   $user = newuser();
  201.   $pass = newpass("$user");
  202.   $host = newhosts();
  203.  
  204.   print "#"x70;
  205.   print "\n\n";
  206.   print "That was it ... here is an overview of what you gave to me:\n";
  207.   print "The database name    : $db\n";
  208.   print "The username         : $user\n";
  209. #  print "The password        : $pass\n";
  210.   print "The host(s)        : $host\n";
  211.   print "#"x70;
  212.   print "\n\n";
  213.   print "Are you pretty sure you would like to implement this [yes/no]: ";
  214.   my $no = <STDIN>;
  215.   chomp($no);
  216.   if ($no =~ /n/i) {
  217.     print "Okay .. that was it then ... See ya\n\n";
  218.     return(0);
  219.   } else {
  220.     print "Okay ... let's go then ...\n\n";
  221.   }
  222.  
  223.   if ($todo == 2) {
  224.     # create the database
  225.     if ($db) {
  226.     my $sth = $dbh->do("CREATE DATABASE $db") || $dbh->errstr;
  227.     } else {
  228.       print STDERR "What do you want? You wanted to create new database and add new user, right?\n";
  229.       die "But then specify databasename, please\n";
  230.   }
  231.   }
  232.  
  233.   if ( ( !$todo ) or not ( $todo =~ m/^[2-7]$/ ) ) {
  234.     print STDERR "Sorry, select option $todo isn't known inside the program .. See ya\n";
  235.     quit();
  236.   }
  237.  
  238.   my @hosts = split(/,/,$host);
  239.   if (!$user) {
  240.      die "username not specified: $user\n";
  241.   }
  242.   if (!$db) {
  243.      die "databasename is not specified nor *\n";
  244.   }
  245.   foreach $host (@hosts) {
  246.     # user privileges: SELECT
  247.     if (($todo == 2) || ($todo == 3)) {
  248.       $sth = $dbh->do("GRANT SELECT ON $db.* TO $user@\"$host\" IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  249.     } elsif ($todo == 4) {
  250.       # user privileges: SELECT,INSERT,UPDATE,DELETE
  251.       $sth = $dbh->do("GRANT SELECT,INSERT,UPDATE,DELETE ON $db.* TO $user@\"$host\" IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  252.     } elsif ($todo == 5) {
  253.       # user privileges: SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,LOCK TABLES,CREATE TEMPORARY TABLES
  254.       $sth = $dbh->do("GRANT SELECT,INSERT,UPDATE,DELETE,CREATE,DROP,INDEX,LOCK TABLES,CREATE TEMPORARY TABLES ON $db.* TO $user@\"$host\" IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  255.     } elsif ($todo == 6) {
  256.        # all privileges
  257.        $sth = $dbh->do("GRANT ALL ON $db.* TO \'$user\'\@\'$host\' IDENTIFIED BY \'$pass\'") || die $dbh->errstr;
  258.     } elsif ($todo == 7) {
  259.        # all privileges set to N
  260.        $sth = $dbh->do("REVOKE ALL ON *.* FROM \'$user\'\@\'$host\'") || die $dbh->errstr;
  261.     }
  262.     }
  263.   $dbh->do("FLUSH PRIVILEGES") || print STDERR "Can't flush privileges\n";
  264.   print "Everything is inserted and mysql privileges have been reloaded.\n\n";
  265. }
  266.  
  267. ###
  268. # ask for a new database name
  269. ###
  270. sub newdatabase {
  271.   my ($answer,$good,$db);
  272.   print "\n\nWhich database would you like to add: ";
  273.   while (<STDIN>) {
  274.     $answer = $_;
  275.     $good = 0;
  276.     chomp($answer);
  277.     if ($answer) {
  278.       my $sth = $dbh->prepare("SHOW DATABASES") || die $dbh->errstr;
  279.       $sth->execute || die $dbh->errstr;
  280.       while (my @r = $sth->fetchrow_array) {
  281.         if ($r[0] eq $answer) {
  282.           print "\n\nSorry, this database name is already in use; try something else: ";
  283.           $good = 1;
  284.         }
  285.       }
  286.     } else {
  287.       print "You must type something ...\nTry again: ";
  288.       next;
  289.     }
  290.     last if ($good == 0);
  291.   }
  292.   $db = $answer;
  293.   print "The new database $db will be created\n";
  294.   return($db);
  295. }
  296.  
  297. ###
  298. # select a database
  299. ###
  300. sub database {
  301.   my ($answer,$good,$db);
  302.   print "\n\nWhich database from existing databases would you like to select: \n";
  303.   print "You can choose from: \n";
  304.   my $sth = $dbh->prepare("show databases") || die $dbh->errstr;
  305.   $sth->execute || die $dbh->errstr;
  306.   while (my @r = $sth->fetchrow_array) {
  307.     print "  - $r[0] \n";
  308.   }
  309.   print "Which database will it be (case sensitive). Type * for any: \n";
  310.   while (<STDIN>) {
  311.     $answer = $_;
  312.     $good = 0;
  313.     chomp($answer);
  314.     if ($answer) {
  315.       if ($answer eq "*") {
  316.         print "OK, the user entry will NOT be limited to any database";
  317.         return("*");
  318.       }
  319.       my $sth = $dbh->prepare("show databases") || die $dbh->errstr;
  320.       $sth->execute || die $dbh->errstr;
  321.       while (my @r = $sth->fetchrow_array) {
  322.         if ($r[0] eq $answer) {
  323.           $good = 1;
  324.           $db = $r[0];
  325.           last;
  326.         }
  327.       }
  328.     } else {
  329.         print "Type either database name or * meaning any databasename. That means";
  330.         print " any of those above but also any which will be created in future!";
  331.         print " This option gives a user chance to operate on databse mysql, which";
  332.         print " contains privilege settings. That is really risky!\n";
  333.       next;
  334.     }
  335.     if ($good == 1) {
  336.       last;
  337.     } else {
  338.       print "You must select one from the list.\nTry again: ";
  339.       next;
  340.     }
  341.   }
  342.   print "The database $db will be used.\n";
  343.   return($db);
  344. }
  345.  
  346. ###
  347. # ask for a new username
  348. ###
  349. sub newuser
  350. {
  351.   my $user = "";
  352.   my $answer = "";
  353.  
  354.   print "\nWhat username is to be created: ";
  355.   while(<STDIN>)
  356.   {
  357.     $answer = $_;
  358.     chomp($answer);
  359.     if ($answer)
  360.     {
  361.       $user = $answer;
  362.     }
  363.     else
  364.     {
  365.       print "You must type something ...\nTry again: ";
  366.       next;
  367.     }
  368.     last;
  369.   }
  370.   print "Username = $user\n";
  371.   return($user);
  372. }
  373.  
  374. ###
  375. # ask for a user which is already in the user table
  376. ###
  377. sub user
  378. {
  379.   my ($answer,$user);
  380.  
  381.   print "\nFor which user do you want to specify a password: ";
  382.   while(<STDIN>)
  383.   {
  384.     $answer = $_;
  385.     chomp($answer);
  386.     if ($answer)
  387.     {
  388.       my $sth = $dbh->prepare("select User from user where User = '$answer'") || die $dbh->errstr;
  389.       $sth->execute || die $dbh->errstr;
  390.       my @r = $sth->fetchrow_array;
  391.       if ($r[0])
  392.       {
  393.         $user = $r[0];
  394.       }
  395.       else
  396.       {
  397.        print "Sorry, user $answer isn't known in the user table.\nTry again: ";
  398.        next;
  399.      }
  400.     }
  401.     else
  402.     {
  403.       print "You must type something ...\nTry again: ";
  404.       next;
  405.     }
  406.     last;
  407.   }
  408.   print "Username = $user\n";
  409.   return($user);
  410. }
  411.  
  412. ###
  413. # ask for a new password
  414. ###
  415. sub newpass
  416. {
  417.   my ($user) = @_;
  418.   my ($pass,$answer,$good,$yes);
  419.  
  420.   print "Would you like to set a password for $user [y/n]: ";
  421.   $yes = <STDIN>;
  422.   chomp($yes);
  423.   if ($yes =~ /y/)
  424.   {
  425.     system "stty -echo";
  426.     print "What password do you want to specify for $user: ";
  427.     while(<STDIN>)
  428.     {
  429.       $answer = $_;
  430.       chomp($answer);
  431.       system "stty echo";
  432.       print "\n";
  433.       if ($answer)
  434.       {
  435.         system "stty -echo";
  436.         print "Type the password again: ";
  437.         my $second = <STDIN>;
  438.         chomp($second);
  439.         system "stty echo";
  440.         print "\n";
  441.         if ($answer ne $second)
  442.         {
  443.           print "Passwords aren't the same; we begin from scratch again.\n";
  444.           system "stty -echo";
  445.           print "Password please: ";
  446.           next;
  447.         }
  448.         else
  449.         {
  450.           $pass = $answer;
  451.         }
  452.       }
  453.       else
  454.       {
  455.         print "You must type something ...\nTry again: ";
  456.         next;
  457.       }
  458.       last;
  459.     }
  460. #    print "The password for $user is $pass.\n";
  461.   }
  462.   else
  463.   {
  464.     print "We won't set a password so the user doesn't have to use it\n";
  465.     $pass = "";
  466.   }
  467.   return($pass);
  468. }
  469.  
  470. ###
  471. # ask for new hosts
  472. ###
  473. sub newhosts
  474. {
  475.   my ($host,$answer,$good);
  476.  
  477.   print "We now need to know from what host(s) the user will connect.\n";
  478.   print "Keep in mind that % means 'from any host' ...\n";
  479.   print "The host please: ";
  480.   while(<STDIN>)
  481.   {
  482.     $answer = $_;
  483.     chomp($answer);
  484.     if ($answer)
  485.     {
  486.       $host .= ",$answer";
  487.       print "Would you like to add another host [yes/no]: ";
  488.       my $yes = <STDIN>;
  489.       chomp($yes);
  490.       if ($yes =~ /y/i)
  491.       {
  492.         print "Okay, give us the host please: ";
  493.         next;
  494.       }
  495.       else
  496.       {
  497.         print "Okay we keep it with this ...\n";
  498.       }
  499.     }
  500.     else
  501.     {
  502.       print "You must type something ...\nTry again: ";
  503.       next;
  504.     }
  505.     last;
  506.   }
  507.   $host =~ s/^,//;
  508.   print "The following host(s) will be used: $host.\n";
  509.   return($host);
  510. }
  511.  
  512. ###
  513. # ask for a host which is already in the user table
  514. ###
  515. sub hosts
  516. {
  517.   my ($user) = @_;
  518.   my ($answer,$good,$host);
  519.  
  520.   print "We now need to know which host for $user we have to change.\n";
  521.   print "Choose from the following hosts: \n";
  522.   $user = $dbh->quote($user);
  523.   my $sth = $dbh->prepare("select Host,User from user where User = $user") || die $dbh->errstr;
  524.   $sth->execute || die $dbh->errstr;
  525.   while (my @r = $sth->fetchrow_array)
  526.   {
  527.     print "  - $r[0] \n";
  528.   }
  529.   print "The host please (case sensitive): ";
  530.   while(<STDIN>)
  531.   {
  532.     $answer = $_;
  533.     chomp($answer);
  534.     if ($answer)
  535.     {
  536.       $sth = $dbh->prepare("select Host,User from user where Host = '$answer' and User = $user") || die $dbh->errstr;
  537.       $sth->execute || die $dbh->errstr;
  538.       my @r = $sth->fetchrow_array;
  539.       if ($r[0])
  540.       {
  541.         $host = $answer;
  542.         last;
  543.       }
  544.       else
  545.       {
  546.         print "You have to select a host from the list ...\nTry again: ";
  547.         next;
  548.       }
  549.     }
  550.     else
  551.     {
  552.       print "You have to type something ...\nTry again: ";
  553.       next;
  554.     }
  555.     last;
  556.   }
  557.   print "The following host will be used: $host.\n";
  558.   return($host);
  559. }
  560.  
  561. ###
  562. # a nice quit (first disconnect and then exit
  563. ###
  564. sub quit
  565. {
  566.   $dbh->disconnect;
  567.   exit(0);
  568. }
  569.  
  570. ###
  571. # Read variables password, port and socket from .my.cnf under the client
  572. # or perl groups
  573. ###
  574.  
  575. sub read_my_cnf
  576. {
  577.   open(TMP,$ENV{'HOME'} . "/.my.cnf") || return 1;
  578.   while (<TMP>)
  579.   {
  580.     if (/^\[(client|perl)\]/i)
  581.     {
  582.       while ((defined($_=<TMP>)) && !/^\[\w+\]/)
  583.       {
  584.     print $_;
  585.     if (/^host\s*=\s*(\S+)/i)
  586.     {
  587.       $opt_host = $1;
  588.     }
  589.     elsif (/^user\s*=\s*(\S+)/i)
  590.     {
  591.       $opt_user = $1;
  592.     }
  593.     elsif (/^password\s*=\s*(\S+)/i)
  594.     {
  595.       $opt_password = $1;
  596.     }
  597.     elsif (/^port\s*=\s*(\S+)/i)
  598.     {
  599.       $opt_port = $1;
  600.     }
  601.     elsif (/^socket\s*=\s*(\S+)/i)
  602.     {
  603.       $opt_socket = $1;
  604.     }
  605.       }
  606.     }
  607.   }
  608.   close(TMP);
  609. }
  610.  
  611. ###
  612. # the help text
  613. ###
  614. sub usage
  615. {
  616.   print <<EOL;
  617. ----------------------------------------------------------------------
  618.                  The permission setter for MySQL.
  619.                       version: $version
  620.  
  621.                  made by: Luuk de Boer <luuk\@wxs.nl>
  622. ----------------------------------------------------------------------
  623.  
  624. The permission setter is a little program which can help you add users
  625. or databases or change passwords in MySQL. Keep in mind that we don't
  626. check permissions which already been set in MySQL. So if you can't
  627. connect to MySQL using the permission you just added, take a look at
  628. the permissions which have already been set in MySQL.
  629.  
  630. The permission setter first reads your .my.cnf file in your Home
  631. directory if it exists.
  632.  
  633. Options for the permission setter:
  634.  
  635. --help        : print this help message and exit.
  636.  
  637. The options shown below are used for making the connection to the MySQL
  638. server. Keep in mind that the permissions for the user specified via
  639. these options must be sufficient to add users / create databases / set
  640. passwords.
  641.  
  642. --user        : is the username to connect with.
  643. --password    : the password of the username.
  644. --host        : the host to connect to.
  645. --socket    : the socket to connect to.
  646. --port        : the port number of the host to connect to.
  647.  
  648. If you don't give a password and no password is set in your .my.cnf
  649. file, then the permission setter will ask for a password.
  650.  
  651.  
  652. EOL
  653. exit(0);
  654. }
  655.