]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - sys/dev/cfi/cfi_core.c
Copy libevent sources to contrib
[FreeBSD/FreeBSD.git] / sys / dev / cfi / cfi_core.c
1 /*-
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 2007, Juniper Networks, Inc.
5  * Copyright (c) 2012-2013, SRI International
6  * All rights reserved.
7  *
8  * Portions of this software were developed by SRI International and the
9  * University of Cambridge Computer Laboratory under DARPA/AFRL contract
10  * (FA8750-10-C-0237) ("CTSRD"), as part of the DARPA CRASH research
11  * programme.
12  *
13  * Redistribution and use in source and binary forms, with or without
14  * modification, are permitted provided that the following conditions
15  * are met:
16  * 1. Redistributions of source code must retain the above copyright
17  *    notice, this list of conditions and the following disclaimer.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  * 3. Neither the name of the author nor the names of any co-contributors
22  *    may be used to endorse or promote products derived from this software
23  *    without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
26  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
27  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
28  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
29  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
30  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
31  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
32  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
33  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35  * SUCH DAMAGE.
36  */
37
38 #include <sys/cdefs.h>
39 __FBSDID("$FreeBSD$");
40
41 #include "opt_cfi.h"
42
43 #include <sys/param.h>
44 #include <sys/systm.h>
45 #include <sys/bus.h>
46 #include <sys/conf.h>
47 #include <sys/endian.h>
48 #include <sys/kenv.h>
49 #include <sys/kernel.h>
50 #include <sys/malloc.h>   
51 #include <sys/module.h>
52 #include <sys/rman.h>
53 #include <sys/sysctl.h>
54
55 #include <machine/bus.h>
56
57 #include <dev/cfi/cfi_reg.h>
58 #include <dev/cfi/cfi_var.h>
59
60 static void cfi_add_sysctls(struct cfi_softc *);
61
62 extern struct cdevsw cfi_cdevsw;
63
64 char cfi_driver_name[] = "cfi";
65 devclass_t cfi_devclass;
66 devclass_t cfi_diskclass;
67
68 uint32_t
69 cfi_read_raw(struct cfi_softc *sc, u_int ofs)
70 {
71         uint32_t val;
72
73         ofs &= ~(sc->sc_width - 1);
74         switch (sc->sc_width) {
75         case 1:
76                 val = bus_space_read_1(sc->sc_tag, sc->sc_handle, ofs);
77                 break;
78         case 2:
79                 val = bus_space_read_2(sc->sc_tag, sc->sc_handle, ofs);
80                 break;
81         case 4:
82                 val = bus_space_read_4(sc->sc_tag, sc->sc_handle, ofs);
83                 break;
84         default:
85                 val = ~0;
86                 break;
87         }
88         return (val);
89 }
90
91 uint32_t
92 cfi_read(struct cfi_softc *sc, u_int ofs)
93 {
94         uint32_t val;
95         uint16_t sval;
96
97         ofs &= ~(sc->sc_width - 1);
98         switch (sc->sc_width) {
99         case 1:
100                 val = bus_space_read_1(sc->sc_tag, sc->sc_handle, ofs);
101                 break;
102         case 2:
103                 sval = bus_space_read_2(sc->sc_tag, sc->sc_handle, ofs);
104 #ifdef CFI_HARDWAREBYTESWAP
105                 val = sval;
106 #else
107                 val = le16toh(sval);
108 #endif
109                 break;
110         case 4:
111                 val = bus_space_read_4(sc->sc_tag, sc->sc_handle, ofs);
112 #ifndef CFI_HARDWAREBYTESWAP
113                 val = le32toh(val);
114 #endif
115                 break;
116         default:
117                 val = ~0;
118                 break;
119         }
120         return (val);
121 }
122
123 static void
124 cfi_write(struct cfi_softc *sc, u_int ofs, u_int val)
125 {
126
127         ofs &= ~(sc->sc_width - 1);
128         switch (sc->sc_width) {
129         case 1:
130                 bus_space_write_1(sc->sc_tag, sc->sc_handle, ofs, val);
131                 break;
132         case 2:
133 #ifdef CFI_HARDWAREBYTESWAP
134                 bus_space_write_2(sc->sc_tag, sc->sc_handle, ofs, val);
135 #else
136                 bus_space_write_2(sc->sc_tag, sc->sc_handle, ofs, htole16(val));
137
138 #endif
139                 break;
140         case 4:
141 #ifdef CFI_HARDWAREBYTESWAP
142                 bus_space_write_4(sc->sc_tag, sc->sc_handle, ofs, val);
143 #else
144                 bus_space_write_4(sc->sc_tag, sc->sc_handle, ofs, htole32(val));
145 #endif
146                 break;
147         }
148 }
149
150 /*
151  * This is same workaound as NetBSD sys/dev/nor/cfi.c cfi_reset_default()
152  */
153 static void
154 cfi_reset_default(struct cfi_softc *sc)
155 {
156
157         cfi_write(sc, 0, CFI_BCS_READ_ARRAY2);
158         cfi_write(sc, 0, CFI_BCS_READ_ARRAY);
159 }
160
161 uint8_t
162 cfi_read_qry(struct cfi_softc *sc, u_int ofs)
163 {
164         uint8_t val;
165  
166         cfi_write(sc, CFI_QRY_CMD_ADDR * sc->sc_width, CFI_QRY_CMD_DATA); 
167         val = cfi_read(sc, ofs * sc->sc_width);
168         cfi_reset_default(sc);
169         return (val);
170
171
172 static void
173 cfi_amd_write(struct cfi_softc *sc, u_int ofs, u_int addr, u_int data)
174 {
175
176         cfi_write(sc, ofs + AMD_ADDR_START, CFI_AMD_UNLOCK);
177         cfi_write(sc, ofs + AMD_ADDR_ACK, CFI_AMD_UNLOCK_ACK);
178         cfi_write(sc, ofs + addr, data);
179 }
180
181 static char *
182 cfi_fmtsize(uint32_t sz)
183 {
184         static char buf[8];
185         static const char *sfx[] = { "", "K", "M", "G" };
186         int sfxidx;
187
188         sfxidx = 0;
189         while (sfxidx < 3 && sz > 1023) {
190                 sz /= 1024;
191                 sfxidx++;
192         }
193
194         sprintf(buf, "%u%sB", sz, sfx[sfxidx]);
195         return (buf);
196 }
197
198 int
199 cfi_probe(device_t dev)
200 {
201         char desc[80];
202         struct cfi_softc *sc;
203         char *vend_str;
204         int error;
205         uint16_t iface, vend;
206
207         sc = device_get_softc(dev);
208         sc->sc_dev = dev;
209
210         sc->sc_rid = 0;
211         sc->sc_res = bus_alloc_resource_any(dev, SYS_RES_MEMORY, &sc->sc_rid,
212             RF_ACTIVE);
213         if (sc->sc_res == NULL)
214                 return (ENXIO);
215
216         sc->sc_tag = rman_get_bustag(sc->sc_res);
217         sc->sc_handle = rman_get_bushandle(sc->sc_res);
218
219         if (sc->sc_width == 0) {
220                 sc->sc_width = 1;
221                 while (sc->sc_width <= 4) {
222                         if (cfi_read_qry(sc, CFI_QRY_IDENT) == 'Q')
223                                 break;
224                         sc->sc_width <<= 1;
225                 }
226         } else if (cfi_read_qry(sc, CFI_QRY_IDENT) != 'Q') {
227                 error = ENXIO;
228                 goto out;
229         }
230         if (sc->sc_width > 4) {
231                 error = ENXIO;
232                 goto out;
233         }
234
235         /* We got a Q. Check if we also have the R and the Y. */
236         if (cfi_read_qry(sc, CFI_QRY_IDENT + 1) != 'R' ||
237             cfi_read_qry(sc, CFI_QRY_IDENT + 2) != 'Y') {
238                 error = ENXIO;
239                 goto out;
240         }
241
242         /* Get the vendor and command set. */
243         vend = cfi_read_qry(sc, CFI_QRY_VEND) |
244             (cfi_read_qry(sc, CFI_QRY_VEND + 1) << 8);
245
246         sc->sc_cmdset = vend;
247
248         switch (vend) {
249         case CFI_VEND_AMD_ECS:
250         case CFI_VEND_AMD_SCS:
251                 vend_str = "AMD/Fujitsu";
252                 break;
253         case CFI_VEND_INTEL_ECS:
254                 vend_str = "Intel/Sharp";
255                 break;
256         case CFI_VEND_INTEL_SCS:
257                 vend_str = "Intel";
258                 break;
259         case CFI_VEND_MITSUBISHI_ECS:
260         case CFI_VEND_MITSUBISHI_SCS:
261                 vend_str = "Mitsubishi";
262                 break;
263         default:
264                 vend_str = "Unknown vendor";
265                 break;
266         }
267
268         /* Get the device size. */
269         sc->sc_size = 1U << cfi_read_qry(sc, CFI_QRY_SIZE);
270
271         /* Sanity-check the I/F */
272         iface = cfi_read_qry(sc, CFI_QRY_IFACE) |
273             (cfi_read_qry(sc, CFI_QRY_IFACE + 1) << 8);
274
275         /*
276          * Adding 1 to iface will give us a bit-wise "switch"
277          * that allows us to test for the interface width by
278          * testing a single bit.
279          */
280         iface++;
281
282         error = (iface & sc->sc_width) ? 0 : EINVAL;
283         if (error)
284                 goto out;
285
286         snprintf(desc, sizeof(desc), "%s - %s", vend_str,
287             cfi_fmtsize(sc->sc_size));
288         device_set_desc_copy(dev, desc);
289
290  out:
291         bus_release_resource(dev, SYS_RES_MEMORY, sc->sc_rid, sc->sc_res);
292         return (error);
293 }
294
295 int
296 cfi_attach(device_t dev) 
297 {
298         struct cfi_softc *sc;
299         u_int blksz, blocks;
300         u_int r, u;
301         uint64_t mtoexp, ttoexp;
302 #ifdef CFI_SUPPORT_STRATAFLASH
303         uint64_t ppr;
304         char name[KENV_MNAMELEN], value[32];
305 #endif
306
307         sc = device_get_softc(dev);
308         sc->sc_dev = dev;
309
310         sc->sc_rid = 0;
311         sc->sc_res = bus_alloc_resource_any(dev, SYS_RES_MEMORY, &sc->sc_rid,
312 #ifndef ATSE_CFI_HACK
313             RF_ACTIVE);
314 #else
315             RF_ACTIVE | RF_SHAREABLE);
316 #endif
317         if (sc->sc_res == NULL)
318                 return (ENXIO);
319
320         sc->sc_tag = rman_get_bustag(sc->sc_res);
321         sc->sc_handle = rman_get_bushandle(sc->sc_res);
322
323         /* Get time-out values for erase, write, and buffer write. */
324         ttoexp = cfi_read_qry(sc, CFI_QRY_TTO_ERASE);
325         mtoexp = cfi_read_qry(sc, CFI_QRY_MTO_ERASE);
326         if (ttoexp == 0) {
327                 device_printf(dev, "erase timeout == 0, using 2^16ms\n");
328                 ttoexp = 16;
329         }
330         if (ttoexp > 41) {
331                 device_printf(dev, "insane timeout: 2^%jdms\n", ttoexp);
332                 return (EINVAL);
333         }
334         if (mtoexp == 0) {
335                 device_printf(dev, "max erase timeout == 0, using 2^%jdms\n",
336                     ttoexp + 4);
337                 mtoexp = 4;
338         }
339         if (ttoexp + mtoexp > 41) {
340                 device_printf(dev, "insane max erase timeout: 2^%jd\n",
341                     ttoexp + mtoexp);
342                 return (EINVAL);
343         }
344         sc->sc_typical_timeouts[CFI_TIMEOUT_ERASE] = SBT_1MS * (1ULL << ttoexp);
345         sc->sc_max_timeouts[CFI_TIMEOUT_ERASE] =
346             sc->sc_typical_timeouts[CFI_TIMEOUT_ERASE] * (1ULL << mtoexp);
347
348         ttoexp = cfi_read_qry(sc, CFI_QRY_TTO_WRITE);
349         mtoexp = cfi_read_qry(sc, CFI_QRY_MTO_WRITE);
350         if (ttoexp == 0) {
351                 device_printf(dev, "write timeout == 0, using 2^18ns\n");
352                 ttoexp = 18;
353         }
354         if (ttoexp > 51) {
355                 device_printf(dev, "insane write timeout: 2^%jdus\n", ttoexp);
356                 return (EINVAL);
357         }
358         if (mtoexp == 0) {
359                 device_printf(dev, "max write timeout == 0, using 2^%jdms\n",
360                     ttoexp + 4);
361                 mtoexp = 4;
362         }
363         if (ttoexp + mtoexp > 51) {
364                 device_printf(dev, "insane max write timeout: 2^%jdus\n",
365                     ttoexp + mtoexp);
366                 return (EINVAL);
367         }
368         sc->sc_typical_timeouts[CFI_TIMEOUT_WRITE] = SBT_1US * (1ULL << ttoexp);
369         sc->sc_max_timeouts[CFI_TIMEOUT_WRITE] =
370             sc->sc_typical_timeouts[CFI_TIMEOUT_WRITE] * (1ULL << mtoexp);
371
372         ttoexp = cfi_read_qry(sc, CFI_QRY_TTO_BUFWRITE);
373         mtoexp = cfi_read_qry(sc, CFI_QRY_MTO_BUFWRITE);
374         /* Don't check for 0, it means not-supported. */
375         if (ttoexp > 51) {
376                 device_printf(dev, "insane write timeout: 2^%jdus\n", ttoexp);
377                 return (EINVAL);
378         }
379         if (ttoexp + mtoexp > 51) {
380                 device_printf(dev, "insane max write timeout: 2^%jdus\n",
381                     ttoexp + mtoexp);
382                 return (EINVAL);
383         }
384         sc->sc_typical_timeouts[CFI_TIMEOUT_BUFWRITE] =
385             SBT_1US * (1ULL << cfi_read_qry(sc, CFI_QRY_TTO_BUFWRITE));
386         sc->sc_max_timeouts[CFI_TIMEOUT_BUFWRITE] =
387             sc->sc_typical_timeouts[CFI_TIMEOUT_BUFWRITE] *
388             (1ULL << cfi_read_qry(sc, CFI_QRY_MTO_BUFWRITE));
389
390         /* Get the maximum size of a multibyte program */
391         if (sc->sc_typical_timeouts[CFI_TIMEOUT_BUFWRITE] != 0)
392                 sc->sc_maxbuf = 1 << (cfi_read_qry(sc, CFI_QRY_MAXBUF) |
393                     cfi_read_qry(sc, CFI_QRY_MAXBUF) << 8);
394         else
395                 sc->sc_maxbuf = 0;
396
397         /* Get erase regions. */
398         sc->sc_regions = cfi_read_qry(sc, CFI_QRY_NREGIONS);
399         sc->sc_region = malloc(sc->sc_regions * sizeof(struct cfi_region),
400             M_TEMP, M_WAITOK | M_ZERO);
401         for (r = 0; r < sc->sc_regions; r++) {
402                 blocks = cfi_read_qry(sc, CFI_QRY_REGION(r)) |
403                     (cfi_read_qry(sc, CFI_QRY_REGION(r) + 1) << 8);
404                 sc->sc_region[r].r_blocks = blocks + 1;
405
406                 blksz = cfi_read_qry(sc, CFI_QRY_REGION(r) + 2) |
407                     (cfi_read_qry(sc, CFI_QRY_REGION(r) + 3) << 8);
408                 sc->sc_region[r].r_blksz = (blksz == 0) ? 128 :
409                     blksz * 256;
410         }
411
412         /* Reset the device to a default state. */
413         cfi_write(sc, 0, CFI_BCS_CLEAR_STATUS);
414
415         if (bootverbose) {
416                 device_printf(dev, "[");
417                 for (r = 0; r < sc->sc_regions; r++) {
418                         printf("%ux%s%s", sc->sc_region[r].r_blocks,
419                             cfi_fmtsize(sc->sc_region[r].r_blksz),
420                             (r == sc->sc_regions - 1) ? "]\n" : ",");
421                 }
422         }
423
424         u = device_get_unit(dev);
425         sc->sc_nod = make_dev(&cfi_cdevsw, u, UID_ROOT, GID_WHEEL, 0600,
426             "%s%u", cfi_driver_name, u);
427         sc->sc_nod->si_drv1 = sc;
428
429         cfi_add_sysctls(sc);
430
431 #ifdef CFI_SUPPORT_STRATAFLASH
432         /*
433          * Store the Intel factory PPR in the environment.  In some
434          * cases it is the most unique ID on a board.
435          */
436         if (cfi_intel_get_factory_pr(sc, &ppr) == 0) {
437                 if (snprintf(name, sizeof(name), "%s.factory_ppr",
438                     device_get_nameunit(dev)) < (sizeof(name) - 1) &&
439                     snprintf(value, sizeof(value), "0x%016jx", ppr) <
440                     (sizeof(value) - 1))
441                         (void) kern_setenv(name, value);
442         }
443 #endif
444
445         device_add_child(dev, "cfid", -1);
446         bus_generic_attach(dev);
447
448         return (0);
449 }
450
451 static void
452 cfi_add_sysctls(struct cfi_softc *sc)
453 {
454         struct sysctl_ctx_list *ctx;
455         struct sysctl_oid_list *children;
456
457         ctx = device_get_sysctl_ctx(sc->sc_dev);
458         children = SYSCTL_CHILDREN(device_get_sysctl_tree(sc->sc_dev));
459
460         SYSCTL_ADD_UINT(ctx, children, OID_AUTO,
461             "typical_erase_timout_count",
462             CTLFLAG_RD, &sc->sc_tto_counts[CFI_TIMEOUT_ERASE],
463             0, "Number of times the typical erase timeout was exceeded");
464         SYSCTL_ADD_UINT(ctx, children, OID_AUTO,
465             "max_erase_timout_count",
466             CTLFLAG_RD, &sc->sc_mto_counts[CFI_TIMEOUT_ERASE], 0,
467             "Number of times the maximum erase timeout was exceeded");
468         SYSCTL_ADD_UINT(ctx, children, OID_AUTO,
469             "typical_write_timout_count",
470             CTLFLAG_RD, &sc->sc_tto_counts[CFI_TIMEOUT_WRITE], 0,
471             "Number of times the typical write timeout was exceeded");
472         SYSCTL_ADD_UINT(ctx, children, OID_AUTO,
473             "max_write_timout_count",
474             CTLFLAG_RD, &sc->sc_mto_counts[CFI_TIMEOUT_WRITE], 0,
475             "Number of times the maximum write timeout was exceeded");
476         if (sc->sc_maxbuf > 0) {
477                 SYSCTL_ADD_UINT(ctx, children, OID_AUTO,
478                     "typical_bufwrite_timout_count",
479                     CTLFLAG_RD, &sc->sc_tto_counts[CFI_TIMEOUT_BUFWRITE], 0,
480                     "Number of times the typical buffered write timeout was "
481                     "exceeded");
482                 SYSCTL_ADD_UINT(ctx, children, OID_AUTO,
483                     "max_bufwrite_timout_count",
484                     CTLFLAG_RD, &sc->sc_mto_counts[CFI_TIMEOUT_BUFWRITE], 0,
485                     "Number of times the maximum buffered write timeout was "
486                     "exceeded");
487         }
488 }
489
490 int
491 cfi_detach(device_t dev)
492 {
493         struct cfi_softc *sc;
494
495         sc = device_get_softc(dev);
496
497         destroy_dev(sc->sc_nod);
498         free(sc->sc_region, M_TEMP);
499         bus_release_resource(dev, SYS_RES_MEMORY, sc->sc_rid, sc->sc_res);
500         return (0);
501 }
502
503 static int
504 cfi_wait_ready(struct cfi_softc *sc, u_int ofs, sbintime_t start,
505     enum cfi_wait_cmd cmd)
506 {
507         int done, error, tto_exceeded;
508         uint32_t st0 = 0, st = 0;
509         sbintime_t now;
510
511         done = 0;
512         error = 0;
513         tto_exceeded = 0;
514         while (!done && !error) {
515                 /*
516                  * Save time before we start so we always do one check
517                  * after the timeout has expired.
518                  */
519                 now = sbinuptime();
520
521                 switch (sc->sc_cmdset) {
522                 case CFI_VEND_INTEL_ECS:
523                 case CFI_VEND_INTEL_SCS:
524                         st = cfi_read(sc, ofs);
525                         done = (st & CFI_INTEL_STATUS_WSMS);
526                         if (done) {
527                                 /* NB: bit 0 is reserved */
528                                 st &= ~(CFI_INTEL_XSTATUS_RSVD |
529                                         CFI_INTEL_STATUS_WSMS |
530                                         CFI_INTEL_STATUS_RSVD);
531                                 if (st & CFI_INTEL_STATUS_DPS)
532                                         error = EPERM;
533                                 else if (st & CFI_INTEL_STATUS_PSLBS)
534                                         error = EIO;
535                                 else if (st & CFI_INTEL_STATUS_ECLBS)
536                                         error = ENXIO;
537                                 else if (st)
538                                         error = EACCES;
539                         }
540                         break;
541                 case CFI_VEND_AMD_SCS:
542                 case CFI_VEND_AMD_ECS:
543                         st0 = cfi_read(sc, ofs);
544                         st = cfi_read(sc, ofs);
545                         done = ((st & 0x40) == (st0 & 0x40)) ? 1 : 0;
546                         break;
547                 }
548
549                 if (tto_exceeded ||
550                     now > start + sc->sc_typical_timeouts[cmd]) {
551                         if (!tto_exceeded) {
552                                 tto_exceeded = 1;
553                                 sc->sc_tto_counts[cmd]++;
554 #ifdef CFI_DEBUG_TIMEOUT
555                                 device_printf(sc->sc_dev,
556                                     "typical timeout exceeded (cmd %d)", cmd);
557 #endif
558                         }
559                         if (now > start + sc->sc_max_timeouts[cmd]) {
560                                 sc->sc_mto_counts[cmd]++;
561 #ifdef CFI_DEBUG_TIMEOUT
562                                 device_printf(sc->sc_dev,
563                                     "max timeout exceeded (cmd %d)", cmd);
564 #endif
565                         }
566                 }
567         }
568         if (!done && !error)
569                 error = ETIMEDOUT;
570         if (error)
571                 printf("\nerror=%d (st 0x%x st0 0x%x)\n", error, st, st0);
572         return (error);
573 }
574
575 int
576 cfi_write_block(struct cfi_softc *sc)
577 {
578         union {
579                 uint8_t         *x8;
580                 uint16_t        *x16;
581                 uint32_t        *x32;
582         } ptr, cpyprt;
583         register_t intr;
584         int error, i, neederase = 0;
585         uint32_t st;
586         u_int wlen;
587         sbintime_t start;
588
589         /* Intel flash must be unlocked before modification */
590         switch (sc->sc_cmdset) {
591         case CFI_VEND_INTEL_ECS:
592         case CFI_VEND_INTEL_SCS:
593                 cfi_write(sc, sc->sc_wrofs, CFI_INTEL_LBS);
594                 cfi_write(sc, sc->sc_wrofs, CFI_INTEL_UB);
595                 cfi_write(sc, sc->sc_wrofs, CFI_BCS_READ_ARRAY);
596                 break;
597         }
598
599         /* Check if an erase is required. */
600         for (i = 0; i < sc->sc_wrbufsz; i++)
601                 if ((sc->sc_wrbuf[i] & sc->sc_wrbufcpy[i]) != sc->sc_wrbuf[i]) {
602                         neederase = 1;
603                         break;
604                 }
605
606         if (neederase) {
607                 intr = intr_disable();
608                 start = sbinuptime();
609                 /* Erase the block. */
610                 switch (sc->sc_cmdset) {
611                 case CFI_VEND_INTEL_ECS:
612                 case CFI_VEND_INTEL_SCS:
613                         cfi_write(sc, sc->sc_wrofs, CFI_BCS_BLOCK_ERASE);
614                         cfi_write(sc, sc->sc_wrofs, CFI_BCS_CONFIRM);
615                         break;
616                 case CFI_VEND_AMD_SCS:
617                 case CFI_VEND_AMD_ECS:
618                         cfi_amd_write(sc, sc->sc_wrofs, AMD_ADDR_START,
619                             CFI_AMD_ERASE_SECTOR);
620                         cfi_amd_write(sc, sc->sc_wrofs, 0, CFI_AMD_BLOCK_ERASE);
621                         break;
622                 default:
623                         /* Better safe than sorry... */
624                         intr_restore(intr);
625                         return (ENODEV);
626                 }
627                 intr_restore(intr);
628                 error = cfi_wait_ready(sc, sc->sc_wrofs, start, 
629                     CFI_TIMEOUT_ERASE);
630                 if (error)
631                         goto out;
632         } else
633                 error = 0;
634
635         /* Write the block using a multibyte write if supported. */
636         ptr.x8 = sc->sc_wrbuf;
637         cpyprt.x8 = sc->sc_wrbufcpy;
638         if (sc->sc_maxbuf > sc->sc_width) {
639                 switch (sc->sc_cmdset) {
640                 case CFI_VEND_INTEL_ECS:
641                 case CFI_VEND_INTEL_SCS:
642                         for (i = 0; i < sc->sc_wrbufsz; i += wlen) {
643                                 wlen = MIN(sc->sc_maxbuf, sc->sc_wrbufsz - i);
644
645                                 intr = intr_disable();
646
647                                 start = sbinuptime();
648                                 do {
649                                         cfi_write(sc, sc->sc_wrofs + i,
650                                             CFI_BCS_BUF_PROG_SETUP);
651                                         if (sbinuptime() > start + sc->sc_max_timeouts[CFI_TIMEOUT_BUFWRITE]) {
652                                                 error = ETIMEDOUT;
653                                                 goto out;
654                                         }
655                                         st = cfi_read(sc, sc->sc_wrofs + i);
656                                 } while (! (st & CFI_INTEL_STATUS_WSMS));
657
658                                 cfi_write(sc, sc->sc_wrofs + i,
659                                     (wlen / sc->sc_width) - 1);
660                                 switch (sc->sc_width) {
661                                 case 1:
662                                         bus_space_write_region_1(sc->sc_tag,
663                                             sc->sc_handle, sc->sc_wrofs + i,
664                                             ptr.x8 + i, wlen);
665                                         break;
666                                 case 2:
667                                         bus_space_write_region_2(sc->sc_tag,
668                                             sc->sc_handle, sc->sc_wrofs + i,
669                                             ptr.x16 + i / 2, wlen / 2);
670                                         break;
671                                 case 4:
672                                         bus_space_write_region_4(sc->sc_tag,
673                                             sc->sc_handle, sc->sc_wrofs + i,
674                                             ptr.x32 + i / 4, wlen / 4);
675                                         break;
676                                 }
677
678                                 cfi_write(sc, sc->sc_wrofs + i,
679                                     CFI_BCS_CONFIRM);
680
681                                 intr_restore(intr);
682
683                                 error = cfi_wait_ready(sc, sc->sc_wrofs + i,
684                                     start, CFI_TIMEOUT_BUFWRITE);
685                                 if (error != 0)
686                                         goto out;
687                         }
688                         goto out;
689                 default:
690                         /* Fall through to single word case */
691                         break;
692                 }
693
694         }
695
696         /* Write the block one byte/word at a time. */
697         for (i = 0; i < sc->sc_wrbufsz; i += sc->sc_width) {
698
699                 /* Avoid writing unless we are actually changing bits */
700                 if (!neederase) {
701                         switch (sc->sc_width) {
702                         case 1:
703                                 if(*(ptr.x8 + i) == *(cpyprt.x8 + i))
704                                         continue;
705                                 break;
706                         case 2:
707                                 if(*(ptr.x16 + i / 2) == *(cpyprt.x16 + i / 2))
708                                         continue;
709                                 break;
710                         case 4:
711                                 if(*(ptr.x32 + i / 4) == *(cpyprt.x32 + i / 4))
712                                         continue;
713                                 break;
714                         }
715                 }
716
717                 /*
718                  * Make sure the command to start a write and the
719                  * actual write happens back-to-back without any
720                  * excessive delays.
721                  */
722                 intr = intr_disable();
723
724                 start = sbinuptime();
725                 switch (sc->sc_cmdset) {
726                 case CFI_VEND_INTEL_ECS:
727                 case CFI_VEND_INTEL_SCS:
728                         cfi_write(sc, sc->sc_wrofs + i, CFI_BCS_PROGRAM);
729                         break;
730                 case CFI_VEND_AMD_SCS:
731                 case CFI_VEND_AMD_ECS:
732                         cfi_amd_write(sc, 0, AMD_ADDR_START, CFI_AMD_PROGRAM);
733                         break;
734                 }
735                 switch (sc->sc_width) {
736                 case 1:
737                         bus_space_write_1(sc->sc_tag, sc->sc_handle,
738                             sc->sc_wrofs + i, *(ptr.x8 + i));
739                         break;
740                 case 2:
741                         bus_space_write_2(sc->sc_tag, sc->sc_handle,
742                             sc->sc_wrofs + i, *(ptr.x16 + i / 2));
743                         break;
744                 case 4:
745                         bus_space_write_4(sc->sc_tag, sc->sc_handle,
746                             sc->sc_wrofs + i, *(ptr.x32 + i / 4));
747                         break;
748                 }
749                 
750                 intr_restore(intr);
751
752                 error = cfi_wait_ready(sc, sc->sc_wrofs, start,
753                    CFI_TIMEOUT_WRITE);
754                 if (error)
755                         goto out;
756         }
757
758         /* error is 0. */
759
760  out:
761         cfi_reset_default(sc);
762
763         /* Relock Intel flash */
764         switch (sc->sc_cmdset) {
765         case CFI_VEND_INTEL_ECS:
766         case CFI_VEND_INTEL_SCS:
767                 cfi_write(sc, sc->sc_wrofs, CFI_INTEL_LBS);
768                 cfi_write(sc, sc->sc_wrofs, CFI_INTEL_LB);
769                 cfi_write(sc, sc->sc_wrofs, CFI_BCS_READ_ARRAY);
770                 break;
771         }
772         return (error);
773 }
774
775 #ifdef CFI_SUPPORT_STRATAFLASH
776 /*
777  * Intel StrataFlash Protection Register Support.
778  *
779  * The memory includes a 128-bit Protection Register that can be
780  * used for security.  There are two 64-bit segments; one is programmed
781  * at the factory with a unique 64-bit number which is immutable.
782  * The other segment is left blank for User (OEM) programming.
783  * The User/OEM segment is One Time Programmable (OTP).  It can also
784  * be locked to prevent any further writes by setting bit 0 of the
785  * Protection Lock Register (PLR).  The PLR can written only once.
786  */
787
788 static uint16_t
789 cfi_get16(struct cfi_softc *sc, int off)
790 {
791         uint16_t v = bus_space_read_2(sc->sc_tag, sc->sc_handle, off<<1);
792         return v;
793 }
794
795 #ifdef CFI_ARMEDANDDANGEROUS
796 static void
797 cfi_put16(struct cfi_softc *sc, int off, uint16_t v)
798 {
799         bus_space_write_2(sc->sc_tag, sc->sc_handle, off<<1, v);
800 }
801 #endif
802
803 /*
804  * Read the factory-defined 64-bit segment of the PR.
805  */
806 int 
807 cfi_intel_get_factory_pr(struct cfi_softc *sc, uint64_t *id)
808 {
809         if (sc->sc_cmdset != CFI_VEND_INTEL_ECS)
810                 return EOPNOTSUPP;
811         KASSERT(sc->sc_width == 2, ("sc_width %d", sc->sc_width));
812
813         cfi_write(sc, 0, CFI_INTEL_READ_ID);
814         *id = ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(0)))<<48 |
815               ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(1)))<<32 |
816               ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(2)))<<16 |
817               ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(3)));
818         cfi_write(sc, 0, CFI_BCS_READ_ARRAY);
819         return 0;
820 }
821
822 /*
823  * Read the User/OEM 64-bit segment of the PR.
824  */
825 int 
826 cfi_intel_get_oem_pr(struct cfi_softc *sc, uint64_t *id)
827 {
828         if (sc->sc_cmdset != CFI_VEND_INTEL_ECS)
829                 return EOPNOTSUPP;
830         KASSERT(sc->sc_width == 2, ("sc_width %d", sc->sc_width));
831
832         cfi_write(sc, 0, CFI_INTEL_READ_ID);
833         *id = ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(4)))<<48 |
834               ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(5)))<<32 |
835               ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(6)))<<16 |
836               ((uint64_t)cfi_get16(sc, CFI_INTEL_PR(7)));
837         cfi_write(sc, 0, CFI_BCS_READ_ARRAY);
838         return 0;
839 }
840
841 /*
842  * Write the User/OEM 64-bit segment of the PR.
843  * XXX should allow writing individual words/bytes
844  */
845 int
846 cfi_intel_set_oem_pr(struct cfi_softc *sc, uint64_t id)
847 {
848 #ifdef CFI_ARMEDANDDANGEROUS
849         register_t intr;
850         int i, error;
851         sbintime_t start;
852 #endif
853
854         if (sc->sc_cmdset != CFI_VEND_INTEL_ECS)
855                 return EOPNOTSUPP;
856         KASSERT(sc->sc_width == 2, ("sc_width %d", sc->sc_width));
857
858 #ifdef CFI_ARMEDANDDANGEROUS
859         for (i = 7; i >= 4; i--, id >>= 16) {
860                 intr = intr_disable();
861                 start = sbinuptime();
862                 cfi_write(sc, 0, CFI_INTEL_PP_SETUP);
863                 cfi_put16(sc, CFI_INTEL_PR(i), id&0xffff);
864                 intr_restore(intr);
865                 error = cfi_wait_ready(sc, CFI_BCS_READ_STATUS, start,
866                     CFI_TIMEOUT_WRITE);
867                 if (error)
868                         break;
869         }
870         cfi_write(sc, 0, CFI_BCS_READ_ARRAY);
871         return error;
872 #else
873         device_printf(sc->sc_dev, "%s: OEM PR not set, "
874             "CFI_ARMEDANDDANGEROUS not configured\n", __func__);
875         return ENXIO;
876 #endif
877 }
878
879 /*
880  * Read the contents of the Protection Lock Register.
881  */
882 int 
883 cfi_intel_get_plr(struct cfi_softc *sc, uint32_t *plr)
884 {
885         if (sc->sc_cmdset != CFI_VEND_INTEL_ECS)
886                 return EOPNOTSUPP;
887         KASSERT(sc->sc_width == 2, ("sc_width %d", sc->sc_width));
888
889         cfi_write(sc, 0, CFI_INTEL_READ_ID);
890         *plr = cfi_get16(sc, CFI_INTEL_PLR);
891         cfi_write(sc, 0, CFI_BCS_READ_ARRAY);
892         return 0;
893 }
894
895 /*
896  * Write the Protection Lock Register to lock down the
897  * user-settable segment of the Protection Register.
898  * NOTE: this operation is not reversible.
899  */
900 int 
901 cfi_intel_set_plr(struct cfi_softc *sc)
902 {
903 #ifdef CFI_ARMEDANDDANGEROUS
904         register_t intr;
905         int error;
906         sbintime_t start;
907 #endif
908         if (sc->sc_cmdset != CFI_VEND_INTEL_ECS)
909                 return EOPNOTSUPP;
910         KASSERT(sc->sc_width == 2, ("sc_width %d", sc->sc_width));
911
912 #ifdef CFI_ARMEDANDDANGEROUS
913         /* worthy of console msg */
914         device_printf(sc->sc_dev, "set PLR\n");
915         intr = intr_disable();
916         binuptime(&start);
917         cfi_write(sc, 0, CFI_INTEL_PP_SETUP);
918         cfi_put16(sc, CFI_INTEL_PLR, 0xFFFD);
919         intr_restore(intr);
920         error = cfi_wait_ready(sc, CFI_BCS_READ_STATUS, start,
921             CFI_TIMEOUT_WRITE);
922         cfi_write(sc, 0, CFI_BCS_READ_ARRAY);
923         return error;
924 #else
925         device_printf(sc->sc_dev, "%s: PLR not set, "
926             "CFI_ARMEDANDDANGEROUS not configured\n", __func__);
927         return ENXIO;
928 #endif
929 }
930 #endif /* CFI_SUPPORT_STRATAFLASH */