VolumeManager.cpp revision fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77 
1/*
2 * Copyright (C) 2008 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <stdio.h>
18#include <stdlib.h>
19#include <string.h>
20#include <errno.h>
21#include <fcntl.h>
22#include <sys/stat.h>
23#include <sys/types.h>
24#include <sys/mount.h>
25
26#include <linux/kdev_t.h>
27
28#define LOG_TAG "Vold"
29
30#include <openssl/md5.h>
31
32#include <cutils/log.h>
33
34#include <sysutils/NetlinkEvent.h>
35
36#include "VolumeManager.h"
37#include "DirectVolume.h"
38#include "ResponseCode.h"
39#include "Loop.h"
40#include "Fat.h"
41#include "Devmapper.h"
42#include "Process.h"
43#include "Asec.h"
44
45VolumeManager *VolumeManager::sInstance = NULL;
46
47VolumeManager *VolumeManager::Instance() {
48    if (!sInstance)
49        sInstance = new VolumeManager();
50    return sInstance;
51}
52
53VolumeManager::VolumeManager() {
54    mDebug = false;
55    mVolumes = new VolumeCollection();
56    mActiveContainers = new AsecIdCollection();
57    mBroadcaster = NULL;
58    mUsbMassStorageEnabled = false;
59    mUsbConnected = false;
60
61    readInitialState();
62}
63
64void VolumeManager::readInitialState() {
65    FILE *fp;
66    char state[255];
67
68    /*
69     * Read the initial mass storage enabled state
70     */
71    if ((fp = fopen("/sys/devices/virtual/usb_composite/usb_mass_storage/enable", "r"))) {
72        if (fgets(state, sizeof(state), fp)) {
73            mUsbMassStorageEnabled = !strncmp(state, "1", 1);
74        } else {
75            SLOGE("Failed to read usb_mass_storage enabled state (%s)", strerror(errno));
76        }
77        fclose(fp);
78    } else {
79        SLOGD("USB mass storage support is not enabled in the kernel");
80    }
81
82    /*
83     * Read the initial USB connected state
84     */
85    if ((fp = fopen("/sys/devices/virtual/switch/usb_configuration/state", "r"))) {
86        if (fgets(state, sizeof(state), fp)) {
87            mUsbConnected = !strncmp(state, "1", 1);
88        } else {
89            SLOGE("Failed to read usb_configuration switch (%s)", strerror(errno));
90        }
91        fclose(fp);
92    } else {
93        SLOGD("usb_configuration switch is not enabled in the kernel");
94    }
95}
96
97VolumeManager::~VolumeManager() {
98    delete mVolumes;
99    delete mActiveContainers;
100}
101
102char *VolumeManager::asecHash(const char *id, char *buffer, size_t len) {
103    static const char* digits = "0123456789abcdef";
104
105    unsigned char sig[MD5_DIGEST_LENGTH];
106
107    if (buffer == NULL) {
108        SLOGE("Destination buffer is NULL");
109        errno = ESPIPE;
110        return NULL;
111    } else if (id == NULL) {
112        SLOGE("Source buffer is NULL");
113        errno = ESPIPE;
114        return NULL;
115    } else if (len < MD5_ASCII_LENGTH_PLUS_NULL) {
116        SLOGE("Target hash buffer size < %d bytes (%d)",
117                MD5_ASCII_LENGTH_PLUS_NULL, len);
118        errno = ESPIPE;
119        return NULL;
120    }
121
122    MD5(reinterpret_cast<const unsigned char*>(id), strlen(id), sig);
123
124    char *p = buffer;
125    for (int i = 0; i < MD5_DIGEST_LENGTH; i++) {
126        *p++ = digits[sig[i] >> 4];
127        *p++ = digits[sig[i] & 0x0F];
128    }
129    *p = '\0';
130
131    return buffer;
132}
133
134void VolumeManager::setDebug(bool enable) {
135    mDebug = enable;
136    VolumeCollection::iterator it;
137    for (it = mVolumes->begin(); it != mVolumes->end(); ++it) {
138        (*it)->setDebug(enable);
139    }
140}
141
142int VolumeManager::start() {
143    return 0;
144}
145
146int VolumeManager::stop() {
147    return 0;
148}
149
150int VolumeManager::addVolume(Volume *v) {
151    mVolumes->push_back(v);
152    return 0;
153}
154
155void VolumeManager::notifyUmsAvailable(bool available) {
156    char msg[255];
157
158    snprintf(msg, sizeof(msg), "Share method ums now %s",
159             (available ? "available" : "unavailable"));
160    SLOGD(msg);
161    getBroadcaster()->sendBroadcast(ResponseCode::ShareAvailabilityChange,
162                                    msg, false);
163}
164
165void VolumeManager::handleSwitchEvent(NetlinkEvent *evt) {
166    const char *devpath = evt->findParam("DEVPATH");
167    const char *name = evt->findParam("SWITCH_NAME");
168    const char *state = evt->findParam("SWITCH_STATE");
169
170    if (!name || !state) {
171        SLOGW("Switch %s event missing name/state info", devpath);
172        return;
173    }
174
175    bool oldAvailable = massStorageAvailable();
176    if (!strcmp(name, "usb_configuration")) {
177        mUsbConnected = !strcmp(state, "1");
178        SLOGD("USB %s", mUsbConnected ? "connected" : "disconnected");
179        bool newAvailable = massStorageAvailable();
180        if (newAvailable != oldAvailable) {
181            notifyUmsAvailable(newAvailable);
182        }
183    } else {
184        SLOGW("Ignoring unknown switch '%s'", name);
185    }
186}
187void VolumeManager::handleUsbCompositeEvent(NetlinkEvent *evt) {
188    const char *function = evt->findParam("FUNCTION");
189    const char *enabled = evt->findParam("ENABLED");
190
191    if (!function || !enabled) {
192        SLOGW("usb_composite event missing function/enabled info");
193        return;
194    }
195
196    if (!strcmp(function, "usb_mass_storage")) {
197        bool oldAvailable = massStorageAvailable();
198        mUsbMassStorageEnabled = !strcmp(enabled, "1");
199        SLOGD("usb_mass_storage function %s", mUsbMassStorageEnabled ? "enabled" : "disabled");
200        bool newAvailable = massStorageAvailable();
201        if (newAvailable != oldAvailable) {
202            notifyUmsAvailable(newAvailable);
203        }
204    }
205}
206
207void VolumeManager::handleBlockEvent(NetlinkEvent *evt) {
208    const char *devpath = evt->findParam("DEVPATH");
209
210    /* Lookup a volume to handle this device */
211    VolumeCollection::iterator it;
212    bool hit = false;
213    for (it = mVolumes->begin(); it != mVolumes->end(); ++it) {
214        if (!(*it)->handleBlockEvent(evt)) {
215#ifdef NETLINK_DEBUG
216            SLOGD("Device '%s' event handled by volume %s\n", devpath, (*it)->getLabel());
217#endif
218            hit = true;
219            break;
220        }
221    }
222
223    if (!hit) {
224#ifdef NETLINK_DEBUG
225        SLOGW("No volumes handled block event for '%s'", devpath);
226#endif
227    }
228}
229
230int VolumeManager::listVolumes(SocketClient *cli) {
231    VolumeCollection::iterator i;
232
233    for (i = mVolumes->begin(); i != mVolumes->end(); ++i) {
234        char *buffer;
235        asprintf(&buffer, "%s %s %d",
236                 (*i)->getLabel(), (*i)->getMountpoint(),
237                 (*i)->getState());
238        cli->sendMsg(ResponseCode::VolumeListResult, buffer, false);
239        free(buffer);
240    }
241    cli->sendMsg(ResponseCode::CommandOkay, "Volumes listed.", false);
242    return 0;
243}
244
245int VolumeManager::formatVolume(const char *label) {
246    Volume *v = lookupVolume(label);
247
248    if (!v) {
249        errno = ENOENT;
250        return -1;
251    }
252
253    return v->formatVol();
254}
255
256int VolumeManager::getAsecMountPath(const char *id, char *buffer, int maxlen) {
257    char asecFileName[255];
258    snprintf(asecFileName, sizeof(asecFileName), "%s/%s.asec", Volume::SEC_ASECDIR, id);
259
260    memset(buffer, 0, maxlen);
261    if (access(asecFileName, F_OK)) {
262        errno = ENOENT;
263        return -1;
264    }
265
266    snprintf(buffer, maxlen, "%s/%s", Volume::ASECDIR, id);
267    return 0;
268}
269
270int VolumeManager::createAsec(const char *id, unsigned int numSectors,
271                              const char *fstype, const char *key, int ownerUid) {
272    struct asec_superblock sb;
273    memset(&sb, 0, sizeof(sb));
274
275    sb.magic = ASEC_SB_MAGIC;
276    sb.ver = ASEC_SB_VER;
277
278    if (numSectors < ((1024*1024)/512)) {
279        SLOGE("Invalid container size specified (%d sectors)", numSectors);
280        errno = EINVAL;
281        return -1;
282    }
283
284    if (lookupVolume(id)) {
285        SLOGE("ASEC id '%s' currently exists", id);
286        errno = EADDRINUSE;
287        return -1;
288    }
289
290    char asecFileName[255];
291    snprintf(asecFileName, sizeof(asecFileName), "%s/%s.asec", Volume::SEC_ASECDIR, id);
292
293    if (!access(asecFileName, F_OK)) {
294        SLOGE("ASEC file '%s' currently exists - destroy it first! (%s)",
295             asecFileName, strerror(errno));
296        errno = EADDRINUSE;
297        return -1;
298    }
299
300    /*
301     * Add some headroom
302     */
303    unsigned fatSize = (((numSectors * 4) / 512) + 1) * 2;
304    unsigned numImgSectors = numSectors + fatSize + 2;
305
306    if (numImgSectors % 63) {
307        numImgSectors += (63 - (numImgSectors % 63));
308    }
309
310    // Add +1 for our superblock which is at the end
311    if (Loop::createImageFile(asecFileName, numImgSectors + 1)) {
312        SLOGE("ASEC image file creation failed (%s)", strerror(errno));
313        return -1;
314    }
315
316    char idHash[33];
317    if (!asecHash(id, idHash, sizeof(idHash))) {
318        SLOGE("Hash of '%s' failed (%s)", id, strerror(errno));
319        unlink(asecFileName);
320        return -1;
321    }
322
323    char loopDevice[255];
324    if (Loop::create(idHash, asecFileName, loopDevice, sizeof(loopDevice))) {
325        SLOGE("ASEC loop device creation failed (%s)", strerror(errno));
326        unlink(asecFileName);
327        return -1;
328    }
329
330    char dmDevice[255];
331    bool cleanupDm = false;
332
333    if (strcmp(key, "none")) {
334        // XXX: This is all we support for now
335        sb.c_cipher = ASEC_SB_C_CIPHER_TWOFISH;
336        if (Devmapper::create(idHash, loopDevice, key, numImgSectors, dmDevice,
337                             sizeof(dmDevice))) {
338            SLOGE("ASEC device mapping failed (%s)", strerror(errno));
339            Loop::destroyByDevice(loopDevice);
340            unlink(asecFileName);
341            return -1;
342        }
343        cleanupDm = true;
344    } else {
345        sb.c_cipher = ASEC_SB_C_CIPHER_NONE;
346        strcpy(dmDevice, loopDevice);
347    }
348
349    /*
350     * Drop down the superblock at the end of the file
351     */
352
353    int sbfd = open(loopDevice, O_RDWR);
354    if (sbfd < 0) {
355        SLOGE("Failed to open new DM device for superblock write (%s)", strerror(errno));
356        if (cleanupDm) {
357            Devmapper::destroy(idHash);
358        }
359        Loop::destroyByDevice(loopDevice);
360        unlink(asecFileName);
361        return -1;
362    }
363
364    if (lseek(sbfd, (numImgSectors * 512), SEEK_SET) < 0) {
365        close(sbfd);
366        SLOGE("Failed to lseek for superblock (%s)", strerror(errno));
367        if (cleanupDm) {
368            Devmapper::destroy(idHash);
369        }
370        Loop::destroyByDevice(loopDevice);
371        unlink(asecFileName);
372        return -1;
373    }
374
375    if (write(sbfd, &sb, sizeof(sb)) != sizeof(sb)) {
376        close(sbfd);
377        SLOGE("Failed to write superblock (%s)", strerror(errno));
378        if (cleanupDm) {
379            Devmapper::destroy(idHash);
380        }
381        Loop::destroyByDevice(loopDevice);
382        unlink(asecFileName);
383        return -1;
384    }
385    close(sbfd);
386
387    if (strcmp(fstype, "none")) {
388        if (strcmp(fstype, "fat")) {
389            SLOGW("Unknown fstype '%s' specified for container", fstype);
390        }
391
392        if (Fat::format(dmDevice, numImgSectors)) {
393            SLOGE("ASEC FAT format failed (%s)", strerror(errno));
394            if (cleanupDm) {
395                Devmapper::destroy(idHash);
396            }
397            Loop::destroyByDevice(loopDevice);
398            unlink(asecFileName);
399            return -1;
400        }
401        char mountPoint[255];
402
403        snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::ASECDIR, id);
404        if (mkdir(mountPoint, 0777)) {
405            if (errno != EEXIST) {
406                SLOGE("Mountpoint creation failed (%s)", strerror(errno));
407                if (cleanupDm) {
408                    Devmapper::destroy(idHash);
409                }
410                Loop::destroyByDevice(loopDevice);
411                unlink(asecFileName);
412                return -1;
413            }
414        }
415
416        if (Fat::doMount(dmDevice, mountPoint, false, false, ownerUid,
417                         0, 0000, false)) {
418            SLOGE("ASEC FAT mount failed (%s)", strerror(errno));
419            if (cleanupDm) {
420                Devmapper::destroy(idHash);
421            }
422            Loop::destroyByDevice(loopDevice);
423            unlink(asecFileName);
424            return -1;
425        }
426    } else {
427        SLOGI("Created raw secure container %s (no filesystem)", id);
428    }
429
430    mActiveContainers->push_back(strdup(id));
431    return 0;
432}
433
434int VolumeManager::finalizeAsec(const char *id) {
435    char asecFileName[255];
436    char loopDevice[255];
437    char mountPoint[255];
438
439    snprintf(asecFileName, sizeof(asecFileName), "%s/%s.asec", Volume::SEC_ASECDIR, id);
440
441    char idHash[33];
442    if (!asecHash(id, idHash, sizeof(idHash))) {
443        SLOGE("Hash of '%s' failed (%s)", id, strerror(errno));
444        return -1;
445    }
446
447    if (Loop::lookupActive(idHash, loopDevice, sizeof(loopDevice))) {
448        SLOGE("Unable to finalize %s (%s)", id, strerror(errno));
449        return -1;
450    }
451
452    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::ASECDIR, id);
453    // XXX:
454    if (Fat::doMount(loopDevice, mountPoint, true, true, 0, 0, 0227, false)) {
455        SLOGE("ASEC finalize mount failed (%s)", strerror(errno));
456        return -1;
457    }
458
459    if (mDebug) {
460        SLOGD("ASEC %s finalized", id);
461    }
462    return 0;
463}
464
465int VolumeManager::renameAsec(const char *id1, const char *id2) {
466    char *asecFilename1;
467    char *asecFilename2;
468    char mountPoint[255];
469
470    asprintf(&asecFilename1, "%s/%s.asec", Volume::SEC_ASECDIR, id1);
471    asprintf(&asecFilename2, "%s/%s.asec", Volume::SEC_ASECDIR, id2);
472
473    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::ASECDIR, id1);
474    if (isMountpointMounted(mountPoint)) {
475        SLOGW("Rename attempt when src mounted");
476        errno = EBUSY;
477        goto out_err;
478    }
479
480    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::ASECDIR, id2);
481    if (isMountpointMounted(mountPoint)) {
482        SLOGW("Rename attempt when dst mounted");
483        errno = EBUSY;
484        goto out_err;
485    }
486
487    if (!access(asecFilename2, F_OK)) {
488        SLOGE("Rename attempt when dst exists");
489        errno = EADDRINUSE;
490        goto out_err;
491    }
492
493    if (rename(asecFilename1, asecFilename2)) {
494        SLOGE("Rename of '%s' to '%s' failed (%s)", asecFilename1, asecFilename2, strerror(errno));
495        goto out_err;
496    }
497
498    free(asecFilename1);
499    free(asecFilename2);
500    return 0;
501
502out_err:
503    free(asecFilename1);
504    free(asecFilename2);
505    return -1;
506}
507
508#define UNMOUNT_RETRIES 5
509#define UNMOUNT_SLEEP_BETWEEN_RETRY_MS (1000 * 1000)
510int VolumeManager::unmountAsec(const char *id, bool force) {
511    char asecFileName[255];
512    char mountPoint[255];
513
514    snprintf(asecFileName, sizeof(asecFileName), "%s/%s.asec", Volume::SEC_ASECDIR, id);
515    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::ASECDIR, id);
516
517    char idHash[33];
518    if (!asecHash(id, idHash, sizeof(idHash))) {
519        SLOGE("Hash of '%s' failed (%s)", id, strerror(errno));
520        return -1;
521    }
522
523    return unmountLoopImage(id, idHash, asecFileName, mountPoint, force);
524}
525
526int VolumeManager::unmountImage(const char *fileName, bool force) {
527    char mountPoint[255];
528
529    char idHash[33];
530    if (!asecHash(fileName, idHash, sizeof(idHash))) {
531        SLOGE("Hash of '%s' failed (%s)", fileName, strerror(errno));
532        return -1;
533    }
534
535    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::LOOPDIR, idHash);
536
537    return unmountLoopImage(fileName, idHash, fileName, mountPoint, force);
538}
539
540int VolumeManager::unmountLoopImage(const char *id, const char *idHash,
541        const char *fileName, const char *mountPoint, bool force) {
542    if (!isMountpointMounted(mountPoint)) {
543        SLOGE("Unmount request for %s when not mounted", id);
544        errno = EINVAL;
545        return -1;
546    }
547
548    int i, rc;
549    for (i = 1; i <= UNMOUNT_RETRIES; i++) {
550        rc = umount(mountPoint);
551        if (!rc) {
552            break;
553        }
554        if (rc && (errno == EINVAL || errno == ENOENT)) {
555            SLOGI("Container %s unmounted OK", id);
556            rc = 0;
557            break;
558        }
559        SLOGW("%s unmount attempt %d failed (%s)",
560              id, i, strerror(errno));
561
562        int action = 0; // default is to just complain
563
564        if (force) {
565            if (i > (UNMOUNT_RETRIES - 2))
566                action = 2; // SIGKILL
567            else if (i > (UNMOUNT_RETRIES - 3))
568                action = 1; // SIGHUP
569        }
570
571        Process::killProcessesWithOpenFiles(mountPoint, action);
572        usleep(UNMOUNT_SLEEP_BETWEEN_RETRY_MS);
573    }
574
575    if (rc) {
576        errno = EBUSY;
577        SLOGE("Failed to unmount container %s (%s)", id, strerror(errno));
578        return -1;
579    }
580
581    int retries = 10;
582
583    while(retries--) {
584        if (!rmdir(mountPoint)) {
585            break;
586        }
587
588        SLOGW("Failed to rmdir %s (%s)", mountPoint, strerror(errno));
589        usleep(UNMOUNT_SLEEP_BETWEEN_RETRY_MS);
590    }
591
592    if (!retries) {
593        SLOGE("Timed out trying to rmdir %s (%s)", mountPoint, strerror(errno));
594    }
595
596    if (Devmapper::destroy(idHash) && errno != ENXIO) {
597        SLOGE("Failed to destroy devmapper instance (%s)", strerror(errno));
598    }
599
600    char loopDevice[255];
601    if (!Loop::lookupActive(idHash, loopDevice, sizeof(loopDevice))) {
602        Loop::destroyByDevice(loopDevice);
603    } else {
604        SLOGW("Failed to find loop device for {%s} (%s)", fileName, strerror(errno));
605    }
606
607    AsecIdCollection::iterator it;
608    for (it = mActiveContainers->begin(); it != mActiveContainers->end(); ++it) {
609        if (!strcmp(*it, id)) {
610            free(*it);
611            mActiveContainers->erase(it);
612            break;
613        }
614    }
615    if (it == mActiveContainers->end()) {
616        SLOGW("mActiveContainers is inconsistent!");
617    }
618    return 0;
619}
620
621int VolumeManager::destroyAsec(const char *id, bool force) {
622    char asecFileName[255];
623    char mountPoint[255];
624
625    snprintf(asecFileName, sizeof(asecFileName), "%s/%s.asec", Volume::SEC_ASECDIR, id);
626    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::ASECDIR, id);
627
628    if (isMountpointMounted(mountPoint)) {
629        if (mDebug) {
630            SLOGD("Unmounting container before destroy");
631        }
632        if (unmountAsec(id, force)) {
633            SLOGE("Failed to unmount asec %s for destroy (%s)", id, strerror(errno));
634            return -1;
635        }
636    }
637
638    if (unlink(asecFileName)) {
639        SLOGE("Failed to unlink asec '%s' (%s)", asecFileName, strerror(errno));
640        return -1;
641    }
642
643    if (mDebug) {
644        SLOGD("ASEC %s destroyed", id);
645    }
646    return 0;
647}
648
649int VolumeManager::mountAsec(const char *id, const char *key, int ownerUid) {
650    char asecFileName[255];
651    char mountPoint[255];
652
653    snprintf(asecFileName, sizeof(asecFileName), "%s/%s.asec", Volume::SEC_ASECDIR, id);
654    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::ASECDIR, id);
655
656    if (isMountpointMounted(mountPoint)) {
657        SLOGE("ASEC %s already mounted", id);
658        errno = EBUSY;
659        return -1;
660    }
661
662    char idHash[33];
663    if (!asecHash(id, idHash, sizeof(idHash))) {
664        SLOGE("Hash of '%s' failed (%s)", id, strerror(errno));
665        return -1;
666    }
667
668    char loopDevice[255];
669    if (Loop::lookupActive(idHash, loopDevice, sizeof(loopDevice))) {
670        if (Loop::create(idHash, asecFileName, loopDevice, sizeof(loopDevice))) {
671            SLOGE("ASEC loop device creation failed (%s)", strerror(errno));
672            return -1;
673        }
674        if (mDebug) {
675            SLOGD("New loop device created at %s", loopDevice);
676        }
677    } else {
678        if (mDebug) {
679            SLOGD("Found active loopback for %s at %s", asecFileName, loopDevice);
680        }
681    }
682
683    char dmDevice[255];
684    bool cleanupDm = false;
685    int fd;
686    unsigned int nr_sec = 0;
687
688    if ((fd = open(loopDevice, O_RDWR)) < 0) {
689        SLOGE("Failed to open loopdevice (%s)", strerror(errno));
690        Loop::destroyByDevice(loopDevice);
691        return -1;
692    }
693
694    if (ioctl(fd, BLKGETSIZE, &nr_sec)) {
695        SLOGE("Failed to get loop size (%s)", strerror(errno));
696        Loop::destroyByDevice(loopDevice);
697        close(fd);
698        return -1;
699    }
700
701    /*
702     * Validate superblock
703     */
704    struct asec_superblock sb;
705    memset(&sb, 0, sizeof(sb));
706    if (lseek(fd, ((nr_sec-1) * 512), SEEK_SET) < 0) {
707        SLOGE("lseek failed (%s)", strerror(errno));
708        close(fd);
709        Loop::destroyByDevice(loopDevice);
710        return -1;
711    }
712    if (read(fd, &sb, sizeof(sb)) != sizeof(sb)) {
713        SLOGE("superblock read failed (%s)", strerror(errno));
714        close(fd);
715        Loop::destroyByDevice(loopDevice);
716        return -1;
717    }
718
719    close(fd);
720
721    if (mDebug) {
722        SLOGD("Container sb magic/ver (%.8x/%.2x)", sb.magic, sb.ver);
723    }
724    if (sb.magic != ASEC_SB_MAGIC || sb.ver != ASEC_SB_VER) {
725        SLOGE("Bad container magic/version (%.8x/%.2x)", sb.magic, sb.ver);
726        Loop::destroyByDevice(loopDevice);
727        errno = EMEDIUMTYPE;
728        return -1;
729    }
730    nr_sec--; // We don't want the devmapping to extend onto our superblock
731
732    if (strcmp(key, "none")) {
733        if (Devmapper::lookupActive(idHash, dmDevice, sizeof(dmDevice))) {
734            if (Devmapper::create(idHash, loopDevice, key, nr_sec,
735                                  dmDevice, sizeof(dmDevice))) {
736                SLOGE("ASEC device mapping failed (%s)", strerror(errno));
737                Loop::destroyByDevice(loopDevice);
738                return -1;
739            }
740            if (mDebug) {
741                SLOGD("New devmapper instance created at %s", dmDevice);
742            }
743        } else {
744            if (mDebug) {
745                SLOGD("Found active devmapper for %s at %s", asecFileName, dmDevice);
746            }
747        }
748        cleanupDm = true;
749    } else {
750        strcpy(dmDevice, loopDevice);
751    }
752
753    if (mkdir(mountPoint, 0777)) {
754        if (errno != EEXIST) {
755            SLOGE("Mountpoint creation failed (%s)", strerror(errno));
756            if (cleanupDm) {
757                Devmapper::destroy(idHash);
758            }
759            Loop::destroyByDevice(loopDevice);
760            return -1;
761        }
762    }
763
764    if (Fat::doMount(dmDevice, mountPoint, true, false, ownerUid, 0,
765                     0222, false)) {
766//                     0227, false)) {
767        SLOGE("ASEC mount failed (%s)", strerror(errno));
768        if (cleanupDm) {
769            Devmapper::destroy(idHash);
770        }
771        Loop::destroyByDevice(loopDevice);
772        return -1;
773    }
774
775    mActiveContainers->push_back(strdup(id));
776    if (mDebug) {
777        SLOGD("ASEC %s mounted", id);
778    }
779    return 0;
780}
781
782/**
783 * Mounts an image file <code>img</code>.
784 */
785int VolumeManager::mountImage(const char *img, const char *key, int ownerUid) {
786    char mountPoint[255];
787
788#if 0
789    struct stat imgStat;
790    if (stat(img, &imgStat) != 0) {
791        SLOGE("Could not stat '%s': %s", img, strerror(errno));
792        return -1;
793    }
794
795    if (imgStat.st_uid != ownerUid) {
796        SLOGW("Image UID does not match requestor UID (%d != %d)",
797                imgStat.st_uid, ownerUid);
798        return -1;
799    }
800#endif
801
802    char idHash[33];
803    if (!asecHash(img, idHash, sizeof(idHash))) {
804        SLOGE("Hash of '%s' failed (%s)", img, strerror(errno));
805        return -1;
806    }
807
808    snprintf(mountPoint, sizeof(mountPoint), "%s/%s", Volume::LOOPDIR, idHash);
809
810    if (isMountpointMounted(mountPoint)) {
811        SLOGE("Image %s already mounted", img);
812        errno = EBUSY;
813        return -1;
814    }
815
816    char loopDevice[255];
817    if (Loop::lookupActive(idHash, loopDevice, sizeof(loopDevice))) {
818        if (Loop::create(idHash, img, loopDevice, sizeof(loopDevice))) {
819            SLOGE("Image loop device creation failed (%s)", strerror(errno));
820            return -1;
821        }
822        if (mDebug) {
823            SLOGD("New loop device created at %s", loopDevice);
824        }
825    } else {
826        if (mDebug) {
827            SLOGD("Found active loopback for %s at %s", img, loopDevice);
828        }
829    }
830
831    char dmDevice[255];
832    bool cleanupDm = false;
833    int fd;
834    unsigned int nr_sec = 0;
835
836    if ((fd = open(loopDevice, O_RDWR)) < 0) {
837        SLOGE("Failed to open loopdevice (%s)", strerror(errno));
838        Loop::destroyByDevice(loopDevice);
839        return -1;
840    }
841
842    if (ioctl(fd, BLKGETSIZE, &nr_sec)) {
843        SLOGE("Failed to get loop size (%s)", strerror(errno));
844        Loop::destroyByDevice(loopDevice);
845        close(fd);
846        return -1;
847    }
848
849    close(fd);
850
851    if (strcmp(key, "none")) {
852        if (Devmapper::lookupActive(idHash, dmDevice, sizeof(dmDevice))) {
853            if (Devmapper::create(idHash, loopDevice, key, nr_sec,
854                                  dmDevice, sizeof(dmDevice))) {
855                SLOGE("ASEC device mapping failed (%s)", strerror(errno));
856                Loop::destroyByDevice(loopDevice);
857                return -1;
858            }
859            if (mDebug) {
860                SLOGD("New devmapper instance created at %s", dmDevice);
861            }
862        } else {
863            if (mDebug) {
864                SLOGD("Found active devmapper for %s at %s", img, dmDevice);
865            }
866        }
867        cleanupDm = true;
868    } else {
869        strcpy(dmDevice, loopDevice);
870    }
871
872    if (mkdir(mountPoint, 0755)) {
873        if (errno != EEXIST) {
874            SLOGE("Mountpoint creation failed (%s)", strerror(errno));
875            if (cleanupDm) {
876                Devmapper::destroy(idHash);
877            }
878            Loop::destroyByDevice(loopDevice);
879            return -1;
880        }
881    }
882
883    if (Fat::doMount(dmDevice, mountPoint, true, false, ownerUid, 0,
884                     0227, false)) {
885        SLOGE("Image mount failed (%s)", strerror(errno));
886        if (cleanupDm) {
887            Devmapper::destroy(idHash);
888        }
889        Loop::destroyByDevice(loopDevice);
890        return -1;
891    }
892
893    mActiveContainers->push_back(strdup(img));
894    if (mDebug) {
895        SLOGD("Image %s mounted", img);
896    }
897    return 0;
898}
899
900int VolumeManager::mountVolume(const char *label) {
901    Volume *v = lookupVolume(label);
902
903    if (!v) {
904        errno = ENOENT;
905        return -1;
906    }
907
908    return v->mountVol();
909}
910
911int VolumeManager::shareAvailable(const char *method, bool *avail) {
912
913    if (strcmp(method, "ums")) {
914        errno = ENOSYS;
915        return -1;
916    }
917
918    *avail = massStorageAvailable();
919    return 0;
920}
921
922int VolumeManager::shareEnabled(const char *label, const char *method, bool *enabled) {
923    Volume *v = lookupVolume(label);
924
925    if (!v) {
926        errno = ENOENT;
927        return -1;
928    }
929
930    if (strcmp(method, "ums")) {
931        errno = ENOSYS;
932        return -1;
933    }
934
935    if (v->getState() != Volume::State_Shared) {
936        *enabled = false;
937    } else {
938        *enabled = true;
939    }
940    return 0;
941}
942
943int VolumeManager::simulate(const char *cmd, const char *arg) {
944
945    if (!strcmp(cmd, "ums")) {
946        if (!strcmp(arg, "connect")) {
947            notifyUmsAvailable(true);
948        } else if (!strcmp(arg, "disconnect")) {
949            notifyUmsAvailable(false);
950        } else {
951            errno = EINVAL;
952            return -1;
953        }
954    } else {
955        errno = EINVAL;
956        return -1;
957    }
958    return 0;
959}
960
961int VolumeManager::shareVolume(const char *label, const char *method) {
962    Volume *v = lookupVolume(label);
963
964    if (!v) {
965        errno = ENOENT;
966        return -1;
967    }
968
969    /*
970     * Eventually, we'll want to support additional share back-ends,
971     * some of which may work while the media is mounted. For now,
972     * we just support UMS
973     */
974    if (strcmp(method, "ums")) {
975        errno = ENOSYS;
976        return -1;
977    }
978
979    if (v->getState() == Volume::State_NoMedia) {
980        errno = ENODEV;
981        return -1;
982    }
983
984    if (v->getState() != Volume::State_Idle) {
985        // You need to unmount manually befoe sharing
986        errno = EBUSY;
987        return -1;
988    }
989
990    dev_t d = v->getDiskDevice();
991    if ((MAJOR(d) == 0) && (MINOR(d) == 0)) {
992        // This volume does not support raw disk access
993        errno = EINVAL;
994        return -1;
995    }
996
997    int fd;
998    char nodepath[255];
999    snprintf(nodepath,
1000             sizeof(nodepath), "/dev/block/vold/%d:%d",
1001             MAJOR(d), MINOR(d));
1002
1003    if ((fd = open("/sys/devices/platform/usb_mass_storage/lun0/file",
1004                   O_WRONLY)) < 0) {
1005        SLOGE("Unable to open ums lunfile (%s)", strerror(errno));
1006        return -1;
1007    }
1008
1009    if (write(fd, nodepath, strlen(nodepath)) < 0) {
1010        SLOGE("Unable to write to ums lunfile (%s)", strerror(errno));
1011        close(fd);
1012        return -1;
1013    }
1014
1015    close(fd);
1016    v->handleVolumeShared();
1017    return 0;
1018}
1019
1020int VolumeManager::unshareVolume(const char *label, const char *method) {
1021    Volume *v = lookupVolume(label);
1022
1023    if (!v) {
1024        errno = ENOENT;
1025        return -1;
1026    }
1027
1028    if (strcmp(method, "ums")) {
1029        errno = ENOSYS;
1030        return -1;
1031    }
1032
1033    if (v->getState() != Volume::State_Shared) {
1034        errno = EINVAL;
1035        return -1;
1036    }
1037
1038    dev_t d = v->getDiskDevice();
1039
1040    int fd;
1041    char nodepath[255];
1042    snprintf(nodepath,
1043             sizeof(nodepath), "/dev/block/vold/%d:%d",
1044             MAJOR(d), MINOR(d));
1045
1046    if ((fd = open("/sys/devices/platform/usb_mass_storage/lun0/file", O_WRONLY)) < 0) {
1047        SLOGE("Unable to open ums lunfile (%s)", strerror(errno));
1048        return -1;
1049    }
1050
1051    char ch = 0;
1052    if (write(fd, &ch, 1) < 0) {
1053        SLOGE("Unable to write to ums lunfile (%s)", strerror(errno));
1054        close(fd);
1055        return -1;
1056    }
1057
1058    close(fd);
1059    v->handleVolumeUnshared();
1060    return 0;
1061}
1062
1063int VolumeManager::unmountVolume(const char *label, bool force) {
1064    Volume *v = lookupVolume(label);
1065
1066    if (!v) {
1067        errno = ENOENT;
1068        return -1;
1069    }
1070
1071    if (v->getState() == Volume::State_NoMedia) {
1072        errno = ENODEV;
1073        return -1;
1074    }
1075
1076    if (v->getState() != Volume::State_Mounted) {
1077        SLOGW("Attempt to unmount volume which isn't mounted (%d)\n",
1078             v->getState());
1079        errno = EBUSY;
1080        return -1;
1081    }
1082
1083    cleanupAsec(v, force);
1084
1085    return v->unmountVol(force);
1086}
1087
1088/*
1089 * Looks up a volume by it's label or mount-point
1090 */
1091Volume *VolumeManager::lookupVolume(const char *label) {
1092    VolumeCollection::iterator i;
1093
1094    for (i = mVolumes->begin(); i != mVolumes->end(); ++i) {
1095        if (label[0] == '/') {
1096            if (!strcmp(label, (*i)->getMountpoint()))
1097                return (*i);
1098        } else {
1099            if (!strcmp(label, (*i)->getLabel()))
1100                return (*i);
1101        }
1102    }
1103    return NULL;
1104}
1105
1106bool VolumeManager::isMountpointMounted(const char *mp)
1107{
1108    char device[256];
1109    char mount_path[256];
1110    char rest[256];
1111    FILE *fp;
1112    char line[1024];
1113
1114    if (!(fp = fopen("/proc/mounts", "r"))) {
1115        SLOGE("Error opening /proc/mounts (%s)", strerror(errno));
1116        return false;
1117    }
1118
1119    while(fgets(line, sizeof(line), fp)) {
1120        line[strlen(line)-1] = '\0';
1121        sscanf(line, "%255s %255s %255s\n", device, mount_path, rest);
1122        if (!strcmp(mount_path, mp)) {
1123            fclose(fp);
1124            return true;
1125        }
1126    }
1127
1128    fclose(fp);
1129    return false;
1130}
1131
1132int VolumeManager::cleanupAsec(Volume *v, bool force) {
1133    while(mActiveContainers->size()) {
1134        AsecIdCollection::iterator it = mActiveContainers->begin();
1135        SLOGI("Unmounting ASEC %s (dependant on %s)", *it, v->getMountpoint());
1136        if (unmountAsec(*it, force)) {
1137            SLOGE("Failed to unmount ASEC %s (%s)", *it, strerror(errno));
1138            return -1;
1139        }
1140    }
1141    return 0;
1142}
1143
1144