Code

Replaced in_array calls for gosa-plugins
[gosa.git] / gosa-plugins / fai / admin / fai / class_faiPartitionTable.inc
index ad1cc3a570a61d54356bebe72ba8f9799ad3d702..9ed849507a86cef41340c67ce77f003287dfcf89 100644 (file)
@@ -8,7 +8,7 @@ class faiPartitionTable extends plugin
   var $objectclasses    = array("top","FAIclass","FAIpartitionTable");
 
   var $subAttributes      = array("cn","description");
-  var $subPartAttributes  = array("cn","FAIpartitionNr","FAIpartitionSize","FAImountPoint","FAIfsType","FAIpartitionType","FAImountOptions","FAIfsOptions","FAIpartitionFlags","description","FAIfsCreateOptions","FAIfsTuneOptions","FAIfsOptions","FAIpartitionFlags");
+  var $subPartAttributes  = array("cn","FAIpartitionNr","FAIpartitionSize","FAImountPoint","FAIfsType","FAIpartitionType","FAImountOptions","FAIfsOptions","FAIpartitionFlags","description","FAIfsCreateOptions","FAIfsTuneOptions","FAIfsOptions","FAIpartitionFlags","FAIlvmDevice");
 
   var $sub64coded = array();
   var $subBinary = array();
@@ -46,7 +46,6 @@ class faiPartitionTable extends plugin
         if(!preg_match("/".preg_quote($this->dn, '/')."$/i",$obj['dn'])) continue;
         $objects = array();
         $objects['description']  = "";
-        $objects['status']      = "edited";
 
         // Transform disk type into image later...
         if (!isset($obj['FAIdiskType'])){
@@ -54,11 +53,21 @@ class faiPartitionTable extends plugin
         } else {
           $objects['FAIdiskType']        = $obj['FAIdiskType'];
         }
-        $objects['FAIdiskOption']        = $obj['FAIdiskOption'];
+
+        // Get disk options, without 'count' index. 
+        $objects['FAIdiskOption'] = array();
+        if (isset($obj['FAIdiskOption'])){
+          for($i=0;$i<$obj['FAIdiskOption']['count'];$i++){ 
+            $objects['FAIdiskOption'][] = $obj['FAIdiskOption'][$i];
+          }
+        }
 
         // Transform potential lvm information
         if (isset($obj['FAIlvmDevice'])){
-          $objects['FAIlvmDevice'] = $obj['FAIlvmDevice'];
+          for($i=0;$i<$obj['FAIlvmDevice']['count'];$i++){
+            $name = $obj['FAIlvmDevice'][$i]; 
+            $objects['FAIlvmDevice'][$name] = $name;
+          }
         }
 
         $objects['dn']          = $obj['dn'];
@@ -77,7 +86,6 @@ class faiPartitionTable extends plugin
           if(!preg_match("/".preg_quote($disk['dn'], '/')."$/i",$obj['dn'])) continue;
 
           $objects = array();
-          $objects['status']      = "edited";
           $objects['dn']          = $obj['dn'];
           $objects                = $this->get_object_attributes($objects,$this->subPartAttributes);
           unset($objects['dn']);;
@@ -86,6 +94,11 @@ class faiPartitionTable extends plugin
       }
     }
 
+    /* Keep track of disk changes, by comparing the initial and resulting 
+     *  disk setup.
+     */
+    $this->initial_disks = $this->disks;
+
     $this->is_new = FALSE;
     if($this->dn == "new"){
       $this->is_new =TRUE;
@@ -134,7 +147,7 @@ class faiPartitionTable extends plugin
       if ($this->FAIpartitionMethod == "setup-storage") {
         if(isset($_POST['AddDisk'])) $type = "disk";
         if(isset($_POST['AddRaid'])) $type = "raid";
-        if(isset($_POST['AddAddVolgroup'])) $type = "lvm";
+        if(isset($_POST['AddVolgroup'])) $type = "lvm";
         $this->dialog = new faiDiskEntry($this->config,$this->dn,$this, array(),$type); 
       } else {
         $this->dialog = new faiPartitionTableEntry($this->config,$this->dn,$this); 
@@ -157,131 +170,120 @@ class faiPartitionTable extends plugin
     }
 
     /* Edit entries via GET */
-    $Udisk = null;
+    $s_action = "";
+    $s_entry = "";
     if(isset($_GET['act']) && isset($_GET['id'])){
       if($_GET['act'] == "edit" && isset($this->disks[$_GET['id']])){
-        $Udisk= $_GET['id'];
+        $s_entry= $_GET['id'];
+        $s_action= "edit";
       }
     }
 
     /* New Listhandling */
     foreach($_POST as $name => $value){
-      if(preg_match("/^edit_/",$name)){
-        $entry = preg_replace("/^edit_/","",$name);
-        $Udisk = base64_decode(preg_replace("/_.*/","",$entry));
+      if(preg_match("/^edit_[^_]*_[xy]$/",$name)){
+        $s_entry = preg_replace("/^edit_/","",$name);
+        $s_entry = base64_decode(preg_replace("/_.*/","",$s_entry));
+        $s_action = "edit"; 
         break;
       }
-      if(preg_match("/^delete_/",$name)){
-        $entry = preg_replace("/^delete_/","",$name);
-        $disk = base64_decode(preg_replace("/_.*/","",$entry));
-
-        if (!preg_match("/freeze/i", $this->FAIstate)){
-          if(isset($this->disks[$disk])){
-
-            /* Check for references */
-            $ignore = false;
-            $name = "";
-            foreach($this->disks as $dtest) {
-              // Is raid?
-              $device= null;
-              $name = $dtest['cn'];
-
-              if ($disk == "raid"){
-                $device = "md";
-              } else {
-                $device = $disk;
-
-                // Used by raid?
-                if (isset($this->disks[$name]['partitions'])){
-                  foreach ($this->disks[$name]['partitions'] as $partition) {
-                    if (preg_match("/${disk}\.?[0-9]+/", $partition['FAIpartitionSize'])){
-                      $ignore = true;
-                      break 2;
-                    }
-                  }
-                }
-              }
+      if (!preg_match("/freeze/i", $this->FAIstate) && preg_match("/^delete_[^_]*_/",$name)){
+        $s_entry = preg_replace("/^delete_/","",$name);
+        $s_entry = base64_decode(preg_replace("/_.*/","",$s_entry));
+        $s_action = "remove";
+        break;
+      }
+    }
 
-              // Used by volgroup?
-              if (isset($this->disks[$name]["FAIlvmDevice"])){
-                foreach ($this->disks[$name]["FAIlvmDevice"] as $vg_element) {
-                  if (preg_match("/^${device}\.?[0-9]+$/", $vg_element)){
-                    $ignore = true;
-                    break 2;
-                  }
-                }
+
+    /* Disk remove was requested. 
+     * Now check if the disk is still in use, in this case 
+     *  display a warning message and abort the removal. 
+     * If the disk is not used anymore, then remove it. 
+     */
+    if($s_action == "remove"){
+      if(isset($this->disks[$s_entry])){
+
+        /* Create a list of all partitions that are used in 
+         *  lvm or raid compilations. 
+         */
+        $list = array();
+        foreach($this->disks as $dname => $disk){
+          if($disk['FAIdiskType'] != "disk" && $dname != $s_entry){
+            if($disk['FAIdiskType'] == "lvm"){
+              foreach($disk['FAIlvmDevice'] as $partname){
+                $list[preg_replace("/:.*$/","",$partname)][] = $disk;
               }
             }
-
-            if ($ignore) {
-              msg_dialog::display(_("Error"), sprintf(_("The disk cannot be deleted while it is used in the '%s' disk definition!"), $name), ERROR_DIALOG);
-            } else {
-              if($this->disks[$disk]['status']=="edited"){
-                $this->disks[$disk."-delete"]=$this->disks[$disk];
-                unset($this->disks[$disk]);
-                $disk = $disk."-delete";        
-                $this->disks[$disk]['status']="delete";
-                foreach($this->disks[$disk]['partitions'] as $name => $value ){
-                  if($value['status']=="edited"){
-                    $this->disks[$disk]['partitions'][$name]['status']="delete"; 
-                  }else{
-                    unset($this->disks[$disk]['partitions'][$name]);
-                  }
+            foreach($disk['partitions'] as $partkey => $part){
+              if($disk['FAIdiskType'] == "raid"){
+                foreach(explode(",",$part['FAIpartitionSize']) as $partname){
+                  $list[preg_replace("/:.*$/","",$partname)][] = $disk;
                 }
-              }else{
-                unset($this->disks[$disk]);
               }
-            }
-
+            }  
           }
         }
-        break;
-      }
-    }
-
-    if($Udisk){
-      $usedDiskNames =array();
-      if(isset($this->disks[$Udisk])){
 
-        foreach($this->disks  as $key=>$disk){
-          if($key != $Udisk){
-            $usedDiskNames[]= $key;
+        /* Now that we've a list of all partition references, lets check if
+         *  one of the partitions we are going to remove is still in use.
+         */
+        $used = array();
+        foreach($this->disks[$s_entry]['partitions'] as $part){
+          if(isset($list[$part['cn']])){    
+            foreach($list[$part['cn']] as $disk){
+              $used[$disk['cn']] = $disk['cn'];
+            }
           }
         }
 
-        /* Set object info string, which will be displayed in plugin info line */ 
-        if(isset($this->disks[$Udisk]['dn'])){
-          set_object_info($this->disks[$Udisk]['dn']);
-          $dn = $this->disks[$Udisk]['dn'];
+        /* Skip removal while disk is in use. 
+         */
+        if(count($used)){
+          $used = implode(",",$used);
+          msg_dialog::display(_("Error"), 
+              sprintf(_("The disk cannot be deleted while it is used in the '%s' disk definition!"), 
+                $used), ERROR_DIALOG);
         }else{
-          set_object_info("");
-          $dn = "new";
-        }
 
-        if(isset($this->disks[$Udisk]['FAIdiskType'])){
-          switch($this->disks[$Udisk]['FAIdiskType']){
-            case 'raid': 
-            case 'lvm': 
-            case 'disk': 
-              $this->dialog = new faiDiskEntry(
-                  $this->config,$this->dn,$this,$this->disks[$Udisk], 
-                  $this->disks[$Udisk]['FAIdiskType']); 
-              break;
-            case 'old': 
-              $this->dialog = new faiPartitionTableEntry(
-                  $this->config,$this->dn,$usedDiskNames,$this->disks[$Udisk]); 
-              break;
-          }
-        }else{
-          $this->dialog = new faiPartitionTableEntry(
-              $this->config,$this->dn,$usedDiskNames,$this->disks[$Udisk]); 
-        }
-        if($this->dialog){
-          $this->dialog->set_acl_base($this->acl_base_for_current_object($dn));
-          $this->dialog->set_acl_category("fai");
-          $this->dialog->FAIstate = $this->FAIstate;
-          $this->is_dialog = true;
-        }
+          /* Everything is ok, we can remove the disk now.
+           */
+          unset($this->disks[$s_entry]);
+        } 
+      } 
+    }
+
+    
+
+    if($s_action == "edit"){
+
+      /* Set object info string, which will be displayed in plugin info line */ 
+      if(isset($this->disks[$s_entry]['dn'])){
+        set_object_info($this->disks[$s_entry]['dn']);
+        $dn = $this->disks[$s_entry]['dn'];
+      }else{
+        set_object_info("");
+        $dn = "new";
+      }
+
+      $type ="old"; 
+      if(isset($this->disks[$s_entry]['FAIdiskType'])){
+        $type = $this->disks[$s_entry]['FAIdiskType'];
+      }
+      if(in_array_strict($type,array('raid','lvm','disk'))){
+        $this->dialog = new faiDiskEntry(
+            $this->config,$this->dn,$this,$this->disks[$s_entry], 
+            $this->disks[$s_entry]['FAIdiskType']); 
+
+      }else{
+        $this->dialog = new faiPartitionTableEntry(
+            $this->config,$this->dn,$this,$this->disks[$s_entry]); 
+      }
+      if($this->dialog){
+        $this->dialog->set_acl_base($this->acl_base_for_current_object($dn));
+        $this->dialog->set_acl_category("fai");
+        $this->dialog->FAIstate = $this->FAIstate;
+        $this->is_dialog = true;
       }
     }
 
@@ -307,19 +309,7 @@ class faiPartitionTable extends plugin
         }else{
           $disk = $this->dialog->save();
           if(isset($disk['rename'])){
-            if($this->disks[$disk['rename']['from']]['status']=="edited"){
-              $this->disks[$disk['rename']['from']]['status']="delete";
-            }else{
-              unset($this->disks[$disk['rename']['from']]);
-            }
-
-            foreach($disk['partitions'] as $key => $val){
-              if($disk['partitions'][$key]['status']!="delete"){
-                $disk['partitions'][$key]['status']= "new";
-              }
-            }
-
-            $disk['status']="new";
+            unset($this->disks[$disk['rename']['from']]);
             $disk['cn']= $disk['rename']['to'];
           }
 
@@ -398,32 +388,25 @@ class faiPartitionTable extends plugin
       $dn = $this->acl_base_for_current_object($dn);
       $acl = $this->ui->get_permissions($dn,"fai/faiPartitionTableEntry");
       if(preg_match("/(r|w)/",$acl)) {
-        if($disk['status'] != "delete"){
 
-          $act .= "<input type='image' src='images/lists/edit.png'   name='edit_%s'    title='"._("edit")."' alt='"._("edit")."'>";
-          if(preg_match("/d/",$acl)){
-            $act .="<input type='image' src='images/lists/trash.png' name='delete_%s'  title='"._("delete")."' alt='"._("delete")."'>";
-          }
+        $act .= "<input type='image' src='images/lists/edit.png'   name='edit_%s'    title='"._("edit")."' alt='"._("edit")."'>";
+        if(preg_match("/d/",$acl) && !preg_match("/freeze/", $this->FAIstate)){
+          $act .="<input type='image' src='images/lists/trash.png' name='delete_%s'  title='"._("delete")."' alt='"._("delete")."'>";
+        }
 
-          $cnt=0;
-          foreach($disk['partitions'] as $val){
-            if($val['status']!="delete"){
-              $cnt ++;
-            }
-          }
+        $cnt= count($disk['partitions']);
 
-          $edit_link = "<a href='?plug=".$_GET['plug']."&amp;act=edit&amp;id=".$key."'>".$key."</a>";
-          $types= array("old" => "plugins/fai/images/fai_partitionTable.png", "disk" => "plugins/fai/images/fai_partitionTable.png",
-                        "raid" => "plugins/fai/images/raid.png", "lvm" => "plugins/ogroups/images/list_ogroup.png");
-          $type = isset($disk['FAIdiskType'])?$types[$disk['FAIdiskType']]:$types['old'];
-          $divlist->AddEntry(array( 
+        $edit_link = "<a href='?plug=".$_GET['plug']."&amp;act=edit&amp;id=".$key."'>".$key."</a>";
+        $types= array("old" => "plugins/fai/images/fai_partitionTable.png", "disk" => "plugins/fai/images/fai_partitionTable.png",
+            "raid" => "plugins/fai/images/raid.png", "lvm" => "plugins/ogroups/images/list_ogroup.png");
+        $type = isset($disk['FAIdiskType'])?$types[$disk['FAIdiskType']]:$types['old'];
+        $divlist->AddEntry(array( 
               array("string"=> "<img border='0' src='".$type."'>", "attach"=>"style='width:16px'"),
               array("string"=> $edit_link, "attach"=>"style='width:100px'"),
               array("string"=> $disk['description']),
               array("string"=> $cnt,  "attach"=>"style='width:16px'"),
               array("string"=>str_replace("%s",base64_encode($key),$act),
                 "attach"=>"style='border-right: 0px;width:50px;text-align:right;'")));
-        }
       }
     }
     $smarty->assign("Entry_divlist",$divlist->DrawList());
@@ -433,6 +416,15 @@ class faiPartitionTable extends plugin
   }
 
 
+  function getUsedDiskNames()
+  {
+    $ret = array();
+    foreach($this->disks as $disk){
+      $ret[] = $disk['cn'];
+    }
+    return($ret);  
+  }
+
 
   /* Delete me, and all my subtrees
    */
@@ -446,7 +438,7 @@ class faiPartitionTable extends plugin
 
     FAI::prepare_to_save_FAI_object($use_dn,array(),true);
     new log("remove","fai/".get_class($this),$use_dn,$this->attributes);   
-    foreach($this->disks as $disk){
+    foreach($this->initial_disks as $disk){
       $disk_dn = "cn=".$disk['cn'].",".$this->dn;
       $use_dn = preg_replace("/".preg_quote(FAI::get_release_dn($this->dn), '/')."/i",$release, $disk_dn);
       FAI::prepare_to_save_FAI_object($use_dn,array(),true);
@@ -463,14 +455,14 @@ class faiPartitionTable extends plugin
    */
   function save_object()
   {
-    if (preg_match("/freeze/", $this->FAIstate)) return;
-    plugin::save_object();
-    foreach($this->attributes as $attrs){
-      if(isset($_POST[$attrs])){
-        $this->$attrs = $_POST[$attrs];
+    if(isset($_POST['FAIpartitionTablePosted'])){
+      if (preg_match("/freeze/", $this->FAIstate)) return;
+      plugin::save_object();
+      foreach($this->attributes as $attrs){
+        if(isset($_POST[$attrs])){
+          $this->$attrs = $_POST[$attrs];
+        }
       }
-    }
-    if(isset($_POST['faiPartitionTable'])){
       if(!count($this->disks)){
         if(isset($_POST['mode'])){
           $this->FAIpartitionMethod = "setup-storage";
@@ -505,8 +497,8 @@ class faiPartitionTable extends plugin
   /* Save to LDAP */
   function save()
   {
-
     plugin::save();
+
     /* Save current settings.
      * 1 : We must save the partition table, with its description and cn 
      * 2 : Append Disk with cn and  description.
@@ -519,10 +511,15 @@ class faiPartitionTable extends plugin
 
     if($this->initially_was_account){
       new log("modify","fai/".get_class($this),$this->dn,$this->attributes);
+      @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$this->dn , "Saving disk: ");
     }else{
       new log("create","fai/".get_class($this),$this->dn,$this->attributes);
+      @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$this->dn , "Adding disk: ");
     }
  
+    // Prepare disks to be saved - The 'status' attribute is added here.
+    $this->prepareDiskToBeSave(); 
     /* Sort entries, because we must delete entries with status="delete" first */
     $order = array();
     foreach($this->disks as $key => $disk){
@@ -536,24 +533,15 @@ class faiPartitionTable extends plugin
       }
     }
 
+
     /* Append all disks to ldap */
     foreach($order as $cn=>$disk){
+
       $disk_dn                    = "cn=".$disk['cn'].",".$this->dn;
+      $short_dn                   = "cn=".$disk['cn'].",...";
       $disk_attrs['cn']           =  $disk['cn'];
-      $disk_attrs['description']  =  $disk['description'];
-  
-      if(isset($disk['FAIdiskType'])){
-        $disk_attrs['FAIdiskType']  =  $disk['FAIdiskType']; 
-      }
-      if(isset($disk['FAIdiskOption'])){
-        $disk_attrs['FAIdiskOption']  =  $disk['FAIdiskOption']; 
-      }
-
-      if(empty($disk_attrs['description']) && $disk['status'] == "edited"){
-        $disk_attrs['description'] = array();
-      }
-
       $disk_attrs['objectClass']  =  array("top","FAIclass","FAIpartitionDisk");
+  
 
       if($disk['status']=="new"){
         $ldap->cat($disk_dn,array("objectClass"));
@@ -562,16 +550,44 @@ class faiPartitionTable extends plugin
         }
       }
 
+      foreach(array("description","FAIdiskType","FAIdiskOption","FAIlvmDevice") as $attr){
+        if($disk['status'] == "new"){
+          if(isset($disk_attrs[$attr])) unset($disk_attrs[$attr]);
+          if(isset($disk[$attr]) && !empty($disk[$attr])){
+            if(is_array($disk[$attr])){
+              $disk_attrs[$attr] = array_values($disk[$attr]);
+            }else{
+              $disk_attrs[$attr] = $disk[$attr];
+            }
+          }
+        }else{
+          if(isset($disk[$attr]) && !empty($disk[$attr])){
+            if(is_array($disk[$attr])){
+              $disk_attrs[$attr] = array_values($disk[$attr]);
+            }else{
+              $disk_attrs[$attr] = $disk[$attr];
+            }
+          }else{
+            $disk_attrs[$attr] = array();
+          }
+        }
+      }
+
+
       /* Tag object */
       $this->tag_attrs($disk_attrs, $disk_dn, $this->gosaUnitTag);
 
       if($disk['status'] == "delete"){
+        @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$short_dn , "Removing disk: ");
         FAI::prepare_to_save_FAI_object($disk_dn,array(),true);
         $this->handle_post_events("remove");
+        unset($this->disks[$cn]);
       }elseif($disk['status'] == "edited"){
+        @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$short_dn , "Updating disk: ");
         FAI::prepare_to_save_FAI_object($disk_dn,$disk_attrs);
         $this->handle_post_events("modify");
       }elseif($disk['status']=="new"){
+        @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$short_dn , "Adding disk: ");
         FAI::prepare_to_save_FAI_object($disk_dn,$disk_attrs);
         $this->handle_post_events("add");
       }
@@ -579,7 +595,7 @@ class faiPartitionTable extends plugin
       if($disk['status']!="delete")
 
       /* Add all partitions */
-      foreach($disk['partitions'] as $key => $partition){
+      foreach($disk['partitions'] as $pkey => $partition){
         $partition_attrs = array();
 
         foreach($partition as $key => $value){
@@ -590,9 +606,10 @@ class faiPartitionTable extends plugin
           }
         }
 
-        $partition_dn= "FAIpartitionNr=".$partition_attrs['FAIpartitionNr'].",".$disk_dn;      
+        $partition_dn= "FAIpartitionNr=".$partition_attrs['FAIpartitionNr'].",".$disk_dn;     
+        $short_dn= "FAIpartitionNr=".$partition_attrs['FAIpartitionNr'].",...";
+     
         $partition_attrs['objectClass']= array("top","FAIclass","FAIpartitionEntry");
-        $partition_attrs['cn']= $partition_attrs['FAIpartitionNr'];
         
         unset($partition_attrs['status']);
         unset($partition_attrs['old_cn']);
@@ -611,61 +628,152 @@ class faiPartitionTable extends plugin
         /* Tag object */
         $this->tag_attrs($partition_attrs, $partition_dn, $this->gosaUnitTag);
 
+        
         if($partition['status'] == "delete"){
+          @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$short_dn , "Removing partition: ");
           FAI::prepare_to_save_FAI_object($partition_dn,array(),true);
           $this->handle_post_events("remove");
+          unset($this->disks[$cn]['partitions'][$pkey]);
         }elseif($partition['status'] == "edited"){
+          @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$short_dn , "Updating partition: ");
           FAI::prepare_to_save_FAI_object($partition_dn,$partition_attrs);
           $this->handle_post_events("modify");
         }elseif($partition['status']=="new"){
+          @DEBUG (DEBUG_FAI, __LINE__, __FUNCTION__, __FILE__,$short_dn , "Adding partition: ");
           FAI::prepare_to_save_FAI_object($partition_dn,$partition_attrs);
           $this->handle_post_events("add");
         }
+        
+        // We do not need the status flag any longer 
+        if(isset($this->disks[$cn]['partitions'][$pkey]['status'])){
+          unset($this->disks[$cn]['partitions'][$pkey]['status']);
+        }
+      }
+      // We do not need the status flag any longer 
+      if(isset($this->disks[$cn]['status'])){
+        unset($this->disks[$cn]['status']);
       }
     }
+    $this->initial_disks = $this->disks;
     $this->handle_post_events("add");
   }
 
 
-  function PrepareForCopyPaste($source)
+  function prepareDiskToBeSave()
   {
-    plugin::PrepareForCopyPaste($source);
+    foreach($this->disks as $id => $disk){
+      
+      /* Correct FAIpartitionNr.
+       * If we've only primary partition then set the partition numbers from  
+       *  1 to 4, else set the primary from 1 to 3 and logical >= 5 
+       * 
+       */
+      if(!isset($disk['partitions'])){
+        $disk['partitions'] = array();
+      }
+      $newSetup = array();
+
+      if($disk['FAIdiskType'] == "disk"){
+        $primary = $logical = array();
+        foreach($disk['partitions'] as $partid => $part){
+          if($part['FAIpartitionType'] == "primary"){
+            $primary[$partid] = $part;
+          }elseif($part['FAIpartitionType'] == "logical"){
+            $logical[$partid] = $part;
+          }else{
+            trigger_error("Fatal: unknown disk type? ".$part['FAIpartitionType']); 
+          }
+        }
+        $cnt = 1;
+        foreach($primary as $part){
+          $part['FAIpartitionNr'] = $cnt;
+          $part['cn'] = $disk['cn'].$cnt;
+          $newSetup[$cnt] = $part;
+          $cnt ++;
+        } 
+        $cnt = 5;
+        foreach($logical as $part){
+          $part['FAIpartitionNr'] = $cnt;
+          $part['cn'] = $disk['cn'].$cnt;
+          $newSetup[$cnt] = $part;
+          $cnt ++;
+        }
+        $this->disks[$disk['cn']]['partitions'] = $newSetup; 
+      }
+    }
 
-    /* Read all leaf objects of this object (For FAIscript this would be FAIscriptEntry)
-     */
-    $res = FAI::get_all_objects_for_given_base($source['dn'],"(&(objectClass=FAIclass)(objectClass=FAIpartitionDisk))");
-    foreach($res as $obj){
+    # FAIpartitionNr have to be used as index for this->disks
+    #  else the next operation will fail. 
 
-      /* Skip not relevant objects */
-      if(!preg_match("/".preg_quote($source['dn'], '/')."$/i",$obj['dn'])) continue;
+    /* Check if there are disks to be removed, edited or added.
+     * We compare the initial disk setup with the current setup and
+     *  and add a status flag, which will then be used to perform the 
+     *  correct action - add, edited, remove.
+     */
+    foreach($this->disks as $key => $disk){
 
-      $objects = array();
-      $objects['description']  = "";
-      $objects['status']      = "edited";
-      $objects['dn']          = $obj['dn'];
-      $objects                = $this->get_object_attributes($objects,$this->subAttributes);
-      $this->disks[$objects['cn']] = $objects;
-      $this->disks[$objects['cn']]['partitions'] = array();
+      // - A complete NEW disk
+      if(!isset($this->initial_disks[$disk['cn']])){
+        $this->disks[$key]['status'] = "new";
+        foreach($disk['partitions'] as $pkey => $part){
+          $this->disks[$disk['cn']]['partitions'][$pkey]['status'] = "new";
+        }
+      }else{
+    
+        // - Disk was "EDITED" 
+        $this->disks[$key]['status'] = "edited";
+        foreach($disk['partitions'] as $pkey => $part){
+
+          // - Check whether partition is "NEW" or "EDITED" 
+          if(!isset($this->initial_disks[$key]['partitions'][$pkey])){
+            $this->disks[$key]['partitions'][$pkey]['status'] = "new";
+          }else{
+            $this->disks[$key]['partitions'][$pkey]['status'] = "edited";
+          }
+        }
+      }
+    }
+     
+    /* Check which partitions havbe to be removed. 
+     * (They intially existed, but are now gone.)
+     */ 
+    foreach($this->initial_disks as $ikey => $idisk){
+      
+      // - Complete disk was REMOVED.
+      if(!isset($this->disks[$idisk['cn']])){
+        $this->disks[$idisk['cn']] = $idisk;
+        $this->disks[$idisk['cn']]['status'] = "delete";
+        foreach($idisk['partitions'] as $pkey=>$part){
+          $this->disks[$idisk['cn']]['partitions'][$pkey] = $part;
+          $this->disks[$idisk['cn']]['partitions'][$pkey]["status"] = "delete";
+        }
+      }else{
+        foreach($idisk['partitions'] as $pkey=>$part){
+          if(!isset($this->disks[$idisk['cn']]['partitions'][$pkey])){
+            $this->disks[$idisk['cn']]['partitions'][$pkey] = $part;
+            $this->disks[$idisk['cn']]['partitions'][$pkey]["status"] = "delete";
+          }
+        }
+      }
     }
+  }
+  
 
-    /* read all partitions for each disk
-     */
-    foreach($this->disks as $name => $disk){
-      $res = FAI::get_all_objects_for_given_base($disk['dn'],"(&(objectClass=FAIclass)(objectClass=FAIpartitionEntry))");
-      foreach($res as $obj){
+  function PrepareForCopyPaste($source)
+  {
+    plugin::PrepareForCopyPaste($source);
 
-        /* Skip not relevant objects */
-        if(!preg_match("/".preg_quote($disk['dn'], '/')."$/i",$obj['dn'])) continue;
+    /* Read all leaf objects of this object (For FAIscript this would be FAIscriptEntry)
+     */
 
-        $objects = array();
-        $objects['status']      = "edited";
-        $objects['dn']          = $obj['dn'];
-        $objects                = $this->get_object_attributes($objects,$this->subPartAttributes);
-        unset($objects['dn']);;
-        $this->disks[$name]['partitions'][$objects['FAIpartitionNr']] = $objects;
-      }
-    }
-    ksort($this->disks);
+    /* To be sure to copy disks and partitions correctly, just create a 
+     *  new PartitionTable object and use the 'disks' attribute 
+     *  from this it. This is much easier and less code.
+     */
+    $obj = new faiPartitionTable($this->config, $source['dn']);
+    $this->disks = $obj->disks;
   }