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