fjernet heng ved forsøk på drop av ikke-eksisterende databaser.

fjernet heng ved forsøk på create av eksisterende databaser.
fjernet mulighet for buffer overflows i databasenavnet.

mysql-dbadm.c

@@ -1,5 +1,5 @@
 /*
- * @(#) $Header: /tmp/cvs/mysql-admutils/mysql-dbadm.c,v 1.11 2003-04-08 18:31:53 lkarsten Exp $
+ * @(#) $Header: /tmp/cvs/mysql-admutils/mysql-dbadm.c,v 1.12 2004-11-16 20:56:51 lkarsten Exp $
  *
  * mysql-dbadm.c
  *
@@ -87,9 +87,19 @@ usage()
 int
 create(MYSQL *pmysql, char *db)
 {
-  if (mysql_create_db(pmysql, db))
+  // hvis man forsøker å dra mysql_create_db() på en database som
+  // allerede finnes, så henger bare hele sql-kallet. Vi må derfor
+  // forsøke å selecte databasen først. Ettersom man alltid er root,
+  // går jo dette vanligvis bra.
+  // finner ut om denne finnes fra før.
+  if (!mysql_select_db(pmysql, db)) {
+      return dberror(pmysql, "Database '%s' already exists.", db);
+  } 
+  mysql_select_db(pmysql, "mysql");
+  // oppretter databasen.
+  if (mysql_create_db(pmysql, db)) {
       return dberror(pmysql, "Cannot create database '%s'.", db);
-
+  }
   fprintf(stderr, "Database '%s' created.\n", db);
   return 0;
 }
@@ -104,6 +114,12 @@ drop(MYSQL *pmysql, char *db)
   if (mysql_query(pmysql, query))
     dberror(pmysql, "Failed to delete permissions for database '%s'.", db);
 
+  if (mysql_select_db(pmysql, db)) {
+      dberror(pmysql, "Database '%s' doesn't exists.", db);
+      return 0;
+  } 
+  mysql_select_db(pmysql, "mysql");
+
   if (mysql_drop_db(pmysql, db))
     return dberror(pmysql, "Cannot drop database '%s'.", db);
 
@@ -406,6 +422,7 @@ main(int argc, char *argv[])
   enum { c_create, c_drop, c_editperm, c_show } command;
   MYSQL mysql;
   char **dblist, **p;
+  char *db;
 
   program_name = argv[0];
 
@@ -464,37 +481,38 @@ main(int argc, char *argv[])
 	}
       free(dblist);
     }
-  else
+  else {
-    {
+      db = malloc(64); 
       /* for each supplied database name, perform the requested action */
-      for (i = 2; i < argc; i++)
+      for (i = 2; i < argc; i++) {
-	{
+          // HE HE
-	  if (! (owner(argv[i]) || member(argv[i]))) 
+          strncpy(db, argv[i], 32);
-	    {
+          db[33] = '\0';
+
+	  if (! (owner(db) || member(db))) {
 	      dberror(NULL, "You are not the owner of '%s'.  Skipping.",
-		      argv[i]);
+                      db);
 	      continue;
 	  }
 
-	  switch (command)
+	  switch (command) {
-	    {
 	    case c_create:
-	      create(&mysql, argv[i]);
+	      create(&mysql, db);
 	      break;
 	    case c_drop:
-	      drop(&mysql, argv[i]);
+	      drop(&mysql, db);
 	      break;
 	    case c_editperm:
-	      editperm(&mysql, argv[i]);
+	      editperm(&mysql, db);
 	      break;
 	    case c_show:
-	      show(&mysql, argv[i]);
+	      show(&mysql, db);
 	      break;
 	    default:
 	      return dberror(NULL, "This point should never be reached!");
 	  }
-	}
+        } // for
-    } 
+    }  // else
 
   mysql_reload(&mysql);
   mysql_close(&mysql);