#include <asm/machdep.h>
 #include <linux/uaccess.h>
 #include <asm/rtas.h>
+#include <asm/rtas-work-area.h>
 
 static struct workqueue_struct *pseries_hp_wq;
 
        struct property *property;
        struct property *last_property = NULL;
        struct cc_workarea *ccwa;
+       struct rtas_work_area *work_area;
        char *data_buf;
        int cc_token;
        int rc = -1;
        if (cc_token == RTAS_UNKNOWN_SERVICE)
                return NULL;
 
-       data_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
-       if (!data_buf)
-               return NULL;
+       work_area = rtas_work_area_alloc(SZ_4K);
+       data_buf = rtas_work_area_raw_buf(work_area);
 
        ccwa = (struct cc_workarea *)&data_buf[0];
        ccwa->drc_index = drc_index;
        ccwa->zero = 0;
 
        do {
-               /* Since we release the rtas_data_buf lock between configure
-                * connector calls we want to re-populate the rtas_data_buffer
-                * with the contents of the previous call.
-                */
-               spin_lock(&rtas_data_buf_lock);
-
-               memcpy(rtas_data_buf, data_buf, RTAS_DATA_BUF_SIZE);
-               rc = rtas_call(cc_token, 2, 1, NULL, rtas_data_buf, NULL);
-               memcpy(data_buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);
-
-               spin_unlock(&rtas_data_buf_lock);
-
-               if (rtas_busy_delay(rc))
-                       continue;
+               do {
+                       rc = rtas_call(cc_token, 2, 1, NULL,
+                                      rtas_work_area_phys(work_area), NULL);
+               } while (rtas_busy_delay(rc));
 
                switch (rc) {
                case COMPLETE:
        } while (rc);
 
 cc_error:
-       kfree(data_buf);
+       rtas_work_area_free(work_area);
 
        if (rc) {
                if (first_dn)