o 3'fH@sUdZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZddlmZddlmZddlmZddlmZd d gegd Zeed <d ZeeZdZeeZdZdZdZ dZ!e"e#Z$ddZ%ddZ&ddZ'dIddZ(dIddZ)de*de e*dee*e*fd e+fd!d"Z,dJd#d$Z-d%e*d&e*d dfd'd(Z.dKd)d*Z/d+d,Z0d e ee*ee*e*fee*ffd-d.Z1d/ee ee e*dee*e*fd0ee*e*fd1ee e*d2ed eee*f d3d4Z2d/eee*d eee*fd5d6Z3d/eee*d7e*dee*e*fd0ee*e*fd2ed eee*f d8d9Z4d:eee*d eee*fd;d<Z5d:eee*d dfd=d>Z6d?e+d@e+dAee*d dfdBdCZ7dDe*dEed2edFe8d df dGdHZ9dS)Lz-Mounts: Configure mount points and swap filesN)DictListOptionalTuplecast)subputil)Cloud)Config) MetaSchema) PER_INSTANCE cc_mountsall)iddistros frequencyactivate_by_schema_keysmetaz&^([x]{0,1}[shv]d[a-z][0-9]*|sr[0-9]+)$z^.+:.*z /etc/fstabzcomment=cloudconfigi@cCs8|dvrdSdD]}||r|ddkrdSqdS)N)amirootswapT) ephemeralebs:F) startswithfind)nameenumnamer %sNz+Ignoring nonexistent default named mount %sFz&Device %s already defined in fstab: %sT)r:r;)rErFrGr r r!sanitized_devname_is_valids rIc Cs`|dur td}|d}d||d}|dur|dur|}n2|rKt|}|j|j}||d<|dur>tt|d|}n||dkrJt|d}n|durQ|}||d<|dtkr^|}n|d tkridt}n t t |tt}t||}||d <i} | D]\} } t | trd | t| | <q| | | <qtd | d | d | d| d| d|S)Ntotalna)availmax_inmemrMg?maxsizez%s MBzCsuggest %s swap for %s memory with '%s' disk given max=%s [max=%s]'rOrN)r read_meminfor'statvfsf_frsizef_bfreeminintGBroundmathsqrtitems isinstanceMBr:r;) memsizemaxsizefsyssugg_maxinforUrMminsizerSpinfokvr r r!suggested_swapsizesN            rjfnamerScsdfdd}tj|}t|t|ddkr0tddd|gtd d |gd krAtd krA|||d nz|||dWntjy^t d|||d Ynwtj |rkt |dz td|gWdStjyt |w)zSize is in MiB.z6Failed to create swapfile '%s' of size %sMB via %s: %sc std|||dkrddd||g}n|dkr%ddd|dd |g}ntd z tj|d d WdStjyP}zt||||t|d}~ww) Nz3Creating swapfile in '%s' on fstype '%s' using '%s' fallocatez-lz%sMddz if=/dev/zerozof=%szbs=1Mzcount=%sz:Missing dependency: 'dd' and 'fallocate' are not availableT)capture)r:r;rProcessExecutionErrorrerdel_file)rkrSmethodcmdeerrmsgfstyper r! create_swaps4 z$create_swapfile..create_swapr9btrfstruncatez-s0chattrz+Cxfs)rPrmrlz4fallocate swap creation failed, will attempt with ddimkswapN)r'r(dirnamer ensure_dirget_mount_inforkernel_versionror:rer)chmodrp)rkrSrwswap_dirr rtr!create_swapfiles0      rcCstj|}t|dkr2ztd}Wnty%t dYdSwt |t |||d}tt |d}|sCt ddStj tj dt||gd |S) z fname: full path string of filename to setup size: the size to create. set to "auto" for recommended maxsize: the maximum size autorJz)Not creating swap: failed to read meminfoN)rcrbrarz'Not creating swap: suggested size was 0zSetting up swap file)msgfuncargs)r'r(rr&lowerrrTIOErrorr:r;rrjrYlog_timer)rkrSrbrramibsizer r r!setup_swapfiles.     rc CsZt|ts tddS|dd}|dd}|dd}|r"|s)tddStj|rstjd s=td ||Sz$t d  D]}| |d rYtd ||WSqEtd |Wnt yrtd||YSwz t|tr|dkrt |}t|trt |}t|||dWSt y}z td|WYd}~dSd}~ww)zahandle the swap config, calling setup_swap if necessary. return None or (filename, size) z%input for swap config was not a dict.Nfilenamez /swap.imgrSrrbzno need to setup swapz /proc/swapsz:swap file %s exists, but no /proc/swaps exists, being safe zswap file %s already in usez+swap file %s exists, but not in /proc/swapsz.swap file %s exists. Error reading /proc/swapsr)rkrSrbzfailed to setup swap: %s)r_dictr:warningr=r;r'r(r)rload_text_file splitlinesr Exceptionr& human2bytesr)swapcfgrkrSrblinersr r r!handle_swapcfg5sP              rcCslg}i}g}tjtr1ttD]}t|vr||q| }|r0|||d<||q|||fS)asParse /etc/fstab. Parse fstab, ignoring any lines containing "comment=cloudconfig". :return: A 3-tuple containing: - A list of lines exactly as they appear in fstab - A dictionary with key being the first token in the line and value being the entire line - A list of any lines that were ignored due to "comment=cloudconfig" r) r'r(r) FSTAB_PATHrrr MNT_COMMENTappendr2) fstab_linesrG fstab_removedrtoksr r r! parse_fstabes      rmountsdevice_aliasesdefault_fieldscloudc Csg}|D]Y}t|tstd|qt|d}t||j|d}t|||r0|g|dd} n|} t| D]\} } | durE|| | | <q6t| | | | <q6| |t | d7} | | q|S)aSanitize mounts to ensure we can work with devices in config. Specifically: - Ensure the mounts configuration is a list of lists - Transform and sanitize device names - Ensure all tokens are strings - Add default options to any lines without options z%Mount option not a list, ignoring: %srrAr9N) r_listr:rr&rDdevice_name_to_devicerI enumeratelenr) rrGrrr updated_linesrstartsanitized_devname updated_lineindextokenr r r!sanitize_mounts_configurations&     rcCslg}g}|dddD]#}|ddus|d|vr)td|d||dq ||q |dddS)zRemove any entries that have a device name that doesn't exist. If the second field of a mount line is None (not the string, the value), we skip it along with any other entries that came before it that share the same device name. Nrr9rz$Skipping nonexistent device named %s)r:r;r)ractlist dev_denylistrr r r!remove_nonexistent_devicess rdefault_mount_optionsc st|}ddd|ddggdfD]5d}t||j|d}t|||s&qtt|d<tfd d |D}|rAt d |q| q|S) zAdd default mounts to the user provided mount configuration. Add them only if no other entry has the same device name r8z/mntrrz2)rnonerswrzrzrrc3s |] }|ddkVqdS)rNr ).0cfgm default_mountr r! s z,add_default_mounts_to_cfg..z-Not including %s, already previously included) copydeepcopyrDrrIrr&anyr:r;r) rrrGrr new_mountsrrFdefault_already_existsr rr!add_default_mounts_to_cfgs&     rrcCsdd|DS)z=Add "comment=cloudconfig" to the mount options of each entry.cSs8g|]}|dd|ddtg|ddqS)N,rP)rrentryr r r! s*zadd_comment..r rr r r! add_commentsrcCs(tdd|DrtddgdSdS)z1Call 'swapon -a' if any entry has a swap fs type.css|] }|ddkVqdS)rKrNr rr r r!rsz*activate_swap_if_needed..swapon-aN)rrrr r r!activate_swap_if_neededsr uses_systemd changes_madedirscCsfd}|rd}nddtD}tt||}|r/tddg|r1tddgd Sd Sd S) zCall 'mount -a' if needed. If changes were made, always call 'mount -a'. Otherwise, call 'mount -a' if any of the directories in the mount list are not already mounted. FTcSsh|] }d|vr|dqS) mountpointr )rvalr r r! s z"mount_if_needed..mountr systemctlz daemon-reloadN)rrvaluesboolset differencer)rrrdo_mount mount_pointsr r r!mount_if_neededs rrcfgrc s|j}|r dnd}ddd|ddg}|d|}|dg}td |t\} } } |d i} t|| | ||} t| || | |} t| } t | } t |d i}|r^| |d d d ddgt | dkrktddSdd| D}dd| D}|D]}zt |Wq{tyt td|Yq{wdd|Ddd| DfddDfddD}| |dd| }t t||rtd|ntdt| t|t||dS)z Handle the mounts configuration.z:defaults,nofail,x-systemd.after=cloud-init.service,_netdevzdefaults,nobootwaitNrrzrmount_default_fieldsrzmounts configuration is %srrrrrz No modifications to fstab neededcSsg|]}d|qS) )r3rr r r!r5szhandle..cSs"g|] }|ddr|dqS)r9r0)r)rdr r r!r7s"z Failed to make '%s' config-mountcSg|]}|ddqSrrreplacernr r r!r?cSrrrrr r r!r@rcg|] }|vrd|qS)z- r )rdrop)saddsr r!rBscr)z+ r )radd)sdropsr r!rBsz%s  zChanges to fstab: %szNo changes to /etc/fstab made.)distrorr=r:r;rrrrrrrrrrrlogexcextendr3 write_filerrrr)rrrrrrhardcoded_defaultsrrrrGrr updated_cfgswapfile cfg_linesrrsopscontentsr )rrr!handle s`             r)N)NNN)NN):__doc__rloggingr\r'retypingrrrrr cloudinitrrcloudinit.cloudr cloudinit.configr cloudinit.config.schemar cloudinit.settingsr r__annotations__DEVICE_NAME_FILTERcompiler>NETWORK_NAME_FILTERr#rrr`rZ getLogger__name__r:r"r%r/r7rDr&rrIrjrrrrrrrrrrrrr r r r!s         1  9 >"(0     ".     "$ "